OpenSlides/openslides/motions/migrations/0002_misc_features.py

# -*- coding: utf-8 -*-
# Generated by Django 1.10.4 on 2016-12-11 21:23
from __future__ import unicode_literals

import django.db.models.deletion
import jsonfield.fields
from django.conf import settings
from django.db import migrations, models

import openslides.utils.models


def change_label_of_state(apps, schema_editor):
    """
    Changes the label of former state "commited a bill" to "refered to committee".
    """
    # We get the model from the versioned app registry;
    # if we directly import it, it will be the wrong version.
    State = apps.get_model('motions', 'State')

    try:
        state = State.objects.get(name='commited a bill')
    except State.DoesNotExist:
        # State does not exists, there is nothing to change.
        pass
    else:
        state.name = 'refered to committee'
        state.action_word = 'Refer to committee'
        state.save(skip_autoupdate=True)


def add_recommendation_labels(apps, schema_editor):
    """
    Adds recommendation labels to some of the built-in states.
    """
    # We get the model from the versioned app registry;
    # if we directly import it, it will be the wrong version.
    State = apps.get_model('motions', 'State')

    name_label_map = {
        'accepted': 'Acceptance',
        'rejected': 'Rejection',
        'not decided': 'No decision',
        'permitted': 'Permission',
        'adjourned': 'Adjournment',
        'not concerned': 'No concernment',
        'refered to committee': 'Referral to committee',
        'rejected (not authorized)': 'Rejection (not authorized)',
    }
    for state in State.objects.all():
        if name_label_map.get(state.name):
            state.recommendation_label = name_label_map[state.name]
            state.save(skip_autoupdate=True)


class Migration(migrations.Migration):

    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
        ('motions', '0001_initial'),
    ]

    operations = [
        migrations.CreateModel(
            name='MotionBlock',
            fields=[
                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
                ('title', models.CharField(max_length=255)),
            ],
            options={
                'default_permissions': (),
            },
            bases=(openslides.utils.models.RESTModelMixin, models.Model),
        ),
        migrations.CreateModel(
            name='MotionChangeRecommendation',
            fields=[
                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
                ('rejected', models.BooleanField(default=False)),
                ('type', models.PositiveIntegerField(default=0)),
                ('line_from', models.PositiveIntegerField()),
                ('line_to', models.PositiveIntegerField()),
                ('text', models.TextField(blank=True)),
                ('creation_time', models.DateTimeField(auto_now=True)),
                ('author', models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL)),
                ('motion_version', models.ForeignKey(
                    on_delete=django.db.models.deletion.CASCADE, related_name='change_recommendations', to='motions.MotionVersion')),
            ],
            options={
                'default_permissions': (),
            },
            bases=(openslides.utils.models.RESTModelMixin, models.Model),
        ),
        migrations.AlterModelOptions(
            name='motion',
            options={
                'default_permissions': (),
                'ordering': (
                    'identifier',
                ),
                'permissions': (
                    ('can_see', 'Can see motions'),
                    ('can_create', 'Can create motions'),
                    ('can_support', 'Can support motions'),
                    ('can_see_and_manage_comments', 'Can see and manage comments'),
                    ('can_manage', 'Can manage motions')
                ),
                'verbose_name': 'Motion',
            },
        ),
        migrations.AddField(
            model_name='motion',
            name='comments',
            field=jsonfield.fields.JSONField(null=True),
        ),
        migrations.AddField(
            model_name='motion',
            name='origin',
            field=models.CharField(blank=True, max_length=255),
        ),
        migrations.AddField(
            model_name='motion',
            name='recommendation',
            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='+', to='motions.State'),
        ),
        migrations.AddField(
            model_name='state',
            name='recommendation_label',
            field=models.CharField(max_length=255, null=True),
        ),
        migrations.AddField(
            model_name='state',
            name='show_recommendation_extension_field',
            field=models.BooleanField(default=False),
        ),
        migrations.AddField(
            model_name='state',
            name='show_state_extension_field',
            field=models.BooleanField(default=False),
        ),
        migrations.AlterField(
            model_name='motion',
            name='state',
            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='+', to='motions.State'),
        ),
        migrations.AddField(
            model_name='motion',
            name='motion_block',
            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='motions.MotionBlock'),
        ),
        migrations.RunPython(
            change_label_of_state
        ),
        migrations.RunPython(
            add_recommendation_labels
        ),
    ]