region: Fix convert commmand

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()