Serializers: warehouse module

1 files changed, 23 insertions, 1 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 31588cde2..336046a86 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -126,7 +126,7 @@ class Warehouse(Address, DocumentItem, GeoItem, QRCodeItem, DashboardFormItem,
     def __str__(self):
         return self.name
 
-    def naturel_key(self):
+    def natural_key(self):
         return (self.external_id, )
 
     def _get_base_image_path(self):
@@ -364,6 +364,12 @@ post_save.connect(post_save_cache, sender=WarehouseDivision)
 post_delete.connect(post_save_cache, sender=WarehouseDivision)
 
 
+class WarehouseDivisionLinkManager(models.Manager):
+    def get_by_natural_key(self, warehouse, division):
+        return self.get(warehouse__external_id=warehouse,
+                        division__txt_idx=division)
+
+
 class WarehouseDivisionLink(models.Model):
     RELATED_SET_NAME = "divisions"
     RELATED_ATTRS = ["order"]
@@ -371,6 +377,7 @@ class WarehouseDivisionLink(models.Model):
     warehouse = models.ForeignKey(Warehouse, related_name='divisions')
     division = models.ForeignKey(WarehouseDivision)
     order = models.IntegerField(_("Order"), default=10)
+    objects = WarehouseDivisionLinkManager()
 
     class Meta:
         ordering = ('warehouse', 'order')
@@ -379,6 +386,9 @@ class WarehouseDivisionLink(models.Model):
     def __str__(self):
         return u"{} - {}".format(self.warehouse, self.division)
 
+    def natural_key(self):
+        return self.warehouse.external_id, self.division.txt_idx
+
 
 class ContainerType(GeneralType):
     length = models.IntegerField(_(u"Length (mm)"), blank=True, null=True)
@@ -865,12 +875,20 @@ m2m_changed.connect(document_attached_changed,
                     sender=Container.documents.through)
 
 
+class ContainerLocalisationManager(models.Manager):
+    def get_by_natural_key(self, container, warehouse, division):
+        return self.get(container__external_id=container,
+                        division__warehouse__external_id=warehouse,
+                        division__division__txt_idx = division)
+
+
 class ContainerLocalisation(models.Model):
     container = models.ForeignKey(Container, verbose_name=_(u"Container"),
                                   related_name='division')
     division = models.ForeignKey(WarehouseDivisionLink,
                                  verbose_name=_(u"Division"))
     reference = models.CharField(_(u"Reference"), max_length=200, default='')
+    objects = ContainerLocalisationManager()
 
     class Meta:
         verbose_name = _(u"Container localisation")
@@ -883,6 +901,10 @@ class ContainerLocalisation(models.Model):
                            str(self.division), self.reference))
         return lbl
 
+    def natural_key(self):
+        return self.container.external_id, self.division.warehouse.external_id,\
+               self.division.division.txt_idx
+
     def save(self, *args, **kwargs):
         super(ContainerLocalisation, self).save(*args, **kwargs)
         self.container.skip_history_when_saving = True