Container: import_get_location for parent container

1 files changed, 16 insertions, 1 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index ecf33bb87..b51f33176 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -33,7 +33,8 @@ from django.template.defaultfilters import slugify
 from ishtar_common.utils import ugettext_lazy as _, pgettext_lazy
 
 from django.apps import apps
-from ishtar_common.data_importer import post_importer_action
+from ishtar_common.data_importer import post_importer_action, \
+    pre_importer_action
 from ishtar_common.model_managers import ExternalIdManager, UUIDModelManager
 from ishtar_common.models import Document, GeneralType, get_external_id, \
     LightHistorizedItem, OwnPerms, Address, Person, post_save_cache, \
@@ -939,6 +940,20 @@ class Container(DocumentItem, Merge, LightHistorizedItem, QRCodeItem, GeoItem,
     def natural_key(self):
         return (self.uuid, )
 
+    @classmethod
+    @pre_importer_action
+    def import_get_location(cls, context, value):
+        if context.get("container_type", None) and context.get(
+                "reference", None):
+            try:
+                context["location"] = Warehouse.objects.get(external_id=value)
+                return
+            except Warehouse.DoesNotExist:
+                pass
+        for k in list(context.keys()):
+            if k != "import_get_location":
+                context.pop(k)
+
     def _generate_cached_label(self):
         return self.precise_location