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()