More automation for YouTube: volume management (WiP).

[audio.git] / src / archive / migrations / 0012_url_to_slug.py

diff --git a/src/archive/migrations/0012_url_to_slug.py b/src/archive/migrations/0012_url_to_slug.py
new file mode 100644 (file)
index 0000000..d880c95
--- /dev/null
+++ b/src/archive/migrations/0012_url_to_slug.py
@@ -0,0 +1,30 @@
+# Generated by Django 3.0.6 on 2020-05-29 11:57
+
+from django.db import migrations
+
+
+def url_to_slug(apps, schema_editor):
+    Audiobook = apps.get_model('archive', 'Audiobook')
+    for a in Audiobook.objects.all():
+        a.slug = a.url.rstrip().rstrip('/').rsplit('/', 1)[-1]
+        a.save()
+
+def slug_to_url(apps, schema_editor):
+    Audiobook = apps.get_model('archive', 'Audiobook')
+    for a in Audiobook.objects.all():
+        a.url = 'https://wolnelektury.pl/katalog/lektura/{}/'.format(a.slug)
+        a.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archive', '0011_audiobook_slug'),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            url_to_slug,
+            slug_to_url
+        )
+    ]