OpenSlides/openslides/motions/migrations/0003_motion_comments.py
2017-09-12 14:58:43 +02:00

61 lines
2.0 KiB
Python

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.db import migrations
def change_motions_comments(apps, schema_editor):
"""
Index the comments fields in the config. Changing from an array to a dict with the
ids as keys. CHange all motions from an array for comments to a dict with the comments
field id as key to link motion comments and comments fields.
"""
# We get the model from the versioned app registry;
# if we directly import it, it will be the wrong version.
ConfigStore = apps.get_model('core', 'ConfigStore')
Motion = apps.get_model('motions', 'Motion')
try:
config_comments_fields = ConfigStore.objects.get(key='motions_comments').value
except ConfigStore.DoesNotExist:
config_comments_fields = [] # The old default: An empty list.
comments_fields = {}
for index, field in enumerate(config_comments_fields):
comments_fields[index+1] = field
max_index = len(config_comments_fields)-1
try:
db_value = ConfigStore.objects.get(key='motions_comments')
except ConfigStore.DoesNotExist:
db_value = ConfigStore(key='motions_comments')
db_value.value = comments_fields
# We cannot provide skip_autoupdate=True here, becuase this object is a fake object. It does *not*
# inherit from the RESTModelMixin, so the save() methos from base_model.py (django's default)
# gets called. This is because we are in the core app and try to save a core model. See
# comments in PR #3376.
db_value.save()
for motion in Motion.objects.all():
comments = {}
for index, comment in enumerate(motion.comments or []):
if index > max_index:
break
comments[index+1] = comment
motion.comments = comments
motion.save(skip_autoupdate=True)
class Migration(migrations.Migration):
dependencies = [
('motions', '0002_misc_features'),
]
operations = [
migrations.RunPython(
change_motions_comments
),
]