netbox-docker/startup_scripts/010_groups.py

28 lines
657 B
Python

import sys
from users.models import AdminGroup, AdminUser
from startup_script_utils import load_yaml
groups = load_yaml("/opt/netbox/initializers/groups.yml")
if groups is None:
sys.exit()
for params in groups:
groupname=params['name']
group, created = AdminGroup.objects.get_or_create(
name=groupname
)
if created:
print("👥 Created group", groupname)
for username in params.get("users", []):
user = AdminUser.objects.get(username=username)
if user:
group.user_set.add(user)
print(" 👤 Assigned user %s to group %s" % (username, AdminGroup.name))
group.save()