Fix rights for finds and contextrecords

1 files changed, 9 insertions, 1 deletions

diff --git a/archaeological_finds/models.py b/archaeological_finds/models.py
index e4a247941..32b4334ff 100644
--- a/archaeological_finds/models.py
+++ b/archaeological_finds/models.py
@@ -22,7 +22,7 @@ import datetime
 from django.conf import settings
 from django.contrib.gis.db import models
 from django.core.urlresolvers import reverse
-from django.db.models import Max
+from django.db.models import Max, Q
 from django.utils.translation import ugettext_lazy as _, ugettext
 
 from ishtar_common.models import GeneralType, ImageModel, BaseHistorizedItem, \
@@ -479,6 +479,14 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
             new.base_finds.add(base_find)
         return new
 
+    @classmethod
+    def get_query_owns(cls, user):
+        return Q(base_finds__context_record__operation__scientist=user.
+                 ishtaruser.person) |\
+            Q(base_finds__context_record__operation__in_charge=user.
+              ishtaruser.person) |\
+            Q(history_creator=user)
+
     def save(self, *args, **kwargs):
         super(Find, self).save(*args, **kwargs)
         q = self.base_finds