U = el@sddlZddlZddlmZddlmZmZmZm Z m Z ddl Z ddl Tddl mZddlmZmZejZejZGdddejZejejdS) N) lru_cache)IteratorListNoReturnTupleUnion)*) cmf_calendar)MO relativedeltac @s eZdZeeeeedddZedDeeeeedddZedEeeeee ed d d Z edFeeeee ee ed d dZ edGeeeeeeee edddZ edHeeeeeeee edddZ edeee efdddZedeedddZedIdeeeedddZeedJeeeed!d"d#ZedKeedejjfeeed$d%d&ZedLdeeeeefed'd(d)Zedeeed*d+d,Z edMdeeed.d/d0Z!d1d2Z"d3d4Z#d5d6Z$d7d8Z%d9d:Z&d;d<Z'd=d>Z(ee)j*ddd?d@dAZ+edBdCZ,dS)N CmfCalendar) date_startdate_endreturnccs0tt||jdD]}|t|VqdS)u Генератор диапазона дат Args: date_start (Date): начальная дата date_end (Date): конечная дата Yields: Iterator[Date]: дата N)rangeintdaysdatetime timedelta)r rir./common/models/cmf_calendar.py _date_ranges zCmfCalendar._date_rangeN)date month_dayposition day_of_weekrc sz|r2tjjd}||kr$|}j|d}nDt}|jj}fdd|D}|dkrn||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: дата rdaycs$g|]}|jjkr|qSr)month).0Zweekrrrr 9sz=CmfCalendar._get_day_by_position_in_month..)clZ monthrangeyearr replaceZCalendarZmonthdatescalendar) rrrrZ days_in_monthrcalendarweeksZ day_of_weeksrr"r_get_day_by_position_in_monthsz)CmfCalendar._get_day_by_position_in_monthr) start_dateend_daterepeatsperiodrcCsRg}|dk rd}|}|rN|dkr*|d8}n ||kr4qN|||t|d7}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]: список дат NTrr)appendr )r,r-r.r/r current_daterrr every_day>s  zCmfCalendar.every_day)r,r-r.r/weekdaysrc Csg}|dk rd}|ttdd}|r|D]N}|t|d}||krp|dkrV|d8}n||krfd}qz|||s*qzq*|t|d7}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]: список дат NTr%weekdayrF)r*)r r r1) r,r-r.r/r4rr2r6rrrr every_week`s$  zCmfCalendar.every_week)r,r-r.r/rrrrc Csg}|dk rd}|}|rtj||||d} | |krX|dkrD|d8}n | |krNq|| |jd|} |j| d} | dd} |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]: список дат NTrrrr )r'r )r r+r1r r'r() r,r-r.r/rrrrr2rr r'rrr every_months(   zCmfCalendar.every_month)r,r-r.r rrrrc Cszg}|dk rd}|j|dd}|rvtj||||d} | |krb|dkrN|d8}n | |krXqv|| |j|jdd}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_ NTrr rr8)r')r(r r+r1r') r,r-r.r rrrrr2rrrr every_years"  zCmfCalendar.every_yearzmodels.CmfCalendarExclude) exclusionrc sDg}jj}jj}j}j}jdkrHjdkrHd}||||}njdkrpd}|j|||jj d}n̈jdkrd}|j|||jj j d }njd krd }|j ||||j d }nzjd krd}|j ||||j j d}nPjdkr(d}fddt dD}|j|||||d}nd}|||||}||fS)uw Расчитывает и возвращает исключительные дни и приоритет исключения Приоритеты исключений (7-самый высокий, 1-самый низкий): 7 - Ежедневно (каждый день) 6 - Ежегодно (в указанный день) 5 - Ежегодно (по позиции дня недели месяца) 4 - Ежемесячно (в указанный день) 3 - Ежемесячно (по позиции дня недели) 2 - Еженедельно 1 - Ежедневно (каждый N день) Args: exclusion (CmfCalendarExclude): исключение Returns: Tuple[int, List[Date]]: приоритет, список дат r3rZevery_year_day)r rZevery_year_week_day)r rrZevery_month_dayr$)rZevery_month_week_day)rrr7cs"g|]}td|dr|qS)rF)getattr)r!rr=rrr#'sz3CmfCalendar._calc_excluded_days..)r4)period_start_datevalueperiod_end_dater/Z repeat_timesZ repeat_typer3r<r rZmonth_week_positionZmonth_day_weekr:rr7) clsr=rr,r-r/r.priorityr4rrDr_calc_excluded_dayssP         zCmfCalendar._calc_excluded_days)r)rrcsd}g}|d}tjj|ddgdd|gdd|ggd }d }|D]*}||\} } || krH| |krH| }|}qH|r|jj} |jj} | d kr|d d |j Dntj j |ddgdd|gdd|ggd }|rt ||djdkr| ddg|j}t ||dj} t ||dj} | d krD|fdd |j Dtjj ||d} | stj||d} | _|j| _|j| _|j| _| | _| | _|| _| dS)u Пересчет указанного дня календаря Args: calendar (CmfCalendar): объект календаря date (Date): дата Returns: NoReturn Nrr intervals.*rE<=rG>=parentfieldsfilterrworkcss*|]"}|jjd|jjdgVqdS%H:%MN from_timerFstrftimeto_timer!intervalrrr Ssz*CmfCalendar.recalc_date.._typedefaultdefault_workweek.*default_workweek.intervals.*_intervals_total_minutesc3s6|].}|jjkr|jjd|jjdgVqdSrSday_weekrFrVrWrXrYr5rrr[es rOr)r6modelsCmfCalendarExcludelistrJ exclude_typerFintervals_total_minutesextend intervalsCmfCalendarWorkWeekgetrC load_fieldsdefault_workweekCmfCalendarDayrbr'r rday_typeinterval_total_minutes interval_jsonsave)rHr)rrrjday_num exclusionsZexclusion_priorityr=rIZ excluded_daysrprq calendar_dayrr5r recalc_date0s^     zCmfCalendar.recalc_dateT)r)r r back_recalcrcCsh|dkr$tj}|jddd}|dkrF|tddtjdd}|||D]}|||qRdS)u$ Пересчет дней календаря Если интервал не задан, делает пересчет от начала текущего года +2 года, итого 3 года Args: calendar (CmfCalendar): объект календаря date_start (Date, optional): начальная дата date_end (Date, optional): конечная дата back_recalc (bool, optional): пересчет назад только текущего года Nrr;rA)Zyearsr0)rtodayrr(r rrrw)rHr)r rrxryrrrrrecalc_calendar_dayuszCmfCalendar.recalc_calendar_dayF) calendar_idrdetailrcCsHtjjdd|gdd|ggdddgd}|s.dS|jj|jj|jjidS) NZ parent_id=rrprqrrrQrP)typerjrhr|)rdrorlrprFrrrq)rHr{rr|rrrr_get_dayszCmfCalendar._get_day)r)rr|rcCs t|d}|j|||d}|S)u< Получает день календаря Args: calendar (Union[str, CmfCalendar, CmfType]): календарь date (Date): дата detail (bool, optional): Returns: day (dict): информация о дне календаря r )r|)ZcmfutilZget_obj_id_by_anyr)rHr)rr|r{rrrrget_days zCmfCalendar.get_day)r)r'rc sN|dkrtj}nt|tjs,t|tjr2|j}tt|dd}tt|dd}t}|}||kr||j|j|j ddgd||<|tj dd7}q`| dd g|j }t jj|d d gd d d|gd d|ggdd|gdd|gggd}|D]n\} } | d} t|| dj} t|| dj} | | d<| | d<| dkrfdd|jD| d<q|D]}|jj}||jkrh|d} t|| dj} | dkr|tj dd7}qtt|| dj} ||}|r | |d<| |d<fdd|jD|d<|tj dd7}qtqht jj|d d gd d d|gd d|ggdd|gdd|gggd}fdd|D}t|D]n\}}}g}|jj} |jj} | dkrdd|jD}|D].} || }|r| |d<| |d<||d<qqz|D]V\} } t jj|| d }|st j|| d }| D]\}}t|||q&|qdS)!uQ Расчитывает все дни года с учетом рабочих недель и исключений и сохраняет их в БД Args: calendar (CmfCalendar): объект календаря year (Union[str, int, Date, Datetime], optional): год. Defaults to None. Nrr9r)rbr'r rrprqrrr0r^r_rrKORrErMrLrGrNrr\r`rprqrRcs4g|],}|jjkr|jjd|jjdgqSrTrarYr5rrr#s z-CmfCalendar.calc_one_year..rrr]cs4g|],}|jjkr|jjd|jjdgqSrrarYr5rrr#s csg|]}||fqSr)rJ)r!r=rHrrr#scss*|]"}|jjd|jjdgVqdSrSrUrYrrrr[sz,CmfCalendar.calc_one_year..rc)rnowr' isinstancerrdictr6r rrrmrnrdrkrfitemsrCrFrjrErGrlresortedrgrhrosetattrrs)rHr)r'Zstart_date_yearZ end_date_yearZ days_of_yearr2rnZ workweeksrZday_datartrprqZworkweekZcurrent_period_dateZ day_of_yearruZexcluded_dates_Zdatesr=rjrvZ field_namerFr)rHr6r calc_one_years                zCmfCalendar.calc_one_year)r)from_dtto_dtrcCsdd}d}ttjt|jpdd}|}|}||kr|||} | ddkrl|tjdd7}q8| d D]} tj| dd } tj| dd } tjj|| |d } tjj|| |d }|tddkr|tjdd7}|||| |}||7}qt|tjdd7}q8|S) u Получает продолжительность рабочего времени в минутах из указанного диапазона дат Args: calendar (CmfCalendar): объект календаря from_dt (datetime): дата и время начала диапазона to_dt (datetime): дата и время конца диапазона Returns: minutes_sum (int): сумма минут cSs8t||}t||}||kr0t||dSdSdS)N<r)maxminr total_seconds)rrinterval_start interval_endstartendrrr intersect=s   z3CmfCalendar.get_duration_minutes..intersectrZsecondsrrRrr0rjrTtzinfo) rtimezonerrrrstrptimetimecombine)rHr)rrrZ minutes_sumtzr2r-rrZrVrXrrZinterrrrget_duration_minutes0s*       z CmfCalendar.get_duration_minutesr)r)rdurationrcCs|s|Sttjt|jpdd}d}|dkrz*CmfCalendar.get_schedule..)rfrQrnrj)selfrschedulerrr get_scheduleszCmfCalendar.get_schedulecCs|jr dSdS)NTF)ru)rrrrris_exclude_dayszCmfCalendar.is_exclude_daycCs.|jr*t|jd|d}|dkr*dSdS)Nrr\ZweekendTF)rnrCr6)rrrprrris_weekend_days zCmfCalendar.is_weekend_dayc Cstjj}||}tddd}td}|D],}||jjkrN|jj}||jjkr6|jj}q6tjj |||d}tjj |||d}||fS)N;rr) rrZ astimezonerrrrVrFrXr) rr2rrZ_minZ_maxitemworkday_start_timeworkday_end_timerrrset_worktime_borders     zCmfCalendar.set_worktime_bordercCs|ddddddg|}d}|js(|S||kr|jdddd}||rX||sR||\}}||krz|krnn||kr|krnnq||kr||kr||kr||kr|t||d t||d 7}n||kr*||kr*|t||d 7}n||krX||krX|t||d 7}nj||krp|krnn|t||d 7}n4||kr|krPnn|t||d 7}q||kr ||kr |d t||d 7}q||kr||kr4|krRnn|t||d 7}nN||krv|t||d 7}n*||krP|t||||d 7}n||kr||kr|krnn|d t||d 7}nb||kr*|d t||d t||d 7}n&||kr|d t||d 7}nH||kr|d t||d 7}n|t||d 7}|t j d d 7}|jdddd}q(|S) u~ Вычисляет величину паузы попадающее в нерабочие интервалы rndefault_workweek.intervals#default_workweek.intervals.day_weekru$default_workweek.intervals.from_time"default_workweek.intervals.to_timer)ZhourZminutesecondrirr0) rmrnr(rrrrrrrr)rrstopr2Znon_working_timeZ begin_dayrrrrrcalc_non_working_pause_durationsx  $ $      z+CmfCalendar.calc_non_working_pause_durationcCs|ddddddg|js0t||dS|}d}||kr||rX||s||\}}||krz|krnn4||kr|krnn|t||d7}qt||ddkr4t||ddkrnLt||ddkr|t||d7}n|t||d7}nLt||ddkrh|t||d7}n|t||d7}|tjd d 7}q8|S) u Вычисляет величину паузы попадающее в рабочие интервалы чтобы правильно скорректировать elapsed_time rnrrrurrrrrr0) rmrnrrrrrrr)rpause_interval_start_timepause_interval_stop_timer2pauserrrrrcalc_work_pause_durations@   z$CmfCalendar.calc_work_pause_durationc Cs^tjjdddgdddgdddgdd|ggddd d d d d gd}|D]}|j|qBdS)u Обработчик смотрит только активные циклы (не на триггерной паузе и не фуллтайм) Zbreachedr}FalseZpausedZ stop_timeZNULLzsla_goal.calendarZsla_goalcodeZwithin_calendar_hoursrrlast_time_updater~N)rdZCmfSDeskSlaCyclerfrZset_nowrs)rZcyclescyclerrrcalendar_handlerHs  zCmfCalendar.calendar_handler)Z only_onceZ system_taskcOs2tjjddddddgd}|D]}t|qdS)Nrnrrrurr)rP)rdr rfr)_args_kwargsZ calendarsr)rrrcelery_hourly_hookZszCmfCalendar.celery_hourly_hookcCst|jdSr)Zcmf_deferred_taskrrrrr hourly_hookszCmfCalendar.hourly_hook)NNN)NNr)NNrN)NNrNNN)NNNNNN)NNT)F)F)N)r)-__name__ __module__ __qualname__ staticmethodDaterrrr+rr3r7r:r< classmethodrrJrrwboolrzrrrrrZcmfrPZCmfTyperDatetimerrrrrrrrrrZ celery_appZtaskrrrrrrr s! *3/@D}74T3  -r )r)r&r functoolsrtypingrrrrrZpytzZ cmf.includeZ common.fieldsr Zdateutil.relativedeltar r rrrr ZAPPZHOOK_CRON_HOURLYr1rrrrrs$