B
    fU                 @   sB   d dl T d dlZd dlmZ d dlmZ G dd dejjj	Z	dS )    )*N)	CmfEntity)CmfActiveEntityc                   s   e Zd ZdZejjjjddddg Ze	d-dd	Z
d.d
dZd/ddZ fddZdd Zdd Zedd Z fddZ fddZdd Zdd Zdd  Zd0d!d"Zeed#dd$d%d& Zd1 fd'd(	Zd)d* Zd2d+d,Z  ZS )3CmfWorkflowTstatus_listget_default_statusget_related_projects_inforeplaceNFc             K   sl  |s|st ddd |s dg}nd|kr2|d |s|dd|g}|rL||g}tjjf ||d|}	|rx|	jdd	 d
 |	S tj|ddgd}
tjj	dd|
j
jgddddgdd}g }|stjjf dd|gdddgg|dd|}t|}xb|jD ]X}|r|jsq|jjs,|
jjjdd |jD kr||
r|js@q||j qW t|}|jdd	 d
 |S )Nz6DEV: FATAL get_status_list if not obj and not workflowT)abortordernoworkflowz==)filterfieldsc             S   s
   | j dkS )Nclosed)code)s r   ./common/models/cmf_workflow.py<lambda>!       z-CmfWorkflow.get_status_list.<locals>.<lambda>)keystatus)r   id=transztrans.status_to.ordernoztrans.status_fromztrans.trans_sdesk_portal_allow)r   r   cache_inmemoryallow_empty_transitionc             S   s   g | ]}|j jqS r   )r   value).0tr   r   r   
<listcomp>;   s    z/CmfWorkflow.get_status_list.<locals>.<listcomp>c             S   s   | j jS )N)r   r   )r   r   r   r   r   A   r   )	cmf_alertappendmodels	CmfStatuslistsortZcmfutilZget_obj_by_idr   getr   r   setr   trans_sdesk_portal_allowstatus_fromZis_nullr   r   trans_filter_ok	status_toadd)clsZobj_idZworkflow_idr   closed_lastr)   r   kwargs
get_filterresobjr   Zvalid_statusesr   r   r   r   get_status_list   sL    



$
zCmfWorkflow.get_status_listc             O   sD   dd| g}|r||g}t jj|d|i|}|r@|jdd d |S )Nr   z==r   c             S   s
   | j dkS )Nr   )r   )r   r   r   r   r   K   r   z)CmfWorkflow.status_list.<locals>.<lambda>)r   )r#   r$   r%   r&   )selfr   r/   argsr0   r1   r2   r   r   r   r   D   s    
zCmfWorkflow.status_listOPENc          	      s   dg}d }|r$| j dd|g|d}np rN| j |dd}|jd fddd	 nF|d
krzdg}| j ddd
gdddgg|d}|s| j dd|g|dd}|r|d S |rtd|  d| d| d  d S )Nr   r   z==)r   order_byT)r8   r/   c                s>   | j  j k| j jk| j jk| j jk jdko:| jdkfS )NZ	IN_REVIEWCLOSED)r   r   namestatus_type)r   )r   r   r   r   ]   s
    



z0CmfWorkflow.get_default_status.<locals>.<lambda>)reverser   r9   z-ordernor;   r   )r   r8   r/   r   u+   Отсутствует статус для z status_type=z, status_code=z	, status=)r   r&   ZCmfError)r5   r;   Zstatus_coder   Zraise_errorr8   r2   r   )r   r   r   P   s4    



zCmfWorkflow.get_default_statusc                s   t   dg S )Nr   )supersave_preload_fields)r5   )	__class__r   r   r>   ~   s    zCmfWorkflow.save_preload_fieldsc                s  | j ddg i  x| j jD ]z}|ddg | }d|_|j|_| |_|  | |< x:|jD ]0}|dddg | }d|_||_	|  qdW qW x<| j j
D ].}|ddg | }d|_| |_ fd	d
|jD |_|jjr |j |_|  |dddg x@|jD ]6}|ddg | }d|_| |_||_|  qW x@|jD ]6}	|	ddg |	 }
d|
_| |
_||
_|
  q\W x@|jD ]6}|ddg | }d|_| |_||_|  qW qW d S )Nstatusesr   z**approve_groupsFZ	approversZmaster_approversr*   c                s   g | ]} | qS r   r   )r   Z_st)status_mappingr   r   r       s    zACmfWorkflow.scaffold_statuses_and_transitions.<locals>.<listcomp>trans_filterstrans_validatorstrans_actionsZwz_value_obj)templateZload_fieldsr@   Zclonesystemr   r   saverA   parentr   r*   r,   Zis_not_nullrC   
