o Ci+@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZmZdd lmZdd lmZmZdd lmZddlmZGdddeZGdddeZ GdddeZ!dS))APIView)Responsestatus)swagger_auto_schema) transaction)Q) csrf_exempt)models)role_serializers)datetimetimezone)token_function)require_permissionCustomIsAuthenticated)method_decorator)OAuth2Authenticationc@sHeZdZegZegZejZ e e de ejde ejddZdS) CreateRolez role.create) request_bodyc Cs|jj}tjjj|d}|js!|jjdd s!t ddit j dS|j |jd}|jdd r|jd }|jd d }|jd g}tjjj||d rYt dd|dit jdStjjj|d}tjjj|||d} | j|t d| j| jdd|Ddt jdSt ddit jdS)Nidadminnamemessagez&You are not authorized to create rolesr)dataT)raise_exceptionr description permissions) name__iexact created_byzRole "z" already existsid__in)rrr!zRole created successfullycSsg|]}|jqSr).0pr$r$+/var/www/Datamplify/authentication/roles.py Asz#CreateRole.post..)rrole_id role_namerzSerializer Error)userr auth_models UserProfileobjectsget is_superuserrolesfilterexistsrrHTTP_403_FORBIDDENserializer_classris_validvalidated_dataRoleHTTP_406_NOT_ACCEPTABLE PermissioncreatersetrHTTP_201_CREATEDHTTP_400_BAD_REQUEST) selfrequestuser_idr+ serializerrrpermission_idsrroler$r$r'postsB     zCreateRole.postN)__name__ __module__ __qualname__rauthentication_classesrpermission_classesauth_serializersCreateRoleSerializerr5rrrr ratomicrEr$r$r$r'rs  rc@s0eZdZegZegZeede ddZ dS)RoleList role.viewcCs|jj}tjjj|d}|js!|jjdd s!t ddit j dStj jtddt|jd Bd }tj|dd }t d t|j|jd t jdS)zC Get all roles created by the logged-in Admin user rrrrz$You are not authorized to view rolesrT) is_default)r!z -created_at)manyzRoles fetched successfully)rcountr1)r+rr,r-r.r/r0r1r2r3rrr4r8rorder_byrKRoleListSerializerlenr HTTP_200_OK)r?r@rAr+r1rBr$r$r'r/Ps &z RoleList.getN) rFrGrHrrIrrJrrr r/r$r$r$r'rNLs  rNc@sleZdZegZegZeede ddZ eede e j ddZ eede e j dd Zd S) RoleDetailrOcCs|jj}tjjj|d}|js!|jjdd s!t ddit j dSz tj jj||d}Wntj jy@t ddit jdYSwt|}t d |jd t jdS) zQ Get detailed info of a single Role (created by logged-in Admin) rrrrz+You are not authorized to view role detailsr)rr!z$Role not found or not created by youz!Role details fetched successfullyrrD)r+rr,r-r.r/r0r1r2r3rrr4r8 DoesNotExistHTTP_404_NOT_FOUNDrKRoleDetailSerializerrrV)r?r@r)rAr+rDrBr$r$r'r/us,  zRoleDetail.getz role.editc Cs|jj}z tjjj|d}Wntjjy"tdditj dYSwt |j jt |kr6tdditj dS|j d|j}|j d|j}|j dg}||_||_||rgtjjj|d }|j|t|} td | j d tjdS) z8 Update a custom role (only by creator) rrRole not foundrz*You are not authorized to update this rolerrrr"zRole updated successfullyrX)r+rr,r8r.r/rYrrrZstrr!r4rrrsaver:r2rr<rKr[rV) r?r@r)rArDrrpermissions_idsrrBr$r$r'puts&  zRoleDetail.putz role.deletecCs|jj}z tjjj|d}Wntjjy"tdditj dYSwt |j jt |kr6tdditj dS| tdditjdS)z8 Delete a custom role (only by creator) rrr\rz*You are not authorized to delete this rolezRole deleted successfully)r+rr,r8r.r/rYrrrZr]r!r4deleterV)r?r@r)rArDr$r$r'raszRoleDetail.deleteN)rFrGrHrrIrrJrrr r/rrMr`rar$r$r$r'rWqs    %rWN)"rest_framework.viewsrrest_framework.responserrest_frameworkrdrf_yasg.utilsr django.dbrdjango.db.modelsrdjango.views.decorators.csrfr authenticationr r,r rKr r tzauthentication.utilsrauthentication.permissionsrrdjango.utils.decoratorsr&oauth2_provider.contrib.rest_frameworkrrrNrWr$r$r$r's"            :%