U DjNe @sddlZddlZddlmZddlmZmZmZm Z m Z ddl Z ddl Tddl mZddlmZmZddd d d d d dddddd ZejZejZejZGdddejZejejdS)N) lru_cache)IteratorListNoReturnTupleUnion)*) cmf_calendar)MO relativedeltauянвuфевuмарuапрuмаяuиюнuиюлuавгuсенuоктuнояuдек)  c sPeZdZeeeeedddZedGeeeeedddZedHeeeee 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&d!d!d8d9d:Z'd;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]: дата 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# Asz=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 )r5r6r7r8r current_dater"r"r# every_dayFs  zCmfCalendar.every_day)r5r6r7r8weekdaysrc 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)r3)r r r:) r5r6r7r8r=rr;r?r)r"r"r# every_weekhs$  zCmfCalendar.every_week)r5r6r7r8r&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)r0r*)rr4r:r*r0r1) r5r6r7r8r&r'r(rr;r)r*r0r"r"r# every_months(   zCmfCalendar.every_month)r5r6r7r*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)rA)r0)r1rr4r:r0) r5r6r7r*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)rr%todaycombinetimer cmf_alert)rErFrJr"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<)modelsrrN total_seconds)rErFr"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(r@r cs"g|]}td|dr|qS)r(F)getattr)r+r!rSr"r#r-_sz2CmfCalendar.calc_excluded_days..)r=)period_start_datevalueperiod_end_dater8Z repeat_times repeat_typer<rDr*r&month_week_positionmonth_day_weekrBrr@) clsrSrpriorityr5r6r8r7r=r"rUr#calc_excluded_days&sT         zCmfCalendar.calc_excluded_days)r2r%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.*rY!=rV<=rX>=parentfieldsfilterrworkcss*|]"}|jjd|jjdgVqdS%H:%MNrErWstrftimerFr+intervalr"r"r# sz*CmfCalendar.recalc_date.._typedefaultdefault_workweek.*default_workweek.intervals.*_intervals_total_minutesc3s6|].}|jjkr|jjd|jjdgVqdSrhday_weekrWrErkrFrlr>r"r#rns rdr%)r?rPCmfCalendarExcludelistr^ exclude_typerWintervals_total_minutesextend intervalsCmfCalendarWorkWeekgetrT load_fieldsdefault_workweekCmfCalendarDayrur0r*r)day_typeinterval_total_minutes interval_jsonsave)r\r2r%r)r|day_num exclusionsZexclusion_priorityrSr]Z excluded_daysrr calendar_dayr"r>r# recalc_datehsj      zCmfCalendar.recalc_dateT)r2rr 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 rCr)yearsr9)rrJr%r1r r r$r)r\r2rrrrJr%r"r"r#recalc_calendar_dayszCmfCalendar.recalc_calendar_dayF)r2r%detailrcCs~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): информация о дне календаря rZ parent_id=r%rrrrfredirtyreu День z%d.%m.%Yu* не найден в календаре "uO". Попробуйте запустить пересчет календаря.TrH)typer|rzr) ZcmfutilZget_obj_id_by_anyrPrr~Z get_obj_by_idrMnamerrWrr)r\r2r%rZ calendar_idr)r"r"r#get_days zCmfCalendar.get_day)r2r0rc 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)rur0r*r)rrrr9rqrrrr_ORrVrbrarXrcr)rorsrrrgcs4g|],}|jjkr|jjd|jjdgqSrirtrlr>r"r#r-*s z-CmfCalendar.calc_one_year..rrpcs4g|],}|jjkr|jjd|jjdgqSrrtrlr>r"r#r-@s rYr`cs"g|]}|jr||fqSr")rYr^)r+rS)r\r"r#r-]scSs(g|] }|jjd|jjdgqSrrjrlr"r"r#r-fs)rdr0reTZ only_datarv)!rnowr0 isinstancer%rdictr?r*r)r rrrPr}rxitemsrTrWr|rVrXr~rwsortedryrzrpopsortsetattr is_changedr)r\r2r0Zstart_date_yearZ end_date_yearZ days_of_yearr;r workweeksr%Zday_datarrrworkweekZcurrent_period_dater|Z day_of_yearrZexcluded_dates_datesrSZ calendar_daysrZ field_namerWr")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 до 10TrHrcalc_num_yearsNrrFr)rMrrrrr0rrrrPrrr%Z calc_untilr cache_clearr)selfrr current_yearZ from_yearZto_yearr0r"r"r#recalc_calendars    zCmfCalendar.recalc_calendar)r2from_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)NrOr)maxminrrQ)rrinterval_start interval_endstartendr"r"r# intersects   z3CmfCalendar.get_duration_minutes..intersectrZsecondsrrgr r9r|ritzinfo) rtimezoner rr%rstrptimerLrK)r\r2rrrZ minutes_sumtzr;r6r)rmrErFrrZinterr"r"r#get_duration_minutess*       z CmfCalendar.get_duration_minutes)r2rdurationrcCs|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 r r Zcontribr2z.jsonrzutf-8)encodingr0uГод (uL) указанный в файле не соответствует году u в имени файла cSsi|]}|d|dqS)idrr")r+holidayr"r"r# fsz-CmfCalendar.load_holidays..holidays transfersrr%.z%Y.%m.%d rdrrYr<r8rrVrXF)rfis_autorrNcs|dkS)Nfromr~itemZdate_r"r#z+CmfCalendar.load_holidays..cs|dkS)Ntorrrr"r#rrrru за Zexclude_z%Y_%m_%d:z ()rT) rYrrrVrXr*r&r[rZrrZLIKE%r)rfreZinclude_deleted)rdrr_r)rdrurer|zNOT INz_%rrf)rr)2 itertoolsrpathlibrrrr0ZconfigZ CMF_FOLDERstrZ with_suffixexistsopenZjsonloadsreadr~ZloggingZwarningrr%r) MONTHS_SHORTr*rPrwsgetnextrfrr?rrZ cmf_deletedrrrrr|ZCmfCalendarWorkWeekIntervalrxZCmfCalendarExcludeIntervalrErFZinterval_minutesZ_calc_intervals_total_minutesZ_calc_exclude_typer:delete _set_as_dirty)&rrrZ day_typesrrr0Z file_pathfZ calendar_dataZ calendar_yearZ exclude_codesrrrZ date_datar%Zdate_strZexcluderrrZtransferZ transfer_fromZ transfer_toZ transfer_daterZ exclude_dataZfieldrWrr?Zexclude_intervalsZworkweek_intervalsr|Zexclude_intervalZworkweek_intervalZexcludesr"rr# load_holidays@s                $                zCmfCalendar.load_holidayscCsd|_tjjdS)u Помечает, что календарь необходимо пересчитать и сбрасывает кэш дней TN)rrPrrrrr"r"r#rszCmfCalendar._set_as_dirtycCs|jjs dS|jstdddd|jd}tjjdd|jgd d dggd }|rtd |_|jdd d|jd|}t|dS)uX Изменяет флаг "Календарь по умолчанию" NuДля сброса флага "Календарь по умолчанию" вы должны установить его у любого другого календаря.TrHufВо всех новых проектах будет использоваться календарь ""rr` is_defaultrrFru^Сбросили флаг "Календарь по умолчанию" у календаря "z". ) rrrMrrPrr~rr)rmsgZcurrent_default_calendarr"r"r#_set_is_defaults"  zCmfCalendar._set_is_defaultcCs |jjrd|_|jjrd|_dS)u} Исправляет нулевые значения, если приходит null выставляет 0 rN)rZis_nullrrr"r"r#fix_null_valueszCmfCalendar.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)Ntemplaterz workweeks.*z exclusions.*)rreFTrr_uПо умолчанию)rdrrr)roZweekendru\Количество лет расчета не должно быть больше 10 летrH)#Zis_newrTrPrr~rrrrZclonerdsystemrrr|rVrXrrrrZ astimezonerrZ utcoffsetrQr}rrrrrMrrsuper) rargskwargsZtemplate_calendarZtemplate_workweekZ new_workweekZtemplate_intervalZ new_intervalZtemplate_exclusionZ new_exclusionrrr! __class__r"r#rsV             zCmfCalendar.savecsh|dg|jrtdddtjj|dgd}|rZddd |D}td |ddtj||S) NruНельзя удалить календарь по умолчанию. Сначала назначьте любой другой календарь, как календарь по умолчанию.TrHr)r2rez, cSsg|]}d|jdqS)r)r)r+Zprojectr"r"r#r-csz&CmfCalendar.delete..ukНевозможно удалить календарь, он используется в проектах: ) rrrMrPZ CmfProjectrxjoinrr)rrrZprojectsZ projects_strrr"r#rZs  zCmfCalendar.delete)NNN)NNr )NNr N)NNr NNN)NNNNNN)NN)NN)NNT)F)N)rF)r)/__name__ __module__ __qualname__ staticmethodDaterr$rr4rr<r@rBrDTimerDatetimerNrR classmethodr^rrboolrrrrZcmfreZCmfTyperrrrrrrZcmf_deferred_jobrrrrrrr __classcell__r"r"rr#rs! *3/ AM!&74  3 =r)r2r/r functoolsrtypingrrrrrZpytzZ cmf.includeZ common.fieldsr Zdateutil.relativedeltar r rrLrr%rrrZAPPZHOOK_CRON_HOURLYr:rr"r"r"r#sB  [