o Ci @sPdZddlmZddlmZddlmZmZddlm Z m Z GdddeZ dS) zY Management command to seed RBAC permissions and roles Usage: python manage.py seed_rbac ) BaseCommand) transaction) PermissionRole)DEFAULT_PERMISSIONS DEFAULT_ROLESc@seZdZdZddZdS)Commandz1Seed RBAC permissions and roles into the databasec Os|j|jdt|jdd}d}tD]\}}}}tjj |||dd\} } | r5|d7}q|d7}q|j|jd|d|d |jd d} d} t D]?\} }t jj | |d |d d d\}} |ddkr~|j tjntjj|dd}|j || r| d7} qY| d7} qY|j|jd| d| d|j|jd|jdtj|jdt j|jdt jD]}|j }|jd|jd|d qWddS1swYdS)NzStarting RBAC seeding...zCreating permissions...r)namemodule)codedefaultsu ✓ Created z permissions, updated z permissionszCreating roles... descriptionis_system_role)rdefault)r r permissionsALL)code__inz roles, updated z rolesz === RBAC Seeding Complete ===zTotal Permissions: z Total Roles: z Roles created:u • z )stdoutwritestyleSUCCESSratomicrrobjectsupdate_or_createritemsrrsetallfiltercountr )selfargsoptions created_perms updated_permsr r categoryrpermcreated created_roles updated_roles role_name role_configroleperms perm_countr/L/var/www/Datamplify/authentication/management/commands/create_permissions.pyhandles^              "zCommand.handleN)__name__ __module__ __qualname__helpr1r/r/r/r0r s rN) __doc__django.core.management.baser django.dbrauthentication.modelsrr authentication.permission_modelsrrrr/r/r/r0s