Add divisions to warehouses

1 files changed, 15 insertions, 1 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 71b5bf6d7..579c448ae 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -35,7 +35,7 @@ post_delete.connect(post_save_cache, sender=WarehouseType)
 
 
 class Warehouse(Address, OwnPerms):
-    name = models.CharField(_(u"Name"), max_length=40)
+    name = models.CharField(_(u"Name"), max_length=200)
     warehouse_type = models.ForeignKey(WarehouseType,
                                        verbose_name=_(u"Warehouse type"))
     person_in_charge = models.ForeignKey(
@@ -60,6 +60,20 @@ class Warehouse(Address, OwnPerms):
         return u"%s (%s)" % (self.name, unicode(self.warehouse_type))
 
 
+class WarehouseDivision(models.Model):
+    name = models.CharField(_(u"Name"), max_length=200)
+    order = models.IntegerField(_(u"Order"), default=1)
+    warehouse = models.ForeignKey(Warehouse, verbose_name=_(u"Warehouse"),
+                                  related_name='divisions')
+
+    class Meta:
+        verbose_name = _(u"Warehouse division")
+        verbose_name_plural = _(u"Warehouse divisions")
+
+    def __unicode__(self):
+        return u"%s (%s)" % (self.name, unicode(self.warehouse))
+
+
 class ContainerType(GeneralType):
     length = models.IntegerField(_(u"Length (mm)"), blank=True, null=True)
     width = models.IntegerField(_(u"Width (mm)"), blank=True, null=True)