U e @sddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z ddl Z ddlTddlmZddlmZmZdd d d d d ddddddd ZejZejZejZGdddejZdS)N) lru_cache)deque)IteratorListNoReturnTupleUnion)*) cmf_calendar)MO relativedeltauянвuфевuмарuапрuмаяuиюнuиюлuавгuсенuоктuнояuдек)  c sneZdZeeeeedddZedNeeeeedddZedOeeeee ed d d Z edPeeeee ee ed d dZ edQeeeeeeee edddZ edReeeeeeee edddZ edSeeeeefdddZedTeeedddZedeee efdddZedeeddd ZedUdeeeed"d#d$ZeedVeedejjfeeed&d'd(Z edWdeeeeefed)d*d+Z!dXeeed-d.d/Z"eeed0d1d2Z#edeeed3d4d5Z$edYdeeed6d7d8Z%d9d:Z&ee'd!d!d;d<d=d>Z(d?d@Z)dAdBZ*dCdDZ+dEdFZ,fdGdHZ-dZdJdKZ.fdLdMZ/Z0S)[ CmfCalendar) date_startdate_endreturnccs0tt||jdD]}|t|VqdS)u Генератор диапазона дат Args: date_start (Date): начальная дата date_end (Date): конечная дата Yields: Iterator[Date]: дата r N)rangeintdaysdatetime timedelta)rrir#./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).0Zweekr&r)r#r$ Bsz=CmfCalendar._get_day_by_position_in_month..r)clZ monthrangeyearr+replaceZCalendarZmonthdatescalendar) r&r'r(r)Z days_in_monthr*calendarweeksZ day_of_weeksr#r-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]: список дат NTr r)appendr )r6r7r8r9r current_dater#r#r$ every_dayGs  zCmfCalendar.every_day)r6r7r8r9weekdaysrc 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/weekdayr F)r4)r r r;) r6r7r8r9r>rr<r@r*r#r#r$ every_weekis$  zCmfCalendar.every_week)r6r7r8r9r'r(r)rc 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]: список дат NTr'r(r)r r)r1r+)rr5r;r+r1r2) r6r7r8r9r'r(r)rr<r*r+r1r#r#r$ every_months(   zCmfCalendar.every_month)r6r7r8r+r'r(r)rc 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_ NTr r+r*rB)r1)r2rr5r;r1) r6r7r8r+r'r(r)rr<r*r#r#r$ 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)r r&todaycombinetimer! cmf_alert)rFrGrKr#r#r$normalize_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<)modelsrrO total_seconds)rFrGr#r#r$get_interval_minutess 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]]: приоритет, список дат Nr=r rZevery_year_dayr)r+r'Zevery_year_week_dayr)r+r(r)Zevery_month_dayr)r'Zevery_month_week_dayr)r(r)rArcs"g|]}td|dr|qS)r)F)getattr)r,r"rTr#r$r.`sz2CmfCalendar.calc_excluded_days..)r>)period_start_datevalueperiod_end_dater9Z repeat_times repeat_typer=rEr+r'month_week_positionmonth_day_weekrCrrA) clsrTrpriorityr6r7r9r8r>r#rVr$calc_excluded_days'sT         zCmfCalendar.calc_excluded_days)r3r&rcsd}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 Nr*r intervals.*rZ!=rW<=rY>=parentfieldsfilterrworkcss*|]"}|jjd|jjdgVqdS%H:%MNrFrXstrftimerGr,intervalr#r#r$ sz*CmfCalendar.recalc_date.._typedefaultdefault_workweek.*default_workweek.intervals.*_intervals_total_minutesc3s6|].}|jjkr|jjd|jjdgVqdSriday_weekrXrFrlrGrmr?r#r$ros rer&)r@rQCmfCalendarExcludelistr_ exclude_typerXintervals_total_minutesextend intervalsCmfCalendarWorkWeekgetrU load_fieldsdefault_workweekCmfCalendarDayrvr1r+r*day_typeinterval_total_minutes interval_jsonsave)r]r3r&r*r}day_num exclusionsZexclusion_priorityrTr^Z excluded_daysrr calendar_dayr#r?r$ recalc_dateisj      zCmfCalendar.recalc_dateT)r3rr 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): пересчет назад только текущего года Nr rDr)yearsr:)r rKr&r2r r!r%r)r]r3rrrrKr&r#r#r$recalc_calendar_dayszCmfCalendar.recalc_calendar_dayF)r3r&detailrcCsxt|d}tjjdd|gdd|ggdddgd}|sdtj|d gd }td |d d |jddd|j|j |j idS)u< Получает день календаря Args: calendar (Union[str, CmfCalendar, CmfType]): календарь date (Date): дата detail (bool, optional): Returns: day (dict): информация о дне календаря rZ parent_id=r&rrrrgrfdirtyrfu День z%d.%m.%Yu* не найден в календаре "uO". Попробуйте запустить пересчет календаря.TrI)typer}r{r) ZcmfutilZget_obj_id_by_anyrQrsgetZ get_obj_by_idrNnamerrr)r]r3r&rZ calendar_idr*r#r#r$get_days zCmfCalendar.get_day)r3r1rc 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. Nr rr)rvr1r+r*rrrr:rrrsr r`ORrWrcrbrYrdr*rprtrrrhcs4g|],}|jjkr|jjd|jjdgqSrjrurmr?r#r$r.*s z-CmfCalendar.calc_one_year..rrqcs4g|],}|jjkr|jjd|jjdgqSrrurmr?r#r$r.@s rZracs"g|]}|jr||fqSr#)rZr_)r,rT)r]r#r$r.]scSs(g|] }|jjd|jjdgqSrrkrmr#r#r$r.fs)rer1rfTZ only_datarw)!r nowr1 isinstancer&rdictr@r+r*r!rrrQr~ryitemsrUrXr}rWrYrrxsortedrzr{rpopsortsetattr is_changedr)r]r3r1Zstart_date_yearZ end_date_yearZ days_of_yearr<r workweeksr&Zday_datarrrworkweekZcurrent_period_dater}Z day_of_yearrZexcluded_dates_datesrTZ calendar_daysrZ field_namerXr#)r]r@r$ 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. rru@Параметр back_years должен быть от 0 до 10TrIrcalc_num_yearsNrrFr)rNrrr rr1rrrrQrrr&Z calc_untilr cache_clearr)selfrr current_yearZ from_yearZto_yearr1r#r#r$recalc_calendars    zCmfCalendar.recalc_calendar)from_dtto_dtrcCsXd}ttjt|jpdd}|}|}tjjdd|gdddgdd|gdd |ggdd d gd }|D]}|j|kr|j|kr||jd 7}qj|j D]} tj | dd } tj | dd } tjj |j| |d} tjj |j| |d} | t ddkr| tjdd7} t || }t|| }||krDt||nd}||7}qqj|S)u Получает дельту рабочего времени в секундах между начальной и конечной датой и временем Args: from_dt (datetime): дата и время начала диапазона to_dt (datetime): дата и время конца диапазона Returns: work_timedelta (int): дельта рабочего времени в секундах rZsecondsrerrrhr&rcrbrrrrPrjr tzinfor:)r timezoner!rr&rQrZslistrrstrptimerMrLmaxminrR)rrrZwork_timedeltatzr6r7 work_dayswork_dayrnrFrGinterval_start interval_endstartendZ time_periodr#r#r$get_work_timedeltas8      zCmfCalendar.get_work_timedelta)r3rrrcCsRdd}d}ttjt|jpdd}|}|}||krN|||} | ddkrl|tjdd7}q8||kr||kr|| d 7}|tjdd7}q8| d D]} tj| dd } tj| dd } tjj|| |d } tjj|| |d }|tddkr$|tjdd7}|||| |}||7}q|tjdd7}q8|S) u Получает продолжительность рабочего времени в минутах из указанного диапазона дат Args: calendar (CmfCalendar): объект календаря from_dt (datetime): дата и время начала диапазона to_dt (datetime): дата и время конца диапазона Returns: minutes_sum (int): сумма минут cSs8t||}t||}||kr0t||dSdSdS)NrPr)rrrrR)rrrrrrr#r#r$ intersects   z3CmfCalendar.get_duration_minutes..intersectrrrrhr r:r{r}rjr) r rr!rr&rrrMrL)r]r3rrrZ minutes_sumrr<r7r*rnrFrGrrZinterr#r#r$get_duration_minutess2        z CmfCalendar.get_duration_minutes)r3rdurationrcCs|s|Sttjt|jpdd}d}|dkrt |d d d }t |} W5QRX| d} || krtd| d|d|q>g} dd| dD} | d} | d}|D].}|dtj| dd}|jdt|jd|j}tjjdd|gdddgdddgd d!d|gd"d|gggd#d$}|r~q|d%}|d&}|r| |}n||}d'}ttfd(d)| d'}ttfd*d)| d'}|r|d+}n|r|d,}|rFtj| d|d}|d-|jdt|jd|j7}d.|d/d0|j}d||d1|d2|||j|j||jdd3d4d5 }tjjd6d7d8|gd9gd4d:}|stj|d;}|D]\}}t |||q|j!rd#|_!|j"r|j#d4d<|dkr|$d=g|$d>g|j%}|r>|}n|}|j&} tj'j(||d9gd?}!|| |!}"|"D]\\}#}$|$r|#stj)|d;}#|$j*|#_*|$j+|#_+|$j,|#_,n |#rd4|#_!|#j"rn|#j#d4d<qnn,|$d@g|j&D]}#d4|#_!|#j#d4d<q|-|.|j#d4d<| /|qtjj(d6dA| gd6d7d.|dBgdCdd4ggdD}%|%D]}|j0d4dEqTq>|$dFg|1|j#d4d<d'S)Gu Метод для кнопки "Загрузить праздники". Добавляет в исключения календаря выходные/празничные дни из файла в формате JSON. Исключения добавляются на текущий и следующий год (при наличии соответствующих файлов). - файлы с данными находятся в каталоге /opt/eva-app/cmf/contrib/calendar/ - имя файла должено содержать только год (например, 2024.json) Добавления не происходит, если пользователь уже добавил исключение на такую же дату. При повторном выполнении метода происходит перезагрузка/восстановление/удаление, если поменялись данные в файле. r) zip_longest)PathuВыходной деньuРабочий день)r rr Zcontribr3z.jsonrzutf-8)encodingr1uГод (uL) указанный в файле не соответствует году u в имени файла cSsi|]}|d|dqS)idrr#)r,holidayr#r#r$ sz-CmfCalendar.load_holidays..holidays transfersrr&.z%Y.%m.%d rerrZr=r9rrWrYF)rgis_autorrNcs|dkS)NfromritemZdate_r#r$z+CmfCalendar.load_holidays..cs|dkS)Ntorrrr#r$rrrru за Zexclude_z%Y_%m_%d:z ()rT) rZrrrWrYr+r'r\r[rrZLIKE%r )rgrfZinclude_deleted)rerr`r)rervrfr}zNOT INz_%rrg)rr)2 itertoolsrpathlibrr rr1ZconfigZ CMF_FOLDERstrZ with_suffixexistsopenZjsonloadsreadrZloggingZwarningrr&r* MONTHS_SHORTr+rQrxrnextrgrr@rrZ cmf_deletedrrrrr}ZCmfCalendarWorkWeekIntervalryZCmfCalendarExcludeIntervalrFrGZinterval_minutesZ_calc_intervals_total_minutesZ_calc_exclude_typer;delete _set_as_dirty)&rrrZ day_typesrrr1Z file_pathfZ calendar_dataZ calendar_yearZ exclude_codesrrrZ date_datar&Zdate_strZexcluderrrZtransferZ transfer_fromZ transfer_toZ transfer_daterZ exclude_dataZfieldrXrr@Zexclude_intervalsZworkweek_intervalsr}Zexclude_intervalZworkweek_intervalZexcludesr#rr$ load_holidaysms                $                zCmfCalendar.load_holidayscCsd|_tjjdS)u Помечает, что календарь необходимо пересчитать и сбрасывает кэш дней TN)rrQrrrrr#r#r$rszCmfCalendar._set_as_dirtycCs|jjs dS|jstdddd|jd}tjjdd|jgd d dggd }|rtd |_|jdd d|jd|}t|dS)uX Изменяет флаг "Календарь по умолчанию" NuДля сброса флага "Календарь по умолчанию" вы должны установить его у любого другого календаря.TrIufВо всех новых проектах будет использоваться календарь ""rra is_defaultrrFru^Сбросили флаг "Календарь по умолчанию" у календаря "z". ) rrrNrrQrrrr)rmsgZcurrent_default_calendarr#r#r$_set_is_default's"  zCmfCalendar._set_is_defaultcCs |jjrd|_|jjrd|_dS)u} Исправляет нулевые значения, если приходит null выставляет 0 rN)rZis_nullrrr#r#r$fix_null_value@szCmfCalendar.fix_null_valuec s|jrvt|ddrtjj|jdddgd}|j|_|j|_|jD]h}| }||_ d|_ |j dd| d g|jD]}| }||_ |j ddqz|jsD|jsD||_qD|jD]P}| } || _ | j dd| d g|jD]}| }| |_ |j ddqqnntjj} ttj| |_tj|d d } td D]} t| d | ddqH| j dd| |_||t|jdkrtddd|jj r|!t"j ||S)Ntemplater z workweeks.*z exclusions.*)rrfFTrr`uПо умолчанию)rerrr*rpZweekendru\Количество лет расчета не должно быть больше 10 летrI)#Zis_newrUrQrrrrrrZcloneresystemrrr}rWrYrrr rZ astimezonerrZ utcoffsetrRr~rrrrrNrrsuper) rargskwargsZtemplate_calendarZtemplate_workweekZ new_workweekZtemplate_intervalZ new_intervalZtemplate_exclusionZ new_exclusionrrr" __class__r#r$rJsV             zCmfCalendar.savec csd}|}tg}||kr|s^|tjjdd|gdd|gdddggdddd d gdgdd gd |}|Vt|jj tj t j d d}|d 7}qdS)Nrrez==r&rcrrhrrzparent.timezoner)rgrfZorder_byslicer r:)rr|rQrrypopleftDatetimerLr&rXrrMr r!)rr6limitnZ next_datebufferrr#r#r$rs" &zCmfCalendar.work_dayscs|dg|jrtdddtjj|dgd}|rZddd |D}td |ddtjj|d }|rtd |d ddt j ||S)NruНельзя удалить календарь по умолчанию. Сначала назначьте любой другой календарь, как календарь по умолчанию.TrIr)r3rfz, cSsg|]}d|jdqS)r)r)r,Zprojectr#r#r$r.sz&CmfCalendar.delete..ukНевозможно удалить календарь, он используется в проектах: )r3uYНевозможно удалить календарь, он используется у u пользователей) rrrNrQZ CmfProjectryjoinZ CmfPersoncountrr)rrrZprojectsZ projects_strZ person_countrr#r$rs"   zCmfCalendar.delete)NNN)NNr )NNr N)NNr NNN)NNNNNN)NN)NN)NNT)F)N)rF)r)r)1__name__ __module__ __qualname__ staticmethodDaterr%rr5rr=rArCrETimerrrOrS classmethodr_rrboolrrrrZcmfrfZCmfTyperrrrrrrrZcmf_deferred_jobrrrrrrrr __classcell__r#r#rr$rs! *3/ AM &.64  3 = r)r3r0r functoolsr collectionsrtypingrrrrrZpytzZ cmf.includeZ common.fieldsr Zdateutil.relativedeltar r rrMrr&r rrr#r#r#r$s2