aboutsummaryrefslogtreecommitdiffstats
path: root/pydis_site/apps/api/migrations
diff options
context:
space:
mode:
authorGravatar Leon Sandøy <[email protected]>2019-10-12 00:07:45 +0200
committerGravatar GitHub <[email protected]>2019-10-12 00:07:45 +0200
commitf357084f15d27b2dea141eccdb86cde61c495ce6 (patch)
treed4a7cef40dcefcb4f3873e159127ee1b310af7d1 /pydis_site/apps/api/migrations
parentUpdate favicons (rounded corners, other improvements) (diff)
parentMerge pull request #280 from python-discord/fix-home-responsive (diff)
Merge branch 'master' into favicons
Diffstat (limited to 'pydis_site/apps/api/migrations')
-rw-r--r--pydis_site/apps/api/migrations/0044_migrate_nominations_from_infraction_to_nomination_model.py64
-rw-r--r--pydis_site/apps/api/migrations/0045_add_plural_name_for_log_entry.py17
2 files changed, 81 insertions, 0 deletions
diff --git a/pydis_site/apps/api/migrations/0044_migrate_nominations_from_infraction_to_nomination_model.py b/pydis_site/apps/api/migrations/0044_migrate_nominations_from_infraction_to_nomination_model.py
new file mode 100644
index 00000000..a56450c0
--- /dev/null
+++ b/pydis_site/apps/api/migrations/0044_migrate_nominations_from_infraction_to_nomination_model.py
@@ -0,0 +1,64 @@
+# Generated by Django 2.2.5 on 2019-09-30 12:15
+import logging
+
+from django.db import migrations
+from django.db.models import Q
+
+log = logging.getLogger('nomination_migration')
+
+
+def migrate_nominations_to_new_model(apps, schema_editor):
+ """
+ Migrations nominations from the infraction model to the nomination model.
+
+ This migration works by replaying the nomination history in chronological order, adding and
+ ending nominations as we've recorded them.
+ """
+ Infraction = apps.get_model('api', 'Infraction')
+ Nomination = apps.get_model('api', 'Nomination')
+
+ all_nominations = (
+ Q(reason__startswith="Helper nomination:") | Q(reason__startswith="Unwatched (talent-pool):")
+ )
+
+ for infraction in Infraction.objects.filter(all_nominations).order_by('inserted_at'):
+ if infraction.reason.startswith("Helper nomination:"):
+ if Nomination.objects.filter(user=infraction.user, active=True).exists():
+ log.error(
+ f"User `{infraction.user.id}` already has an active nomination, aborting."
+ )
+ continue
+ nomination = Nomination(
+ user=infraction.user,
+ inserted_at=infraction.inserted_at,
+ reason=infraction.reason[19:], # Strip "Helper nomination: " prefix
+ actor=infraction.actor,
+ active=True,
+ )
+ nomination.save()
+ infraction.delete()
+ elif infraction.reason.startswith("Unwatched (talent-pool):"):
+ if not Nomination.objects.filter(user=infraction.user, active=True).exists():
+ log.error(
+ f"User `{infraction.user.id}` has no active nomination, can't end it!"
+ )
+ continue
+ nomination = Nomination.objects.get(user=infraction.user, active=True)
+ nomination.end_reason = infraction.reason[25:] # Strip "Unwatched (talent-pool):"
+ nomination.ended_at = infraction.inserted_at
+ nomination.active = False
+ nomination.save()
+ infraction.delete()
+ else:
+ log.error(f"I don't understand this infraction: {infraction}")
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('api', '0043_infraction_hidden_warnings_to_notes'),
+ ]
+
+ operations = [
+ migrations.RunPython(migrate_nominations_to_new_model),
+ ]
diff --git a/pydis_site/apps/api/migrations/0045_add_plural_name_for_log_entry.py b/pydis_site/apps/api/migrations/0045_add_plural_name_for_log_entry.py
new file mode 100644
index 00000000..6b9933d8
--- /dev/null
+++ b/pydis_site/apps/api/migrations/0045_add_plural_name_for_log_entry.py
@@ -0,0 +1,17 @@
+# Generated by Django 2.2.3 on 2019-10-11 17:48
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('api', '0044_migrate_nominations_from_infraction_to_nomination_model'),
+ ]
+
+ operations = [
+ migrations.AlterModelOptions(
+ name='logentry',
+ options={'verbose_name_plural': 'Log entries'},
+ ),
+ ]