U
    e                    @   sT   d dl T d dlmZmZ d dlmZ d dlZd dlT d dlm	Z	 G dd de	Z	dS )    )*)ROUND_UP
ROUND_DOWN)	timedeltaN)CmfGanttTaskc                       s  e Zd Zedd Z fddZedd Zedd Zed	d
 Z	edd Z
dd Zdd Zdd Zdd Zdd ZdrddZdd ZdsddZdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zdtd/d0Zd1d2 Zdud3d4Zd5d6 Zd7d8 Zdvd9d:Z d;d< Z!dwd=d>Z"d?d@ Z#dAdB Z$dCdD Z%dEdF Z&dGdH Z'dIdJ Z(dKdL Z)dMdN Z*dOdP Z+dxdQdRZ,dSdT Z-dUdV Z.dWdX Z/dYdZ Z0d[d\ Z1d]d^ Z2dyd_d`Z3dadb Z4dcdd Z5dedf Z6dgdh fdidj
Z7d.dk fdldm
Z8 fdndoZ9 fdpdqZ:  Z;S )zr   c              	   C   s   t  }|rPddddddddd	g	}t jjd |j|d
}|rP|D ]}|| ||< q>|jrl|jjjdkrl|j|_|rv||_|j	  |j|_||_
|j|_t  |jddd W 5 Q R X |S )Nsched_start_datesched_finish_datesched_duration
sched_cost
sched_workconstrain_start_typeconstrain_start_dateconstrain_finish_typeconstrain_finish_date)Z	parent_idtask_idfieldsZ
CmfProjectF)emitnotify)modelsr   getidparentvalue
class_nameprojectgantt_projectloadtaskparent_taskcmfutilZdisable_notifysave)clsr   baseline
gantt_taskZ
fieldsListop_gantt_taskZfield r%   (./modules/gantt/models/cmf_gantt_task.pycreate_gantt_task
   s,      

zCmfGanttTask.create_gantt_taskc                    s    dddddddg}t   | S )Nr   r   task.logic_prefixztask.parent_task.logic_prefixztask.has_child_tasksztask.plan_start_dateztask.plan_end_date)supersave_preload_fields)selfr   	__class__r%   r&   r*   #   s    	z CmfGanttTask.save_preload_fieldsc                 C   s2   | j jrd S | j | j d }t|jtdtdS )Nd   z1.11Zrounding)r
   is_null_perform_completeDecimalquantizer   r+   resr%   r%   r&   _perform_cost1   s    zCmfGanttTask._perform_costc                 C   s:   | j jrd S | j | j d }t|jtdtd}t|S Nr.   1r/   )r	   r0   r1   r2   r3   r   intr4   r%   r%   r&   _perform_duration8   s
    zCmfGanttTask._perform_durationc                 C   s:   | j jrd S | j | j d }t|jtdtd}t|S r7   )r   r0   r1   r2   r3   r   r9   r4   r%   r%   r&   _perform_work@   s
    zCmfGanttTask._perform_workc                 C   s   | j js| jjrdS ttj| j   }| j ttjkr@dS | j| j   }t|dkrj|| d }nd}|dkrzd}t	|j
t	dtd}t|S )Nr   r.   r8   r/   )r   r0   r   datetimeZnowtimezoneZutcZtotal_secondsr9   r2   r3   r   )r+   Zdelta_now_time_secZdelta_finish_time_secZperform_completer5   r%   r%   r&   r1   H   s    zCmfGanttTask._perform_completec                 C   s   d\}}}| j jrH| j jdkrH| j jj }||   | \}}}|}| jdkrx| j	}|rp|rpt
||}n|pv|}|}| jdkr| j}|r|rt||}n|p|}d }| jr| j}n|}|||fS )NNNNtask.gantt_project)0-const3-after)r@   2-before)r   r   logic_prefixr$   r   load_fieldsr*   get_parent_plan_limitr   r   maxr   r   minconst_durationr	   )r+   Zparent_start_limitZparent_finish_limitZparent_duration_limitparent_gantt_taskstart_limitfinish_limitduration_limitr%   r%   r&   rE   a   s,    


