U
    vuhG                     @   s   d dl Zd dlZd dlmZ d dlmZ d dlmZm	Z	m
Z
mZmZ d dlZd dlT d dlmZ d dlmZmZ dd	d
ddddddddddZejZejZejZG dd dejZdS )    N)	lru_cache)deque)IteratorListNoReturnTupleUnion)*)cmf_calendar)MOrelativedeltau   янвu   февu   марu   апрu   маяu   июнu   июлu   авгu   сенu   октu   нояu   дек)                        	   
         c                       s  e Zd Zejjddg Zeeeee dddZ	edQee
e
e
eddd	ZedReee
e
ee dddZedSeee
e
ee
 ee dddZedTeee
e
e
e
e
ee dddZedUeee
e
e
e
e
ee dddZedVeeeeef dddZedWeee
dddZedee
ee f dddZed eed d!d"ZedXd eeeed$d%d&ZeedYeed ej j!f eee"d(d)d*Z#edZd eee
eef ed+d,d-Z$d[e
eed.d/d0Z%eee
d1d2d3Z&ed\d eee
d4d5d6Z'ed]d ee
ed8d9d:Z(ed;d< Z)ed=d> Z*ee+d#d#d?d@dAdB Z,dCdD Z-dEdF Z.dGdH Z/dIdJ Z0 fdKdLZ1dMdN Z2 fdOdPZ3  Z4S )^CmfCalendarrecalc_calendarload_holidays)
date_startdate_endreturnc                 c   s0   t t||  jd D ]}| t| V  qdS )u   
        Генератор диапазона дат

        Args:
            date_start (Date): начальная дата
            date_end (Date): конечная дата

        Yields:
            Iterator[Date]: дата
        r   N)rangeintdaysdatetime	timedelta)r   r   i r%   ./common/models/cmf_calendar.py_date_range   s    zCmfCalendar._date_rangeN)date	month_daypositionday_of_weekr   c           	         sz   |r2t  j jd }||k r$|} j|d}nDt  }| j j} fdd|D }|dk rn|| n|d }|S )u:  
        Вспомогательный метод для every_month и every_year
        Возвращает дату по указанному дню месяца или по позиции дня недели месяца

        Args:
            date (Date): дата
            month_day (int, optional): день месяца (1-31). Defaults to None.
            position (int, optional): позиция недели месяца. Defaults to None.
                                      (0-первая, 1-вторая, 2-третья, 3-четвертая, 4-последняя)
            day_of_week (int, optional): день недели. Defaults to None.
                                         (0-ПН, 1-ВТ, 2-СР, 3-ЧТ ,4-ПТ ,5-СБ, 6-ВС)

        Returns:
            Date: дата
        r   )dayc                    s$   g | ]}| j  j kr| qS r%   )month).0Zweekr(   r+   r%   r&   
<listcomp>G   s      z=CmfCalendar._get_day_by_position_in_month.<locals>.<listcomp>r   )clZ
monthrangeyearr-   replaceZCalendarZmonthdatescalendar)	r(   r)   r*   r+   Zdays_in_monthr,   calendarweeksZday_of_weeksr%   r/   r&   _get_day_by_position_in_month-   s    z)CmfCalendar._get_day_by_position_in_monthr   )
start_dateend_daterepeatsperiodr   c                 C   sR   g }|dk	rd}| }|rN|dkr*|d8 }n
||kr4qN| | |t|d7 }q|S )u  
        Возвращает список дат для типа повторения "Ежедневно"

        Args:
            start_date (Date): начальная дата
            end_date (Date, optional): конечная дата. Defaults to None.
            repeats (int, optional): количество повторений. Defaults to None.
            period (int, optional): периодичность (каждый N день). Defaults to 1.

            Обязателен один из двух аргументов, либо end_date, либо repeats

        Returns:
            List[Date]: список дат
        NTr   r!   )appendr   )r8   r9   r:   r;   r!   current_dater%   r%   r&   	every_dayL   s    

zCmfCalendar.every_day)r8   r9   r:   r;   weekdaysr   c           	      C   s   g }|dk	rd}| t tdd }|r|D ]N}|t |d }|| krp|dkrV|d8 }n||krfd} qz|| |s* qzq*|t |d7 }q"|S )u]  
        Возвращает список дат для типа повторения "Еженедельно"

        Args:
            start_date (Date): начальная дата
            end_date (Date, optional): конечная дата. Defaults to None.
            repeats (int, optional): количество повторений. Defaults to None.
            period (int, optional): периодичность (каждая N неделя). Defaults to 1.
            weekdays (List[int], optional): список дней недели. Defaults to None.
                                            (0-ПН, 1-ВТ, 2-СР, 3-ЧТ ,4-ПТ ,5-СБ, 6-ВС)

            Обязателен один из двух аргументов, либо end_date, либо repeats

        Returns:
            List[Date]: список дат
        NTr1   weekdayr   F)r6   )r   r   r=   )	r8   r9   r:   r;   r@   r!   r>   rB   r,   r%   r%   r&   
