diff --git a/grandlyon/scripts/mark-admin-roles.py b/grandlyon/scripts/mark-admin-roles.py index 66afc58..6bbcf23 100644 --- a/grandlyon/scripts/mark-admin-roles.py +++ b/grandlyon/scripts/mark-admin-roles.py @@ -8,13 +8,11 @@ import sys from django.db.models.query import Q -from django_rbac.utils import get_role_model, get_ou_model +from authentic2.a2_rbac.models import OrganizationalUnit, Role -Ou = get_ou_model() -Role = get_role_model() name_dst = sys.argv[1] -ou = Ou.objects.get(name=name_dst) +ou = OrganizationalUnit.objects.get(name=name_dst) role = Role.objects.get(name=u'Administrateur %s' % name_dst, ou=ou) for subrole in Role.objects.filter(Q(slug='_a2-hobo-superuser', ou=ou) |