🗃️ database: add associated basket to treatments - is exhibition for treatments types

2 files changed, 84 insertions, 0 deletions

diff --git a/archaeological_finds/migrations/0131_ishtar_users.py b/archaeological_finds/migrations/0131_ishtar_users.py
index 8d8772343..4b6baa73c 100644
--- a/archaeological_finds/migrations/0131_ishtar_users.py
+++ b/archaeological_finds/migrations/0131_ishtar_users.py
@@ -57,4 +57,29 @@ class Migration(migrations.Migration):
             name='thumbnail',
             field=models.ImageField(blank=True, help_text=ishtar_common.utils.max_size_help, max_length=255, null=True, upload_to=ishtar_common.utils.get_image_path),
         ),
+        migrations.AlterField(
+            model_name='basefind',
+            name='ishtar_users',
+            field=models.ManyToManyField(blank=True, related_name='basefind_associated', to='ishtar_common.IshtarUser'),
+        ),
+        migrations.AlterField(
+            model_name='find',
+            name='ishtar_users',
+            field=models.ManyToManyField(blank=True, related_name='find_associated', to='ishtar_common.IshtarUser'),
+        ),
+        migrations.AlterField(
+            model_name='property',
+            name='ishtar_users',
+            field=models.ManyToManyField(blank=True, related_name='property_associated', to='ishtar_common.IshtarUser'),
+        ),
+        migrations.AlterField(
+            model_name='treatment',
+            name='ishtar_users',
+            field=models.ManyToManyField(blank=True, related_name='treatment_associated', to='ishtar_common.IshtarUser'),
+        ),
+        migrations.AlterField(
+            model_name='treatmentfile',
+            name='ishtar_users',
+            field=models.ManyToManyField(blank=True, related_name='treatmentfile_associated', to='ishtar_common.IshtarUser'),
+        ),
     ]
diff --git a/archaeological_finds/migrations/0132_exhibitions_basket_for_treatment.py b/archaeological_finds/migrations/0132_exhibitions_basket_for_treatment.py
new file mode 100644
index 000000000..874a01744
--- /dev/null
+++ b/archaeological_finds/migrations/0132_exhibitions_basket_for_treatment.py
@@ -0,0 +1,59 @@
+# Generated by Django 2.2.24 on 2024-11-19 17:49
+
+from django.db import migrations, models
+
+
+MAJ_SQL1 = "UPDATE archaeological_finds_treatmentfile SET basket_id=associated_basket_id;"
+MAJ_SQL2 = "UPDATE archaeological_finds_historicaltreatmentfile SET basket_id=associated_basket_id;"
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_finds', '0131_ishtar_users'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='basefind',
+            options={'verbose_name': 'Base find', 'verbose_name_plural': 'Base finds'},
+        ),
+        migrations.AddField(
+            model_name='historicaltreatmentfile',
+            name='basket_id',
+            field=models.PositiveIntegerField(blank=True, null=True, verbose_name='Basket ID'),
+        ),
+        migrations.AddField(
+            model_name='treatmentfile',
+            name='basket_id',
+            field=models.PositiveIntegerField(blank=True, null=True, verbose_name='Basket ID'),
+        ),
+        migrations.RunSQL(MAJ_SQL1),
+        migrations.RunSQL(MAJ_SQL2),
+        migrations.RemoveField(
+            model_name='historicaltreatmentfile',
+            name='associated_basket',
+        ),
+        migrations.RemoveField(
+            model_name='treatmentfile',
+            name='associated_basket',
+        ),
+        migrations.RenameField(model_name="treatmentfile", old_name="basket_id", new_name="associated_basket_id"),
+        migrations.RenameField(model_name="historicaltreatmentfile", old_name="basket_id", new_name="associated_basket_id"),
+
+        migrations.AddField(
+            model_name='treatment',
+            name='associated_basket_id',
+            field=models.PositiveIntegerField(blank=True, null=True, verbose_name='Basket ID'),
+        ),
+        migrations.AddField(
+            model_name='historicaltreatment',
+            name='associated_basket_id',
+            field=models.PositiveIntegerField(blank=True, null=True, verbose_name='Basket ID'),
+        ),
+        migrations.AddField(
+            model_name='treatmentfiletype',
+            name='is_exhibition',
+            field=models.BooleanField(default=False, verbose_name='Is an exhibition'),
+        ),
+    ]