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

WIP - program migrations

parent 7ff1f43d
Branches
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
from django.db import migrations
from shared.blocks import SocialLinkBlock
from django.db import migrations, transaction
from district.blocks import ProgramGroupWithCandidatesBlock
from shared.blocks import SocialLinkBlock
def migrate_programs(apps, schema_editor):
......@@ -21,15 +22,20 @@ def migrate_programs(apps, schema_editor):
"district", "DistrictPostElectionStrategyPage"
)
DistrictNewProgramPage = apps.get_model(
"district", "DistrictNewProgramPage"
)
DistrictNewProgramPage = apps.get_model("district", "DistrictNewProgramPage")
Locale = apps.get_model("wagtailcore", "Locale")
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")
root_page_content_type = ContentType.objects.get(
app_label="district", model="districtelectionrootpage"
)
......@@ -37,12 +43,19 @@ def migrate_programs(apps, schema_editor):
app_label="district", model="districtelectioncampaignpage"
)
people_page_content_type = ContentType.objects.get(
app_label="district", model="districtpeoplepage"
)
program_page_content_type = ContentType.objects.get(
app_label="district", model="districtelectionprogrampage"
)
program_post_election_strategy_content_type = ContentType.objects.get(
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
def get_children_of_type(model, parent_page, content_type):
......@@ -57,22 +70,21 @@ def migrate_programs(apps, schema_editor):
# Title
"title": "", # Done
"order": 0,
# Candidate / program lists
"candidate_pages": [],
"candidate_blocks": [],
# Program lists
"program_points": [],
"post_election_strategies": [],
# Candidates
"candidate_list_number": "",
"candidate_list_title": "",
"candidate_pages": [],
"candidate_blocks": [], # Page creation done
"combined_candidates": [],
"primary_candidate_count": 0,
# Program
"program_title": "",
"program_is_inline": False,
"program_content_before": "", # Done
# Misc.
"funding_info": "",
}
......@@ -101,33 +113,36 @@ def migrate_programs(apps, schema_editor):
"candidate_list_title"
] = campaign_page.candidate_list_title
election_data[
"program_title"
] = campaign_page.program_point_list_title
election_data["program_title"] = campaign_page.program_point_list_title
election_data[
"program_is_inline"
] = campaign_page.show_program_points_inline
election_data["program_content_before"] = campaign_page.content
election_data["funding_info"] = campaign_page.campaign_funding_info
position = 0
for candidate_list_block in campaign_page.candidates.get_prep_value():
candidate_list = candidate_list_block["value"]
for position, candidate in enumerate(
candidate_list["candidate_list"]
):
election_data["primary_candidate_count"] = candidate_list[
"candidate_list_big_count"
]
for candidate in candidate_list["candidate_list"]:
position = position + 1
if candidate["type"] == "person_page":
election_data["candidate_pages"].append(
{
"position": position,
"page": DistrictPersonPage.objects.filter(
id=candidate["value"]
).first(),
}
)
candidate_data = {
"position": position,
"page": DistrictPersonPage.objects.filter(
id=candidate["value"]
).first(),
}
election_data["candidate_pages"].append(candidate_data)
election_data["combined_candidates"].append(candidate_data)
continue
if candidate["type"] == "person_blocks":
......@@ -147,7 +162,6 @@ def migrate_programs(apps, schema_editor):
"guarantor_page": program_page.guarantor,
"image": program_page.image,
"perex": program_page.perex,
"content": program_page.content,
"funding_info": program_page.campaign_funding_info,
}
)
......@@ -162,89 +176,178 @@ def migrate_programs(apps, schema_editor):
election_data["post_election_strategies"].append(
{
"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
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
for candidate_block in election_data["candidate_blocks"]:
if parent_people_page is None:
break # We can't do anything here
social_links = []
if candidate_block["value"]["facebook_url"]:
if candidate_block["data"]["facebook_url"]:
social_links.append(
SocialLinkBlock().to_python({
"icon": "ico--facebook",
"text": "Facebook",
"link": candidate_block["value"]["facebook_url"]
})
SocialLinkBlock().to_python(
{
"icon": "ico--facebook",
"text": "Facebook",
"link": candidate_block["data"]["facebook_url"],
}
)
)
if candidate_block["value"]["instagram_url"]:
if candidate_block["data"]["instagram_url"]:
social_links.append(
SocialLinkBlock().to_python({
"icon": "ico--instagram",
"text": "Instagram",
"link": candidate_block["value"]["instagram_url"]
})
SocialLinkBlock().to_python(
{
"icon": "ico--instagram",
"text": "Instagram",
"link": candidate_block["data"]["instagram_url"],
}
)
)
if candidate_block["value"]["twitter_url"]:
if candidate_block["data"]["twitter_url"]:
social_links.append(
SocialLinkBlock().to_python({
"icon": "ico--twitter",
"text": "Twitter",
"link": candidate_block["value"]["twitter_url"]
})
SocialLinkBlock().to_python(
{
"icon": "ico--twitter",
"text": "Twitter",
"link": candidate_block["data"]["twitter_url"],
}
)
)
if candidate_block["value"]["youtube_url"]:
if candidate_block["data"]["youtube_url"]:
social_links.append(
SocialLinkBlock().to_python({
"icon": "ico--youtube",
"text": "YouTube",
"link": candidate_block["value"]["youtube_url"]
})
SocialLinkBlock().to_python(
{
"icon": "ico--youtube",
"text": "YouTube",
"link": candidate_block["data"]["youtube_url"],
}
)
)
if candidate_block["value"]["flickr_url"]:
if candidate_block["data"]["flickr_url"]:
social_links.append(
SocialLinkBlock().to_python({
"icon": "ico--flickr",
"text": "Flickr",
"link": candidate_block["value"]["flickr_url"]
})
SocialLinkBlock().to_python(
{
"icon": "ico--flickr",
"text": "Flickr",
"link": candidate_block["data"]["flickr_url"],
}
)
)
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,
max_path_page = (
Page.objects.filter(
path__startswith=people_page.path,
depth=parent_people_page.depth + 1,
)
.order_by("-path")
.first()
)
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(
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_block = ProgramGroupWithCandidatesBlock().to_python({
"title": election_data["title"],
"preamble_content": election_data["program_content_before"],
# Update numchild on the home page
home_page.numchild = home_page.numchild + 1
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
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