z"CmfGanttTask.get_parent_plan_limitc                 C   s  d  } }}| j jsdS tjjdd| j g|  dD ]}|jdkr`|sN|j}n|jr`t||j}|j	dkr|sv|j
}n|j
rt||j
}|jr|s|j}n|jrt||j}|j jr4| \}}}|s|}n|rt||}|s|}n|rt||}|s|}q4|r4t||}q4|||fS )Nr>   task.parent_task=filterr   )r@   4-before)r@   1-after)r   has_child_tasksr   r   listr*   r   r   rG   r   r   rF   rH   rL   get_childrens_plan_limit)r+   rJ   rK   rL   r#   Zch_start_limitZch_finish_limitZch_duration_limitr%   r%   r&   rU      sH    




z%CmfGanttTask.get_childrens_plan_limitc                 C   sB   t | ddrd S d| _|  \| _| _| _|  \| _| _| _	d S )N_calc_plan_limits_cache_doneFT)
getattrrV   rE   _plan_start_limit_min_plan_finish_limit_max_plan_duration_limit_maxrU   _plan_start_limit_max_plan_finish_limit_min_plan_duration_limit_minr+   r%   r%   r&   _calc_plan_limits_cache   s    z$CmfGanttTask._calc_plan_limits_cachec                 C   sF   | j r,| jr,tj|  | j j| jj| _n| j jrB| jjrBd| _d S Nr   )	r   r   r   CmfCalendarget_duration_minutes_get_calendarr   r	   oldr^   r%   r%   r&   _calc_sched_duration   s      z!CmfGanttTask._calc_sched_durationc                 C   sF   | j r,| jr,tj|  | j j| jj| _n| j jrB| jjrBd| _d S r`   )	actual_start_dateactual_finish_dater   ra   rb   rc   r   actual_durationrd   r^   r%   r%   r&   _calc_actual_duration   s      z"CmfGanttTask._calc_actual_durationNc                 C   s  | j js| j dkrd | _d S | jjr0| jjr0d S | jr| jdkr|dkrtj|  | jj	t
| j }| jj	}| jdkr|r||k rtd|dddd	 | jd
kr|r||krtd|dddd	 || _d S | jr| jdkr|dkrtj|  | jj	t
| j  }| jj	}| jdkrH|rH||k rHtd|dddd	 | jdkrz|rz||krztd|dddd	 || _d S | jdkr| jdkr| jrtd d S )Nr   r@   
start_daterR   um   Плановая дата окончания раньше ограничения даты окончания %d.%m.%Yu   . Увеличьте длительность или измените плановую дату начала на более позднюю.TabortrB   uk   Плановая дата окончания позже ограничения даты окончания u   . Уменьшите длительность или измените плановую дату начала на более раннюю.finish_daterA   ua   Плановая дата начала раньше ограничения даты начала u   . Уменьшите длительность или измените плановую дату окончания на более позднюю.rQ   u_   Плановая дата начала позже ограничения даты начала u   . Увеличьте длительность или измените плановую дату окончания на более раннюю.u   Плановый период может не соответствовать плановой длительности, т.к. все переменные зафиксированы)r	   r0   r   r   r   r   ra   get_date_by_durationrc   r   r9   r   	cmf_alertr   r   rH   )r+   offsetr   r   r   r   r%   r%   r&   _calc_sched_period   sr      

  
 zCmfGanttTask._calc_sched_periodc                 C   s   | j js| j dkrd | _d S | jjr0| jjr0d S | jrZtj|  | jjt	| j | _d S | jrtj|  | jjt	| j  | _d S d S r`   )
rh   r0   rg   rf   r   ra   ro   rc   r   r9   r^   r%   r%   r&   _calc_actual_period  s&        
z CmfGanttTask._calc_actual_periodc           
      C   s  | j jsdS | jdkrdS | jdkrtjjdd| j gdgd}|sFdS t|}d}|D ]6}|r||j|jkr|||jj	pvd7 }qV||jj	pd7 }qV|| }n | jd	krztj
