Manage container localisation

1 files changed, 18 insertions, 1 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 7d37bd664..57068f374 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -97,7 +97,7 @@ post_delete.connect(post_save_cache, sender=ContainerType)
 
 
 class Container(LightHistorizedItem):
-    TABLE_COLS = ['reference', 'container_type', 'location']
+    TABLE_COLS = ['reference', 'container_type__label', 'location__name']
 
     # search parameters
     EXTRA_REQUEST_KEYS = {
@@ -121,3 +121,20 @@ class Container(LightHistorizedItem):
         lbl = u" - ".join((self.reference, unicode(self.container_type),
                            unicode(self.location)))
         return lbl
+
+
+class ContainerLocalisation(models.Model):
+    container = models.ForeignKey(Container, verbose_name=_(u"Container"))
+    division = models.ForeignKey(WarehouseDivisionLink,
+                                 verbose_name=_(u"Division"))
+    reference = models.CharField(_(u"Reference"), max_length=200, default='')
+
+    class Meta:
+        verbose_name = _(u"Container localisation")
+        verbose_name_plural = _(u"Container localisations")
+        unique_together = ('container', 'division')
+
+    def __unicode__(self):
+        lbl = u" - ".join((unicode(self.container),
+                           unicode(self.division), self.reference))
+        return lbl