Warehouse: fix division form int and save

1 files changed, 10 insertions, 0 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index bf7ea1eb6..bbcdd93d7 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -106,6 +106,14 @@ class Warehouse(Address, DashboardFormItem, OwnPerms):
         return u"{}/{}".format(self.SLUG, slugify(self.name))
 
     @property
+    def location_types(self):
+        return [
+            wd.division.label
+            for wd in WarehouseDivisionLink.objects.filter(
+                warehouse=self).order_by('order').all()
+        ]
+
+    @property
     def associated_filename(self):
         return datetime.date.today().strftime('%Y-%m-%d') + '-' + \
             slugify(unicode(self))
@@ -254,6 +262,8 @@ post_delete.connect(post_save_cache, sender=WarehouseDivision)
 
 
 class WarehouseDivisionLink(models.Model):
+    RELATED_SET_NAME = "divisions"
+    RELATED_ATTRS = ["order"]
     RELATIVE_MODELS = {Warehouse: 'warehouse'}
     warehouse = models.ForeignKey(Warehouse, related_name='divisions')
     division = models.ForeignKey(WarehouseDivision)