Manage treatment state and effective execution

1 files changed, 33 insertions, 0 deletions

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)
+    ]