every_weekn   s$    

zCmfCalendar.every_week)r8   r9   r:   r;   r)   r*   r+   r   c                 C   s   g }|dk	rd}| }|rt j||||d}	|	| krX|dkrD|d8 }n
|	|krNq||	 |jd | }
|j|
d  }|
d d }
|j||
d}q|S )u  
        Возвращает список дат для типа повторения "Ежемесячно"
        Расчитывает даты в двух режимах:
            1 - по указанному дню, если установлен параметр month_day
            2 - по позиции дня недели, если установлены параметры position и day_of_week

        Args:
            start_date (Date): начальная дата
            end_date (Date, optional): конечная дата. Defaults to None.
            repeats (int, optional): количество повторений. Defaults to None.
            period (int, optional): периодичность (каждый N месяц). Defaults to 1.
            month_day (int, optional): день месяца (1-31). Defaults to None.
            position (int, optional): позиция недели месяца. Defaults to None.
                                      (0-первая, 1-вторая, 2-третья, 3-четвертая, 4-последняя)
            day_of_week (int, optional): день недели. Defaults to None.
                                         (0-ПН, 1-ВТ, 2-СР, 3-ЧТ ,4-ПТ ,5-СБ, 6-ВС)

            Обязателен один из двух аргументов, либо end_date, либо repeats

        Returns:
            List[Date]: список дат
        NTr)   r*   r+   r   r   )r3   r-   )r   r7   r=   r-   r3   r4   )r8   r9   r:   r;   r)   r*   r+   r!   r>   r,   r-   r3   r%   r%   r&   every_month   s(     

zCmfCalendar.every_month)r8   r9   r:   r-   r)   r*   r+   r   c           
      C   sz   g }|dk	rd}| j |dd}|rvtj||||d}	|	| krb|dkrN|d8 }n
|	|krXqv||	 |j |jd d}q|S )u  
        Возвращает список дат для типа повторения "Ежегодно"
        Расчитывает даты в двух режимах:
            1 - по указанному дню, если установлены параметры month и month_day
            2 - по позиции дня недели месяца, если установлены параметры month, position и day_of_week

        Args:
            start_date (Date): начальная дата
            end_date (Date, optional): конечная дата. Defaults to None.
            repeats (int, optional): количество повторений. Defaults to None.
            month (int, optional): месяц (1-Янв, 2-Фев, ..., 12-Дек). Defaults to None.
            month_day (int, optional): день месяца (1-31). Defaults to None.
            position (int, optional): позиция недели месяца. Defaults to None.
                                      (0-первая, 1-вторая, 2-третья, 3-четвертая, 4-последняя)
            day_of_week (int, optional): день недели. Defaults to None.
                                         (0-ПН, 1-ВТ, 2-СР, 3-ЧТ ,4-ПТ ,5-СБ, 6-ВС)

            Обязателен один из двух аргументов, либо end_date, либо repeats

        Returns:
            List[Date]: _description_
        NTr   r-   r,   rD   )r3   )r4   r   r7   r=   r3   )
r8   r9   r:   r-   r)   r*   r+   r!   r>   r,   r%   r%   r&   
every_year   s"     

zCmfCalendar.every_year)	from_timeto_timer   c                 C   sf   t j }t j || } t j ||}t dd| krJ|t jdd7 }| |kr^tddd | |fS )u  
        Нормализует интервал времени и проверяет его на корректность.
        Так как у datetime нет времени 24:00,
        то время окончания 00:00 конвертируется в 00:00 следующего дня,
        как будто это интервал, например, 18:00 - 24:00

        Args:
            from_time (Time, optional): время начала интервала. Defaults to None.
            to_time (Time, optional): время окончания интервала. Defaults to None.

        Returns:
            Tuple[Datetime, Datetime]: дата и время начала, дата и время окончания
        r      Zhoursuo   Время начала должно быть раньше времени окончания интервалаTabort)r"   r(   todaycombinetimer#   	cmf_alert)rH   rI   rN   r%   r%   r&   normalize_time_interval   s    
