Add archaeological sites to models (refs #1586)

1 files changed, 22 insertions, 0 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index 59cb46589..814924a01 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -75,6 +75,26 @@ class Period(GeneralType) :
     def __unicode__(self):
         return self.label
 
+class ArchaeologicalSite(BaseHistorizedItem):
+    reference = models.CharField(_(u"Reference"), max_length=20)
+    name = models.CharField(_(u"Name"), max_length=200,
+                            null=True, blank=True)
+    class Meta:
+        verbose_name = _(u"Archaeological site")
+        verbose_name_plural = _(u"Archaeological sites")
+        permissions = (
+          ("view_archaeologicalsite",
+           ugettext(u"Can view all Archaeological site")),
+          ("view_own_archaeologicalsite",
+           ugettext(u"Can view own Archaeological site")),
+          ("add_own_archaeologicalsite",
+           ugettext(u"Can add own Archaeological site")),
+          ("change_own_archaeologicalsite",
+           ugettext(u"Can change own Archaeological site")),
+          ("delete_own_archaeologicalsite",
+           ugettext(u"Can delete own Archaeological site")),
+        )
+
 class Operation(BaseHistorizedItem, OwnPerms):
     TABLE_COLS = ['year_index', 'operation_type', 'remains', 'towns',
                   'associated_file_short_label', 'start_date',
@@ -137,6 +157,8 @@ class Operation(BaseHistorizedItem, OwnPerms):
     comment = models.TextField(_(u"Comment"), null=True, blank=True)
     cached_label = models.CharField(_(u"Cached name"), max_length=500,
                                     null=True, blank=True)
+    archaeological_sites = models.ManyToManyField(ArchaeologicalSite,
+                                        verbose_name=_(u"Archaeological sites"))
     history = HistoricalRecords()
 
     class Meta: