a ^ΌhM@s<ddlmZddlmZddlmZmZmZmZddl m Z m Z m Z m Z mZddlmZddlmZmZddlmZddlmZdd lmZddlmZmZddlmZdd lmZdd lmZdd lmZm Z dd lm!Z!ddl"m#Z#Gdddeej$Z%Gdddeej$Z&Gdddeej$Z'Gdddeej$Z(dS) defaultdict)JRPUser)Project ProjectMember ProjectModule ProjectTask)ProjectMemberSerializerProjectModuleSerializerProjectSerializerProjectTaskRetriveSerializerProjectTaskSerializer)ListAndSerializerMixin)viewsets permissions)IsAuthenticated)action)Response)openapi)swagger_auto_schema)MultiPartParser FormParserstatus)Countc@s`eZdZeZeeeedZeZej dZ e gZ ddZddZeddgd d d d Zd S)ProjectViewSetupdateretrievezget item by id createlist -created_atcCs|j|jjjddSz= Set the created_by user when creating a new DepartmentType. ) created_byNsaverequestuseridself serializerr+@E:\Server_2_sachin\app\JRP\backend_erp\prj_mgt\views\projects.pyperform_create"szProjectViewSet.perform_createcCs|j|jjjddSz9 Set the updated_by user when updating a DepartmentType. ) updated_byNr#r(r+r+r,perform_update&szProjectViewSet.perform_updateFGETDDL ListdetailmethodsnamecOs2tj}|j|ddd}dd|jD}t|S)NT read_onlymanycSsg|]}|d|ddqS)r' project_namer'r6r+.0itemr+r+r, 3z&ProjectViewSet.ddl..)robjectsalldefault_serializer_classdatar)r)r%argskwargs data_listdata_list_serializr filtered_datar+r+r,ddl*s  zProjectViewSet.ddlN)__name__ __module__ __qualname__r serializer_classserializer_classesrCrrArBorder_byquerysetrpermission_classesr-r0rrJr+r+r+r,rsrc@seZdZeZeeeedZeZej dZ e gZ ddZddZeddgd d eejd ejd ejd gdddZeejd ejd ejd gdddZdS)ProjectMembersViewSetrr cCs|j|jjjddSr!r#r(r+r+r,r-Esz$ProjectMembersViewSet.perform_createcCs|j|jjjddSr.r#r(r+r+r,r0Isz$ProjectMembersViewSet.perform_updateFr1r2r3 project_idproject_id Query Param descriptiontypemanual_parameterscOsB|jdd}td|tjj|dd}dd|D}t|S)NrTrTmembercSsHg|]@}|jj|jjr8|jjd|jjd|jdn|jjdqS) z ()r;)r\r' first_name last_nameroleemailr<r+r+r,r?ds0z-ProjectMembersViewSet.ddl..) query_paramsgetprintrrAfilterselect_relatedrr)r%rErFrTrGrIr+r+r,rJMs zProjectMembersViewSet.ddlc Os|jdd}tjj|ddd}td|g}|D]\}td|||j|j |j |j |j j|j j d|jj|jjd|jj|jjdd q6t|S) NrTr[projectr\rGr;r])r'r6rb)Z member_idra joined_at is_activerir&)rcrdrrArfrgreappendr'rarjrkrir:r\r_r`rbr)r)r%rErFrTmembersZ member_listr\r+r+r,rms(   zProjectMembersViewSet.listN)rKrLrMr rNrOrCrrArBrPrQrrRr-r0rrr ParameterIN_QUERY TYPE_INTEGERrJrr+r+r+r,rS6s@  rSc@seZdZeZeeeedZeZej dZ e gZ ddZddZeddgd d eejd ejd ejd gdddZeejd ejd ejd gdddZdS)ProjectModuleViewSetrr cCs|j|jjjddSr!r#r(r+r+r,r-sz#ProjectModuleViewSet.perform_createcCs|j|jjjddSr.r#r(r+r+r,r0sz#ProjectModuleViewSet.perform_updateFr1r2r3rTrUrVrYcOs<|jdd}td|tjj|d}dd|D}t|S)NrTr[cSsg|]}|j|jdqSr;r'titler<r+r+r,r?sz,ProjectModuleViewSet.ddl..)rcrdrerrArfrrhr+r+r,rJs zProjectModuleViewSet.ddlc Os|jdd}|r(tjj|dd}ntjd}td|g}|D]2}||j |j |j |j |j j |j j|j j dqJt|S)NrTr[rirG) module_idZ module_titleZmodule_descriptionactiverTr:Zproject_description)rcrdrrArfrgrBrerlr'rtrWrvrir:r)r)r%rErFrTrGZ module_datamoduler+r+r,rs"  zProjectModuleViewSet.listN)rKrLrMr rNrOrCrrArBrPrQrrRr-r0rrrrnrorprJrr+r+r+r,rqs@  rqc @seZdZeZeeeedZeZej dZ e gZ eejdejejdddgddd Zd d Zedd gd deejdejdejdgdddZeejdejdejdgdddZddZedd gd deejdejdejdgdddZedd gd deejdejdejdgdddZdd lmZedd gd!deejd"ejd#ejdejd$ejd%ejdejd&ejd'ejdgdd(d)Z d*S)+ProjectTasksViewSetrr attachmentszMultiple filesF)rXrWrequiredrYcCs|j|jjjddSr!r#r(r+r+r,r-s z"ProjectTasksViewSet.perform_createcCs|j|jjjddSr.r#r(r+r+r,r0sz"ProjectTasksViewSet.perform_updater1r2r3ruzmodule_id Query ParamrVcOs<|jdd}td|tjj|d}dd|D}t|S)NrurTrucSsg|]}|j|jdqSrrrsr<r+r+r,r?.sz+ProjectTasksViewSet.ddl..)rcrdrerrArfr)r)r%rErFrurGrIr+r+r,rJs zProjectTasksViewSet.ddlcOs\|jdd}td||r2tjj|dd}ntjd}|j|ddd}t |j S)NrurTr{rwTr7) rcrdrerrArfrgrBrCrrD)r)r%rErFrurGrHr+r+r,r8s zProjectTasksViewSet.listcOsZ|}||}|j}dd|jD|d<t|jdd|jD|d<t|S)NcSsg|] }|jqSr+)r')r=r&r+r+r,r?[r@z0ProjectTasksViewSet.retrieve..Zassigned_member_idscSsg|] }|jjqSr+)fileurl)r= attachmentr+r+r,r?]r@Zattachment_files) get_objectget_serializerrDassigned_membersrBreryr)r)r%rErFtaskr*rDr+r+r,rUs  zProjectTasksViewSet.retrieveuser_idzuser_id Query ParamcCsz|jdd}Wn$tjy6tddiddYS0tjj|dd}g}|D]8}t |}| |j j j |j j j|j jd|jigd qTt|S) Nrr4zUser not found.r)rmodule__project task_datarTr: module_nametasks)rcrdr DoesNotExistrrrArfrgr rlrwrir'r:rtrD)r)r%rr project_data tasks_objrr+r+r,get_task_list_by_user`s* z)ProjectTasksViewSet.get_task_list_by_userrTcCsz|jdd}Wn$tjy6tddiddYS0tjj|dd}g}|D]8}t |}| |j j j |j j j|j jd|jigd qTt|S) NrTr4zproject_id not found.rr)Zmodule__project__idrrr)rcrdrrrrrArfrgr rlrwrir'r:rtrD)r)r%rTrrrrr+r+r,get_task_list_by_projects* z,ProjectTasksViewSet.get_task_list_by_projectrrzproject_task_report Listrzstatus Query Param filter_byz9filter_by Query Param (project,user,targer_date,priority) filter_valuezfilter_value Query Paramc Osp|jd}|jd}|jd}t||dvrFtdditjdStj}|dkrf|j d d}n|d krz|j d d}|d kr|r|j |d }d }nL|dkr|r|j |d}d}n.|dkrd}n |dkrd}ntdditjdSt t } | ddD]X} | jD]H} | | j| jjj| jj| j| j| j| jrF| jdnddq qdd| D} t| S)Nrrr) In Progress CompletederrorzInvalid statusrrdonerri)Z%module__project__project_name__iexactZmodule__project__project_namer&)Z$assigned_members__first_name__iexactZassigned_members__email target_dateend_datepriorityzInvalid or missing group_byrrz%d-%m-%Y)rirwrtrWrrcSsg|]\}}||dqS))r6rr+)r=r6Z task_listr+r+r,r? r@z;ProjectTasksViewSet.project_task_report..)rcrdrerrHTTP_400_BAD_REQUESTrrArBexcluderfrrrgprefetch_relatedrr_rlrwrir:rtrWrrstrftimeitemsvalues)r)r%rErFZ status_filtergroup_byrrQZ group_fieldZgroupedrr\Z response_dataZ report_datar+r+r,project_task_reportsJ           z'ProjectTasksViewSet.project_task_reportN)!rKrLrMr rNrOrCrrArBrPrQrrRrrrnIN_FORM TYPE_FILEr-r0rrorprJrrrr collectionsr TYPE_STRINGrr+r+r+r,rxs     ! ! rxN))rrhrms.models.userrZprj_mgt.models.projectsrrrrZprj_mgt.serializers.projectsr r r r r hrms.views._views_mixinsrrest_frameworkrrZrest_framework.permissionsrrest_framework.decoratorsrrest_framework.responserdrf_yasgrZdrf_yasg.utilsrZrest_framework.parsersrrrdjango.db.modelsr ModelViewSetrrSrqrxr+r+r+r,s&           #b^