jd
d| j gddgd}|sdS d}d}	|D ]}|r:|j|j jkr:|j ddg ||j jj	pd7 }|j jjdkrj|	|j jj	p4d7 }	q||jj	pHd7 }|jjdkr|	|jj	pfd7 }	q|	| d }n | jdkrdS | jdkrdS | j|k r|| _dS )uB   
        updated_ch_gantt_task - указан, когда
        N0-work
1-completerM   rN   actual_completerO   r   2-story_pointsr   agile_story_pointszstatus.status_typeCLOSEDr.   3-costz4-fixed)r   rS   actual_complete_typer   r   rT   lenr   rv   r   CmfTaskrD   rx   statusstatus_type)
r+   updated_ch_gantt_taskZch_gantt_task_listZtotal_countZsum_actual_completecZres_actual_completeZch_task_listZtotal_story_pointsZfinished_story_pointsr%   r%   r&   _calc_summarry_actual_complete,  sR    

z+CmfGanttTask._calc_summarry_actual_completec                 C   sv  |    |   | jdkr,| j| _d| _n| jdkr| j}| jrV| jrVt| j| j}|r| jrj| j|k r| jrtd|j	dd || _n| jdkr| j}| jr| j
rt| j| j
}|r| jr| j|kr| jrtd|j	dd || _nN| js| j| _| jr| jrt| j| j| _| jr:| j
r:t| j| j
| _| j
rd| jrd| j| j
krdtd	dd
 | jr| jr| j| jk rtddd
 | jjr8| j
r| jj| j
k r8| jr| jj| jkr8| js| jj| _n\| jdkrnN| jdkr| jj| jk r8| jj| _n&| jdkr8| jj| jkr8| jj| _| jr^| jr^| js^| jdd n|    | jdd d S )Nr@   TrA   uA   Плановая дата начала перемещена на rk   uR   , так как была раньше ограничения даты началаrQ   uP   , так как была позже ограничения даты началаu~   Нельзя указать Дату начала больше зафиксированной у дочерней задачиrl   u   Нельзя указать Дату начала меньше зафиксированной у родительской задачи)1-earlyrQ   )z2-latterrA   rn   rq   r   
date_field)re   r_   r   r   r   lock_sched_start_daterX   rF   rp   r   r[   rG   r   Zplan_start_dater	   lock_sched_finish_daterr   _change_parent_start_date)r+   r   r%   r%   r&   #_do_constrain_start_date_is_changedf  sl    





z0CmfGanttTask._do_constrain_start_date_is_changedc           
      C   s   g }| j  }ddddddg}| jjr`| jjdkr`dd	|gd
d	| jgdddgg}tjj||d}dd	| jgdd	dgg}tj	j|dgd}tt
dd |}dd	|gd
d|gg}tjj||d}	||	 S )Nr   r   r   r   r   r   r?   r   rN   r   r(   z!=rO   Zout_linkzrelation_type.codezsystem.additional_parentin_linkc                 S   s
   | j jjS N)r   r   r   )Zrelr%   r%   r&   <lambda>      z6CmfGanttTask._get_parent_gantt_tasks.<locals>.<lambda>IN)r   r   r   r   rC   parent_task_idr   r   rT   ZCmfRelationOptionmap)
r+   Znormal_parentsr"   r   rP   Zadditional_filterZadditional_parents_relationsZadditional_parents_tasks_idsZadditional_parents_filterZadditional_parentsr%   r%   r&   _get_parent_gantt_tasks  s,    
 

