146 lines
5.0 KiB
Python
146 lines
5.0 KiB
Python
# Generated by Finn Stutzenstein on 2019-10-29 10:55
|
|
|
|
from decimal import Decimal
|
|
|
|
from django.db import migrations, transaction
|
|
|
|
|
|
def change_pollmethods(apps, schema_editor):
|
|
""" yn->YN, yna->YNA """
|
|
AssignmentPoll = apps.get_model("assignments", "AssignmentPoll")
|
|
pollmethod_map = {
|
|
"yn": "YN",
|
|
"yna": "YNA",
|
|
"votes": "votes",
|
|
}
|
|
for poll in AssignmentPoll.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
|
|
"""
|
|
Assignment = apps.get_model("assignments", "Assignment")
|
|
for assignment in Assignment.objects.all():
|
|
for i, poll in enumerate(assignment.polls.order_by("pk").all()):
|
|
poll.title = str(i + 1)
|
|
poll.save(skip_autoupdate=True)
|
|
|
|
|
|
def set_onehunderd_percent_bases(apps, schema_editor):
|
|
AssignmentPoll = apps.get_model("assignments", "AssignmentPoll")
|
|
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="assignments_poll_100_percent_base")
|
|
value = base_map[config.value]
|
|
except (ConfigStore.DoesNotExist, KeyError):
|
|
value = "YNA"
|
|
|
|
for poll in AssignmentPoll.objects.all():
|
|
if poll.pollmethod == "votes" and value in ("YN", "YNA"):
|
|
poll.onehundred_percent_base = "votes"
|
|
elif poll.pollmethod == "YN" and value == "YNA":
|
|
poll.onehundred_percent_base = "YN"
|
|
else:
|
|
poll.onehundred_percent_base = value
|
|
poll.save(skip_autoupdate=True)
|
|
|
|
|
|
def set_majority_methods(apps, schema_editor):
|
|
AssignmentPoll = apps.get_model("assignments", "AssignmentPoll")
|
|
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="assignments_poll_default_majority_method")
|
|
value = majority_map[config.value]
|
|
except (ConfigStore.DoesNotExist, KeyError):
|
|
value = "simple"
|
|
|
|
for poll in AssignmentPoll.objects.all():
|
|
poll.majority_method = value
|
|
poll.save(skip_autoupdate=True)
|
|
|
|
|
|
def convert_votes(apps, schema_editor):
|
|
AssignmentVote = apps.get_model("assignments", "AssignmentVote")
|
|
value_map = {
|
|
"Yes": "Y",
|
|
"No": "N",
|
|
"Abstain": "A",
|
|
"Votes": "Y",
|
|
}
|
|
for vote in AssignmentVote.objects.all():
|
|
vote.value = value_map[vote.value]
|
|
vote.save(skip_autoupdate=True)
|
|
|
|
|
|
def convert_votesabstain(apps, schema_editor):
|
|
AssignmentPoll = apps.get_model("assignments", "AssignmentPoll")
|
|
AssignmentVote = apps.get_model("assignments", "AssignmentVote")
|
|
for poll in AssignmentPoll.objects.all():
|
|
if poll.votesabstain is not None and poll.votesabstain > Decimal(0):
|
|
with transaction.atomic():
|
|
option = poll.options.first()
|
|
vote = AssignmentVote(
|
|
option=option, value="A", weight=poll.votesabstain
|
|
)
|
|
vote.save(skip_autoupdate=True)
|
|
|
|
|
|
def convert_votesno(apps, schema_editor):
|
|
AssignmentPoll = apps.get_model("assignments", "AssignmentPoll")
|
|
AssignmentVote = apps.get_model("assignments", "AssignmentVote")
|
|
for poll in AssignmentPoll.objects.all():
|
|
if poll.votesno is not None and poll.votesno > Decimal(0):
|
|
with transaction.atomic():
|
|
option = poll.options.first()
|
|
vote = AssignmentVote(option=option, value="N", weight=poll.votesno)
|
|
vote.save(skip_autoupdate=True)
|
|
|
|
|
|
def set_correct_state(apps, schema_editor):
|
|
""" if poll.published, set state to published """
|
|
AssignmentPoll = apps.get_model("assignments", "AssignmentPoll")
|
|
AssignmentVote = apps.get_model("assignments", "AssignmentVote")
|
|
for poll in AssignmentPoll.objects.all():
|
|
# Polls, that are published (old field) but have no votes, will be
|
|
# left at the created state...
|
|
if AssignmentVote.objects.filter(option__poll__pk=poll.pk).exists():
|
|
if poll.published:
|
|
poll.state = 4 # published
|
|
else:
|
|
poll.state = 3 # finished
|
|
poll.save(skip_autoupdate=True)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
("assignments", "0008_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(convert_votesabstain),
|
|
migrations.RunPython(convert_votesno),
|
|
migrations.RunPython(set_correct_state),
|
|
]
|