OpenSlides/server/openslides/motions/migrations/0034_voting_2.py
Finn Stutzenstein f9d19db9e2
Update black
2021-04-26 09:00:04 +02:00

109 lines
3.4 KiB
Python

# Generated by Finn Stutzenstein on 2019-10-30 13:43
from django.db import migrations
def change_pollmethods(apps, schema_editor):
"""yn->YN, yna->YNA"""
MotionPoll = apps.get_model("motions", "MotionPoll")
pollmethod_map = {
"yn": "YN",
"yna": "YNA",
}
for poll in MotionPoll.objects.all():
poll.pollmethod = pollmethod_map.get(poll.pollmethod, "YNA")
poll.save(skip_autoupdate=True)
def set_poll_titles(apps, schema_editor):
"""
Sets titles to their indexes
"""
Motion = apps.get_model("motions", "Motion")
for motion in Motion.objects.all():
for i, poll in enumerate(motion.polls.order_by("pk").all()):
poll.title = str(i + 1)
poll.save(skip_autoupdate=True)
def set_onehunderd_percent_bases(apps, schema_editor):
MotionPoll = apps.get_model("motions", "MotionPoll")
ConfigStore = apps.get_model("core", "ConfigStore")
base_map = {
"YES_NO_ABSTAIN": "YNA",
"YES_NO": "YN",
"VALID": "valid",
"CAST": "cast",
"DISABLED": "disabled",
}
try:
config = ConfigStore.objects.get(key="motions_poll_100_percent_base")
value = base_map[config.value]
except (ConfigStore.DoesNotExist, KeyError):
value = "YNA"
for poll in MotionPoll.objects.all():
# The pollmethod is new (default is YNA), so we do not need
# to check, if the 100% base is valid.
poll.onehundred_percent_base = value
poll.save(skip_autoupdate=True)
def set_majority_methods(apps, schema_editor):
MotionPoll = apps.get_model("motions", "MotionPoll")
ConfigStore = apps.get_model("core", "ConfigStore")
majority_map = {
"simple_majority": "simple",
"two-thirds_majority": "two_thirds",
"three-quarters_majority": "three_quarters",
"disabled": "disabled",
}
try:
config = ConfigStore.objects.get(key="motions_poll_default_majority_method")
value = majority_map[config.value]
except (ConfigStore.DoesNotExist, KeyError):
value = "simple"
for poll in MotionPoll.objects.all():
poll.majority_method = value
poll.save(skip_autoupdate=True)
def convert_votes(apps, schema_editor):
MotionVote = apps.get_model("motions", "MotionVote")
value_map = {
"Yes": "Y",
"No": "N",
"Abstain": "A",
}
for vote in MotionVote.objects.all():
if vote.value not in value_map.values():
vote.value = value_map[vote.value]
vote.save(skip_autoupdate=True)
def set_correct_state(apps, schema_editor):
"""If there are votes, set the state to finished"""
MotionPoll = apps.get_model("motions", "MotionPoll")
MotionVote = apps.get_model("motions", "MotionVote")
for poll in MotionPoll.objects.all():
if MotionVote.objects.filter(option__poll__pk=poll.pk).exists():
poll.state = 3 # finished
poll.save(skip_autoupdate=True)
class Migration(migrations.Migration):
dependencies = [
("motions", "0033_voting_1"),
]
operations = [
migrations.RunPython(change_pollmethods),
migrations.RunPython(set_poll_titles),
migrations.RunPython(set_onehunderd_percent_bases),
migrations.RunPython(set_majority_methods),
migrations.RunPython(convert_votes),
migrations.RunPython(set_correct_state),
]