z$CmfGanttTask._get_parent_gantt_tasksc                 C   s6   d }|  ddg | jr | jj}|p4tjjddgdS )Nzproject.calendarzproject.calendar.timezonezsystem:defaultr=   )coder   )rD   r   calendarr   ra   r   )r+   r   r%   r%   r&   rc     s
    zCmfGanttTask._get_calendarc                 C   s   | j dkr| jrtddd | jrT| jrT| jdkrT| jjrT| jj| jk rTtddd | jr| jr| jdkr| jjr| jj| jkrtddd | jr| jr| js| j	d	d
 n| 
  d S )Nr@   u   Нельзя менять дату начала, если зафиксирована дата окончания и длительностьTrl   rA   ua   Ошибка: Плановая дата начала меньше даты ограниченияrQ   ua   Ошибка: Плановая дата начала больше даты ограниченияrn   r   )r   rH   rp   r   r   r   newr	   r   rr   re   r^   r%   r%   r&   _do_sched_start_date_is_changed  s"    z,CmfGanttTask._do_sched_start_date_is_changedc                 C   s   | j dkr| jrtddd | jrF| jdkrF| jj| jk rFtddd | jrp| jdkrp| jj| jkrptddd | jr| jr| jr| j	d	d
 n| 
  d S )Nr@   u   Нельзя менять дату окончания, если зафиксирована дата начала и длительностьTrl   rR   ug   Ошибка: Плановая дата окончания меньше даты ограниченияrB   ug   Ошибка: Плановая дата окончания больше даты ограниченияrj   r   )r   rH   rp   r   r   r   r   r	   r   rr   re   r^   r%   r%   r&    _do_sched_finish_date_is_changed  s    z-CmfGanttTask._do_sched_finish_date_is_changedc                 C   s   |   }|D ]}|jdkr$td q| | r|| rD| | || k r|jdkrt|| rt| | || krttd  d S |jdkr|| r| | || k rtd  d S | | ||< |  |j|d qd S )Nr@   ^   Дата родительской задачи является зафиксированнойrQ   u   Предупреждение: Плановая дата начала дочерней задачи больше плановой даты начала родительской задачиrA   u   Предупреждение: Плановая дата начала дочерней задачи меньше плановой даты начала родительской задачиr   )r   r   rp   r    r   )r+   r   parentsrI   r%   r%   r&   r     s&    
 z&CmfGanttTask._change_parent_start_datec                 C   s   |   }|D ]}|jdkr$td q| | r|| rD| | || kr|jdkrt|| rt| | || k rttd  d S | | }|jdkr|| r| | || krtd |jr||jkr|j}|||< |  || q| jdk| _d S )Nr@   r   rR   u   Предупреждение: Плановая дата окончания дочерней задачи меньше плановой даты окончания родительской задачиrB   u   Предупреждение: Плановая дата окончания дочерней задачи больше плановой даты окончания родительской задачи)r   r   rp   r   r    _change_parent_finish_dater   r   )r+   r   r   rI   Z
date_valuer%   r%   r&   r      s,    
 z'CmfGanttTask._change_parent_finish_datec                 C   sn  |    |   | jdkr,| j| _d| _n| jdkr| j}| jrV| jrVt| j| j}|r| jrj| j|k r| jrtd|j	dd || _n| jdkr| j}| jr| jrt
| j| j}|r| jr| j|kr| jrtd|j	dd || _nN| js| j| _| jr| jrt| j| j| _| jr:| jr:t
| j| j| _| jr| jrd| j| jkrdtd	dd
 | jr| j| jk rtddd
 | jjr0| jr| jj| jk r0| jr| jj| jkr0| js| jj| _n\| jdkrnN| jdkr
| jj| jkr0| jj| _n&| jdkr0| jj| jk r0| jj| _| jrV| jrH| jrV| jdd n|    | jdk| _d S )Nr@   TrR   uG   Плановая дата окончания перемещена на rk   uX   , так как была раньше ограничения даты окончанияrB   uV   , так как была позже ограничения даты окончанияu   Нельзя указать Дату окончания больше зафиксированной у родительской задачиrl   u   Нельзя указать Дату окончания меньше зафиксированной у дочерней задачиrj   r   )re   r_   r   r   r   r   r\   rF   rp   r   rG   rY   r   Zplan_end_dater	   r   rH   rr   )r+   r   r%   r%   r&   $_do_constrain_finish_date_is_changed  sn    





z1CmfGanttTask._do_constrain_finish_date_is_changedFc                 C   s   | j dkr | jdkr tddd |   | jrF| j| jkrFtddd | jrd| j| jk rdtddd |   |s| jr| js| j| _	| j
dd n| j| j	k rtddd | jd	d
 d S )Nr@   u   Нельзя менять длительность, если зафиксирована дата начала и дата окончанияTrl   u   Нельзя указать Длительность больше зафиксированной у родительской задачиu   Нельзя указать Длительность меньше зафиксированной у дочерней задачи)from_sched_duration_changedu   Нельзя указать Длительность меньше чем Трудозатраты, когда установлена опция Фиксированные Ресурсы и Трудозатраты зафиксированыr   r   )r   r   rp   r_   rZ   r	   r]   rr   
