o ^h @sHddlmZddlmZddlmZmZddlmZGdddeZ dS)) BaseCommand) Permission) DepartmentEmployee) CustomUserc@s eZdZdZddZddZdS)CommandzXGrant superuser privileges to Management department employees and ensure all permissionscCs$|jdtdd|jdddddS)Nz --usernamez/Grant superuser privileges to specific username)typehelpz--all-management store_truezAGrant superuser privileges to all Management department employees)actionr ) add_argumentstr)selfparserrW/var/www/html/optinet_system/accounts/management/commands/grant_superuser_privileges.py add_arguments s zCommand.add_argumentsc Os|d}|d}|rHz#tjj|d}d|_d|_d|_||j|j d|Wntj yG|j|j d|YdSw|rzHt jjdd }tjj|d d }d }|D]$} | jrd| j_d| j_d| j_| j|d 7}|jd| jq^|j|j d|dWnt j y|j|j dYn w|j|j dtjjdd} |jd| D]}|jd|jd|pddqdS)Nusernameall_management)rTadminz.Successfully granted superuser privileges to: zUser not found: Management)nameactive) departmentemployment_statusrz!Granted superuser privileges to: zSuccessfully updated z0 Management employees with superuser privileges!z8Management department not found. Please create it first.z4Please specify either --username or --all-management) is_superuserz Current Superusers:z- z (zNo name))getrobjectsris_staffrolesavestdoutwritestyleSUCCESS DoesNotExistERRORrrfilteruser full_nameWARNINGr get_full_name) rargsoptionsrrr*management_deptmanagement_employees updated_countemployee superusersrrrhandlesf      &zCommand.handleN)__name__ __module__ __qualname__r rr5rrrrrs rN) django.core.management.baserdjango.contrib.auth.modelsr hr.modelsrraccounts.modelsrrrrrrs