z#CmfCalendar.normalize_time_intervalc                 C   s.   | r|sdS t j| |\} }||   d S )u?  
        Вычисляет в минутах интервал времени

        Args:
            from_time (Time, optional): время начала. Defaults to None.
            to_time (Time, optional): время окончания. Defaults to None.

        Returns:
            int: минуты
        r   <   )modelsr   rR   total_seconds)rH   rI   r%   r%   r&   get_interval_minutes  s    z CmfCalendar.get_interval_minuteszmodels.CmfCalendarExclude)	exclusionr   c           	         sb  g }d} j j} jj} j} j} jdkrN jdkrNd}| |||}n jdkrvd}| j||| j j	d}n jdkrd	}| j||| j j
 jd
}n jdkrd}| j|||| j	d}n jdkrd}| j|||| j
 jd}nh jdkr.d} fddtdD }| j|||||d}n, jdkrZ jdkrZd}| ||||}||fS )uw  
        Расчитывает и возвращает исключительные дни и приоритет исключения
        Приоритеты исключений (7-самый высокий, 1-самый низкий):
            7 - Ежедневно (каждый день)
            6 - Ежегодно (в указанный день)
            5 - Ежегодно (по позиции дня недели месяца)
            4 - Ежемесячно (в указанный день)
            3 - Ежемесячно (по позиции дня недели)
            2 - Еженедельно
            1 - Ежедневно (каждый N день)

        Args:
            exclusion (CmfCalendarExclude): исключение

        Returns:
            Tuple[int, List[Date]]: приоритет, список дат
        Nr?   r   r   Zevery_year_dayr   )r-   r)   Zevery_year_week_dayr   )r-   r*   r+   Zevery_month_dayr   )r)   Zevery_month_week_dayr   )r*   r+   rC   r   c                    s"   g | ]}t  d | dr|qS )r+   F)getattr)r.   r$   rW   r%   r&   r0   e  s      z2CmfCalendar.calc_excluded_days.<locals>.<listcomp>)r@   )period_start_datevalueperiod_end_dater;   Zrepeat_timesrepeat_typer?   rG   r-   r)   month_week_positionmonth_day_weekrE   r   rC   )	clsrW   r!   priorityr8   r9   r;   r:   r@   r%   rY   r&   calc_excluded_days,  sT    

 



 
zCmfCalendar.calc_excluded_days)r5   r(   r   c                    s  d}g }|   d  }tjj|ddgdddgdd|gd	d
|ggd}d}|D ]4}| |\}	}
|	dkrlqP||
krP|	|krP|	}|}qP|r|jj}|jj}|dkr|dd |j	D  ntj
j|ddgdd|gd	d
|ggd}|rt|| djdkr|ddg |j}t|| dj}t|| dj}|dkrX| fdd|j	D  tjj||d}|stj||d} |_|j|_|j|_|j|_||_||_||_|  |S )u   
        Пересчет указанного дня календаря

        Args:
            calendar (CmfCalendar): объект календаря
            date (Date): дата

        Returns:
            NoReturn
        Nr,   r	   intervals.*r]   !=rZ   <=r\   >=parentfieldsfilterr   workc                 s   s*   | ]"}|j jd |jjd gV  qdS %H:%MNrH   r[   strftimerI   r.   intervalr%   r%   r&   	<genexpr>  s   z*CmfCalendar.recalc_date.<locals>.<genexpr>_typedefaultdefault_workweek.*default_workweek.intervals.*_intervals_total_minutesc                 3   s6   | ].}|j j kr|jjd |jjd gV  qdS rl   day_weekr[   rH   ro   rI   rp   rA   r%   r&   rr     s    rh   r(   )rB   rT   CmfCalendarExcludelistrb   exclude_typer[   intervals_total_minutesextend	intervalsCmfCalendarWorkWeekgetrX   load_fieldsdefault_workweekCmfCalendarDayry   r3   r-   r,   day_typeinterval_total_minutesinterval_jsonsave)r`   r5   r(   r,   r   day_num
exclusionsZexclusion_priorityrW   ra   Zexcluded_daysr   r   calendar_dayr%   rA   r&   recalc_daten  sj    




