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