Year summaries.

[wolnelektury.git] / src / club / migrations / 0023_completed_at.py

diff --git a/src/club/migrations/0023_completed_at.py b/src/club/migrations/0023_completed_at.py
new file mode 100644 (file)
index 0000000..d23c289
--- /dev/null
+++ b/src/club/migrations/0023_completed_at.py
@@ -0,0 +1,26 @@
+import json
+from django.db import migrations
+
+
+def populate_completed_at(apps, schema_editor):
+    PayUOrder = apps.get_model('club', 'PayUOrder')
+    for order in PayUOrder.objects.filter(status='COMPLETED'):
+        for n in order.notification_set.order_by('received_at'):
+            if json.loads(n.body)['order']['status'] == 'COMPLETED':
+                order.completed_at = n.received_at
+                order.save()
+                break
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('club', '0022_payuorder_completed_at'),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            populate_completed_at,
+            migrations.RunPython.noop,
+        )
+    ]