U
    d                     @   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dd ZdYddZdd Zdd ZdZd!d"Zd[d#d$Zd%d& Zd'd( Zd\d)d*Zd+d, Zd]d-d.Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Z d;d< Z!d=d> Z"d^d?d@Z#dAdB Z$dCdD Z%dEdF Z&dGdH Z'dIdJ Z(dKdL Z)d_dMdNZ*dOdP Z+dQdR Z,dSdT Z-dUdV fdWdX
Z.  Z/S )`r   c              	   C   sj   t  }|jr$|jjjdkr$|j|_|r6||_|j|_||_|j|_t	
  |jddd W 5 Q R X |S )NZ
CmfProjectF)emitnotify)modelsr   parentvalue
class_nameZprojectZgantt_projecttaskparent_taskcmfutilZdisable_notifysave)clsr   Zbaseline
gantt_task 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   ztask.logic_prefixztask.parent_task.logic_prefixztask.has_child_tasksztask.plan_start_dateztask.plan_end_date)supersave_preload_fields)selffields	__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)
sched_costis_null_perform_completeDecimalquantizer   r   resr   r   r   _perform_cost'   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   )sched_durationr   r    r!   r"   r   intr#   r   r   r   _perform_duration.   s
    zCmfGanttTask._perform_durationc                 C   s:   | j jrd S | j | j d }t|jtdtd}t|S r&   )
sched_workr   r    r!   r"   r   r)   r#   r   r   r   _perform_work6   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   r'   r   )sched_start_dater   sched_finish_datedatetimenowtimezoneZutctotal_secondsr)   r!   r"   r   )r   Zdelta_now_time_secZdelta_finish_time_secZperform_completer$   r   r   r   r    >   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r6   2-before)r   r   logic_prefixop_gantt_taskloadload_fieldsr   get_parent_plan_limitconstrain_start_typeconstrain_start_datemaxconstrain_finish_typeconstrain_finish_date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   r>   W   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 )Nr3   task.parent_task=filterr   r6   4-beforer6   1-after)r   has_child_tasksr	   r   listr   r?   r@   rD   rB   rC   rA   rE   rI   get_childrens_plan_limit)r   rG   rH   rI   r   Zch_start_limitZch_finish_limitZch_duration_limitr   r   r   rT   {   sH    




z%CmfGanttTask.get_childrens_plan_limitc                 C   sB   t | ddrd S d| _|  \| _| _| _|  \| _| _| _	d S )N_calc_plan_limits_cache_doneFT)
getattrrU   r>   _plan_start_limit_min_plan_finish_limit_max_plan_duration_limit_maxrT   _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   s0  | j r| jsd S | j| j  }| d d d }tt|d d}|d d }| jdkr| jdkr| jr|| j	r|t
ddd	 || _d S | jd
kr| jdkr| jr| j|k r| jr| j	rt
ddd	 t
d || _d S | jdkr,| jdkr,| jr| j|kr,| jr| j	rt
ddd	 t
d || _d S d S )N<      着&?      r6   u   Несовместимые ограничения Плановых Дат и фиксированной ДлительностиTabortrN   rP   u   Увеличили длительность, т.к. Плановые Даты ограничивают её минимальный размерr5   r8   u   Уменьшили длительность, т.к. Плановые Даты ограничивают её максимальный размер)r@   rC   r2   rA   mathceilr?   rB   r(   rE   	cmf_alert)r   delta
delta_days	work_daysr(   r   r   r   _calc_sched_duration   s2    z!CmfGanttTask._calc_sched_durationc                 C   sj   | j rP| jrP| j| j  }| d d d }tt|d d}|d d | _n| j jrf| jjrfd| _d S )Nr_   r`   ra   rb   rc   r   )actual_start_dateactual_finish_dater2   rA   rf   rg   actual_durationold)r   ri   rj   rk   r   r   r   _calc_actual_duration   s    z"CmfGanttTask._calc_actual_durationc                 C   s   | j jrd S | jjr | jjr d S | j d d }t|d }| jr| jdkr| jt|d | _|dkr| jjj	ddt
  jd	}| j|kr|tdd7 }|| _d S | jr| jdkr| jt|d | _d S | jdkr| jdkr| jrtd
 d S )Nrc   r_   ra   r6   )Zdaysrb      r   )ZhourZminutetzinfou   Плановый период может не соответствовать плановой длительности, т.к. все переменные зафиксированы)r(   r   r-   r.   rf   rg   rB   r   r   replacer/   r0   Z
astimezoners   r?   rE   rh   )r   rk   Zcalendar_daysZ
today_dater   r   r   _calc_sched_period   s*    
zCmfGanttTask._calc_sched_periodNc           
      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-completerJ   rK   actual_completerL   r   2-story_pointsr   agile_story_pointszstatus.status_typeCLOSEDr   3-costz4-fixed)r   rR   actual_complete_typer	   r   rS   lenidrx   r   ZCmfTaskr=   rz   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   s  |    |   | jdkr$| j| _n| jdkrT| j| _| jr| jrt| j| j| _np| jdkr| j| _| jr| jrt| j| j| _n@| j| _| jr| jrt| j| j| _| jr| jrt| j| j| _| jr| j| jkrt	ddd | jr| j| jk rt	ddd | j
