X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/c226012e4114e09fd37cd384a2aada77476815cd..deb806113c76384876cd4969dc20c1e76ddabfef:/src/social/migrations/0006_legacy_group.py

diff --git a/src/social/migrations/0006_legacy_group.py b/src/social/migrations/0006_legacy_group.py
new file mode 100644
index 000000000..f6881a322
--- /dev/null
+++ b/src/social/migrations/0006_legacy_group.py
@@ -0,0 +1,26 @@
+from django.db import migrations
+
+
+def legacy_group(apps, schema_editor):
+    Cite = apps.get_model('social', 'Cite')
+    BannerGroup = apps.get_model('social', 'BannerGroup')
+
+    traditional = BannerGroup.objects.create(name='Tradycyjne cytaty')
+    banners = BannerGroup.objects.create(name='Bannery')
+
+    Cite.objects.exclude(book=None).update(group=traditional)
+    Cite.objects.filter(book=None).update(group=banners)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('social', '0005_auto_20190318_1309'),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            legacy_group,
+            migrations.RunPython.noop
+        )
+    ]