diff --git a/region/management/commands/convert_region_to_district.py b/region/management/commands/convert_region_to_district.py index d5b740de2e2f8be20d357a405c5fe426e79005bb..90042a4ae541648a64489fd40f8f89056b87dadb 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()