zCmfCalendar.recalc_dateT)r5   r   r   back_recalcr   c                 C   sh   |dkr$t j   }|jddd}|dkrF|tdd t jdd }| ||D ]}| || qRdS )u$  
        Пересчет дней календаря
        Если интервал не задан, делает пересчет от начала текущего года +2 года, итого 3 года

        Args:
            calendar (CmfCalendar): объект календаря
            date_start (Date, optional): начальная дата
            date_end (Date, optional): конечная дата
            back_recalc (bool, optional): пересчет назад только текущего года
        Nr   rF   r   )yearsr<   )r"   rN   r(   r4   r   r#   r'   r   )r`   r5   r   r   r   rN   r(   r%   r%   r&   recalc_calendar_day  s    zCmfCalendar.recalc_calendar_dayF)r5   r(   detailr   c                 C   sx   t |d}tjjdd|gdd|ggdddgd}|sdt j|d	gd
}td|dd|j ddd |j|j	|j
i dS )u<  
        Получает день календаря

        Args:
            calendar (Union[str, CmfCalendar, CmfType]): календарь
            date (Date): дата
            detail (bool, optional):

        Returns:
            day (dict): информация о дне календаря
        r   Z	parent_id=r(   r   r   r   rj   ri   dirty)ri   u	   День z%d.%m.%Yu*    не найден в календаре "uO   ". Попробуйте запустить пересчет календаря.TrL   )typer   r~   r   )ZcmfutilZget_obj_id_by_anyrT   r   sgetZget_obj_by_idrQ   namer   r   r   )r`   r5   r(   r   Zcalendar_idr,   r%   r%   r&   get_day  s    zCmfCalendar.get_day)r5   r3   r   c                    s   |dkrt j  j}nt|t js.t|t j r4|j}t t|dd}t t|dd}t }|}||kr| |j|j|j	ddg d||< |t j
dd7 }qb|dd	g |j}tjj|d
dgddd|gdd|ggdd|gdd|ggdd|gdd|gggd}| D ]r\}	}
|	 d }t|| dj}t|| dj}||
d< ||
d< |dkrfdd|jD |
d< q|D ]}|jj}||jkrd}g }| d }t|| dj}|dkr|t j
dd7 }q|dkrt|| dj}fdd|jD }||}|r:||d< ||d< ||d< |t j
dd7 }qqtjj|d
dgdddgddd|gdd|ggdd|gdd|ggdd|gdd|ggggd} fdd|D }t|D ]r\}}}d}g }|jj}|dkr |jj}d d |jD }|D ].}	||	}|r||d< ||d< ||d< qqtjj||d
gd!}|D ]`}||jjd}
|
rP|
dg   |
 D ]\}}t||| q|jrP|j d"d# qP| D ]@\}	}
tj||	d$}|
 D ]\}}t||| q|   qdS )%uQ  
        Расчитывает все дни года с учетом рабочих недель и исключений и сохраняет их в БД

        Args:
            calendar (CmfCalendar): объект календаря
            year (Union[str, int, Date, Datetime], optional): год. Defaults to None.
        Nr   r      r   )ry   r3   r-   r,   r   r   r   r<   ru   rv   r	   rc   ORrZ   rf   re   r\   rg   r,   rs   rw   r   r   rk   c                    s4   g | ],}|j j kr|jjd |jjd gqS rm   rx   rp   rA   r%   r&   r0   /  s    z-CmfCalendar.calc_one_year.<locals>.<listcomp>r   rt   c                    s4   g | ],}|j j kr|jjd |jjd gqS r   rx   rp   rA   r%   r&   r0   E  s    r]   rd   c                    s"   g | ]}|j r ||f qS r%   )r]   rb   )r.   rW   )r`   r%   r&   r0   b  s   c                 S   s(   g | ] }|j jd |jjd gqS r   rn   rp   r%   r%   r&   r0   k  s   )rh   r3   ri   TZ	only_datarz   )!r"   nowr3   
isinstancer(   r    dictrB   r-   r,   r#   r   r   rT   r   r|   itemsrX   r[   r   rZ   r\   r   r{   sortedr}   r~   r   popsortsetattr
is_changedr   )r`   r5   r3   Zstart_date_yearZend_date_yearZdays_of_yearr>   r   	workweeksr(   Zday_datar   r   r   workweekZcurrent_period_dater   Zday_of_yearr   Zexcluded_dates_datesrW   Zcalendar_daysr   Z
field_namer[   r%   )r`   rB   r&   calc_one_year  s    

	










