Warehouse: link warehouse to an organization - manage address dependencies

1 files changed, 32 insertions, 1 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 97898e5c1..3dd3f08e8 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -31,7 +31,7 @@ from ishtar_common.data_importer import post_importer_action
 from ishtar_common.model_managers import ExternalIdManager
 from ishtar_common.models import Document, GeneralType, get_external_id, \
     LightHistorizedItem, OwnPerms, Address, Person, post_save_cache, \
-    DashboardFormItem, ShortMenuItem, \
+    DashboardFormItem, ShortMenuItem, Organization, OrganizationType, \
     document_attached_changed, SearchAltName, DynamicRequest, GeoItem, \
     QRCodeItem, SearchVectorConfig, DocumentItem
 from ishtar_common.model_merging import merge_model_objects
@@ -89,6 +89,9 @@ class Warehouse(Address, DocumentItem, GeoItem, QRCodeItem, DashboardFormItem,
     person_in_charge = models.ForeignKey(
         Person, on_delete=models.SET_NULL, related_name='warehouse_in_charge',
         verbose_name=_(u"Person in charge"), null=True, blank=True)
+    organization = models.ForeignKey(
+        Organization, blank=True, null=True, related_name='warehouses',
+        verbose_name=_("Organization"), on_delete=models.SET_NULL)
     comment = models.TextField(_(u"Comment"), null=True, blank=True)
     associated_divisions = models.ManyToManyField(
         'WarehouseDivision', verbose_name=_("Divisions"), blank=True,
@@ -104,6 +107,7 @@ class Warehouse(Address, DocumentItem, GeoItem, QRCodeItem, DashboardFormItem,
     external_id = models.TextField(_(u"External ID"), blank=True, null=True)
     auto_external_id = models.BooleanField(
         _(u"External ID is set automatically"), default=False)
+    SUB_ADDRESSES = ["organization", "person_in_charge"]
 
     class Meta:
         verbose_name = _(u"Warehouse")
@@ -125,6 +129,33 @@ class Warehouse(Address, DocumentItem, GeoItem, QRCodeItem, DashboardFormItem,
     def _get_base_image_path(self):
         return u"{}/{}".format(self.SLUG, self.external_id)
 
+    def create_attached_organization(self):
+        """
+        Create an attached organization from warehouse fields
+        """
+        dct_orga = {}
+        for k in Address.FIELDS:
+            dct_orga[k] = getattr(self, k)
+
+        q = OrganizationType.objects.filter(txt_idx="warehouse")
+        if q.count():
+            orga_type = q.all()[0]
+        else:
+            orga_type, __ = OrganizationType.objects.get_or_create(
+                txt_idx="undefined",
+                defaults={"label": _("Undefined")}
+            )
+        dct_orga["organization_type"] = orga_type
+        dct_orga["name"] = self.name
+        orga = Organization.objects.create(**dct_orga)
+        self.organization = orga
+        for k in Address.FIELDS:
+            if k == "alt_address_is_prefered":
+                setattr(self, k, False)
+            else:
+                setattr(self, k, None)
+        self.save()
+
     @property
     def location_types(self):
         return [