From 3f2631afa118976a01a4c674f14484daa94058ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jan=20Bedna=C5=99=C3=ADk?= <jan.bednarik@gmail.com>
Date: Tue, 17 May 2022 19:46:39 +0200
Subject: [PATCH] region: Fix convert commmand

---
 .../commands/convert_region_to_district.py           | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/region/management/commands/convert_region_to_district.py b/region/management/commands/convert_region_to_district.py
index d5b740de..90042a4a 100644
--- a/region/management/commands/convert_region_to_district.py
+++ b/region/management/commands/convert_region_to_district.py
@@ -3,7 +3,6 @@ from django.db import transaction
 from wagtail.core.models import Page, Site
 
 from district.models import (
-    DistrictAfterElectionPage,
     DistrictArticlePage,
     DistrictArticlesPage,
     DistrictArticleTag,
@@ -12,8 +11,6 @@ from district.models import (
     DistrictCrossroadPage,
     DistrictCustomPage,
     DistrictElectionBasePage,
-    DistrictElectionPage,
-    DistrictElectionPointPage,
     DistrictHomePage,
     DistrictPeoplePage,
     DistrictPersonPage,
@@ -23,7 +20,6 @@ from district.models import (
     DistrictTagsPage,
 )
 from region.models import (
-    RegionAfterElectionPage,
     RegionArticlePage,
     RegionArticlesPage,
     RegionArticleTag,
@@ -32,8 +28,6 @@ from region.models import (
     RegionCrossroadPage,
     RegionCustomPage,
     RegionElectionBasePage,
-    RegionElectionPage,
-    RegionElectionPointPage,
     RegionHomePage,
     RegionPeoplePage,
     RegionPersonPage,
@@ -58,9 +52,6 @@ migration_map = {
     RegionPersonPage: DistrictPersonPage,
     RegionPeoplePage: DistrictPeoplePage,
     RegionElectionBasePage: DistrictElectionBasePage,
-    RegionAfterElectionPage: DistrictAfterElectionPage,
-    RegionElectionPointPage: DistrictElectionPointPage,
-    RegionElectionPage: DistrictElectionPage,
     RegionProgramPage: DistrictProgramPage,
     RegionCenterPage: DistrictCenterPage,
     RegionCrossroadPage: DistrictCrossroadPage,
@@ -111,6 +102,9 @@ def handle_child_pages(region_root_page, district_root_page):
         child_specific = child.specific
         page_class = child_specific.__class__
 
+        if page_class not in migration_map:
+            continue
+
         DistrictChildPage = migration_map[page_class]
 
         tmp_data_dict = child_specific.__dict__.copy()
-- 
GitLab