jr| jr&| j
j| jk r| jr>| j
j| jkr| jsR| j
j| _n\| jdkr`nN| jdkr| j
j| jk r| j
j| _n&| jd	kr| j
j| jkr| j
j| _|   | jdk| _d S )
Nr6   r7   rO   u~   Нельзя указать Дату начала больше зафиксированной у дочерней задачиTrd   u   Нельзя указать Дату начала меньше зафиксированной у родительской задачи)1-earlyrO   )z2-latterr7   )rl   r^   r?   r@   r-   rW   rA   rZ   rD   rh   r   Zplan_start_dateru   lock_sched_start_dater]   r   r   r   #_do_constrain_start_date_is_changedQ  sT    





z0CmfGanttTask._do_constrain_start_date_is_changedc                 C   s  |    |   | jdkr$| j| _n| jdkrT| j| _| jr| jrt| j| j| _np| jdkr| j| _| jr| jrt| j| j| _n@| j| _| jr| jrt| j| j| _| jr| jrt| j| j| _| jr| jr| j| jkrt	ddd | jr| j| jk rt	ddd | j
jr| jr.| j
j| jk r| jrF| j
j| jkr| jsZ| j
j| _n\| jdkrhnN| jdkr| j
j| jkr| j
j| _n&| jdkr| j
j| jk r| j
j| _|   | jdk| _d S )Nr6   rQ   r9   u   Нельзя указать Дату окончания больше зафиксированной у родительской задачиTrd   u   Нельзя указать Дату окончания меньше зафиксированной у дочерней задачи)rl   r^   rB   rC   r.   r[   rA   rX   rD   rh   r   Zplan_end_dateru   lock_sched_finish_dater]   r   r   r   $_do_constrain_finish_date_is_changed  sV    





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 d S )	Nr6   u   Нельзя менять длительность, если зафиксирована дата начала и дата окончанияTrd   u   Нельзя указать Длительность больше зафиксированной у родительской задачиu   Нельзя указать Длительность меньше зафиксированной у дочерней задачи)from_sched_duration_changedu   Нельзя указать Длительность меньше чем Трудозатраты, когда установлена опция Фиксированные Ресурсы и Трудозатраты зафиксированы)r?   rB   rh   r^   rY   r(   r\   ru   
const_workr+   _do_sched_work_is_changed)r   from_sched_work_changedr   r   r   _do_sched_duration_is_changed  s     
z*CmfGanttTask._do_sched_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tj| j| d S )
Nr   uo   "Фактические трудозатраты" стал ниже нуля, выставляем в ноль.Trd   rv   r   r4   F
user_input)actual_workr   rh   r   rR   actual_myself_workr}   r+   is_not_nullrf   floorrD   rx   $notify_parent_update_actual_completenewrp   r   r:   r   disable_aclr;   r<   r=   r   from_child_add_workr   r	   ZCmfTimeTrackerZ"_op_gantt_actual_work_changed_hook)r   from_timetrackertmp_complete
work_deltarF   r   r   r   _do_actual_work_is_changed  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   "Фактические Затраты" стал ниже нуля, выставляем в ноль.Trd   r|   r   r4   Fr   )actual_costr   rh   r}   r   r   rf   r   rD   rx   r   r   r   r:   r   rp   r   r   r;   r<   r=   r   from_child_add_costr   )r   r   
cost_deltarF   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 )Nrv   r|   )rx   r   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   rp   r   rR   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 Nr   )actual_myself_costr   rp   r   rR   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 8 | jjj }||   || |jdd W 5 Q R X d S )Nu   Нельзя запланировать меньше Трудозатрат, чем запланировано в подзадачах ( < u`   ). Увеличте Трудозатраты или уменьшите их у подзадач.Trd   u   Указано Трудозатрат меньше, чем запланировано в подзадачах. Выставили uo   . Если нужно уменьшить Трудозатраты, уменьшите их у подзадач.)r   u   Нельзя указать Трудозатраты больше чем Длительность, когда установлена опция Фиксированные Ресурсы и Длительность зафиксированаr4   r   Fr   )r+   r   r   rR   agregat_workr   rh   rE   r(   r   const_resourcer   r:   r   rp   r   r   r;   r<   r=   r   from_child_agregate_workr   )r   r   r   rF   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
 8 | jjj }||   || |jd
