Add permissions for treatments sources

1 files changed, 36 insertions, 0 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index 6847029c7..d1b209134 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -579,6 +579,24 @@ class TreatmentSource(Source):
     class Meta:
         verbose_name = _(u"Treatment documentation")
         verbose_name_plural = _(u"Treament documentations")
+        permissions = (
+            ("view_treatmentsource",
+             ugettext(u"Can view all Treatment source")),
+            ("add_treatmentsource",
+             ugettext(u"Can add Treatment source")),
+            ("change_treatmentsource",
+             ugettext(u"Can change Treatment source")),
+            ("delete_treatmentsource",
+             ugettext(u"Can delete Treatment source")),
+            ("view_own_treatmentsource",
+             ugettext(u"Can view own Treatment source")),
+            ("add_own_treatmentsource",
+             ugettext(u"Can add own Treatment source")),
+            ("change_own_treatmentsource",
+             ugettext(u"Can change own Treatment source")),
+            ("delete_own_treatmentsource",
+             ugettext(u"Can delete own Treatment source")),
+        )
 
     @property
     def owner(self):
@@ -597,6 +615,24 @@ class TreatmentFileSource(Source):
     class Meta:
         verbose_name = _(u"Treatment request documentation")
         verbose_name_plural = _(u"Treatment request documentations")
+        permissions = (
+            ("view_filetreatmentsource",
+             ugettext(u"Can view all Treatment request source")),
+            ("add_filetreatmentsource",
+             ugettext(u"Can add Treatment request source")),
+            ("change_filetreatmentsource",
+             ugettext(u"Can change Treatment request source")),
+            ("delete_filetreatmentsource",
+             ugettext(u"Can delete Treatment request source")),
+            ("view_own_filetreatmentsource",
+             ugettext(u"Can view own Treatment request source")),
+            ("add_own_filetreatmentsource",
+             ugettext(u"Can add own Treatment request source")),
+            ("change_own_filetreatmentsource",
+             ugettext(u"Can change own Treatment request source")),
+            ("delete_own_filetreatmentsource",
+             ugettext(u"Can delete own Treatment request source")),
+        )
 
     @property
     def owner(self):