zCmfCalendar.calc_one_year)
back_yearsforcer   c                 C   s   |  dddg |dkr"t| j}|dk s2|dkr>tddd	 | jsL|sLdS tj j}|rd|| n|}|t| j }d}t	||D ]<}|st
t|d
d
}tj| | t
t|dd| _q|| _tjj  d| _| jdd dS )u  
        Пересчитывает дни календаря.
        Пересчет начинается от текущего года вперед на количество лет,
        указанных в параметре calc_num_years календаря.
        Если в параметре back_years указано положительное число, то пересчет начнется с прошлых лет.
        Например:
            back_years = 3
            calc_num_years = 5
            текущий год - 2023
            Будут пересчитаны года с 2020 по 2027 включительно

        Args:
            force (bool, optional): принудительный пересчет дней календаря. Defaults to False.
            back_years (int, optional): количество прошлых лет. Defaults to 0.
        r   calc_num_yearscalc_num_back_yearsNr   r   u@   Параметр back_years должен быть от 0 до 10TrL   r   r   r   Fr   )r   r    r   rQ   r   r"   r   r3   r   r   r(   rT   r   r   
calc_until	calc_fromr   cache_clearr   )selfr   r   current_yearZ	from_yearZto_yearr   r3   r%   r%   r&   r     s(    

zCmfCalendar.recalc_calendar)from_dtto_dtr   c                 C   sX  d}t t jt| jpdd}| }| }tjjdd| gdddgdd|gdd	|ggdd
dgd}|D ]}|j|kr|j|kr||jd 7 }qj|j	D ]}	t j 
|	d d }
t j 
|	d d }t j j|j|
|d}t j j|j||d}| t ddkr|t jdd7 }t||}t||}||k rDt||  nd}||7 }qqj|S )u  
        Получает дельту рабочего времени в секундах между начальной и конечной датой и временем

        Args:
            from_dt (datetime): дата и время начала диапазона
            to_dt (datetime): дата и время конца диапазона

        Returns:
            work_timedelta (int): дельта рабочего времени в секундах
        r   Zsecondsrh   r   r   rk   r(   rf   re   r   r   r   rS   rm   r   tzinfor<   )r"   timezoner#   r    r(   rT   r   Zslistr   r   strptimerP   rO   maxminrU   )r   r   r   Zwork_timedeltatzr8   r9   	work_dayswork_dayrq   rH   rI   interval_startinterval_endstartendZtime_periodr%   r%   r&   get_work_timedelta  s8     	


zCmfCalendar.get_work_timedelta)r5   r   r   r   c                 C   s  dd }d}t t jt|jpdd}| }| }	||	kr| ||}
|oh|| kph|| k}|
d dkr|s|t jdd7 }q8|r|}|
}|d dkr|| kr|t jdd7 }n|| kr|t jdd8 }| ||}q|}
|| kr*|| kr*||
d	 7 }|t jdd7 }q8|
d
 D ]}t j |d d }t j |d d }t j j|||d}t j j|||d}| t ddkr|t jdd7 }|||||}||7 }q2|t jdd7 }q8|S )u  
        Получает продолжительность рабочего времени в минутах из указанного диапазона дат

        Args:
            calendar (CmfCalendar): объект календаря
            from_dt (datetime): дата и время начала диапазона
            to_dt (datetime): дата и время конца диапазона

        Returns:
            minutes_sum (int): сумма минут
        c                 S   s8   t | |}t||}||k r0t||  d S dS d S )NrS   r   )r   r   r    rU   )r   r   r   r   r   r   r%   r%   r&   	intersect  s
    

z3CmfCalendar.get_duration_minutes.<locals>.intersectr   r   r   rk   r   r<   r~   r   rm   r   )	r"   r   r#   r    r(   r   r   rP   rO   )r`   r5   r   r   force_include_endsr   Zminutes_sumr   r>   r9   r,   Zforce_includeZcur_dateZcur_dayrq   rH   rI   r   r   Zinterr%   r%   r&   get_duration_minutes  sH    	
z CmfCalendar.get_duration_minutesr   )r5   r   durationr   c                 C   s  |s|S t t jt|jpdd}d}|dk r<d}t|}| }|dkrl| ||}|sf|d ntt|d }|D ]}	t j 	|	d d
 }
t j 	|	d d
 }t j j||
|d}t j j|||d}|
 t 
ddkr|t jdd	7 }|st||n|}|rt||n|}|||k r:t||  d
 nd8 }|dkrz qPqz|t j|s`dndd	7 }qD|r|t jt|d }n|t jt|d }|S )u$  
        Рассчитывает дату и время на основе длительности от указанной даты и времени

        Args:
            calendar (CmfCalendar): объект календаря
            from_dt (Datetime): дата и время от которой производится расчет
            duration (int, optional): длительность в минутах. Defaults to 0.

        Returns:
            to_dt (Datetime): рассчитанная дата и время
        r   r   FTr   rm   r   r   r<   rS   r1   )Zminutes)r"   r   r#   r    absr(   r   r|   reversedr   rP   rO   r   r   rU   )r`   r5   r   r   r   reverser>   r,   r   rq   rH   rI   r   r   r   r   r   r%   r%   r&   get_date_by_duration/  s8    
