Manage treatment state and effective execution

3 files changed, 106 insertions, 0 deletions

diff --git a/archaeological_finds/migrations/0050_auto_20181211_1509.py b/archaeological_finds/migrations/0050_auto_20181211_1509.py
new file mode 100644
index 000000000..c9928d617
--- /dev/null
+++ b/archaeological_finds/migrations/0050_auto_20181211_1509.py
@@ -0,0 +1,51 @@
+# -*- 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')
+    q = TreatmentState.objects.filter(txt_idx="completed")
+    if q.count():
+        t = q.all()[0]
+        t.executed = True
+        t.save()
+    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, "completed"), (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()
+
+
+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)
+    ]
diff --git a/archaeological_finds/migrations/0051_auto_20181211_1530.py b/archaeological_finds/migrations/0051_auto_20181211_1530.py
new file mode 100644
index 000000000..9bcdb60a7
--- /dev/null
+++ b/archaeological_finds/migrations/0051_auto_20181211_1530.py
@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2018-12-11 15:30
+from __future__ import unicode_literals
+
+import archaeological_finds.models_treatments
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_finds', '0050_auto_20181211_1509'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='treatment',
+            name='treatment_state',
+            field=models.ForeignKey(default=archaeological_finds.models_treatments.TreatmentState.get_default, on_delete=django.db.models.deletion.CASCADE, to='archaeological_finds.TreatmentState', verbose_name='\xc9tat'),
+        ),
+    ]
diff --git a/archaeological_finds/migrations/0052_auto_20181211_1558.py b/archaeological_finds/migrations/0052_auto_20181211_1558.py
new file mode 100644
index 000000000..5c0f63d7c
--- /dev/null
+++ b/archaeological_finds/migrations/0052_auto_20181211_1558.py
@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2018-12-11 15:58
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+def migrate_treatment_states(apps, schema):
+    Treatment = apps.get_model('archaeological_finds', 'Treatment')
+    for t in Treatment.objects.all():
+        t.executed = True
+        t.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_finds', '0051_auto_20181211_1530'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='historicaltreatment',
+            name='executed',
+            field=models.BooleanField(default=False, verbose_name='Treatment have been executed'),
+        ),
+        migrations.AddField(
+            model_name='treatment',
+            name='executed',
+            field=models.BooleanField(default=False, verbose_name='Treatment have been executed'),
+        ),
+        migrations.RunPython(migrate_treatment_states)
+    ]