mirror of
https://github.com/ansible/awx.git
synced 2026-01-11 01:57:35 -03:30
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
from collections import defaultdict
|
|
import os
|
|
import sys
|
|
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
from awx.main.fields import ImplicitRoleField
|
|
from awx.main.models.rbac import Role
|
|
|
|
|
|
role_id = int(os.environ.get('role'))
|
|
role = Role.objects.get(id=role_id)
|
|
|
|
all_roles = {role,}
|
|
graph = defaultdict(set)
|
|
|
|
|
|
# Role parents
|
|
new_parents = {role,}
|
|
while new_parents:
|
|
old_parents = new_parents
|
|
new_parents = set()
|
|
for r in old_parents:
|
|
new_parents |= (set(r.parents.all()) - all_roles)
|
|
for p in r.parents.all():
|
|
graph[p.id].add(r.id)
|
|
all_roles |= new_parents
|
|
|
|
# Role children
|
|
new_children = {role,}
|
|
while new_children:
|
|
old_children = new_children
|
|
new_children = set()
|
|
for r in old_children:
|
|
new_children |= (set(r.children.all()) - all_roles)
|
|
for c in r.children.all():
|
|
graph[r.id].add(c.id)
|
|
all_roles |= new_children
|
|
|
|
|
|
print("digraph G {")
|
|
|
|
for r in sorted(all_roles, key=lambda x: x.id):
|
|
if r.content_type is None:
|
|
print(f' {r.id} [shape=box,label="id={r.id}\lsingleton={r.singleton_name}\l"]')
|
|
else:
|
|
print(f' {r.id} [shape=box,label="id={r.id}\lct={r.content_type}\lobject_id={r.object_id}\lrole_field={r.role_field}\l"]')
|
|
|
|
print()
|
|
for p_id, children in sorted(graph.items()):
|
|
for c_id in children:
|
|
print(f" {p_id} -> {c_id}")
|
|
|
|
print("}")
|