mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Move membership evaluation to SecretRole

Browse Source
This commit is contained in:
Jeremy Stretch
2016-07-07 12:07:02 -04:00
parent c19124fcac
commit 7228801cb0
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/secrets/models.py
View File

@@ -182,6 +182,14 @@ class SecretRole(models.Model):
def get_absolute_url(self):
return "{}?role={}".format(reverse('secrets:secret_list'), self.slug)
def has_member(self, user):
"""
Check whether the given user has belongs to this SecretRole. Note that superusers belong to all roles.
"""
if user.is_superuser:
return True
return user in self.users.all() or user.groups.filter(pk__in=self.groups.all()).exists()
class Secret(CreatedUpdatedModel):
"""
@@ -304,6 +312,4 @@ class Secret(CreatedUpdatedModel):
"""
Check whether the given user has permission to decrypt this Secret.
"""
if user.is_superuser:
return True
return user in self.role.users.all() or user.groups.filter(pk__in=self.role.groups.all()).exists()
return self.role.has_member(user)