transitionrD   rE   )r5   stZst_newZ
approve_grZapprove_gr_newr   Z	trans_newr   Z
filter_newZ	validatorZvalidator_newactionZ
action_newr   )rB   r   !scaffold_statuses_and_transitions   s`    z-CmfWorkflow.scaffold_statuses_and_transitionsc             C   sJ   | j sF| jjs>| jjs>| jjs>| jjs>| jjs>| jjs>| jjrFt	d dS )u   
        Оповещаем пользователя, если выбранные им галочки не будут активированы
        u   Для работы выбранных опций необходимо активировать "Переопределять опции Simple Logic Проекта". Иначе будут применяться настройки ПроектаN)
Zenable_sl_controlZsl_deadline_shift
is_changedZsl_only_owner_approveZsl_deny_no_approveZ)sl_allow_executor_change_nofatal_deadlineZsl_task_need_approveZsl_readonly_closed_taskZsl_task_only_owner_closer!   )r5   r   r   r   sl_consistent_warning   s    z!CmfWorkflow.sl_consistent_warningc             C   s   t tjj| dddS )uP   Возвращает есть ли на данном БП утверждениеT)r   need_approver   )boolr#   r$   r'   )r5   r   r   r   rP      s    zCmfWorkflow.need_approvec                sh   | j r| jstddd | jjr<tj  tj	s<tddd | 
  | j }t j|| |rd|   d S )NuB   Выберите шаблон для Бизнес-процессаT)r
   u   Использование "Действий после сохранения объекта" недоступно в Вашей версии, обратитесь в Техническую поддержку.)is_newrF   r!   Zsave_eval_coderN   r#   ZCmfAccessListZcheck_admin_modeZconfigZBZPYTHON_ENABLEDrO   r=   rH   rM   )r5   r6   r0   rR   )r?   r   r   rH      s    
zCmfWorkflow.savec                s$   | j }t j|| |r |   d S )N)rR   r=   _save_importrM   )r5   r6   r0   rR   )r?   r   r   rS      s    zCmfWorkflow._save_importc             C   s   t jj| dgdd}|S )Ntarget_workflowT)rT   r   r   )r#   CmfSchemeWfRuler%   )r5   Zrulesr   r   r   _get_related_rules   s    zCmfWorkflow._get_related_rulesc                sR   ddddddddg}d	d
d   gg}| fdd|D 7 }tjj||dd}|S )Ndefault_list_workflowdefault_epic_workflowdefault_subproject_workflowdefault_release_workflowdefault_sprint_workflowdefault_document_workflowdefault_task_workflowdefault_subtask_workflowORscheme_wf_rulesINc                s   g | ]}|d  gqS )z==r   )r   field)r5   r   r   r       s    z4CmfWorkflow._get_related_schemes.<locals>.<listcomp>T)r   r   r   )rV   r#   CmfSchemeWfr%   )r5   Zscheme_fieldsfilter_schemeZschemesr   )r5   r   _get_related_schemes   s    z CmfWorkflow._get_related_schemesc                sz  |   }tjjddd|gdd| ggd}ddd  fd	d
|D }tjjddd| gdd| ggdgd}x>|D ]6}|jrn||jj   ||jj d  d7  < qnW tj	jddd| gdd| ggdgd}x>|D ]6}|jr||jj   ||jj d  d7  < qW i }xT|
 D ]H\}	}
|
 kr:|
||	jj< n&tj	j|	dtjj|	dd||	jj< qW |rvt|S d S )Nr_   Z	scheme_wfra   r   z==)r   r   )tasksdocsc                s   i | ]}   |qS r   )copy)r   project)default_scorer   r   
<dictcomp>   s    z9CmfWorkflow.get_related_projects_info.<locals>.<dictcomp>r]   ri   )r   r   rg      rf   )ri   )re   r#   Z
CmfProjectr%   ZCmfDocumentri   
setdefaultr   rh   ZCmfTaskitemsr:   countZjsondumps)r5   Zrelated_schemesZrelated_projectsZprojects_dictZrelated_docsdocZrelated_tasksZtaskZres_dictri   Z
count_dictr   )rj   r   r      s4    







 z%CmfWorkflow.get_related_projects_infoc          	   K   sh   t jjdddddggdd| jjgdd	d
| jj d
ggd}|rLtddd tt j	j| jj|gd d S )NZANDr   ra   openZin_progressr:   z==Zparams_jsonZILIKE%)r   uc   Работа по замене данного бизнес-процесса уже запущенаT)r
   )r6   )
r#   ZCmfDeferredJobro   replace_workflowr:   r   r   r!   Zschedule_deferred_jobr   )r5   new_workflow_idZ_kwargsZreplace_task_existsr   r   r   r	     s    zCmfWorkflow.replaceu(   Смена бизнес-процесса)descriptionZshow_bg_progressbarc                s4  t jj| d t jj|d}t jj dgd}x.|D ]&}|j kr4||_|jdd t  q4W dddd	d
dddg}ddd|gg}| fdd|D 7 }t jj||d}	xF|	D ]>}
x&|D ]}t	|
| krt
|
|| qW |
jdd t  qW ddg}xt D ]}|jdkrq x|jdg fdd|D  |ddgd}|sFP xF|D ]>}x*|D ]"}t	|| krVt
||| qVW |jdd qLW t  qW q W xt D ]}|jdkrqd|jkrx^|jdd gdgddgd}|sP x,|D ]$}|j kr||_|jdd qW t  qW qW d S )N)r   rT   )rT   r   F)ZnotifyrW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   c                s   g | ]}|d  gqS )z==r   )r   rb   )old_workflowr   r   r    ;  s    z0CmfWorkflow.replace_workflow.<locals>.<listcomp>)r   r   r   )rU   rc   c                s   g | ]}|d  gqS )z==r   )r   rb   )rw   r   r   r    L  s    r      )r   r   slice)r$   ZCmfTransActionZCmfTransFilterZCmfTransValidCmfTransdefault_workflowz==)r#   r   r'   rU   r%   rT   rH   Z
cmf_commitrc   getattrsetattrr   iter_subclasses
class_namer   r   r{   )Zold_workflow_idru   r6   r0   Znew_workflowZwf_rulesruleZwf_scheme_fieldsrd   Z
wf_schemesschemerb   Zactive_entity_fieldsmodelZobj_listr3   r   )rw   r   rt   "  sr    








zCmfWorkflow.replace_workflowc       
         sZ  d}g }xFt  D ]:}||jdd| gd7 }|r|s|jdd| gddgd}qW |rr|d td| d	d
d x0tjj| ||dD ]}|j	f |d
d| qW x.tj
j| ||dD ]}|j	f d|i| qW |r>xbtjjddd| gdd| ggddg|dD ]6}	|	j| krd |	_|	j| kr,d |	_|	jd
d qW t j	f ||d| d S )Nr   r   r   )r   
   )r   ry   z...u   Невозможно удалить используемый бизнес процесс. Данный бизнес процесс используется в u    объектахT)r
   )r   include_deletedinclude_system)TEXKOM_db_deleteforce)rT   r   r   r   r_   r]   z==r^   )r   r   r   )Z	only_data)r   r~   ro   Zslistr"   r!   r#   r$   r%   deleterU   rc   r]   r^   rH   r=   )
r5   r   r   r0   Z	ref_countZ	top_usager   r   r   r   )r?   r   r   r   h  s2    

zCmfWorkflow.deletec             C   s   |r|sd S x:t jjddd|jggdd|ggdD ]}||r2|S q2W x6t jjdddd gdd|ggdD ]}||rj|S qjW d S )NTr*   ra   r,   r   )r   r   )r#   rz   r%   r   r+   )r5   r3   r*   r,   Zpossible_transitionr   r   r   calc_transition  s    



zCmfWorkflow.calc_transitionc          	   C   st   d}|sd S |  |||}|s\|j  |jsXtd|j d|j d|j d| dd d S || ||}|S )NFuA   Нет доступного перехода из статуса u    в статус u"    у Бизнес-процесса u    для объекта T)r
   )r   r   loadr!   r:   r   Ztrans_validator_okZtrans_run_actions)r5   r3   r*   r,   Zis_task_transition_runZhas_changesrJ   r   r   r   run_transition  s    
"

zCmfWorkflow.run_transition)NNNFFN)NF)r7   NNT)N)FF)F)__name__
__module____qualname__Z	api_allowcommonr   cmf_workflowr   Zapi_methodsclassmethodr4   r   r   r>   rM   rO   propertyrP   rH   rS   rV   re   r   r	   staticmethodZcmf_deferred_jobrt   r   r   r   __classcell__r   r   )r?   r   r      s4   
 2

.4#
	E!r   )
Zcmf.includeZcommon.fields.cmf_workflowr   Z
cmf.modelsr   Zcommon.models.cmf_active_entityr   r   r   r   r   r   r   r   <module>   s   