&z CmfCalendar.get_date_by_durationc                 C   sV  t jj| | d}ttjt| jp&dd}|d }t|d }|D ]}tj	|d d
 }tj	|d d
 }tjj| ||d}	tjj| ||d}
|	
 t
ddkr|

 t
ddkr dS |

 t
ddkrtjj|
tjj
 |d}
||	kr||
k r dS ||	kr. d	S ||
krH|dkrH d	S |d8 }qHdS )
u=   
        Вычисляет флаг calendar_paused
        r5   r(   r   r   r   r   rm   r   FT)rT   r   r   r(   r"   r   r#   r    lenr   rP   rO   r   )r5   
check_dater,   r   r   r$   rq   rH   rI   r   r   r%   r%   r&   get_sla_cycle_calendar_pausedd  s,    
(



z)CmfCalendar.get_sla_cycle_calendar_pausedc                 C   s  t jj| | d}ttjt| jp&dd}|d }d}t|d }|D ]6}tj	|d d
 }tj	|d d
 }	tjj| ||d}
tjj| |	|d}|

 t
ddkr|
 t
ddkr|tjdd	7 }|jdddd
}|} q|
 t
ddkr,tjj|tjj
 |d}||
krL||k rL|} qn0||
krb|
} qn||kr||dkr| q|d8 }qLtjj|tjj
 |d}|tjdd7 }|stj| |d}|S )uX   
        Вычисляет next_time_update - сл дата пересчета
        r   r   r   r   Nr   rm   r   rK   )ZminutesecondZmicrosecondr<   r5   r   )rT   r   r   r(   r"   r   r#   r    r   r   rP   rO   r4   r   r   get_sla_cycle_next_time_update)r5   r   r,   r   r   resultr$   rq   rH   rI   r   r   r%   r%   r&   r     s@    
(



z*CmfCalendar.get_sla_cycle_next_time_updatez	@minutely)Z	only_onceZ
system_jobZschedulec               	   C   s   t j  } dd| gdddgdddgg}dd	d
ddddddg	}d\}}tjj||||| gd}|sdq||7 }|D ]F}|j|_tj|j	j
|jjd|_tj|j	j
|jjd|j_|  qpqDd S )Nnext_time_updatere   Z	stop_timer   ZNULLcurrentTsla_goalzsla_goal.calendarzsla_goal.calendar.timezonecodeZpause_interval_stop_timeZpause_interval_start_timelast_time_updatecalendar_paused)r   d   )rj   ri   slicer   )r"   r   rT   ZCmfSDeskSlaCycler|   r   r   r   r   r   r5   r[   r   r   r   )r   rj   ri   r   limitZcyclescycler%   r%   r&   celery_minutely_hook  s&    
 z CmfCalendar.celery_minutely_hookc           &         s  ddl m} ddlm} ddd}tj j}||d f}|D ](}|tjdd	t	|
d
}| shq>t|ddd}t| }	W 5 Q R X |	d}
||
krtd|
 d| d|  q>g }dd |	dD }|	d}|	d}|D ].}|d tj|
 d  d }|j dt|j  d|j }tjjdd| gdddgdddgd d!d|gd"d|gggd#d$}|r~q|d%}|d&}|r||}n||}d'}tt fd(d)|d'}tt fd*d)|d'}|r|d+ }n|r |d, }|rFtj|
 d| d }|d-|j dt|j  d|j 7 }d.|d/d0| j }d|| d1| d2|||j|j| |jd d3 d4d5
}tjjd6d7d8| gd9gd4d:}|stj| d;}| D ]\}}t ||| q|j!rd#|_!|j"r|j#d4d< |dkr|$d=g | $d>g | j%}|r>| }n| }|j&} tj'j(||d9gd?}!|| |!}"|"D ]\\}#}$|$r|#stj)|d;}#|$j*|#_*|$j+|#_+|$j,|#_,n|#rd4|#_!|#j"rn|#j#d4d< qnn,|$d@g |j&D ]}#d4|#_!|#j#d4d< q|-  |.  |j#d4d< |/| qtjj(d6dA|gd6d7d.| dBgdCdd4ggdD}%|%D ]}|j0d4dE qTq>| $dFg | 1  | j#d4d< d'S )Gu  
        Метод для кнопки "Загрузить праздники".
        Добавляет в исключения календаря выходные/празничные дни из файла в формате JSON.
        Исключения добавляются на текущий и следующий год (при наличии соответствующих файлов).
            - файлы с данными находятся в каталоге /opt/eva-app/cmf/contrib/calendar/
            - имя файла должено содержать только год (например, 2024.json)
        Добавления не происходит, если пользователь уже добавил исключение на такую же дату.
        При повторном выполнении метода происходит перезагрузка/восстановление/удаление,
        если поменялись данные в файле.
        r   )zip_longest)Pathu   Выходной деньu   Рабочий день)r   r   r   Zcontribr5   z.jsonrzutf-8)encodingr3   u   Год (uL   ) указанный в файле не соответствует году u    в имени файла c                 S   s   i | ]}|d  |d qS )idr   r%   )r.   holidayr%   r%   r&   
