i8UdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z  ddlm Z n#e$r ddlm Z YnwxYwddlZddlmZddlmZmZddlmZmZmZdd lmZdd lmZdd lmZdd l m!Z!erdd lm"Z"ddlm#Z#ddl$m%Z%ddl&m'Z'm(Z(ej)dkre e*e+ezezZ,de-d<n e e*Z,de-d<GddZ.ddgZ/dS)z:Core functionality: querying the calendar for occurrences.) annotationsN) TYPE_CHECKINGClassVar GeneratorOptionalSequence) TypeAlias)ComponentAdapter) DATE_MAX_DT DATE_MIN_DT)BadRuleStringFormatInvalidCalendarPeriodEndBeforeStart) OccurrenceID)Pages)SelectComponents)compare_greater) Component) Occurrence)Series) DateArgumentTime) r T_COMPONENTScZeZdZUdZeeejgZde d<ddl m Z  d8d9dZ e d:dZd;dZejdejdejdejdgZd:dZdd&Zd?d'Zd@d)ZdAd+ZdBd-ZedCd.Z dDdEd7Zd/S)F CalendarQueryaQuery a calendar for occurrences. Functions like :meth:`at`, :meth:`between` andm :meth:`after` can be used to query the selected components. If any malformed icalendar information is found, an :class:`InvalidCalendar` exception is raised. For other bad arguments, you should expect a :class:`ValueError`. Attributes: suppressed_errors: a list of errors to suppress when skip_bad_series is True zClassVar[type[Exception]]suppressed_errorsr)ComponentsWithNameFVEVENTcalendarrkeep_recurrence_attributesbool componentsrskip_bad_seriescx||_|dddkrtdg|_|rt |jnd|_|D]b}t|tr| |}n|}|j | ||jcdS)aACreate an unfoldable calendar from a given calendar. Arguments: calendar: an :class:`icalendar.cal.Calendar` component like :class:`icalendar.cal.Calendar`. keep_recurrence_attributes: Whether to keep attributes that are only used to calculate the recurrence (``RDATE``, ``EXDATE``, ``RRULE``). components: A list of component type names of which the recurrences should be returned. This can also be instances of :class:`SelectComponents`. Examples: ``("VEVENT", "VTODO", "VJOURNAL", "VALARM")`` skip_bad_series: Whether to skip series of components that contain errors. You can use :attr:`CalendarQuery.suppressed_errors` to specify which errors to skip. CALSCALE GREGORIANz'Only Gregorian calendars are supported.N) r#getrseriestupler _skip_errors isinstancestrrextendcollect_series_from)selfr"r#r%r&component_adapter_idcomponent_adapters V/root/projects/butler/venv/lib/python3.11/site-packages/recurring_ical_events/query.py__init__zCalendarQuery.__init__Ds,+E' << K 0 0K ? ?!"KLL L$& =LTE$"8999RT$.   .44 9$($;$;isdigit ValueErrorr?betweenr9 timedeltarA_between_DELTAS)r3r9yearmonthdts r6atzCalendarQuery.ats@ dC  7D dC  J4yyA~~T\\^^~ !J!J!J!JKKKRaR"%%s4!9b'9'93tABBx==ID dH- . . ,<<d++ + dHM * * I<<dX-?Q-G-G-G&GHH H t99>><<a!Q$q'A+q!1DEE E t99>>KD%{{||T2qMD1Ha3CDDD<<ua 04A2FGG G   d # #}}Rdl3t99q=&A!ABBBr8startstop!DateArgument | datetime.timedeltac||}t|tjr||zn||}|||S)aReturn events at a time between start (inclusive) and end (inclusive) Arguments: start: A date specification. See :meth:`to_datetime`. stop: A date specification or a :class:`datetime.timedelta` relative to start. .. warning:: If you pass a :class:`datetime.datetime` to both ``start`` and ``stop``, make sure the :attr:`datetime.datetime.tzinfo` is the same. )rAr/r?rVrW)r3r]r^s r6rUzCalendarQuery.betweensb  ''$ 233 (EDLL!!$'' }}UD)))r8 occurrenceslist[Occurrence]list[Component]c fd|DS)zMap occurrences to components.cDg|]}|jSr*) as_componentr#).0 occurrencer3s r6 z.s:     # #D$C D D   r8r*)r3ras` r6_occurrences_to_componentsz(CalendarQuery._occurrences_to_componentss.    )    r8rendcT||||S)z5Return the occurrences between the start and the end.)rj_occurrences_between)r3r]rks r6rWzCalendarQuery._betweens'..t/H/HPS/T/TUUUr8cg}|jD]\}tj|j5||||dddn #1swxYwY]|S)z4Return the components between the start and the end.N)r, contextlibsuppressr.r1rU)r3r]rkrar,s r6rmz"CalendarQuery._occurrences_betweens(* k ? ?F$T%677 ? ?""6>>%#=#=>>> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s*AA "A earliest_endc#K||}||D]}||jVdS)aIterate over components happening during or after earliest_end. Arguments: earliest_end: A date specification. See :meth:`to_datetime`. Anything happening during or after earliest_end is returned in the order of start time. N)rA_afterrfr#)r3rqrhs r6rEzCalendarQuery.aftersd'' 55 ++l33 K KJ))$*IJJ J J J J K Kr8Generator[Occurrence]c#Ktjd}tjd}d}t}|s ||z}n,#t$rt}t ||rYdSd}YnwxYw|||}||D])}|j|vr|V| |j*t|r|dz n|dz|}|}|dSdS) z@Iterate over occurrences happening during or after earliest_end.r;rGrJFNTrQ) r?rVset OverflowErrorr rrmsortidaddmax) r3rq time_span min_time_spandone result_idsnext_endrarhs r6rszCalendarQuery._aftersF&A...  *2666 (+  $ ')3    &"<::FF   33L(KKK      ) 2 2 = 22$$$$NN:=111!,? A )a-I$L) $ $ $ $ $sA!A/*A/.A/r=cBd}|D]}|dz }|S)zReturn the amount of recurring components in this calendar. .. warning:: Do not use this in production as it generates all occurrences. rr;)rF)r3i_s r6countzCalendarQuery.count#s0   A FAAr8cT|D]}|cStd)zReturn the first recurring component in this calendar. Returns: The first recurring component in this calendar. Raises: IndexError: if the calendar is empty zNo components found.)rF IndexError)r3 components r6firstzCalendarQuery.first/s6  I   /000r8N page_sizeOptional[DateArgument] latest_start next_page_idr0rcf|dn||}|tn||}|rtj|}t ||js||j}g}|D]m}|||j} | |krtj |g|}n2t | j|jrtj ||}nnn+||}n||}t||||j S)atReturn pages for pagination. Args: page_size: the number of components per page earliest_end: the start of the first page All components occur after this date. See :meth:`to_datetime` for possible values. latest_start: the end of the last page All components occur before this date. See :meth:`to_datetime` for possible values. next_page_id: The id of the next page. This is optional for the first page. These are safe to pass outside of the application and back in. N)occurrence_iteratorsizer^r#) rAr r from_stringrr]rsappendrz itertoolschainrr#) r3rrqrrfirst_occurrence_iditeratorlost_occurrencesrhoids r6paginatezCalendarQuery.paginate=sZ* ,3tt9I9I,9W9W '/KKT5E5El5S5S   1".":<"H"H "<1D1JKK 5;;':'@AA#% "*J$++J777$-C111#,?J<#J#J&sy2E2KLL#,?3CX#N#N ;;|44{{<00H ('+'F     r8)Fr F)r"rr#r$r%rr&r$)r9r)rBrC)r]rr^r_)rarbrBrc)r]rrkrrBrc)r]rrkrrBrb)rqrrBrC)rqrrBrt)rBr=)rBr)NNr) rr=rqrrrrr0rBr)__name__ __module__ __qualname____doc__r r icalendarrr__annotations__$recurring_ical_events.selection.namerr7 staticmethodrArFr?rVrXr\rUrjrWrmrErsrpropertyrrr*r8r6rr/s   !4 HGGGGG ,1#. % $$$$$L\>'''' """###1%%%1%%% G2C2C2C2Ch****,    VVVV K K K K$$$$:     1 1 1X 1 04/3 0 0 0 0 0 0 0 r8r)0r __future__rror?rsystypingrrrrrr ImportErrortyping_extensionsr(recurring_ical_events.adapters.componentr recurring_ical_events.constantsr r recurring_ical_events.errorsr rr recurring_ical_events.occurrencerrecurring_ical_events.pagesr$recurring_ical_events.selection.baserrecurring_ical_events.utilrrrrecurring_ical_events.seriesrrecurring_ical_events.typesrr version_infor0typerrr__all__r*r8r6rsm@@@"""""" IIIIIIIIIIIIII,       ,,,++++++++,EEEEEEDDDDDDDD :99999------AAAAAA666666######;;;;;;333333 w&sT2B-C'CFV'VWLWWWWW'smL++++~ ~ ~ ~ ~ ~ ~ ~ B ? +s 0 >>