# -*- coding: utf-8 -*-
# Generated by Django 1.11.10 on 2018-12-11 15:09
from __future__ import unicode_literals

from django.db import migrations, models


def migrate_treatment_states(apps, schema):
    TreatmentState = apps.get_model('archaeological_finds', 'TreatmentState')
    Treatment = apps.get_model('archaeological_finds', 'Treatment')
    for finished in ['completed', 'finished']:
        q = TreatmentState.objects.filter(txt_idx=finished)
        if q.count():
            t = q.all()[0]
            t.executed = True
            t.save()
            break
    for t in TreatmentState.objects.all():
        t.txt_idx = t.txt_idx.replace("_", "-")
        t.save()
    orders = [(10, "planned"), (20, "to-be-confirmed"), (30, "in-progress"),
              (40, finished), (50, "cancelled"), (60, "unknown"),]
    for order, txt_idx in orders:
        q = TreatmentState.objects.filter(txt_idx=txt_idx)
        if not q.count():
            continue
        t = q.all()[0]
        t.order = order
        t.save()
    if TreatmentState.objects.filter(txt_idx=finished).count():
        f = TreatmentState.objects.get(txt_idx=finished)
        for t in Treatment.objects.filter(treatment_state__isnull=True):
            t.treatment_state = f
            t.save()


class Migration(migrations.Migration):

    dependencies = [
        ('archaeological_finds', '0049_auto_20181210_1518'),
    ]

    operations = [
        migrations.AlterModelOptions(
            name='treatmentstate',
            options={'ordering': ('order', 'label'), 'verbose_name': "Type d'\xe9tat de traitement", 'verbose_name_plural': "Types d'\xe9tat de traitement"},
        ),
        migrations.AddField(
            model_name='treatmentstate',
            name='executed',
            field=models.BooleanField(default=False, verbose_name='Treatment is executed'),
        ),
        migrations.AddField(
            model_name='treatmentstate',
            name='order',
            field=models.IntegerField(default=10, verbose_name='Ordre'),
        ),
        migrations.RunPython(migrate_treatment_states)
    ]