<dictcomp>  s      z-CmfCalendar.load_holidays.<locals>.<dictcomp>holidays	transfersr   r(   .z%Y.%m.%d rh   r   r]   r?   r;   r   rZ   r\   F)rj   is_autor   r   Nc                    s   |  d kS )Nfromr   itemZdate_r%   r&   <lambda>.      z+CmfCalendar.load_holidays.<locals>.<lambda>c                    s   |  d kS )Ntor   r   r   r%   r&   r   /  r   r   r   u    за Zexclude_z%Y_%m_%d:z ()r   T)
r]   r   r   rZ   r\   r-   r)   r_   r^   r   r   ZLIKE%r	   )rj   ri   Zinclude_deleted)rh   r   rc   r   )rh   ry   ri   r   zNOT INz_%r   rj   )r   r   )2	itertoolsr   pathlibr   r"   r   r3   ZconfigZ
CMF_FOLDERstrwith_suffixexistsopenZjsonloadsreadr   ZloggingZwarningr   r(   r,   MONTHS_SHORTr-   rT   r{   r   nextrj   r   rB   r   r   Zcmf_deletedr   r   r   r   r   ZCmfCalendarWorkWeekIntervalr|   ZCmfCalendarExcludeIntervalrH   rI   Zinterval_minutesZ_calc_intervals_total_minutesZ_calc_exclude_typer=   delete_set_as_dirty)&r   r   r   Z	day_typesr   r   r3   Z	file_pathfZcalendar_dataZcalendar_yearZexclude_codesr   r   r   Z	date_datar(   Zdate_strZexcluder   r   r   ZtransferZtransfer_fromZtransfer_toZtransfer_dater   Zexclude_dataZfieldr[   r   rB   Zexclude_intervalsZworkweek_intervalsr   Zexclude_intervalZworkweek_intervalZexcludesr%   r   r&   r     s    





  



 $




zCmfCalendar.load_holidaysc                 C   s   d| _ tjj  dS )u   
        Помечает, что календарь необходимо пересчитать и сбрасывает кэш дней
        TN)r   rT   r   r   r   r   r%   r%   r&   r    s    zCmfCalendar._set_as_dirtyc                 C   s   | j jsdS | j stddd d| j d}tjjdd| jgd	d
dggd}|rtd|_ |jdd d|j d| }t| dS )uX   
        Изменяет флаг "Календарь по умолчанию"
        Nu   Для сброса флага "Календарь по умолчанию" вы должны установить его у любого другого календаря.TrL   uf   Во всех новых проектах будет использоваться календарь ""r   rd   
is_defaultr   r   Fr   u^   Сбросили флаг "Календарь по умолчанию" у календаря "z". )	r  r   rQ   r   rT   r   r   r   r   )r   msgZcurrent_default_calendarr%   r%   r&   _set_is_default  s"    
zCmfCalendar._set_is_defaultc                 C   s.   | j jrd| _ | jjrd| _| jjr*d| _dS )u}   
        Исправляет нулевые значения, если приходит null выставляет 0
        r   N)r   Zis_nullr   r   r  r%   r%   r&   fix_null_value  s    zCmfCalendar.fix_null_valuec                    s(  | j r~t| dd rtjj| jdddgd}|j| _|j| _|j| _|j	D ]h}|
 }| |_d|_|jdd |d	g |jD ]}|
 }||_|jdd q|jsL|jsL|| _qL|jD ]P}|
 }	| |	_|	jdd |d	g |jD ]}|
 }|	|_|jdd qqnntj  j}
ttj|
  | _tj| d