const_workr   _do_sched_work_is_changedr   )r+   from_sched_work_changedr%   r%   r&   _do_sched_duration_is_changed[  s"    
z*CmfGanttTask._do_sched_duration_is_changedc                 C   s   |    d S r   )rs   r^   r%   r%   r&   _do_actual_duration_is_changedq  s    z+CmfGanttTask._do_actual_duration_is_changedc              	   C   s(  | j jrd S | j dk r"tddd | jjs2| j | _| jdkr| jjr| jdkrt	
| j | j d }t|d}| jjs|| jkr|| _|   | j jpd| j jpd }| jjr| jjdkrt 8 | jjj }||   || |jdd	 W 5 Q R X |s$| jjr$tj| j| d S )
Nr   uo   "Фактические трудозатраты" стал ниже нуля, выставляем в ноль.Trl   rt   r.   r?   F
user_input)actual_workr0   rp   r   rS   actual_myself_workr{   r   is_not_nullmathfloorrG   rv   $notify_parent_update_actual_completer   rd   r   rC   r   disable_aclr$   r   rD   r*   from_child_add_workr    
is_changedr   ZCmfTimeTrackerZ"_op_gantt_actual_work_changed_hook)r+   from_timetrackertmp_complete
work_deltarI   r%   r%   r&   _do_actual_work_is_changedt  s,    




z'CmfGanttTask._do_actual_work_is_changedc              	   C   s   | j jrd S | j dk r"tddd | jdkr~| jjr~| jdkr~t| j | j d }t|d}| j	jsp|| j	kr~|| _	| 
  | jjr| jjdkr| j jpd| j jpd }t 8 | jjj }||   || |jdd	 W 5 Q R X d S )
Nr   ue   "Фактические Затраты" стал ниже нуля, выставляем в ноль.Trl   rz   r.   r?   Fr   )actual_costr0   rp   r{   r
   r   r   r   rG   rv   r   r   r   rC   r   rd   r   r   r$   r   rD   r*   from_child_add_costr    )r+   r   
cost_deltarI   r%   r%   r&   _do_actual_cost_is_changed  s$    




z'CmfGanttTask._do_actual_cost_is_changedc                 C   s,   | j jrd S | jdkrd S | jdkr(d S d S )Nrt   rz   )rv   r0   r{   r^   r%   r%   r&   _do_actual_complete_is_changed  s    

z+CmfGanttTask._do_actual_complete_is_changedc                 C   sH   | j jp
d| j jpd }| jjs*| j | _n|  j|7  _| j|d d S )Nr   r   )r   r   rd   r   rS   r   r   )r+   r   r   r%   r%   r&   !_do_actual_myself_work_is_changed  s
    
z.CmfGanttTask._do_actual_myself_work_is_changedc                 C   sD   | j jp
d| j jpd }| jjs*| j | _n|  j|7  _|   d S r`   )actual_myself_costr   rd   r   rS   r   r   r+   r   r%   r%   r&   !_do_actual_myself_cost_is_changed  s
    
z.CmfGanttTask._do_actual_myself_cost_is_changedc              	   C   s   | j jrd S | jjrb| j | jk r| jrFtd| j  d| j ddd q| j| _ td| j d nB|s| j r| js| j | _| j	dd | j
r| j| j k rtd	dd | jjr| jjd
kr| j jpd| j jpd }t : | jjj }||   || |jddd W 5 Q R X d S )Nu   Нельзя запланировать меньше Трудозатрат, чем запланировано в подзадачах ( < u`   ). Увеличте Трудозатраты или уменьшите их у подзадач.Trl   u   Указано Трудозатрат меньше, чем запланировано в подзадачах. Выставили uo   . Если нужно уменьшить Трудозатраты, уменьшите их у подзадач.)r   u   Нельзя указать Трудозатраты больше чем Длительность, когда установлена опция Фиксированные Ресурсы и Длительность зафиксированаr?   r   Fr   	only_data)r   r0   r   rS   agregat_workr   rp   rH   r	   r   const_resourcer   rC   r   rd   r   r   r$   r   rD   r*   from_child_agregate_workr    )r+   r   r   rI   r%   r%   r&   r     s2    


