diff options
Diffstat (limited to 'ishtar_common/models.py')
| -rw-r--r-- | ishtar_common/models.py | 25 | 
1 files changed, 15 insertions, 10 deletions
| diff --git a/ishtar_common/models.py b/ishtar_common/models.py index de494e863..a99921740 100644 --- a/ishtar_common/models.py +++ b/ishtar_common/models.py @@ -228,7 +228,7 @@ class OwnPerms(object):      Manage special permissions for object's owner      """      @classmethod -    def get_query_owns(cls, user): +    def get_query_owns(cls, ishtaruser):          """          Query object to get own items          """ @@ -238,7 +238,9 @@ class OwnPerms(object):          """          Check if the current object is owned by the user          """ -        query = self.get_query_owns(user) +        if not hasattr(user, 'ishtaruser'): +            return False +        query = self.get_query_owns(user.ishtaruser)          if not query:              return False          query &= Q(pk=self.pk) @@ -249,7 +251,9 @@ class OwnPerms(object):          """          Check if the user own some items          """ -        query = cls.get_query_owns(user) +        if not hasattr(user, 'ishtaruser'): +            return False +        query = cls.get_query_owns(user.ishtaruser)          if not query:              return False          return cls.objects.filter(query).count() @@ -301,7 +305,9 @@ class OwnPerms(object):          items = []          if hasattr(cls, 'BASKET_MODEL'):              items = list(cls.BASKET_MODEL.objects.filter(user=user).all()) -        query = cls.get_query_owns(user) +        if not hasattr(user, 'ishtaruser'): +            return False +        query = cls.get_query_owns(user.ishtaruser)          if not query and not replace_query:              returned = cls.objects.filter(pk__isnull=True)              if values: @@ -2974,16 +2980,15 @@ class Person(Address, Merge, OwnPerms, ValueGetter):                  fle.save()  # force update of raw_general_contractor      @classmethod -    def get_query_owns(cls, user): +    def get_query_owns(cls, ishtaruser):          return \              Q(operation_scientist_responsability__collaborators__ishtaruser -              =user.ishtaruser) | \ +              =ishtaruser) | \              Q(operation_scientist_responsability__scientist__ishtaruser -              =user.ishtaruser) | \ +              =ishtaruser) | \              Q(operation_collaborator__collaborators__ishtaruser -              =user.ishtaruser) | \ -            Q(operation_collaborator__scientist__ishtaruser -              =user.ishtaruser) +              =ishtaruser) | \ +            Q(operation_collaborator__scientist__ishtaruser=ishtaruser)  post_save.connect(cached_label_changed, sender=Person) | 
