26 lines
645 B
Python
26 lines
645 B
Python
import sys
|
|
|
|
from startup_script_utils import load_yaml
|
|
from users.models import AdminGroup, AdminUser
|
|
|
|
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()
|