z&CmfGanttTask._do_sched_work_is_changedc              	   C   s   | j jrd S | j | jk rX| jr>td| j  d| j ddd n| j| _ td| j d | jjr| jjdkr| j jpvd	| j j	pd	 }t
 : | jjj }||   || |jd
dd W 5 Q R X d S )Nu   Нельзя запланировать меньше Расходов, чем запланировано в подзадачах (r   uX   ). Увеличте Расходов или уменьшите их у подзадач.Trl   u|   Указано Расходов меньше, чем запланировано в подзадачах. Выставили ue   . Если нужно уменьшить Расходы, уменьшите их у подзадач.r?   r   Fr   )r
   r0   agregat_cost
const_costrp   r   r   rC   r   rd   r   r   r$   r   rD   r*   from_child_agregate_costr    r+   r   rI   r%   r%   r&   _do_sched_cost_is_changed  s     

z&CmfGanttTask._do_sched_cost_is_changedc                 C   s"   | j r| jr|   n|   d S r   )rf   rh   rs   ri   r^   r%   r%   r&    _do_actual_start_date_is_changed  s    
z-CmfGanttTask._do_actual_start_date_is_changedc                 C   s(   | j s| jr| jr|   n|   d S r   )rf   rg   rh   rs   ri   r^   r%   r%   r&   !_do_actual_finish_date_is_changed  s    
z.CmfGanttTask._do_actual_finish_date_is_changedc                 C   sF   | j dkr|   n.| j dkr0| jjr0|   n| j dkrB|   d S )Nrt   ru   rw   rz   )r{   r   r   rS   r   r   r^   r%   r%   r&   #_do_actual_complete_type_is_changed  s    



z0CmfGanttTask._do_actual_complete_type_is_changedc                 C   s   | j r|   d S r   )rH   re   r^   r%   r%   r&   _do_const_duration_is_changed'  s    z*CmfGanttTask._do_const_duration_is_changedc                 C   s*   | j rd| _| j| _n| jj| _d | _d S Nr@   )r   r   r   r   defaultr^   r%   r%   r&   $_do_lock_sched_start_date_is_changed-  s
    

z1CmfGanttTask._do_lock_sched_start_date_is_changedc                 C   s*   | j rd| _| j| _n| jj| _d | _d S r   )r   r   r   r   r   r^   r%   r%   r&   %_do_lock_sched_finish_date_is_changed5  s
    

z2CmfGanttTask._do_lock_sched_finish_date_is_changedc           	   	   C   s   | j }| j}| j}| j}t  | jj}|r|j	 }|
|   ||  ||  ||  ||  |jdd | jj}|r|j	 }|
|   || || || || |jdd W 5 Q R X d S )NFr   )r
   r   r   r   r   r   r   rd   r$   r   rD   r*   r   r   r   r   r    r   )	r+   r
   r   r   r   Z
old_parentZold_parent_gantt_taskZ
new_parentZnew_parent_gantt_taskr%   r%   r&   _do_parent_task_is_changed=  s.    






