U 1e2@sddlZddlZddlmZddlmZmZmZm Z m Z ddl Z ddl Tddl mZddlmZmZejZejZejZGdddejZejejdS) N) lru_cache)IteratorListNoReturnTupleUnion)*) cmf_calendar)MO relativedeltac sXeZdZeeeeedddZedJeeeeedddZedKeeeee ed d d Z edLeeeee ee ed d dZ edMeeeeeeee edddZ edNeeeeeeee edddZ edOeeeeefdddZedPeeedddZedeee efdddZedeeddd ZedQdeeeed"d#d$ZeedReedejjfeeed&d'd(Z edSdeeeeefed)d*d+Z!dTd,d-Z"edeeed.d/d0Z#edUdeeed2d3d4Z$d5d6Z%d7d8Z&d9d:Z'd;d<Z(d=d>Z)d?d@Z*dAdBZ+ee,j-d!d!dCdDdEZ.edFdGZ/fdHdIZ0Z1S)V 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 :sz=CmfCalendar._get_day_by_position_in_month..)clZ monthrangeyearr replaceZCalendarZmonthdatescalendar) rrrrZ days_in_monthrcalendarweeksZ day_of_weeksrr"r_get_day_by_position_in_month sz)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_weekas$  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_year) from_timeto_timercCsftj}tj||}tj||}tdd|krJ|tjdd7}||kr^tddd||fS)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)ZhoursuoВремя начала должно быть раньше времени окончания интервалаTabort)rrtodaycombinetimer cmf_alert)r=r>rBrrrnormalize_time_intervals  z#CmfCalendar.normalize_time_intervalcCs.|r|s dStj||\}}||dS)u? Вычисляет в минутах интервал времени Args: from_time (Time, optional): время начала. Defaults to None. to_time (Time, optional): время окончания. Defaults to None. Returns: int: минуты r<)modelsr rF total_seconds)r=r>rrrget_interval_minutes s z CmfCalendar.get_interval_minuteszmodels.CmfCalendarExclude) exclusionrc sbg}d}jj}jj}j}j}jdkrNjdkrNd}||||}n jdkrvd}|j|||jj d}njdkrd }|j|||jj j d }njd krd }|j ||||j d }njdkrd}|j ||||j j d}nhjdkr.d}fddt dD}|j|||||d}n,jdkrZjdkrZd}|||||}||fS)uw Расчитывает и возвращает исключительные дни и приоритет исключения Приоритеты исключений (7-самый высокий, 1-самый низкий): 7 - Ежедневно (каждый день) 6 - Ежегодно (в указанный день) 5 - Ежегодно (по позиции дня недели месяца) 4 - Ежемесячно (в указанный день) 3 - Ежемесячно (по позиции дня недели) 2 - Еженедельно 1 - Ежедневно (каждый N день) Args: exclusion (CmfCalendarExclude): исключение Returns: Tuple[int, List[Date]]: приоритет, список дат Nr3rZevery_year_day)r rZevery_year_week_day)r rrZevery_month_dayr$)rZevery_month_week_day)rrr7cs"g|]}td|dr|qS)rF)getattr)r!rrKrrr#Xsz2CmfCalendar.calc_excluded_days..)r4)period_start_datevalueperiod_end_dater/Z repeat_times repeat_typer3r<r rZmonth_week_positionZmonth_day_weekr:rr7) clsrKrpriorityr,r-r/r.r4rrRrcalc_excluded_dayssT         zCmfCalendar.calc_excluded_days)r)rrcsd}g}|d}tjj|ddgdddgdd|gd d |ggd }d }|D]4}||\} } | dkrlqP|| krP| |krP| }|}qP|r|jj} |jj} | d kr|dd|j Dntj j |ddgdd|gd d |ggd }|rt ||djdkr| ddg|j}t ||dj} t ||dj} | d krX|fdd|j Dtjj ||d} | stj||d} | _|j| _|j| _|j| _| | _| | _|| _| | S)u Пересчет указанного дня календаря Args: calendar (CmfCalendar): объект календаря date (Date): дата Returns: NoReturn Nrr intervals.*rV!=rS<=rU>=parentfieldsfilterrworkcss*|]"}|jjd|jjdgVqdS%H:%MNr=rTstrftimer>r!intervalrrr sz*CmfCalendar.recalc_date.._typedefaultdefault_workweek.*default_workweek.intervals.*_intervals_total_minutesc3s6|].}|jjkr|jjd|jjdgVqdSrcday_weekrTr=rfr>rgr5rrris r_r)r6rHCmfCalendarExcludelistrY exclude_typerTintervals_total_minutesextend intervalsCmfCalendarWorkWeekgetrQ load_fieldsdefault_workweekCmfCalendarDayrpr'r rday_typeinterval_total_minutes interval_jsonsave)rWr)rrrwday_num exclusionsZexclusion_priorityrKrXZ excluded_daysr}r~ calendar_dayrr5r recalc_dateasj      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;rO)Zyearsr0)rrBrr(r rrr)rWr)r rrrBrrrrrecalc_calendar_dayszCmfCalendar.recalc_calendar_dayF)r)rdetailrcCst|d}tjjdd|gdd|ggdddgd}tj|d gd }|sdtd |d d |jddd|jr|td|jd|j j |j j |j j idS)u< Получает день календаря Args: calendar (Union[str, CmfCalendar, CmfType]): календарь date (Date): дата detail (bool, optional): Returns: day (dict): информация о дне календаря r Z parent_id=rr}r~rrar`dirtyr`u День z%d.%m.%Yu* не найден в календаре "uO". Попробуйте запустить пересчет календаря.Tr@uКалендарь "u" не пересчитан, это может привести к некорректному вычислению дат или длительности. Запустите пересчет календаря в настройках.)typerwrur) ZcmfutilZget_obj_id_by_anyrHr|ryZ get_obj_by_idrEnamerr}rTrr~)rWr)rrZ calendar_idrrrrget_days" zCmfCalendar.get_day)r)r'rc s|dkrtjj}nt|tjs.t|tjr4|j}tt|dd}tt|dd}t}|}||kr||j|j|j ddgd||<|tj dd7}qb| dd g|j }t jj|d d gd d d|gd d|ggdd|gdd|ggd d|gdd|gggd}|D]r\} } | d} t|| dj} t|| dj} | | d<| | d<| dkrfdd|jD| d<q|D]}|jj}||jkrd} g}|d} t|| dj} | dkr|tj dd7}q| dkrt|| dj} fdd|jD}||}|r:| |d<| |d<||d<|tj dd7}qqt jj|d d gdddgd d d|gd d|ggdd|gdd|ggd d|gdd|ggggd}fdd|D}t|D]r\}}}d} g}|jj} | dkr|jj} d d|jD}|D].} || }|r| |d<| |d<||d<qqt jj||d gd!}|D]P}||jjd} | rP| D]\}}t|||qr|jrP|jd"d#qP|D]@\} } t j|| d$}| D]\}}t|||q|qdS)%uQ Расчитывает все дни года с учетом рабочих недель и исключений и сохраняет их в БД Args: calendar (CmfCalendar): объект календаря year (Union[str, int, Date, Datetime], optional): год. Defaults to None. Nrr9r)rpr'r rr}r~rr0rlrmrrZORrSr]r\rUr^rrjrnr}r~rbcs4g|],}|jjkr|jjd|jjdgqSrdrorgr5rrr#(s z-CmfCalendar.calc_one_year..rrkcs4g|],}|jjkr|jjd|jjdgqSrrorgr5rrr#>s rVr[csg|]}||fqSr)rY)r!rKrWrrr#[scSs(g|] }|jjd|jjdgqSrrergrrrr#bs)r_r'r`TZ only_datarq) rnowr' isinstancerrdictr6r rrrzr{rHrxrsitemsrQrTrwrSrUryrrsortedrtrur|popsetattr is_changedr)rWr)r'Zstart_date_yearZ end_date_yearZ days_of_yearr2r{Z workweeksrZday_datarr}r~workweekZcurrent_period_daterwZ day_of_yearrZexcluded_dates_ZdatesrKZ calendar_daysrZ field_namerTr)rWr6r calc_one_years              zCmfCalendar.calc_one_yearcCs|ddg|js|sdStjj}|t|j}t||D]&}tj ||t t|dd|_ q@tj j d|_|jdddS) u Пересчитывает дни календаря Args: force (bool, optional): принудительный пересчет дней календаря. Defaults to False. rcalc_num_yearsNr9rFTr)rzrrrr'rrrrHr rrZ calc_untilr cache_clearr)selfforceZ from_yearZto_yearr'rrrrecalc_calendars   zCmfCalendar.recalc_calendar)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)NrGr)maxminrrI)rrinterval_start interval_endstartendrrr intersects   z3CmfCalendar.get_duration_minutes..intersectrZsecondsrrbrr0rwrdtzinfo) rtimezonerrrrstrptimerDrC)rWr)rrrZ minutes_sumtzr2r-rrhr=r>rrZinterrrrget_duration_minutess*       z CmfCalendar.get_duration_minutesr)r)rdurationrcCs|s|Sttjt|jpdd}d}|dkrrrrrrrrrget_date_by_durations8  &z CmfCalendar.get_date_by_durationcsttfdd|jj}|S)Ncs|jtkSN)rpstrr6)xrrrz*CmfCalendar.get_schedule..)rsrar{rw)rrschedulerrr get_scheduleszCmfCalendar.get_schedulecCs|jr dSdS)NTF)r)rrrrris_exclude_day szCmfCalendar.is_exclude_daycCs.|jr*t|jd|d}|dkr*dSdS)NrrjweekendTF)r{rQr6)rrr}rrris_weekend_days zCmfCalendar.is_weekend_dayc Cstjj}||}tddd}td}|D],}||jjkrN|jj}||jjkr6|jj}q6tjj |||d}tjj |||d}||fS)N;rr) rrZ astimezonerrrDr=rTr>rC) 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~ Вычисляет величину паузы попадающее в нерабочие интервалы r{default_workweek.intervals#default_workweek.intervals.day_weekr$default_workweek.intervals.from_time"default_workweek.intervals.to_timer)ZhourZminutesecondrGirr0) rzr{r(rrrrrrIrr)rrstopr2Znon_working_timeZ begin_dayrrrrrcalc_non_working_pause_duration+sx  $ $      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 r{rrrrrrGrrr0) rzr{rrIrrrrr)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 Обработчик смотрит только активные циклы (не на триггерной паузе и не фуллтайм) ZbreachedrFalseZpausedZ stop_timeZNULLzsla_goal.calendarZsla_goalcodeZwithin_calendar_hoursrrlast_time_updaterN)rHZCmfSDeskSlaCyclersrZset_nowr)rZcyclescyclerrrcalendar_handlers  zCmfCalendar.calendar_handler)Z only_onceZ system_taskcOs2tjjddddddgd}|D]}t|qdS)Nr{rrrrrr)rHr rsr)_args_kwargsZ calendarsr)rrrcelery_hourly_hookszCmfCalendar.celery_hourly_hookcCst|jdSr)Zcmf_deferred_taskrrrrr hourly_hookszCmfCalendar.hourly_hookcs|jrHtj|dd}tdD]}t|d|ddq|jdd||_t|jd krbt d dd |jj rpd|_ |j j rtj j tj||S) NuПо умолчанию)r_rrLrrjrTr u\Количество лет расчета не должно быть больше 10 летr@)Zis_newrHrxrrrr{rrrErrr rrsuper)rargskwargsrr __class__rrrs    zCmfCalendar.save)NNN)NNr)NNrN)NNrNNN)NNNNNN)NN)NN)NNT)F)N)F)r)2__name__ __module__ __qualname__ staticmethodDaterrrr+rr3r7r:r<TimerDatetimerFrJ classmethodrYrrboolrrrrZcmfr`ZCmfTyperrrrrrrrrrrrrZ celery_appZtaskrrr __classcell__rrrrr s! *3/ AM& 74T3  - r )r)r&r functoolsrtypingrrrrrZpytzZ cmf.includeZ common.fieldsr Zdateutil.relativedeltar r rDrrrrr ZAPPZHOOK_CRON_HOURLYr1rrrrrs(