fix #3663: changed and updated sort keys

[wolnelektury.git] / src / catalogue / migrations / 0009_auto_20160127_1019.py

diff --git a/src/catalogue/migrations/0009_auto_20160127_1019.py b/src/catalogue/migrations/0009_auto_20160127_1019.py
new file mode 100644 (file)
index 0000000..eebb58b
--- /dev/null
+++ b/src/catalogue/migrations/0009_auto_20160127_1019.py
@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+from sortify import sortify
+
+
+def update_sort_keys(apps, schema_editor):
+    Tag = apps.get_model('catalogue', 'Tag')
+    Book = apps.get_model('catalogue', 'Book')
+    Picture = apps.get_model('picture', 'Picture')
+    ContentType = apps.get_model('contenttypes', 'ContentType')
+    for author in Tag.objects.filter(category='author'):
+        name_parts = author.name.split()
+        sort_key = ' '.join([name_parts[-1]] + name_parts[:-1])
+        author.sort_key = sortify(sort_key.lower())
+        author.save()
+    for model in Book, Picture:
+        ct = ContentType.objects.get_for_model(model)
+        for obj in model.objects.all():
+            authors = Tag.objects.filter(category='author', items__content_type=ct, items__object_id=obj.id)
+            author = authors[0]
+            obj.sort_key_author = author.sort_key
+            obj.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('catalogue', '0008_auto_20151221_1225'),
+        ('picture', '0007_auto_20160125_1709'),
+    ]
+
+    operations = [
+        migrations.RunPython(update_sort_keys),
+    ]