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

update object permission conditionals

This commit is contained in:
ryanmerolle 2021-04-25 11:15:46 -04:00
parent a1e0c52825
commit 4e31218ce6
1 changed files with 24 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
startup_scripts/015_object_permissions.py
View File

@ -10,39 +10,41 @@ if object_permissions is None:
sys.exit()
for params in object_permissions:
for permission_name, permission_details in object_permissions.items():
object_permission, created = ObjectPermission.objects.get_or_create(
name=params["name"],
description=params["description"],
enabled=params["enabled"],
actions=params["actions"],
name=permission_name,
description=permission_details["description"],
enabled=permission_details["enabled"],
actions=permission_details["actions"],
)
# Need to try to pass a list of model_name and app_label for more than the current ALL
# object_types = ContentType.objects.filter(app_label__in=params.pop("object_types"))
# object_permission.object_types.set(ContentType.objects.filter(app_label__in=params.pop("object_types")))
# object_types = ContentType.objects.filter(app_label__in=permission_details["object_types"])
# object_permission.object_types.set(ContentType.objects.filter(app_label__in=permission_details"object_types"]))
object_permission.object_types.set(ContentType.objects.all())
object_permission.save()
print("🔓 Created object permission", object_permission.name)
if permission_details.get("groups", 0):
for groupname in permission_details["groups"]:
group = AdminGroup.objects.get(name=groupname)
for groupname in params.get("groups", []):
group = AdminGroup.objects.get(name=groupname)
if group:
object_permission.groups.add(group)
print(
" 👥 Assigned group %s object permission of %s" % (groupname, object_permission.name)
)
if group:
object_permission.groups.add(group)
print(
" 👥 Assigned group %s object permission of %s" % (groupname, object_permission.name)
)
if permission_details.get("users", 0):
for username in permission_details["users"]:
user = AdminUser.objects.get(username=username)
for username in params.get("users", []):
user = AdminUser.objects.get(username=username)
if user:
object_permission.users.add(user)
print(
" 👤 Assigned user %s object permission of %s" % (username, object_permission.name)
)
if user:
object_permission.users.add(user)
print(
" 👤 Assigned user %s object permission of %s" % (username, object_permission.name)
)
object_permission.save()