src/catalogue/migrations/0019_auto_20171221_1107.py

   1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
   3 #
   4 from django.db import migrations, models
   5
   6
   7 def init_tag_for_books_pictures(apps, schema_editor):
   8     Tag = apps.get_model('catalogue', 'Tag')
   9     db_alias = schema_editor.connection.alias
  10     tag_objects = Tag.objects.using(db_alias).exclude(category='set')
  11     tags_for_books = tag_objects.filter(items__content_type__model__in=('book', 'fragment')).distinct()
  12     tags_for_books.update(for_books=True)
  13     tags_for_pictures = tag_objects.filter(items__content_type__model__in=('picture', 'picturearea')).distinct()
  14     tags_for_pictures.update(for_pictures=True)
  15
  16
  17 class Migration(migrations.Migration):
  18
  19     dependencies = [
  20         ('catalogue', '0018_auto_20171221_1106'),
  21     ]
  22
  23     operations = [
  24         migrations.AddField(
  25             model_name='tag',
  26             name='for_books',
  27             field=models.BooleanField(default=False),
  28         ),
  29         migrations.AddField(
  30             model_name='tag',
  31             name='for_pictures',
  32             field=models.BooleanField(default=False),
  33         ),
  34         migrations.RunPython(init_tag_for_books_pictures, migrations.RunPython.noop)
  35     ]