X-Git-Url: https://git.mdrn.pl/audio.git/blobdiff_plain/576253ba3c8b54dd509de0dbf7373b310fa9b71c..1c72d045af4d7c5607a872a678eadec17498bd6d:/src/youtube/migrations/0016_populate_youtube_relations.py

diff --git a/src/youtube/migrations/0016_populate_youtube_relations.py b/src/youtube/migrations/0016_populate_youtube_relations.py
new file mode 100644
index 0000000..2ea388a
--- /dev/null
+++ b/src/youtube/migrations/0016_populate_youtube_relations.py
@@ -0,0 +1,32 @@
+# Generated by Django 3.1.2 on 2020-10-22 14:15
+
+from django.db import migrations
+
+
+def populate_youtube_relations(apps, schema_editor):
+    YouTube = apps.get_model('youtube', 'YouTube')
+    ThumbnailTemplate = apps.get_model('youtube', 'ThumbnailTemplate')
+    Project = apps.get_model('archive', 'Project')
+
+    try:
+        yt = YouTube.objects.first()
+    except YouTube.DoesNotExist:
+        yt = YouTube.objects.create(name='default')
+
+    Project.objects.filter(youtube=None).update(youtube=yt)
+    ThumbnailTemplate.objects.filter(youtube=None).update(youtube=yt)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('youtube', '0015_auto_20201022_1414'),
+        ('archive', '0019_auto_20201022_1414'),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            populate_youtube_relations,
+            migrations.RunPython.noop
+        )
+    ]