U [Se@sddlZddlZddlmZddlmZmZmZm Z m Z ddl Z ddl Tddl mZddlmZmZejZejZejZGdddejZejejdS) N) lru_cache)IteratorListNoReturnTupleUnion)*) cmf_calendar)MO relativedeltac sVeZdZeeeeedddZedGeeeeedddZedHeeeee ed d d Z edIeeeee ee ed d dZ edJeeeeeeee edddZ edKeeeeeeee edddZ edLeeeeefdddZedMeeedddZedeee efdddZedeeddd ZedNdeeeed"d#d$ZeedOeedejjfeeed&d'd(Z edPdeeeeefed)d*d+Z!dQeeed-d.d/Z"edeeed0d1d2Z#edRdeeed3d4d5Z$d6d7Z%ee&j'd!d!d8d9d:Z(ed;d<Z)d=d>Z*d?d@Z+dAdBZ,fdCdDZ-fdEdFZ.Z/S)S 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: дата r)daycs$g|]}|jjkr|qSr)month).0Zweekrrrr :sz=CmfCalendar._get_day_by_position_in_month..)clZ monthrangeyearrreplaceZCalendarZmonthdatescalendar) 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 r0) r+r,r-r.r3rr1r5rrrr 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*r0rr&r') r+r,r-r.rrrrr1rrr&rrr every_months(   zCmfCalendar.every_month)r+r,r-rrrrrc 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_ NTrrrr7)r&)r'r r*r0r&) r+r,r-rrrrrr1rrrr 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=rArrrnormalize_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 rE 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]]: приоритет, список дат Nr2rZevery_year_day)rrZevery_year_week_day)rrrZevery_month_dayr#)rZevery_month_week_day)rrr6cs"g|]}td|dr|qS)rF)getattr)r rrJrrr"Xsz2CmfCalendar.calc_excluded_days..)r3)period_start_datevalueperiod_end_dater.Z repeat_times repeat_typer2r;rrZmonth_week_positionZmonth_day_weekr9rr6) clsrJrpriorityr+r,r.r-r3rrQrcalc_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.*rU!=rR<=rT>=parentfieldsfilterrworkcss*|]"}|jjd|jjdgVqdS%H:%MNr<rSstrftimer=r intervalrrr sz*CmfCalendar.recalc_date.._typedefaultdefault_workweek.*default_workweek.intervals.*_intervals_total_minutesc3s6|].}|jjkr|jjd|jjdgVqdSrbday_weekrSr<rer=rfr4rrrhs r^r)r5rGCmfCalendarExcludelistrX exclude_typerSintervals_total_minutesextend intervalsCmfCalendarWorkWeekgetrP load_fieldsdefault_workweekCmfCalendarDayror&rrday_typeinterval_total_minutes interval_jsonsave)rVr(rrrvday_num exclusionsZexclusion_priorityrJrWZ excluded_daysr|r} calendar_dayrr4r 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:rN)Zyearsr/)rrArr'r rrr)rVr(r rrrArrrrrecalc_calendar_dayszCmfCalendar.recalc_calendar_dayF)r(rdetailrcCs~t|d}tjjdd|gdd|ggdddgd}tj|d gd }|sdtd |d d |jddd|jj |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}r~r`r_dirtyr_u День z%d.%m.%Yu* не найден в календаре "uO". Попробуйте запустить пересчет календаря.Tr?)typervrtr) ZcmfutilZget_obj_id_by_anyrGr{rxZ get_obj_by_idrDnamer|rSr~r})rVr(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]`}||jjd} | rP| dg| D]\}}t|||q|jrP|j d"d#qP|D]@\} } t j|| d$}| D]\}}t|||q| qdS)%uQ Расчитывает все дни года с учетом рабочих недель и исключений и сохраняет их в БД Args: calendar (CmfCalendar): объект календаря year (Union[str, int, Date, Datetime], optional): год. Defaults to None. Nrr8r)ror&rrr|r}r~r/rkrlrrYORrRr\r[rTr]rrirmr|r}racs4g|],}|jjkr|jjd|jjdgqSrcrnrfr4rrr"#s z-CmfCalendar.calc_one_year..r~rjcs4g|],}|jjkr|jjd|jjdgqSrrnrfr4rrr"9s rUrZcs"g|]}|jr||fqSr)rUrX)r rJrVrrr"VscSs(g|] }|jjd|jjdgqSrrdrfrrrr"_s)r^r&r_TZ only_datarp)!rnowr& isinstancerrdictr5rrrryrzrGrwrritemsrPrSrvrRrTrxrqsortedrsrtr{popsortsetattr is_changedr)rVr(r&Zstart_date_yearZ end_date_yearZ days_of_yearr1rz workweeksrZday_datarr|r}workweekZcurrent_period_datervZ day_of_yearrZexcluded_dates_ZdatesrJZ calendar_daysrZ field_namerSr)rVr5r calc_one_years               zCmfCalendar.calc_one_yearr) back_yearsforcercCs|dks|dkrtddd|ddg|js8|s8dStjj}|rP||n|}|t|j}t||D]&}t j ||t t|d d |_ qlt j jd |_|jdd dS) 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 u@Параметр back_years должен быть от 0 до 10Tr?rcalc_num_yearsNr8rFr)rDryrrrr&rrrrGr rrZ calc_untilr cache_clearr)selfrrZ current_yearZ 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)NrFr)maxminrrH)rrinterval_start interval_endstartendrrr intersects   z3CmfCalendar.get_duration_minutes..intersectrZsecondsrrarr/rvrctzinfo) rtimezonerrrrstrptimerCrB)rVr(rrrZ minutes_sumtzr1r,rrgr<r=rrZinterrrrget_duration_minutess*       z CmfCalendar.get_duration_minutes)r(rdurationrcCs|s|Sttjt|jpdd}d}|dkr.ukНевозможно удалить календарь, он используется в проектах: ) ryrrDrGZ CmfProjectrrjoinrdelete)rrrZprojectsZ projects_strrrrrs  zCmfCalendar.delete)NNN)NNr)NNrN)NNrNNN)NNNNNN)NN)NN)NNT)F)N)rF)r)0__name__ __module__ __qualname__ staticmethodDaterrrr*rr2r6r9r;TimerDatetimerErI classmethodrXrrboolrrrstrZcmfr_ZCmfTyperrrrrrrZ celery_appZtaskrrrrrrr __classcell__rrrrr s! *3/ AM!&74    =r )r(r%r functoolsrtypingrrrrrZpytzZ cmf.includeZ common.fieldsr Zdateutil.relativedeltar r rCrrrrr ZAPPZHOOK_CRON_HOURLYr0rrrrrs&  *