iA.dZddlmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZmZddlmZerdd lmZdd lmZdd lmZd+d e dedefdZeGddZdddddedzdedzdedzdeedzedzff dZdddddedzdefdZdddddedzdeefdZ dddddefdZ!dddddefd Z" d,d"ed#dddedzd$ed%eded#f d&Z# d-ddd%ed'eedzd(ed)edzdeee$dzff d*Z%dS).a( Search operations - Sans-I/O business logic for calendar search. This module contains pure functions that implement search logic without performing any network I/O. Both sync (CalDAVSearcher.search) and async (CalDAVSearcher.async_search) use these same functions. Key functions: - build_search_xml_query(): Build CalDAV REPORT XML query - filter_search_results(): Client-side filtering of search results - determine_search_strategy(): Analyze server features and return search plan - _collation_to_caldav(): Map collation enum to CalDAV identifier )deepcopy) dataclassfield) TYPE_CHECKINGAny)Timezone) Collation)cdavdav)error)Searcher)CalendarObjectResource) FeatureSetT collationcase_sensitivereturnc|tjkr|rdSdS|tjkr|rdSdS|tjkrdSdS)aMap icalendar-searcher Collation enum to CalDAV collation identifier. CalDAV supports collation identifiers from RFC 4790. The default is "i;ascii-casemap" and servers must support at least "i;ascii-casemap" and "i;octet". :param collation: icalendar-searcher Collation enum value :param case_sensitive: Whether the collation should be case-sensitive :return: CalDAV collation identifier string i;octetzi;ascii-casemapzi;unicode-casemap)r SIMPLEUNICODELOCALE)rrs W/root/projects/butler/venv/lib/python3.11/site-packages/caldav/operations/search_ops.py_collation_to_caldavr seI$$$  %9$$ i' ' '  '9&& i& & &! yceZdZUdZdZedzed<dZedzed<dZ eed<e e Z e eed<d Z eed <d Zeed <d Zeed <dS) SearchStrategyzEncapsulates the search strategy decisions based on server capabilities. This dataclass holds all the decisions about how to execute a search, allowing the same logic to be shared between sync and async implementations. N post_filterhacksTsplit_expanded)default_factoryremove_propertiesFremove_category_filterpending_todo_multi_searchretry_with_comptypes)__name__ __module__ __qualname____doc__rbool__annotations__rstrrrsetr!r"r#r$rrrrAs $K###E3: ND#(%"<"<"<s3x<<<$)D(((',t+++"'$&&&&&rrsearcherr featuresrcomp_type_supportN current_hackscurrent_post_filtercd|}|}|js3t|dds"t|ddst|ddr|dkr |s|durd}d}|]t|ddr|jrC|jsg|]}|vj|dk|S)contains)_property_operator).0propexplicit_operatorsr.s r z5_get_explicit_contains_properties..sA     % % %(*Ed*Kz*Y*Y *Y*Y*Yr)r@r<r,rH)r.r/rrKs` @r!_get_explicit_contains_propertiesrMsv4559M9M  +@#%%HH     /   rct|drdSt|js|jo|jS)zCheck if property filters should be removed due to combined search issues. Some servers don't handle combined time-range + property filters properly. search.combined-is-logical-andF)r@r)startendr?r.r/s r,_should_remove_property_filters_for_combinedrSs?=>>u /8<OX5O P PPrct|ddr |jdusdS|do?|do*|d p|dS)zCheck if we need multiple searches for pending todos. Returns True if searching for pending todos and server supports the necessary features for multi-search approach. r4Fz search.textrOz)search.recurrences.includes-implicit.todoz1search.recurrences.includes-implicit.todo.pending)r<r=r@rRs r _needs_pending_todo_multi_searchrUs Hfe , ,1Ku1T1Tu m,,  ! !"B C C %%&QRR R Z$$%XYY rFobjectsrr server_expandcx|s |js|r|s|Sg}|D]$}|js|rD ||| }n&#t$rd|jjD}YnwxYw|sOnd|jjD}|j}d|jD} | |_|D]} t | t r|rT|d} | j} g| _| D]} | | | | n|} | | |s| |&|S)aApply client-side filtering and handle recurrence expansion/splitting. This is a Sans-I/O function - it only processes data without network I/O. :param objects: List of Event/Todo/Journal objects to filter :param searcher: The CalDAVSearcher with filter criteria :param post_filter: Whether to apply the searcher's filter logic. - True: Always apply filters (check_component) - False: Never apply filters, only handle splitting - None: Use default behavior (depends on searcher.expand and other flags) :param split_expanded: Whether to split recurrence sets into multiple separate CalendarObjectResource objects. If False, a recurrence set will be contained in a single object with multiple subcomponents. :param server_expand: Indicates that the server was supposed to expand recurrences. If True and split_expanded is True, splitting will be performed even without searcher.expand being set. :return: Filtered and/or split list of CalendarObjectResource objects ) expand_onlyc<g|]}t|t|Sr- isinstancerrIxs rrLz*_filter_search_results..s9TUW_I`I`rc<g|]}t|t|Sr-r[r]s rrLz*_filter_search_results..s9ZPQS[E\E\rc<g|]}t|t|Sr-r[r]s rrLz*_filter_search_results..s'EEEQZ8-D-DEqEEErT)keep_uid) r>check_component ValueErroricalendar_instance subcomponentsr\rcopy add_componentappend)rVr.rrrWresultofiltereditz_compnew_objnew_itzs r_filter_search_resultsrrs2 8?~- F %% ? k  #33A{?3SS    3A    /=H  EE!/EEE & &D$))  &&$&//2&(#,,B''++++ g&&&&    % % % %  MM!    Ms: AApropsfilters_hackscb ddlm}m}m}|}|} |} t j} |rD|jr|jstj d| t j |j|jz } || g} n| gt|z} tj | z} t jd}d}|j}|r>t!|}t#|dr|jtjjkr|}g}ng}t jdd }t jd d }t jd }t jd |z}t jd |z}t jd |z}t jd t jz}t jdt jz}|d kr||||gn:|dkr|||gn|dkr||g|js|jr2|t j|j|j|js|jrG|t jdt j|j|jzdd||fdd|| fdd|| fg}|D]\}}}}||fn||f}t7||d}|r"|||vrtjd|d||}|rV|jd|kr8|}|dkrt7|dds|jd|_tA||d||vrH|r!|jd|ksJnt j|}tA||d|r|stjd|d|dkrd}|j!D]|} |j!| dkrPt j}!|t j| "|!zd|j#| }"| "}#| $dkrd }#| $d!kr|"j%}$n|"g}$|$D]}"t#|"d"r|"&}"d#}%t#|d$r?| |j'vr6|j(| d}&tS|j'| |&}%t j|"|%%}!|t j|#|!z~|r |r ||z }||z }n|r||z }n|r||z }t j*|z}'t j+| |'gz}(|(|fS)&aBuild a CalDAV calendar-query XML request. This is a Sans-I/O function - it only builds XML without network I/O. :param searcher: CalDAVSearcher instance with search parameters :param server_expand: Ask server to expand recurrences :param props: Additional CalDAV properties to request :param filters: Pre-built filter elements (or None to build from searcher) :param _hacks: Compatibility hack mode :return: Tuple of (xml_element, comp_class) r)EventJournalTodoz!can't expand without a date rangeN VCALENDARtag COMPLETEDT)negate CANCELLEDz NEEDS-ACTIONSTATUSignore_completed1ignore_completed2ignore_completed3VALARMr5VEVENTr4VTODOr6VJOURNALFz.inconsistent search parameters - comp_class = z, want namezunsupported comp class z for searchr8undefr: CATEGORIESr9to_icalr_property_collation)r),caldav.calendarobjectresourcerwrxryr CalendarDatarPrQr ReportErrorExpandlistr Prop CompFilterr;rhasattrr{ TextMatch PropFilter NotDefinedextendrh TimeRange alarm_start alarm_endr<ConsistencyError attributesgetr=setattrrHupperr?lowercatsrr_property_case_sensitiverFilter CalendarQuery))r.rWrsrtrurwrxry AsyncEvent AsyncTodo AsyncJournaldataprops_rJ vcalendar comp_filterr; vNotCompleted vNotCancelled vNeedsActionvStatusNotCompletedvStatusNotCancelledvStatusNeedsActionvStatusNotDefinedvNoCompleteDate comp_mappingsflag comp_name sync_class async_class comp_classesflaggedpropertymatchvalue property_values collation_strr filter_elemroots) r_build_search_xml_queryrs&CBBBBBBBBBJIL    D:~ IX\ I#$GHH H  HNHL999 }$u++% 8:: D ,,IK$J7## 7E " " w{do6I'I'I!KGN;t<<.11L/(33mC/(33mC22\A11DO4E4EEok22T_5F5FFO $$$)<>QRSSSS & & &):;<<<< & & &*+,,,~EEt~hnhlCCDDD x1  OH % %x7KXM_(`(` `   (E:. $ * J6M 5B**0i[(3(; }}*kAZ (D%00  $%*L*H*H,dZddXbdd$J  *;155f==JJ#J&%88.6-1* HdD ) ) )  % % 9"-11&99YFFFFF"oi88 HdD ) ) )X+X$%Vz%V%V%VWWW !!! /CC  &x 0G ; ;O%%E NN4?8>>+;+;<%%Mu FFFty99EABBBB! C&ww [ [ W +--)+K    4"5 5D * r)T)NTF)FNNN)&r(rfr dataclassesrrtypingrr icalendarricalendar_searcher.collationr caldav.elementsr r caldav.libr icalendar_searcherr rrcaldav.compatibility_hintsrr)r+rrtuplerArCrrMrSrUrrtyperr-rrrs  ((((((((%%%%%%%%222222%%%%%%%%6++++++DDDDDD555555ItsB ''''''' ':***Tz*: *  *  4$;d "# ****Z " #Y ( Q Q Q  Q Q Q Q 2 $ DD * +DDD D  D  "# DDDDR " jjjj 9t j j $J j  3t  jjjjjjr