def get_replacement_ballot(self, ballot_id):
replacement_ballot = None
ee = EEHelper()
ee_data = ee.get_data(ballot_id)
if ee_data:
replacement_ballot_id = ee_data["replaced_by"]
if replacement_ballot_id:
replacement_ballot = PostElection.objects.get(
ballot_paper_id=replacement_ballot_id
)
return replacement_ballot
def populate_empty_voting_systems(self):
qs = Election.objects.filter(
voting_system=None, current=True
).prefetch_related("postelection_set")
ee = EEHelper()
for election in qs:
for post_election in election.postelection_set.all():
ballot_data = ee.get_data(post_election.ballot_paper_id)
if "voting_system" in ballot_data:
post_election.voting_system_id = ballot_data[
"voting_system"
]["slug"]
post_election.save()
def import_territories(self):
ee = EEHelper()
post_elections_without_territory = PostElection.objects.filter(
post__territory=""
)
for post_election in post_elections_without_territory:
ee_data = ee.get_data(post_election.ballot_paper_id)
territory = self.extract_territory(ee_data)
self.stdout.write(
"Added territory to {0}: {1}".format(
post_election.ballot_paper_id, territory
)
)
post_election.post.territory = territory
post_election.post.save()
class YNRBallotImporter:
"""
Class for populating local election and ballot models in this
project from YNR.
The class sets up everything needed for show a ballot, including elections,
posts, voting systems, and the person information that show's on a ballot.
(name, candidacy data)
"""
def __init__(
self,
force_update=False,
stdout=sys.stdout,
current_only=False,
exclude_candidacies=False,
force_metadata=False,
force_current_metadata=False,
):
self.stdout = stdout
self.ee_helper = EEHelper()
self.voting_systems = {}
self.election_importer = YNRElectionImporter(self.ee_helper)
self.post_importer = YNRPostImporter(self.ee_helper)
self.force_update = force_update
self.current_only = current_only
self.exclude_candidacies = exclude_candidacies
self.force_metadata = force_metadata
self.force_current_metadata = force_current_metadata
def get_paginator(self, page1):
return JsonPaginator(page1, self.stdout)
def do_import(self, params=None):
default_params = {"page_size": "200"}
if self.current_only:
default_params["current"] = True
if params:
default_params.update(params)
else:
prewarm_current_only = True
if self.force_metadata:
prewarm_current_only = False
self.ee_helper.prewarm_cache(current=prewarm_current_only)
querystring = urlencode(default_params)
if not params and not self.current_only:
# this is a full import, use the cache
url = (settings.YNR_BASE +
"/media/cached-api/latest/ballots-000001.json")
else:
url = settings.YNR_BASE + "/api/next/ballots/?{}".format(
querystring)
pages = self.get_paginator(url)
for page in pages:
self.add_ballots(page)
if not params:
# Don't try to do things like add replaced
# ballots if we've filtered the ballots.
# This is because there's a high chance we've not
# got all the ballots we need yet.
self.run_post_ballot_import_tasks()
self.delete_orphan_posts()
def delete_orphan_posts(self):
"""
This method cleans orphan posts.
This typically gets called at the end of the import process.
"""
return Post.objects.filter(postelection=None).delete()
@transaction.atomic()
def add_ballots(self, results):
for ballot_dict in results["results"]:
print(ballot_dict["ballot_paper_id"])
election = self.election_importer.update_or_create_from_ballot_dict(
ballot_dict)
post = self.post_importer.update_or_create_from_ballot_dict(
ballot_dict)
if not post:
# cant create a ballot without a post so skip to the next one
continue
ballot, created = PostElection.objects.update_or_create(
ballot_paper_id=ballot_dict["ballot_paper_id"],
defaults={
"election": election,
"post": post,
"winner_count": ballot_dict["winner_count"],
"cancelled": ballot_dict["cancelled"],
"locked": ballot_dict["candidates_locked"],
},
)
if ballot.election.current or self.force_metadata:
self.import_metadata_from_ee(ballot)
if not self.exclude_candidacies:
# Now set the nominations up for this ballot
# First, remove any old candidates, this is to flush out candidates
# that have changed. We just delete the `person_post`
# (`membership` in YNR), not the person profile.
ballot.personpost_set.all().delete()
for candidate in ballot_dict["candidacies"]:
person, person_created = Person.objects.update_or_create(
ynr_id=candidate["person"]["id"],
defaults={"name": candidate["person"]["name"]},
)
result = candidate["result"] or {}
PersonPost.objects.create(
post_election=ballot,
person=person,
party_id=candidate["party"]["legacy_slug"],
list_position=candidate["party_list_position"],
elected=result.get("elected"),
votes_cast=result.get("num_ballots", None),
post=ballot.post,
election=ballot.election,
)
if created:
self.stdout.write("Added new ballot: {0}".format(
ballot.ballot_paper_id))
def import_metadata_from_ee(self, ballot):
# First, grab the data from EE
self.set_territory(ballot)
self.set_voting_system(ballot)
self.set_metadata(ballot)
self.set_organisation_type(ballot)
self.set_division_type(ballot)
ballot.save()
def set_territory(self, ballot):
if ballot.post.territory and not self.force_update:
return
ee_data = self.ee_helper.get_data(ballot.ballot_paper_id)
if ee_data and "organisation" in ee_data:
territory = ee_data["organisation"].get("territory_code", "-")
else:
territory = "-"
ballot.post.territory = territory
ballot.post.save()
def set_voting_system(self, ballot):
if ballot.voting_system_id and not self.force_update:
return
ee_data = self.ee_helper.get_data(ballot.ballot_paper_id)
if ee_data and "voting_system" in ee_data:
voting_system_slug = ee_data["voting_system"]["slug"]
if not voting_system_slug in self.voting_systems:
voting_system = VotingSystem.objects.update_or_create(
slug=voting_system_slug,
defaults={"description": ee_data["voting_system"]["name"]},
)[0]
self.voting_systems[voting_system_slug] = voting_system
ballot.voting_system = self.voting_systems[voting_system_slug]
ballot.save()
def set_metadata(self, ballot):
if not self.force_current_metadata:
if ballot.metadata and not self.force_update:
return
ee_data = self.ee_helper.get_data(ballot.ballot_paper_id)
if ee_data:
ballot.metadata = ee_data["metadata"]
def set_organisation_type(self, ballot):
if ballot.post.organization_type and not self.force_update:
return
ee_data = self.ee_helper.get_data(ballot.ballot_paper_id)
if ee_data:
ballot.post.organization_type = ee_data["organisation"][
"organisation_type"]
ballot.post.save()
def set_division_type(self, ballot):
"""
Attempts to set the division_type field from EveryElection
"""
if ballot.post.division_type and not self.force_update:
return
ee_data = self.ee_helper.get_data(ballot.ballot_paper_id)
if not ee_data or not ee_data["division"]:
return
ballot.post.division_type = ee_data["division"].get("division_type")
# ensures the division_type is valid, or will raise a ValidationError
ballot.post.full_clean()
ballot.post.save()
def run_post_ballot_import_tasks(self):
self.attach_cancelled_ballot_info()
def get_replacement_ballot(self, ballot_id):
replacement_ballot = None
ee_data = self.ee_helper.get_data(ballot_id)
if ee_data:
replacement_ballot_id = ee_data["replaced_by"]
if replacement_ballot_id:
try:
replacement_ballot = PostElection.objects.get(
ballot_paper_id=replacement_ballot_id)
except PostElection.DoesNotExist:
pass
return replacement_ballot
def attach_cancelled_ballot_info(self):
# we need to do this as a post-process instead of in the manager
# because if we're going to link 2 PostElection objects together
# we need to be sure that both of them already exist in our DB
cancelled_ballots = PostElection.objects.filter(cancelled=True)
if self.current_only:
cancelled_ballots = cancelled_ballots.filter(
election__current=True)
for cb in cancelled_ballots:
cb.replaced_by = self.get_replacement_ballot(cb.ballot_paper_id)
# Always get metadata, even if we might have it already.
# This is because is self.force_update is False, it might not have
# been imported already
self.set_metadata(cb)
cb.save()
def get_metadata(self, ballot_id):
ee = EEHelper()
ee_data = ee.get_data(ballot_id)
if ee_data:
return ee_data["metadata"]
return None