Skip to content
Snippets Groups Projects
Commit 706a182d authored by Alexa Valentová's avatar Alexa Valentová
Browse files

WIP - program migrations

parent 7ff1f43d
No related branches found
No related tags found
2 merge requests!994Release,!988Redesign
Pipeline #18507 passed
Stage: build
Hide whitespace changes
Inline Side-by-side
district/migrations/0218_auto_20240601_1530.py
+ 183
− 80
View file @ 706a182d
# Generated by Django 5.0.4 on 2024-06-01 13:30 # Generated by Django 5.0.4 on 2024-06-01 13:30
from django.db import migrations from django.db import migrations, transaction
from shared.blocks import SocialLinkBlock
from district.blocks import ProgramGroupWithCandidatesBlock from district.blocks import ProgramGroupWithCandidatesBlock
from shared.blocks import SocialLinkBlock
def migrate_programs(apps, schema_editor): def migrate_programs(apps, schema_editor):
...@@ -21,15 +22,20 @@ def migrate_programs(apps, schema_editor): ...@@ -21,15 +22,20 @@ def migrate_programs(apps, schema_editor):
"district", "DistrictPostElectionStrategyPage" "district", "DistrictPostElectionStrategyPage"
) )
DistrictNewProgramPage = apps.get_model( DistrictNewProgramPage = apps.get_model("district", "DistrictNewProgramPage")
"district", "DistrictNewProgramPage"
) Locale = apps.get_model("wagtailcore", "Locale")
DistrictPersonPage = apps.get_model("district", "DistrictPersonPage") DistrictPersonPage = apps.get_model("district", "DistrictPersonPage")
apps.get_model("wagtailcore", "Page") DistrictPeoplePage = apps.get_model("district", "DistrictPeoplePage")
# Get the default locale
default_locale = Locale.objects.first()
Page = apps.get_model("wagtailcore", "Page")
# Get the content type for DistrictElectionCampaignPage and DistrictElectionRootPage
ContentType = apps.get_model("contenttypes", "ContentType") ContentType = apps.get_model("contenttypes", "ContentType")
root_page_content_type = ContentType.objects.get( root_page_content_type = ContentType.objects.get(
app_label="district", model="districtelectionrootpage" app_label="district", model="districtelectionrootpage"
) )
...@@ -37,12 +43,19 @@ def migrate_programs(apps, schema_editor): ...@@ -37,12 +43,19 @@ def migrate_programs(apps, schema_editor):
app_label="district", model="districtelectioncampaignpage" app_label="district", model="districtelectioncampaignpage"
) )
people_page_content_type = ContentType.objects.get(
app_label="district", model="districtpeoplepage"
)
program_page_content_type = ContentType.objects.get( program_page_content_type = ContentType.objects.get(
app_label="district", model="districtelectionprogrampage" app_label="district", model="districtelectionprogrampage"
) )
program_post_election_strategy_content_type = ContentType.objects.get( program_post_election_strategy_content_type = ContentType.objects.get(
app_label="district", model="districtpostelectionstrategypage" app_label="district", model="districtpostelectionstrategypage"
) )
new_program_page_content_type = ContentType.objects.get(
app_label="district", model="districtnewprogrampage"
)
# Function to get children of a certain type # Function to get children of a certain type
def get_children_of_type(model, parent_page, content_type): def get_children_of_type(model, parent_page, content_type):
...@@ -57,22 +70,21 @@ def migrate_programs(apps, schema_editor): ...@@ -57,22 +70,21 @@ def migrate_programs(apps, schema_editor):
# Title # Title
"title": "", # Done "title": "", # Done
"order": 0, "order": 0,
# Program lists
# Candidate / program lists
"candidate_pages": [],
"candidate_blocks": [],
"program_points": [], "program_points": [],
"post_election_strategies": [], "post_election_strategies": [],
# Candidates # Candidates
"candidate_list_number": "", "candidate_list_number": "",
"candidate_list_title": "", "candidate_list_title": "",
"candidate_pages": [],
"candidate_blocks": [], # Page creation done
"combined_candidates": [],
"primary_candidate_count": 0,
# Program # Program
"program_title": "", "program_title": "",
"program_is_inline": False, "program_is_inline": False,
"program_content_before": "", # Done "program_content_before": "", # Done
# Misc.
"funding_info": "", "funding_info": "",
} }
...@@ -101,33 +113,36 @@ def migrate_programs(apps, schema_editor): ...@@ -101,33 +113,36 @@ def migrate_programs(apps, schema_editor):
"candidate_list_title" "candidate_list_title"
] = campaign_page.candidate_list_title ] = campaign_page.candidate_list_title
election_data[ election_data["program_title"] = campaign_page.program_point_list_title
"program_title"
] = campaign_page.program_point_list_title
election_data[ election_data[
"program_is_inline" "program_is_inline"
] = campaign_page.show_program_points_inline ] = campaign_page.show_program_points_inline
election_data["program_content_before"] = campaign_page.content
election_data["funding_info"] = campaign_page.campaign_funding_info election_data["funding_info"] = campaign_page.campaign_funding_info
position = 0
for candidate_list_block in campaign_page.candidates.get_prep_value(): for candidate_list_block in campaign_page.candidates.get_prep_value():
candidate_list = candidate_list_block["value"] candidate_list = candidate_list_block["value"]
for position, candidate in enumerate( election_data["primary_candidate_count"] = candidate_list[
candidate_list["candidate_list"] "candidate_list_big_count"
): ]
for candidate in candidate_list["candidate_list"]:
position = position + 1 position = position + 1
if candidate["type"] == "person_page": if candidate["type"] == "person_page":
election_data["candidate_pages"].append( candidate_data = {
{ "position": position,
"position": position, "page": DistrictPersonPage.objects.filter(
"page": DistrictPersonPage.objects.filter( id=candidate["value"]
id=candidate["value"] ).first(),
).first(), }
}
) election_data["candidate_pages"].append(candidate_data)
election_data["combined_candidates"].append(candidate_data)
continue continue
if candidate["type"] == "person_blocks": if candidate["type"] == "person_blocks":
...@@ -147,7 +162,6 @@ def migrate_programs(apps, schema_editor): ...@@ -147,7 +162,6 @@ def migrate_programs(apps, schema_editor):
"guarantor_page": program_page.guarantor, "guarantor_page": program_page.guarantor,
"image": program_page.image, "image": program_page.image,
"perex": program_page.perex, "perex": program_page.perex,
"content": program_page.content,
"funding_info": program_page.campaign_funding_info, "funding_info": program_page.campaign_funding_info,
} }
) )
...@@ -162,89 +176,178 @@ def migrate_programs(apps, schema_editor): ...@@ -162,89 +176,178 @@ def migrate_programs(apps, schema_editor):
election_data["post_election_strategies"].append( election_data["post_election_strategies"].append(
{ {
"perex": post_election_strategy_page.perex, "perex": post_election_strategy_page.perex,
"content": post_election_strategy_page.content,
} }
) )
if len(election_data["candidate_pages"]) == 0 and len(election_data["candidate_blocks"]) == 0 and len(election_data["program_points"]) == 0: if (
len(election_data["candidate_pages"]) == 0
and len(election_data["candidate_blocks"]) == 0
and len(election_data["program_points"]) == 0
):
continue # Do nothing for these continue # Do nothing for these
parent_people_page = get_children_of_type(
DistrictPeoplePage, home_page, people_page_content_type
)
parent_people_page = parent_people_page.first()
# Create corresponding pages for the candidate blocks # Create corresponding pages for the candidate blocks
for candidate_block in election_data["candidate_blocks"]: for candidate_block in election_data["candidate_blocks"]:
if parent_people_page is None:
break # We can't do anything here
social_links = [] social_links = []
if candidate_block["value"]["facebook_url"]: if candidate_block["data"]["facebook_url"]:
social_links.append( social_links.append(
SocialLinkBlock().to_python({ SocialLinkBlock().to_python(
"icon": "ico--facebook", {
"text": "Facebook", "icon": "ico--facebook",
"link": candidate_block["value"]["facebook_url"] "text": "Facebook",
}) "link": candidate_block["data"]["facebook_url"],
}
)
) )
if candidate_block["value"]["instagram_url"]: if candidate_block["data"]["instagram_url"]:
social_links.append( social_links.append(
SocialLinkBlock().to_python({ SocialLinkBlock().to_python(
"icon": "ico--instagram", {
"text": "Instagram", "icon": "ico--instagram",
"link": candidate_block["value"]["instagram_url"] "text": "Instagram",
}) "link": candidate_block["data"]["instagram_url"],
}
)
) )
if candidate_block["value"]["twitter_url"]: if candidate_block["data"]["twitter_url"]:
social_links.append( social_links.append(
SocialLinkBlock().to_python({ SocialLinkBlock().to_python(
"icon": "ico--twitter", {
"text": "Twitter", "icon": "ico--twitter",
"link": candidate_block["value"]["twitter_url"] "text": "Twitter",
}) "link": candidate_block["data"]["twitter_url"],
}
)
) )
if candidate_block["value"]["youtube_url"]: if candidate_block["data"]["youtube_url"]:
social_links.append( social_links.append(
SocialLinkBlock().to_python({ SocialLinkBlock().to_python(
"icon": "ico--youtube", {
"text": "YouTube", "icon": "ico--youtube",
"link": candidate_block["value"]["youtube_url"] "text": "YouTube",
}) "link": candidate_block["data"]["youtube_url"],
}
)
) )
if candidate_block["value"]["flickr_url"]: if candidate_block["data"]["flickr_url"]:
social_links.append( social_links.append(
SocialLinkBlock().to_python({ SocialLinkBlock().to_python(
"icon": "ico--flickr", {
"text": "Flickr", "icon": "ico--flickr",
"link": candidate_block["value"]["flickr_url"] "text": "Flickr",
}) "link": candidate_block["data"]["flickr_url"],
}
)
) )
candidate_page = DistrictPersonPage( max_path_page = (
title=candidate_block["data"]["title"], Page.objects.filter(
job=candidate_block["data"]["job"], path__startswith=people_page.path,
profile_image_id=candidate_block["data"]["profile_photo"], depth=parent_people_page.depth + 1,
email=candidate_block["data"]["email"], )
city=candidate_block["data"]["city"], .order_by("-path")
age=candidate_block["data"]["age"], .first()
is_pirate=candidate_block["data"]["is_pirate"],
other_party=candidate_block["data"]["other_party"],
other_party_logo_id=candidate_block["data"]["other_party_logo"],
social_links=social_links,
) )
candidate_page.save() if max_path_page:
max_path = max_path_page.path
next_path_suffix = (
int(max_path[-4:], 36) + 1
) # Base-36 to handle alphanumeric paths
next_path = (
max_path[:-4] + f"{next_path_suffix:04X}"
) # Convert back to base-36
else:
next_path = parent_people_page.path + "0001"
with transaction.atomic():
candidate_page = DistrictPersonPage(
title=candidate_block["data"]["title"],
job=candidate_block["data"]["job"],
profile_image_id=candidate_block["data"]["profile_photo"],
email=candidate_block["data"]["email"],
city=candidate_block["data"]["city"],
age=candidate_block["data"]["age"],
is_pirate=candidate_block["data"]["is_pirate"],
other_party=candidate_block["data"]["other_party"],
other_party_logo_id=candidate_block["data"]["other_party_logo"],
social_links=social_links,
content_type_id=people_page_content_type.id,
locale_id=default_locale.id,
)
candidate_page.numchild = candidate_page.numchild + 1
candidate_page.save(update_fields=["numchild"])
election_data["combined_candidates"].append(
{
"position": candidate_block["position"],
"data": candidate_page,
}
)
max_path_page = (
Page.objects.filter(
path__startswith=home_page.path, depth=home_page.depth + 1
)
.order_by("-path")
.first()
)
if max_path_page:
max_path = max_path_page.path
next_path_suffix = (
int(max_path[-4:], 36) + 1
) # Base-36 to handle alphanumeric paths
next_path = (
max_path[:-4] + f"{next_path_suffix:04X}"
) # Convert back to base-36
else:
next_path = home_page.path + "0001"
with transaction.atomic():
new_program_page = DistrictNewProgramPage( new_program_page = DistrictNewProgramPage(
title="Programy" title="Programy",
slug="programy", # Make sure the slug is unique among siblings
path=next_path,
depth=home_page.depth + 1,
numchild=0,
content_type_id=new_program_page_content_type.id,
locale_id=default_locale.id,
) )
new_program_page.save() new_program_page.save()
new_program_block = ProgramGroupWithCandidatesBlock().to_python({ # Update numchild on the home page
"title": election_data["title"], home_page.numchild = home_page.numchild + 1
"preamble_content": election_data["program_content_before"], home_page.save(update_fields=["numchild"])
}) election_data["combined_candidates"] = sorted(
election_data["combined_candidates"],
key=lambda value: value["position"],
)
print(election_data) new_program_block = ProgramGroupWithCandidatesBlock().to_python(
{
"title": election_data["title"],
"preamble_content": election_data["program_content_before"],
"primary_candidates": [],
"secondary_candidates": [],
}
)
# Stop migration for debugging purposes # Stop migration for debugging purposes
raise ValueError("Stopping migration for debugging purposes") raise ValueError("Stopping migration for debugging purposes")
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment

Pirátský GitLab - Home of the CopyLeft