28 lines
657 B
Python
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()
|