o .j} @sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z e r:ddlmZddlmZmZd Zd Zd Zd ZzddlZdd lmZdd lmZd ZWn ey_Ynwesz%ddlZd Zd ZzddlZd ZWn ey|YnwGd ddejZ Wn eyYnwesesz#ddlZd ZzddlZd ZWn eyYnwGdddejZ Wn eyYnwesesedddl!m"Z"ddl#m$Z$ddl#m%Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2m3Z3m4Z4e5dZ6ej7dkrddl8m9Z9nddlm9Z9Gddde$Z:d-de;d e d!e:fd"d#Z%   d.d$e dBd%e dBd&e;d e d!d'f d(d)ZZ& dded,edeeefdBdefd?d@Z' dded,edeeefdBdefdAdBZ( ddedeeefdBdefdCdDZ)        E ddedBdFe dBdGe dBdHe dIe dJe dKe d&e deeefdBdefdLdMZ*   E ddedBdNe edBd&e deeefdBdef dOdPZ+    E ddedBdQedBd2e edBd&e deeefdBdef dRdSZ,dTdUZ-dVdWZ.ddXe edBddfdYdZZ/dd\d]Z0dd^e1d[de d_fd`daZ2       ddbd_dHe dIe dJe dFe dBdGe dBdce dBdKe dde de defdfdgZ3ddhedBde d[fdidjZ4ddhedBde d[fdkdlZ5ddmdnZ6dedBfdodpZ7de fdqdrZ8de fdsdtZ9dedBfdudvZ:de fdwdxZ;de fdydzZ_http2_proxy_ssl_verify_cert _ssl_cert_timeout is_supported _H2_AVAILABLE _USE_NIQUESTS TypeErrorAttributeError_create_sessioncaldav.davclientrNr2 objectifyr=r@rrunauthrArBbuild_auth_objectr<rCrDrEr,rupdatedictgetrIrJrK)rr=r<r@rrArBrCrDrErr>rFrGrHrIrJrKrLrNurl_strdiscovered_username url_username url_password rate_limitrrrrs/                    zAsyncDAVClient.__init__cCstr+|jpd|jdur|jnd|j|jd}|jdur |j|d<tjdi||_dSz t |jp1dd|_WdSt yEt |_YdSw)z?Create or recreate the async HTTP client with current settings.FNT)http2verifycertrCr<) multiplexedr) _USE_HTTPXrUrWrXrYrVhttpx AsyncClientsessionr r])r client_kwargsrrrr_s   zAsyncDAVClient._create_sessioncs|S)zAsync context manager entry.rrrrr __aenter__&szAsyncDAVClient.__aenter__exc_typeexc_valexc_tbcs|IdHdS)zAsync context manager exit.N)close)rrwrxryrrr __aexit__*szAsyncDAVClient.__aexit__cs<t|drtr|jIdHdS|jIdHdSdS)zClose the async client.rsN)hasattrrprsacloserzrurrrrz3s  zAsyncDAVClient.closemethoddepth extra_headerscCs:i}|dur t||d<|dvrd|d<|r|||S)a Build headers for WebDAV methods. Args: method: HTTP method name. depth: Depth header value (for PROPFIND/REPORT). extra_headers: Additional headers to merge. Returns: Dictionary of headers. NDepth)REPORTPROPFIND PROPPATCH MKCALENDARMKCOLz application/xml; charset="utf-8" Content-Type)r*rd)r~rrrrrr_build_method_headers;s  z$AsyncDAVClient._build_method_headersGETrbodyrate_limit_time_sleptc sz |||||IdHWStjy[}zA|jst|j|j|j}|r,||d7}|dus:|jdur;||jkr;t |IdH| ||||||IdHWYd}~Sd}~ww)a Send an async HTTP request, with optional rate-limit sleep-and-retry. Catches RateLimitError from _async_request. When rate_limit_handle is True and a usable sleep duration is available, sleeps then retries with adaptive backoff (each retry adds half the already-slept time). Stops retrying when rate_limit_max_sleep is exceeded or no sleep duration is available. Otherwise re-raises immediately. N) _async_requestr0RateLimitErrorrIcompute_sleep_secondsretry_after_secondsrJrKasynciosleepr!)rr=r~rrre sleep_secondsrrrr!\s.   zAsyncDAVClient.requestc sf|||||\}}tr!|t||rt|nd||j|jd}n%d}|jdur.|j|ji}|t||r8t|nd||j|j||j|j d }z^|j j di|IdH} trX| j n| j } td| jd| | jdkrd|jdd vr|jsd } | jd rd d || jd D} | r| dd| 7} t| t| |} Wnty.|js|jstr|j j dt|||jdIdH} n"d}|jdur|j|ji}|j j dt|||j||j|j dIdH} tr| j n| j } td| jd| | jdkr'| jd r'|| jd } || |j|d<|j j di|IdH} t| |} Ynwt| jt|| jd|| j| jrY|| jd |||||IdHS| jdkrd | jvr|jr|j j!ddddur|"IdHd|_#|$|j %dd|t||||IdHS| j&dvr|'t|| tj(rt)||||| | S)z Async HTTP request implementation with auth negotiation. Handles connection-abort workaround, 429/503 rate-limit detection, and 401 auth negotiation (including HTTP/2 fallback). N)r~r=contentrrArC) r~r=datarrArCproxiesrmrnzserver responded with  z text/htmlrr?zNo authentication object was provided. HTML was returned when probing the server for supported authentication types. To avoid logging errors, consider passing the auth_type connection parameterzWWW-AuthenticatecSsg|]}|dvr|qS))basicdigestbearerr).0trrr s z1AsyncDAVClient._async_request..z# Supported authentication types: {}z, r)r~r=rrC)r~r=rrCrrmrnz+auth type detection: server responded with rAz Retry-AfterrMF)return_defaults)rir)*_prepare_requestrpr*r1rArCr<schemerDrErsr! reason_phrasereasonlogdebug status_coderrfextract_auth_typesformatjoinwarningr5 Exceptionrrcr0raise_if_rate_limited_should_negotiate_auth_build_auth_from_401rrFrZrzrUr_ set_featurestatus_raise_authorization_errordebug_dump_communication_dump_communication)rr=r~rrurl_objcombined_headersrequest_kwargsrrrmsg auth_typesresponserrrrs              %   zAsyncDAVClient._async_requestpropscsj|dur|s||d}|d||}||pt|jd||IdH}|jdvr3|jr3||_ |S)a Send a PROPFIND request. Args: url: Target URL (defaults to self.url). body: XML properties request (legacy, use props instead). depth: Maximum recursion depth. headers: Additional headers. props: List of property names to request (uses protocol layer). Returns: DAVResponse with results attribute containing parsed PropfindResult list. Nutf-8r) _build_propfind_bodydecoderr!r*r=r_rawparse_propfindresults)rr=rrrr final_headersrrrrpropfind s   zAsyncDAVClient.propfindcs0|d||}||pt|jd||IdHS)a~ Send a REPORT request. Args: url: Target URL (defaults to self.url). body: XML report request. depth: Maximum recursion depth. None means don't send Depth header (required for calendar-multiget per RFC 4791 section 7.9). headers: Additional headers. Returns: DAVResponse rN)rr!r*r=)rr=rrrrrrrreport-s zAsyncDAVClient.reportcs"||p t|jdd|IdHS)z Send an OPTIONS request. Args: url: Target URL (defaults to self.url). headers: Additional headers. Returns: DAVResponse OPTIONSr?N)r!r*r=rr=rrrroptionsDs zAsyncDAVClient.optionsc&|jd|d}||d||IdHS)z Send a PROPPATCH request. Args: url: Target URL (required). body: XML property update request. headers: Additional headers. Returns: DAVResponse rrNrr!rr=rrrrrr proppatchWzAsyncDAVClient.proppatchcr)a0 Send a MKCOL request. MKCOL creates a WebDAV collection. For CalDAV, use mkcalendar instead. Args: url: Target URL (required). body: XML request (usually empty). headers: Additional headers. Returns: DAVResponse rrNrrrrrmkcolkszAsyncDAVClient.mkcolcr)z Send a MKCALENDAR request. Args: url: Target URL (required). body: XML request (usually contains calendar properties). headers: Additional headers. Returns: DAVResponse rrNrrrrr mkcalendarrzAsyncDAVClient.mkcalendarc||d||IdHS)z Send a PUT request. Args: url: Target URL (required). body: Request body (e.g., iCalendar data). headers: Additional headers. Returns: DAVResponse PUTNr!rr=rrrrrputzAsyncDAVClient.putcr)z Send a POST request. Args: url: Target URL (required). body: Request body. headers: Additional headers. Returns: DAVResponse POSTNrrrrrpostrzAsyncDAVClient.postcs||dd|IdHS)z Send a DELETE request. Args: url: Target URL (required). headers: Additional headers. Returns: DAVResponse DELETEr?NrrrrrdeleteszAsyncDAVClient.deletestartendeventtodojournalexpandc sn|j||||||d\} } |d|| } ||pt|jd| d| IdH} | jdvr5| jr5| | _ | S)aR Execute a calendar-query REPORT to search for calendar objects. Args: url: Target calendar URL (defaults to self.url). start: Start of time range filter. end: End of time range filter. event: Include events (VEVENT). todo: Include todos (VTODO). journal: Include journals (VJOURNAL). expand: Expand recurring events. depth: Search depth. headers: Additional headers. Returns: DAVResponse with results containing List[CalendarQueryResult]. )rrrrrrrrNr) _build_calendar_query_bodyrr!r*r=rrrparse_calendar_queryr)rr=rrrrrrrrr_rrrrrcalendar_querys     zAsyncDAVClient.calendar_queryhrefscsb||pg}|d||}||pt|jd|d|IdH}|jdvr/|jr/||_ |S)a Execute a calendar-multiget REPORT to fetch specific calendar objects. Args: url: Target calendar URL (defaults to self.url). hrefs: List of object URLs to retrieve. depth: Search depth. headers: Additional headers. Returns: DAVResponse with results containing List[CalendarQueryResult]. rrNr) _build_calendar_multiget_bodyrr!r*r=rrrrr)rr=rrrrrrrrrcalendar_multigets  z AsyncDAVClient.calendar_multiget sync_tokenc sp|j||d}|d||}||pt|jd|d|IdH}|jdvr6|jr6|} | j |_ | j |_ |S)a Execute a sync-collection REPORT for efficient synchronization. Args: url: Target calendar URL (defaults to self.url). sync_token: Previous sync token (None for initial sync). props: Properties to include in response. depth: Search depth. headers: Additional headers. Returns: DAVResponse with results containing SyncCollectionResult. )rrrrNr) _build_sync_collection_bodyrr!r*r=rrrparse_sync_collectionchangedrr) rr=rrrrrrr sync_resultrrrsync_collections zAsyncDAVClient.sync_collectioncCs ||Sr)_async_value_or_coroutinervaluerrr_value_or_coroutineBrz"AsyncDAVClient._value_or_coroutinecs|SrrrrrrrEsz(AsyncDAVClient._async_value_or_coroutinercCs||}|dkrtrt|j|_dSt|j|_dS|dkr.) rr rrrrr*r=CALENDAR_HOME_SET_PROPSr_make_absolute_urlCALENDAR_LIST_PROPS)rrextract_home_setextract_calendarsrcalendar_home_urlcalendar_infosrr r get_calendarss0       zAsyncDAVClient.get_calendarscalendarinclude_completedkwargsrc sLddlm} | ||||||d} |dur|| _| j|fi| IdH} | S)aSearch a calendar for events, todos, or journals. This method provides a clean interface to calendar search using the operations layer for building queries and processing results. Args: calendar: Calendar to search event: Search for events (VEVENT) todo: Search for todos (VTODO) journal: Search for journals (VJOURNAL) start: Start of date range end: End of date range include_completed: Include completed todos (default: False for todos) expand: Expand recurring events **kwargs: Additional search parameters Returns: List of Event/Todo/Journal objects. Example: # Get all events in January 2024 events = await client.search_calendar( calendar, event=True, start=datetime(2024, 1, 1), end=datetime(2024, 1, 31), ) # Get pending todos todos = await client.search_calendar( calendar, todo=True, include_completed=False, ) r)CalDAVSearcher)rrrrrrN) caldav.searchrr async_search) rrrrrrrrrrrsearcherrrrrsearch_calendars / zAsyncDAVClient.search_calendarrcs\||}|t|j|IdH}|jdkr't|jd|jd|j| | S)a= Search for principals on the server. Instead of returning the current logged-in principal, this method attempts to query for all principals (or principals matching a name). This may or may not work depending on the permissions and implementation of the calendar server. Args: name: Optional name filter to search for specific principals Returns: List of Principal objects found on the server Raises: ReportError: If the server doesn't support principal search Ni,rz - ) _build_principal_search_queryrr*r=rr0 ReportErrorrraw _parse_principal_search_response_find_objects_and_props)rrrrrrrsearch_principalss    z AsyncDAVClient.search_principalscs,ddl}|jdtdd|j|dIdHS)z Deprecated. Use :meth:`search_principals` instead. This method searches for principals on the server. rNz;principals() is deprecated, use search_principals() insteadr) stacklevelr)warningswarnDeprecationWarningr")rrr%rrr principals-szAsyncDAVClient.principalsc|IdHS)z Legacy method. Use :meth:`get_principal` for new code. Returns the Principal object for the authenticated user. N)rrurrrr<zAsyncDAVClient.principalcs\z|IdH}||jIdH}Wnty'|t|jIdH}Ynw|jdS)z Check if the server supports DAV. Returns the DAV header from an OPTIONS request, or None if not supported. NDAV)rrr=rr*rrf)rrrrrrcheck_dav_supportDs  z AsyncDAVClient.check_dav_supportc |IdH}|duod|vS)z Check if the server supports CalDAV. Returns True if the server indicates CalDAV support in DAV header. Nzcalendar-accessr,r dav_headerrrrcheck_cdav_supportQz!AsyncDAVClient.check_cdav_supportcr-)z Check if the server supports RFC6638 scheduling. Returns True if the server indicates scheduling support in DAV header. Nzcalendar-auto-scheduler.r/rrrcheck_scheduling_supportZr2z'AsyncDAVClient.check_scheduling_supportcr))zl Check if the server supports DAV. This is an alias for :meth:`check_dav_support`. Nr.rurrr supports_davcr*zAsyncDAVClient.supports_davcr))zp Check if the server supports CalDAV. This is an alias for :meth:`check_cdav_support`. N)r1rurrrsupports_caldavkr*zAsyncDAVClient.supports_caldavcr))z Check if the server supports RFC6638 scheduling. This is an alias for :meth:`check_scheduling_support`. N)r3rurrrsupports_schedulingsr*z"AsyncDAVClient.supports_scheduling)r?NNNNNNTNNFNTTNNN)rN)NN)rr?Nr)rr?N)Nr?rNN)Nr?rN)r?Nr) NNNFFFFrN)NNrN)NNNrN)rr )FFFNNNF)=r&r'r(r)r<r*__annotations__r=r2r>boolrinttuplerr/rerr_r:rvtype BaseExceptionrr{rz staticmethodrfloatr5r!rlistrrrrrrrrrrrrrrrcrrrrr"r(rr,r1r3r4r5r6rrrrr;ps             # *    #           3  !  # &)=   B    r;proberrc sttfi|}|durtd|r`z,|IdH}td|j|jdd}|s5t dW|St d|W|St y_}z| IdHt d|jd ||d}~ww|S) aB Get an async DAV client instance with configuration from multiple sources. See :func:`caldav.base_client.get_davclient` for full documentation. Args: probe: Verify connectivity with OPTIONS request (default: True). **kwargs: All other arguments passed to base get_davclient. Returns: AsyncDAVClient instance. Raises: ValueError: If no configuration is found. Example:: async with await get_davclient(url="...", username="...", password="...") as client: principal = await client.principal() NztNo configuration found. Provide connection parameters, set CALDAV_URL environment variable, or create a config file.zConnected to CalDAV server: r+r?z:Server did not return DAV header - may not be a DAV serverzServer DAV capabilities: z&Failed to connect to CalDAV server at : )_base_get_davclientr; ValueErrorrrr r=rrfrrrrzr0DAVError)r@rrrr0rrrrr.s, r. calendar_url calendar_name raise_errorsCalendarCollectionc s\ddlm}m}fdd}ztdddi|IdH}Wn ty=}zr(td||WYd}~Sd}~wwz|IdH} | sS|dd ||d WSg} ||} ||} | D]G} d t| vrn| j | d }n| j | d }z| IdH}|dur| |Wq_ty}ztd| d|rWYd}~q_d}~ww| D]4}z| j |dIdH}|r| |Wqty}ztd|d|rӂWYd}~qd}~ww| s| s| sz | IdH}|r|} Wnty}ztd|r WYd}~nd}~ww|| |d WSty-| dddIdHw)a Get calendars from a CalDAV server asynchronously. This is the async version of :func:`caldav.get_calendars`. Args: calendar_url: URL(s) or ID(s) of specific calendars to fetch. calendar_name: Name(s) of specific calendars to fetch by display name. raise_errors: If True, raise exceptions on errors; if False, log and skip. **kwargs: Connection parameters (url, username, password, etc.) Returns: :class:`~caldav.base_client.CalendarCollection` of matching calendars. Use as an async context manager to auto-close the connection:: async with await aio.get_calendars() as calendars: for cal in calendars: print(await cal.get_display_name()) r)rH_normalize_to_listcs(|durtd|rt||S)z)Handle errors based on raise_errors flag.Nz(Problems fetching calendar information: )rr0rC) coro_resulterrmsgrGrr_trys zget_calendars.._tryr@TNzFailed to create async client: zgetting principalr/)cal_url)r zProblems fetching calendar rAr$z$Problems fetching calendar by name 'z': z!Problems fetching all calendars: r)caldav.base_clientrHrIr.rrr0rr*rget_display_nameappendrr{)rErFrGrrHrIrMrrr calendars calendar_urlscalendar_namesrPr display_namecal_nameall_calsrrLrrs          rCalendarResultcs@ddlm}tdi|IdH}|r|dnd}|||jdS)a` Get a single calendar from a CalDAV server asynchronously. This is a convenience function for the common case where only one calendar is needed. Use as an async context manager to auto-close the connection:: async with await aio.get_calendar() as cal: events = await cal.search(event=True) Args: Same as :func:`get_calendars`. Returns: :class:`~caldav.base_client.CalendarResult` wrapping the first matching calendar (or None). Behaves like the calendar via ``__getattr__`` delegation when not used as a context manager. r)rZNrNr)rQrZrr)rrZ collectioncalrrr get_calendars  r])T)NNF)>r)rloggingsyscollections.abcrtypesrtypingrrr urllib.parsercaldav.calendarobjectresourcerrr r rp _USE_HTTPXYZr\r[niquestsr niquests.structuresr ImportErrorhttpxyzrqh2Authrr7r,rQr-r.rBcaldav.compatibility_hintsr/ caldav.libr0caldav.lib.python_utilitiesr1caldav.lib.urlr2caldav.requestsr3caldav.responser4r5r6 getLoggerr version_infotyping_extensionsr:r;r8rr]rrrrs                      2 b