z'CmfGanttTask._do_parent_task_is_changedc                 C   s0  | j r^d| _d| _| jjrZ| jjj rZ| jjjj }|rLt|t	
 | _d| _|   d S | jjrr|   d S | jjr|   d S | jjr|   d S | jjs| jjr|   d S | jjs| jjr|   d S | jjr|   d S | jjr|   d S | jjr|   d S | jjr&|   d S | jjr<|    d S | jjrR| !  d S | j"jrh| #  d S | j$jr~| %  d S | j&jr| '  d S | j(jr| )  d S | j*jr| +  d S | jjr| ,  | j-dd d S | j.jr| /  | j0dd d S | j1jr| 2  d S | j3jr,| 4  d S )Nr   r   r   r   r   )5is_newr	   r   r   r   r$   r   r   rF   r<   Zutcnowr   rr   r   r   r   r   r   r   r   r   r   r   r   r   rh   r   r   r   r   r   rv   r   r   r
   r   rf   r   rg   r   r{   r   rH   r   r   r   r   r   r   r   r   r   r   )r+   Zparent_sched_start_dater%   r%   r&   _save[  s    












zCmfGanttTask._savec                 C   sl   |  |   |   |  j|7  _| jdd |d kr\| j| j |kr\| j| | _|   | jdd d S )NTr   r   )	rD   r*   fix_null_variablesr   r   r   r   r   r    )r+   r   Zremaining_estimater%   r%   r&   timetracker_add_work_hook  s    z&CmfGanttTask.timetracker_add_work_hookc                 C   s(  |rL|j dkrL| jsL| |   |   | j  |   | jdd d S |r|j dkr|r|j dkr| |   |   | j  d| _	| 
  |   | jdd |   d S |r$|j dkr$|r$|j dkr$| |   |   d | _d| _	| 
  |   | jdd |   d S d S )NZIN_PROGRESSTr   ry   r.   r   )r   rf   rD   r*   r   Zset_nowr   r    rg   rv   r   r   r   )r+   Z
old_statusZ
new_statusr%   r%   r&   task_status_is_changed_hook  sF    

z(CmfGanttTask.task_status_is_changed_hookc                 C   s6   |  |   |   |   |   | jdd dS )u   
        В задаче изменились Agile-плановые даты, пересчитаем свои sched-даты, вдруг новые ограничения повлияют на них
        Tr   N)rD   r*   r   r   r   r    r^   r%   r%   r&   task_plan_dates_is_changed_hook  s
    z,CmfGanttTask.task_plan_dates_is_changed_hookc                 C   s    | j jrd| _ |  j |7  _ d S r`   )r   r0   )r+   r   r%   r%   r&   r     s    z CmfGanttTask.from_child_add_workc                 C   s    | j jrd| _ |  j |7  _ d S r`   )r   r0   r   r%   r%   r&   r     s    z CmfGanttTask.from_child_add_costc              	   C   s   | j jrd| _ |  j |7  _ | js,| j | _n2| j | jks^td| j  d| j d| j ddd | jjr| jjdkr| j jp|d| j j	pd }t
 8 | jjj }||   || |jd	d
 W 5 Q R X d S )Nr   u;   Превысили План на Трудозатраты ( <= +   ) в родительской задаче u_   . Увеличте в ней план или введите меньше ТрудозатратTrl   r?   Fr   )r   r0   r   r   rp   r   r   rC   r   rd   r   r   r$   r   rD   r*   r   r    )r+   r   rI   r%   r%   r&   r     s     


z%CmfGanttTask.from_child_agregate_workc              	   C   s   | j jrd| _ |  j |7  _ | js,| j | _n2| j | jks^td| j  d| j d| j ddd | jjr| jjdkr| j jp|d| j j	pd }t
 8 | jjj }||   || |jd	d
 W 5 Q R X d S )Nr   u1   Превысили План на Расходы (r   r   uY   . Увеличте в ней план или введите меньше РасходовTrl   r?   Fr   )r   r0   r   r
   rp   r   r   rC   r   rd   r   r   r$   r   rD   r*   r   r    r   r%   r%   r&   r     s     


z%CmfGanttTask.from_child_agregate_costc                 C   s   |  | dS )u   
        Оповестить родителя, что ему нужно пересчитать свой агрегат actual_complete
        N)r   )r+   r   r%   r%   r&   !from_child_update_actual_complete  s    z.CmfGanttTask.from_child_update_actual_completec              	   C   sV   | j jsdS t 8 | j jj }||   ||  |j	dd W 5 Q R X dS )uP  
        Оповещение родителя.
        Передаем себя, т.к. мы еще можем быть не сохранены и родитель не сможет получить наши данные
        Другой вариант: опочещать через отдельный таск в celery
        NFr   )
