Add Collections models (refs #3396)

1 files changed, 16 insertions, 0 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 69c7ce1d7..c6146e3a7 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -81,6 +81,22 @@ class Warehouse(Address, OwnPerms):
             cached_label_changed(Container, instance=container)
 
 
+class Collection(LightHistorizedItem):
+    name = models.CharField(_(u"Name"), max_length=200,
+                            null=True, blank=True)
+    description = models.TextField(_(u"Description"), null=True, blank=True)
+    warehouse = models.ForeignKey(Warehouse, verbose_name=_(u"Warehouse"),
+                                  related_name='collections')
+
+    class Meta:
+        verbose_name = _(u"Collection")
+        verbose_name_plural = _(u"Collection")
+        ordering = ('name',)
+
+    def __unicode__(self):
+        return self.name
+
+
 class WarehouseDivision(GeneralType):
     class Meta:
         verbose_name = _(u"Warehouse division")