i9tdZddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z ddlZddlmZd d d d d dZeGddZdKdZdLdZdMdZdNdZ dOdPd!Z dQdRd%ZdSd'Z dTdUd,ZdVd.Z dWdXd0ZdYd1Z dZd[d9ZdVd:Zd\d>Z d]d?Z!d^dAZ"d_dBZ# dWd`dDZ$dadFZ% dWdbdJZ&dS)caC CalendarObjectResource operations - Sans-I/O business logic. This module contains pure functions for working with calendar objects (events, todos, journals) without performing any network I/O. Both sync and async clients use these same functions. These functions work on icalendar component objects or raw data strings. ) annotationsN) dataclass)datetime timedeltatimezone)Any)quote)rrulestrCHILDPARENTSIBLING FINISHTOSTART DEPENDENT)r r r z DEPENDS-ONrc<eZdZUdZded<ded<ded<ded<dS)CalendarObjectDataz&Data extracted from a calendar object. str | NoneuidurletagdataN)__name__ __module__ __qualname____doc____annotations___/root/projects/butler/venv/lib/python3.11/site-packages/caldav/operations/calendarobject_ops.pyrr!sB00OOOOOOrrreturnstrcBttjS)z)Generate a new UID for a calendar object.)r uuiduuid4rrr _generate_uidr$+s tz||  rrcHt|ddS)z URL-quote a UID for use in a CalDAV object URL. Slashes are double-quoted (replaced with %2F before percent-encoding) per https://github.com/python-caldav/caldav/issues/143. /z%2F)r replace)rs r _quote_uidr(0s  S%(( ) ))r parent_urlcbt|}|ds|dz }||dS)a) Generate a URL for a calendar object based on its UID. Handles special characters in UID by proper quoting. Args: parent_url: URL of the parent calendar (must end with /) uid: The UID of the calendar object Returns: Full URL for the calendar object r&.ics)r(endswith)r)r quoted_uids r _generate_urlr.:sCCJ   s # #c  ** * * **rpathrc|dsdStjd|}|r|dSdS)z Extract UID from a .ics file path. Args: path: Path like "/calendars/user/calendar/event-uid.ics" Returns: The UID portion, or None if not found r+Nz(/|^)([^/]*).ics$)r,researchgroup)r/matchs r_extract_uid_from_pathr6MsL == t I*D 1 1E {{1~~ 4r componentrgiven_id given_path existing_idtuple[str, str]c^|p|}|s&|d}|rt|}|s&|r$|drt|}|st }d|vr|d|d||r|}nt|dz}||fS)a- Determine the UID and path for a calendar object. This is Sans-I/O logic extracted from CalendarObjectResource._find_id_path(). Priority: 1. given_id parameter 2. existing_id (from object) 3. UID from component 4. UID extracted from path 5. Generate new UID Args: component: icalendar component (VEVENT, VTODO, etc.) given_id: Explicitly provided ID given_path: Explicitly provided path existing_id: ID already set on the object Returns: Tuple of (uid, relative_path) UIDr+)getr r,r6r$popaddr()r7r8r9r:ruid_propr/s r_find_id_and_pathrB_s6  !kC  ==''  h--C 1:1*"5"5f"="=1$Z00 oo  e MM%(#' 9rDTEND end_paramrc(d|vr |djSd|vr||vr||j}|dj}t|tt|tkrjt|ts t|j|j|j}t|ts t|j|j|j}||z Sd|vr2|dj}t|tst dSt dS)a Get duration from a calendar component. According to the RFC, either DURATION or DTEND/DUE should be set, but never both. This function calculates duration from whichever is present. Args: component: icalendar component (VEVENT, VTODO, etc.) end_param: The end parameter name ("DTEND" for events, "DUE" for todos) Returns: Duration as timedelta DURATIONDTSTART)daysr)dt isinstanceryearmonthdayr)r7rDendstartdtstarts r _get_durationrRs"Y$''I)y"8"8 "%)$' c8 $ $ 5((C(C C CeX.. E U[%)DDc8,, =sxCG<<U{II&)'8,, %!$$$ $ Q<<rdatetime | Nonecd|vr |djSd|vr |djSd|vrd|vr|dj|djzSdS)z Get due date from a VTODO component. Handles DUE, DTEND, or DURATION+DTSTART. Args: component: icalendar VTODO component Returns: Due date/datetime, or None if not set DUErCrFrGN)rJr7s r_get_duerWsk "" I  !$$ y Y)%;%;#&:)>)AAA 4rrGduration movable_attrNonecDd|vpd|v}d|v}|r|r||d|dkr|dd|dkr&|d|dj|z dS|dkr&|d|dj|zdSdSd|vr&|d|dj|z dSd|vr&|d|dj|zdSd|vr|d|d|dS)a) Set duration on a component, adjusting other properties as needed. If both DTSTART and DUE/DTEND are set, one must be moved. Args: component: icalendar component to modify duration: New duration movable_attr: Which attribute to move ("DTSTART" or "DUE") rUrFrGN)r?r@rJ)r7rXrYhas_due has_starts r _set_durationr^sky ;J)$;GY&I,9, lD))) 5 MM*d + + + 9 $ $ MM)Yu%5%88%C D D D D D U " " MM%9!5!88!C D D D D D# " )   i5!1!4x!?@@@@@ i   eYy14x?@@@@@  " " MM* % % % j(+++++rboolcx|ddS|dd}|dvrdS|dvrdSdS) z Check if a VTODO component is pending (not completed). Args: component: icalendar VTODO component Returns: True if task is pending, False if completed/cancelled COMPLETEDNFSTATUS NEEDS-ACTION)rcz IN-PROCESST) CANCELLEDra)r>)r7statuss r_is_task_pendingrfsW}}[!!-u ]]8^ 4 4F ///t +++u 4rcompletion_timestampc|tjtj}|dd|dd|d|dS)z Mark a VTODO component as completed. Modifies the component in place. Args: component: icalendar VTODO component completion_timestamp: When the task was completed (defaults to now) Nrbra)rnowrutcr?r@)r7rgs r_mark_task_completedrks_#'|HL99 MM(D!!! MM(K((( MM+344444rc|dd|dd|dd|dd|dddS)zj Mark a VTODO component as not completed. Args: component: icalendar VTODO component reNrbrc completedra)r?r@rVs r_mark_task_uncompletedrn(snMM(D!!! MM(D!!! MM(N+++ MM+t$$$ MM+t$$$$$rTrrule Any | NonerQuse_fixed_deadlines bool | None ignore_countc||d}|dS|td|D}|h|r3d|vr|dj}nT|ptjt j}n3t|d}|ptjt j|z }t|dr| t j}|p|}|r-d|vr)| }| dt| d| }||S) a Calculate the next DTSTART for a recurring task after completion. This implements the logic from Todo._next(). Args: component: icalendar VTODO component with RRULE completion_timestamp: When the task was completed rrule: Override RRULE (default: from component) dtstart: Override DTSTART (default: calculated based on use_fixed_deadlines) use_fixed_deadlines: If True, preserve DTSTART from component. If False, use completion time minus duration. If None, auto-detect from BY* parameters in rrule. ignore_count: If True, ignore COUNT in RRULE Returns: Next DTSTART datetime, or None if no more recurrences NRRULEc3DK|]}|d|VdS)BYN) startswith.0xs r z-_calculate_next_recurrence..Ws3!I!Iall46H6H!I!!I!I!I!I!I!IrrGrU astimezoneCOUNTzutf-8)rQ)r>anyrJrrirrjrRhasattrr}copyr?r to_icaldecodeafter) r7rgrorQrqrsrXts rrule_objs r_calculate_next_recurrencer6sW4 } g&& =4"!!I!IU!I!I!III  VI%%#I.1.L(,x|2L2L$Y66H+Ix|HL/I/IXUGw %%3$$X\22  (B5((  '//88'JJJI ??2  rcd|vrdS|d}|dd}|Mt|tr|dn|}|dkrdSt|tr |dz |d<n|dz |d<dS)z Reduce the COUNT in an RRULE by 1. Args: component: icalendar component with RRULE Returns: False if COUNT was 1 (task should end), True otherwise ruTr~NrrHF)r>rKlist)r7rocount count_vals r_reduce_rrule_countrtsit g E IIgt $ $E  *5$ 7 7BE!HHU >>5 eT " " + 1}E!HH&]E'N 4rrvobject_instanceicalendar_instancecZt|r|ddkp|p|S)z Check if calendar object data is loaded. Args: data: Raw iCalendar data string vobject_instance: vobject instance (if any) icalendar_instance: icalendar instance (if any) Returns: True if data is loaded zBEGIN:rH)r_r)rrrs r_is_calendar_data_loadedrs4 2$**X..2]7G]K] ^ ^^rc|sdS|d|dz|dzdkS)z Check if data contains VEVENT, VTODO, or VJOURNAL. Args: data: Raw iCalendar data string Returns: True if a calendar component is present Fz BEGIN:VEVENTz BEGIN:VTODOzBEGIN:VJOURNALr)r)rs r_has_calendar_componentrsR u >""TZZ %>%>>L\A]A]]   r list[Any]c$d|jDS)z Get all subcomponents except VTIMEZONE. Args: icalendar_instance: icalendar.Calendar instance Returns: List of non-timezone subcomponents cFg|]}t|tj|Sr)rK icalendarTimezonerys r z3_get_non_timezone_subcomponents..s* a a a!z!YM_?`?` aA a a ar) subcomponents)rs r_get_non_timezone_subcomponentsrs b a)7 a a aarct|}|sdS|D]B}t|tjtjtjtjfr|cSCdS)aN Get the primary (non-timezone) component from a calendar. For events/todos/journals, there should be exactly one. For recurrence sets, returns the master component. Args: icalendar_instance: icalendar.Calendar instance Returns: The primary component (VEVENT, VTODO, VJOURNAL, or VFREEBUSY) N)rrKrEventTodoJournalFreeBusy)r componentscomps r_get_primary_componentrsp11CDDJ t   _ini.?AS T   KKK   4rnew_uidc|}|dd|d|p t|S)z Create a copy of a component with a new UID. Args: component: icalendar component to copy new_uid: New UID (generated if not provided) Returns: Copy of the component with new UID r=N)rr?r@r$)r7rnew_comps r_copy_component_with_new_uidrsI~~H LL LL2=??333 OrreltypecZt|S)z Get the reverse relation type for a given relation type. Args: reltype: Relation type (e.g., "PARENT", "CHILD") Returns: Reverse relation type, or None if not defined )RELTYPE_REVERSE_MAPr>upper)rs r_get_reverse_reltypers   " "7==?? 3 33rreltypes set | Nonedict[str, set]cFddlm}|t}|dg}t |t s|g}|D]L}|jdd}|r||vr$||t|Mt|S)z Extract RELATED-TO relations from a component. Args: component: icalendar component reltypes: Optional set of relation types to filter Returns: Dict mapping reltype -> set of UIDs r) defaultdictz RELATED-TORELTYPEr ) collectionsrsetr>rKrparamsr@r dict)r7rrresult relationsrelrs r_extract_relationsrs(''''' [  F lB//I i & & K &&*..H55  x// wCHH%%%% <<r)rr )rr rr )r)r rr rr )r/r rr)NNN) r7rr8rr9rr:rrr;)rC)r7rrDr rr)r7rrrS)rG)r7rrXrrYr rrZ)r7rrr_)N)r7rrgrSrrZ)r7rrrZ)NNNNT)r7rrgrSrorprQrSrqrrrsr_rrS)rrrrrrrr_)rrrr_)rrrr)rrrrp)r7rrrrr)rr rr)r7rrrrr)'r __future__rr2r" dataclassesrrrrtypingr urllib.parser rdateutil.rruler rrr$r(r.r6rBrRrWr^rfrkrnrrrrrrrrrrrrrs#""""" !!!!!!2222222222###### !    ****++++&( !" 66666v(((((V0"!,!,!,!,!,H2-155555* % % % % -1#'+ ;;;;;|8____&    $ b b b b:( 4 4 4 4 r