d}tdD ]}t|d| dd qP|jdd || _|   |   t| jdkrt ddd t| jdk rt ddd t| jdkrt ddd t| jdk rt ddd | jj!s| jj!r| "  t# j||S )Ntemplater	   zworkweeks.*zexclusions.*)r   ri   FTr   rc   u   По умолчаниюrh   r   r   r,   rs   Zweekend   u\   Количество лет расчета не должно быть больше 30 летrL   r   u]   Количество лет расчета не должно быть меньше 1 годаr   un   Количество лет расчета в прошлое не должно быть больше 10 летr   uf   Количество лет расчета в прошлое не должно быть меньше 0)$Zis_newrX   rT   r   r   r  r   r   r   r   Zclonerh   systemr   r   r   rZ   r\   r   r   r"   r   Z
astimezoner   r    Z	utcoffsetrU   r   r   r   r	  r
  rQ   r   r  super)r   argskwargsZtemplate_calendarZtemplate_workweekZnew_workweekZtemplate_intervalZnew_intervalZtemplate_exclusionZnew_exclusionr   r   r$   	__class__r%   r&   r     sd    



zCmfCalendar.savec              	   c   s  dd }d}|}t g }d }d}	d}
|sddd|gdd	d
ggg}|rp|	sX|dd	|g |rp|
sp|dd	|g |tjjdd	| g|gdddddgdgddgd |std| j d| j	  |
 }|j|kr|jd
kr||}|j|_|j|_d}	|j|krH|jd
krD|r,|j|_|j|_n||}|j|_|j|_d}
|jd
krX|}|V  t|jjtj tjdd  }|d7 }q$d S )Nc                 S   s0   t t| D ]}| | jdkr| |   S qd S )Nrk   )r   r   r   )bufferr$   r%   r%   r&   _get_next_work_day  s    z1CmfCalendar.work_days.<locals>._get_next_work_dayr   Fr   r(   rf   r   z==rk   rh   r   r   zparent.timezoner   )rj   ri   Zorder_byr   u   Календарь "u   " рассчитан до Tr   r<   )r   r=   r   rT   r   r|   StopIterationr   r   loadpopleftr(   r   r   r   DatetimerO   r[   r   rP   r"   r#   )r   r8   r   Zfinish_dater  nZ	next_dater  Zprev_work_dayZstart_date_givenZfinish_date_givenZdate_filterr   Znext_work_dayr%   r%   r&   r   
  sZ    	


&zCmfCalendar.work_daysc                    s   |  dg | jrtddd tjj| dgd}|rZddd	 |D }td
| dd tjj| d}|rtd| ddd tj	j| ddgd}|rddd	 |D }td| dd t
 j||S )Nr  u   Нельзя удалить календарь по умолчанию. Сначала назначьте любой другой календарь, как календарь по умолчанию.TrL   r   )r5   ri   z, c                 S   s   g | ]}d |j  d qS )r  )r   )r.   Zprojectr%   r%   r&   r0   T  s     z&CmfCalendar.delete.<locals>.<listcomp>uk   Невозможно удалить календарь, он используется в проектах: )r5   uY   Невозможно удалить календарь, он используется у u    пользователейzparent.namec                 S   s$   g | ]}d |j j d|j d qS )r  z: r  )r.   Zslar%   r%   r&   r0   ^  s     ue   Невозможно удалить календарь, он используется в целях: )r   r  rQ   rT   Z
CmfProjectr|   joinZ	CmfPersoncountZCmfSDeskSlaGoalr  r  )r   r  r  ZprojectsZprojects_strZperson_countZ	sla_goalsZsla_goals_strr  r%   r&   r  K  s.    

zCmfCalendar.delete)NNN)NNr   )NNr   N)NNr   NNN)NNNNNN)NN)NN)NNT)F)N)NF)F)r   )5__name__
__module____qualname__r
   r   Zapi_methodsstaticmethodDater   r'   r    r7   r   r?   rC   rE   rG   Timer   r  rR   rV   classmethodrb   r   r   boolr   r   r   r   Zcmfri   ZCmfTyper   r   r   r   r   r   r   r   r   Zcmf_deferred_jobr   r   r  r	  r
  r   r   r  __classcell__r%   r%   r  r&   r      s            !  
 *          3          /   
AM             .. F   4
(
9 3DAr   )r5   r2   r"   	functoolsr   collectionsr   typingr   r   r   r   r   ZpytzZcmf.includeZcommon.fieldsr
   Zdateutil.relativedeltar   r   r   rP   r"  r(   r!  r  r   r%   r%   r%   r&   <module>   s2           