Mascali/netbox-docker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Permission handling in external function

Browse source 
This move the setting of user and group permissions in the startup
script to its own function.
This commit is contained in:
Tobias Genannt 2020-02-12 08:34:01 +01:00
parent 882f7bcaf2
commit 8d3bd48c7a
4 changed files with 28 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

20
startup_scripts/010_groups.py
View file

@ -1,7 +1,8 @@
from django.contrib.auth.models import Permission, Group, User
from startup_script_utils import load_yaml
import sys
from django.contrib.auth.models import Group, User
from startup_script_utils import load_yaml, set_permissions
groups = load_yaml('/opt/netbox/initializers/groups.yml')
if groups is None:
sys.exit()
@ -19,17 +20,4 @@ for groupname, group_details in groups.items():
user.groups.add(group)
yaml_permissions = group_details.get('permissions', [])
if yaml_permissions:
subject = group.permissions
subject.clear()
for yaml_permission in yaml_permissions:
if '*' in yaml_permission:
permission_filter = '^' + yaml_permission.replace('*','.*') + '$'
permissions = Permission.objects.filter(codename__iregex=permission_filter)
print(" ⚿ Granting", permissions.count(), "permissions matching '" + yaml_permission + "'")
else:
permissions = Permission.objects.filter(codename=yaml_permission)
print(" ⚿ Granting permission", yaml_permission)
for permission in permissions:
subject.add(permission)
set_permissions(group.permissions, yaml_permissions)