r   r   r   r   r$   r   rD   r*   r   r    )r+   rI   r%   r%   r&   r     s    

z1CmfGanttTask.notify_parent_update_actual_completec                 C   s&  | j sp| jrpdd | jddD }|ddddh }|rp| jj  | jjrp| jjjjd	rp| jjj	d
| jd | j
jr| jjrtddd | jjr| jjrtddd | jjs| jjr| jr| jr| j| jk rtddd | jjr| jstddd | jjr"| jjs"| jdkr"tddd d S )Nc                 S   s   h | ]\}}|qS r%   r%   ).0k_r%   r%   r&   	<setcomp>2  s     z0CmfGanttTask.check_user_input.<locals>.<setcomp>T)r   r   r   r   Zgantt_project_idzCmfProject:zPPP-TSK-SCHEDULE)obju   Нельзя менять поле "Фактические трудозатраты" у Summary-задач. Используйте поле "Фактические трудозатраты по собственным ресурсам задачи"rl   u   Нельзя менять поле "Фактические Затраты" у Summary-задач. Используйте поле "Фактические Затраты по собственным ресурсам задачи"u\   Нельзя выставить Дату начала позже Даты окончанияu   Будет доступно в версии Гант 2.0. Сейчас Вы можете использовать декомпозицию для множественных ресурсовr   u   Этот вариант рассчета процента завершения можно использовать только у Summary-задач)r   r   itemsr   Zroot_parentr   r   r   
startswithZcheck_project_role_accessr   rS   rp   r   r   r   r   r{   )r+   Zchanged_fieldsr%   r%   r&   check_user_input0  s8    
zCmfGanttTask.check_user_inputc                 C   s   | j jrd| _ | jjrd| _| jjr*d| _| jjr8d| _| jjrFd| _| jjrTd| _| jjrbd| _| jjrpd| _| j	jr~d| _	| j
jrd| _
| jjrd| _| jjrd| _d S r`   )r   r0   r   r
   r   r	   rh   r   r   r   rv   Zconstrain_slackZconstrain_slack_pctr^   r%   r%   r&   r   M  s0    zCmfGanttTask.fix_null_variablesTr   c                   s@   |r|    |   | jjr*| jjdd |   t j||S )NTr   )r   r   r   r   r    r   r)   )r+   r   argskwargsr,   r%   r&   r    h  s    zCmfGanttTask.save)TEXKOM_db_deletec                   s   |  ddg |rTt| j tjj| jddD ] }t| d |_|jdd q0n&| j	rz| j	j| krz| j	j
sztddd t j|d|i|S )	Nztask.op_gantt_taskztask.cmf_deletedT)r$   Zinclude_deletedr   uY   Нельзя удалять Оперативную гант задачу у задачи!rl   r   )rD   gdebugr   r   r}   rT   r$   r    r   Zcmf_deletedrp   r)   delete)r+   r   r   r   r   r,   r%   r&   r     s    
zCmfGanttTask.deletec                    sB   t  jf | | jrd S td tj| jdd td d S )N_CmfAutomationCrudTrigger startupdateZbefore_save_CmfAutomationCrudTrigger end)	r)   before_save_hookr   r   r   r   CmfAutomationCrudTrigger	crud_hookr   r+   r   r,   r%   r&   r     s    
zCmfGanttTask.before_save_hookc                    sB   t  jf | | jrd S td tj| jdd td d S )Nr   r   Z
after_saver   )	r)   before_save_data_hookr   r   r   r   r   r   r   r   r,   r%   r&   r     s    
z"CmfGanttTask.before_save_data_hook)N)N)F)F)F)F)N)N)<__name__
__module____qualname__classmethodr'   r*   propertyr6   r:   r;   r1   rE   rU   r_   re   ri   rr   rs   r   r   r   rc   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r   r   r   __classcell__r%   r%   r,   r&   r   	   sv   




$-
	
=
:@	B

!



(M
'

r   )
Zcmf.includeZdecimalr   r   r<   r   r   Z
cmf.fieldsZ#modules.gantt.fields.cmf_gantt_taskr   r%   r%   r%   r&   <module>   s   