d W 5 Q R X d S )Nu   Нельзя запланировать меньше Расходов, чем запланировано в подзадачах (r   uX   ). Увеличте Расходов или уменьшите их у подзадач.Trd   u|   Указано Расходов меньше, чем запланировано в подзадачах. Выставили ue   . Если нужно уменьшить Расходы, уменьшите их у подзадач.r4   r   Fr   )r   r   agregat_cost
const_costrh   r   r   r:   r   rp   r   r   r;   r<   r=   r   from_child_agregate_costr   r   r   rF   r   r   r   _do_sched_cost_is_changedN  s     

z&CmfGanttTask._do_sched_cost_is_changedc                 C   s   |    d S Nrq   r]   r   r   r    _do_actual_start_date_is_changedd  s    z-CmfGanttTask._do_actual_start_date_is_changedc                 C   s   |    d S r   r   r]   r   r   r   !_do_actual_finish_date_is_changedg  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 )Nrv   rw   ry   r|   )r}   r   r   rR   r   r   r]   r   r   r   #_do_actual_complete_type_is_changedj  s    



z0CmfGanttTask._do_actual_complete_type_is_changedc                 C   s   | j r|   d S r   )rE   rl   r]   r   r   r   _do_const_duration_is_changedv  s    z*CmfGanttTask._do_const_duration_is_changedc                 C   s   | j rd| _| j| _d S Nr6   )r   r?   r-   r@   r]   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| _d S r   )r   rB   r.   rC   r]   r   r   r   %_do_lock_sched_finish_date_is_changed  s    z2CmfGanttTask._do_lock_sched_finish_date_is_changedc                 C   s  | 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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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 )Nr   r   )-Zis_newr(   r+   r   r   r;   r<   r-   rA   r/   Zutcnowr?   ru   r   
is_changedr   r   r   r@   r   rC   rB   r   r   r   r   r   r   rx   r   r   r   r   rm   r   rn   r   r}   r   rE   r   r   r   r   r   )r   Zparent_sched_start_dater   r   r   _save  sr    








zCmfGanttTask._savec                 C   sl   |  |   |   |  j|7  _| jdd |d kr\| j| j |kr\| j| | _|   | jdd d S )NTr   Z	only_data)	r=   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   r{   r   r   )r   rm   r=   r   r   Zset_nowr   r   rn   rx   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)r=   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   r   )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   r   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_   . Увеличте в ней план или введите меньше ТрудозатратTrd   r4   Fr   )r   r   r   r+   rh   r   r   r:   r   rp   r   r   r;   r<   r=   r   r   r   )r   r   rF   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   . Увеличте в ней план или введите меньше РасходовTrd   r4   Fr   )r   r   r   r   rh   r   r   r:   r   rp   r   r   r;   r<   r=   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_complete6  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<   r=   r   r   r   )r   rF   r   r   r   r   <  s    

z1CmfGanttTask.notify_parent_update_actual_completec                 C   s   | j jr| jjrtddd | jjr8| jjr8tddd | jjsH| jjrl| jrl| jrl| j| jk rltddd | jjr| jstddd | j	jr| jjs| j	dkrtddd d S )	Nu   Нельзя менять поле "Фактические трудозатраты" у Summary-задач. Используйте поле "Фактические трудозатраты по собственным ресурсам задачи"Trd   u   Нельзя менять поле "Фактические Затраты" у Summary-задач. Используйте поле "Фактические Затраты по собственным ресурсам задачи"u\   Нельзя выставить Дату начала позже Даты окончанияu   Будет доступно в версии Гант 2.0. Сейчас Вы можете использовать декомпозицию для множественных ресурсовr   u   Этот вариант рассчета процента завершения можно использовать только у Summary-задач)
r   r   r   rR   rh   r   r@   rC   r   r}   r]   r   r   r   check_user_inputM  s*    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   r   r   r   r   r(   ro   r   r+   r   rx   Zconstrain_slackZconstrain_slack_pctr]   r   r   r   r   a  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   |  s    zCmfGanttTask.save)N)F)F)F)F)N)N)0__name__
__module____qualname__classmethodr   r   propertyr%   r*   r,   r    r>   rT   r^   rl   rq   ru   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   	   s^   




$-
).
:12

"



(?
'
r   )
Zcmf.includeZdecimalr   r   r/   r   rf   Z
cmf.fieldsZ#modules.gantt.fields.cmf_gantt_taskr   r   r   r   r   <module>   s   