Warehouse - model: add collection to container

2 files changed, 26 insertions, 0 deletions

diff --git a/archaeological_warehouse/migrations/0108_container_collection.py b/archaeological_warehouse/migrations/0108_container_collection.py
new file mode 100644
index 000000000..aa5958ddf
--- /dev/null
+++ b/archaeological_warehouse/migrations/0108_container_collection.py
@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.27 on 2021-01-21 16:05
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_warehouse', '0107_auto_20201222_1339'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='container',
+            name='collection',
+            field=models.ForeignKey(blank=True, help_text='Warehouse that own the container', null=True, on_delete=django.db.models.deletion.CASCADE, related_name='collections', to='archaeological_warehouse.Warehouse', verbose_name='Collection'),
+        ),
+    ]
diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index efbdd922d..8273619d9 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -851,6 +851,11 @@ class Container(DocumentItem, Merge, LightHistorizedItem,
         related_name='owned_containers', blank=True, null=True,
         help_text=_("Deprecated - do not use")
     )
+    collection = models.ForeignKey(
+        Warehouse, verbose_name=_("Collection"),
+        related_name='collections', blank=True, null=True,
+        help_text=_("Warehouse that own the container")
+    )
     container_type = models.ForeignKey(ContainerType,
                                        verbose_name=_("Container type"),
                                        related_name="containers")