iQ"dZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZmZmZmZmZmZddlmZmZmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)m*Z*ej+dZ,GddeZ-dS)u Asynchronous JMAP client. Mirrors JMAPClient with all public methods as coroutines. Uses niquests.AsyncSession for HTTP — niquests is a core dependency. ) annotationsN) AsyncSession)build_calendar_getparse_calendar_get)build_event_changesbuild_event_getbuild_event_set_createbuild_event_set_destroybuild_event_set_updateparse_event_changesparse_event_getparse_event_set)build_task_getbuild_task_list_getbuild_task_set_createbuild_task_set_destroybuild_task_set_updateparse_task_list_getparse_task_set)_DEFAULT_USING _TASK_USING_JMAPClientBase) ical_to_jscal) JMAPAuthErrorJMAPMethodError) JMAPCalendarJMAPCalendarObject)Sessionasync_fetch_sessionz caldav.jmapceZdZdZd4dZd5dZd6dZd7d8dZd9dZd:dZ d;dZ dd Z d?d@d!Z dAd"ZdBd%ZdCd&Z dDdEd(ZdFd*ZdGd-ZdHd0ZdId2ZdJd3ZdS)KAsyncJMAPClientawAsynchronous JMAP client for calendar operations. **The JMAP support is experimental, the API may change in minor-releases** Usage:: from caldav.jmap import get_async_jmap_client async with get_async_jmap_client(url="https://jmap.example.com/.well-known/jmap", username="alice", password="secret") as client: calendars = await client.get_calendars() Args: url: URL of the JMAP session endpoint (``/.well-known/jmap``). username: Username for Basic auth. password: Password for Basic auth, or bearer token if no username. auth: A pre-built niquests-compatible auth object. Takes precedence over username/password if provided. auth_type: Force a specific auth type: ``"basic"`` or ``"bearer"``. timeout: HTTP request timeout in seconds. returnc K|SNselfs S/root/projects/butler/venv/lib/python3.11/site-packages/caldav/jmap/async_client.py __aenter__zAsyncJMAPClient.__aenter__Cs  Nonec KdSr%r&)r(exc_typeexc_valexc_tbs r) __aexit__zAsyncJMAPClient.__aexit__Fs tr+rczK|j,t|j|j|jd{V|_|jS)z5Return the cached Session, fetching it on first call.N)authtimeout)_session_cacher url_authr4r's r) _get_sessionzAsyncJMAPClient._get_sessionIs\   &(;tz4<)))######D ""r+N method_calls list[tuple]usinglist[str] | Nonelistc K|d{V}||ntt|d}td|jt |t4d{V}||j||j ddd|j d{V}dddd{Vn#1d{VswxYwY|j dvrt|jd|j d  | |}|d g}|D]>} | \} } } | d kr0| d d} t!|jd| | ?|S)aMPOST a batch of JMAP method calls and return the methodResponses. Args: method_calls: List of 3-tuples ``(method_name, args_dict, call_id)``. using: Capability URN list for the ``using`` field. Defaults to ``_DEFAULT_USING`` (core + calendars). Returns: List of 3-tuples ``(method_name, response_args, call_id)`` from the server's ``methodResponses`` array. Raises: JMAPAuthError: On HTTP 401 or 403. JMAPMethodError: If any methodResponse is an ``error`` response. N)r; methodCallsz"JMAP POST to %s: %d method call(s)zapplication/json)z Content-TypeAccept)jsonr3headersr4)iizHTTP z from API endpointr6reasonmethodResponseserrortype serverErrorzMethod call failed: r6rD error_type)r8rr=logdebugapi_urllenrpostr7r4 status_coderraise_for_statusrAgetr)r(r9r;sessionpayloadhttpresponsedatamethod_responsesresp method_name resp_argscall_idrJs r)_requestzAsyncJMAPClient._requestQs` ))++++++++$/UU^ --   6\IZIZ[[[>>       T!YYZ);GYZZ 'H                             : - -OGx3GGG  !!###}}88$5r::$  D.2 +KGg%%&]]6=AA %=)==)& s:3B?? C  C list[JMAPCalendar]cK|d{V}t|j}||gd{V}|D]2\}}}|dkr&t |}|D]}||_d|_|cS3gS)zFetch all calendars for the authenticated account. Returns: List of :class:`~caldav.jmap.objects.calendar.JMAPCalendar` objects. Nz Calendar/getT)r8r account_idr]r_client _is_async) r(rScall responsesrZr[_ calendarscals r) get_calendarszAsyncJMAPClient.get_calendarss ))++++++++!'"455--//////// )2 ! ! %KAn,,.y99 $))C"&CK$(CMM    - r+ calendar_idstrical_strcK|d{V}t||}t|jd|i}||gd{V}|D]l\}}} |dkr`t |\} } } } } } d| vr||| dd| vrt|jd| ddcSmt|jd) aiCreate a calendar event from an iCalendar string. Args: calendar_id: The JMAP calendar ID to create the event in. ical_str: A VCALENDAR string representing the event. Returns: The server-assigned JMAP event ID. Raises: JMAPMethodError: If the server rejects the create request. N)rinew-0CalendarEvent/setz:CalendarEvent/set response missing created entry for new-0rCidNo CalendarEvent/set response) r8rr r`r]r_raise_set_errorrrM) r(rirkrSjscalrcrdrZr[recreated not_createds r) create_eventzAsyncJMAPClient.create_events:))++++++++hK@@@%g&87E:JKK--//////// )2 . . %KA1113B93M3M0A{Aqk))))';w3GHHH')))#O[w'----2'/:YZZZZr+event_idrcrK|d{V}t|j|g}||gd{V}|D]P\}}}|dkrDt |}|st |jd|dt|ddcSQt |jd ) aFetch a calendar event as an iCalendar string. Args: event_id: The JMAP event ID to retrieve. Returns: A :class:`~caldav.jmap.objects.calendar_object.JMAPCalendarObject` wrapping the raw JSCalendar dict. ``parent`` is ``None`` since no :class:`~caldav.jmap.objects.calendar.JMAPCalendar` is available at the client level. Raises: JMAPMethodError: If the event is not found. NidsCalendarEvent/getzEvent not found: notFoundrIrrWparentNo CalendarEvent/get responserC)r8rr`r]r rrMr) r(rvrSrcrdrZr[reitemss r) get_eventzAsyncJMAPClient.get_events))++++++++w1zBBB--//////// )2 F F %KA111' 22)#O=8==#- *uQxEEEEEE2'/:YZZZZr+cK|d{V}t|}|ddt|j||i}||gd{V}|D]E\}}} |dkr9t |\} } } } } } || vr||| |dSFt|j d)aUpdate a calendar event from an iCalendar string. Args: event_id: The JMAP event ID to update. ical_str: A VCALENDAR string with the updated event data. Raises: JMAPMethodError: If the server rejects the update. NuidrnrprC) r8rpopr r`r]rrqrrM) r(rvrkrSpatchrcrdrZr[re not_updateds r) update_eventzAsyncJMAPClient.update_events))++++++++h'' %%g&88U:KLL--//////// )2   %KA111-z+AsyncJMAPClient._search..s6'D@@@r+)r8_build_event_search_callsr`r]r ) r(rirrrr}rScallsrdrZr[res ` r)_searchzAsyncJMAPClient._searchs))++++++++..w/A;PUWZ\`aa--........ )2   %KA111 / : :2  r+cDK|||||d{VS)aPSearch for calendar events. All parameters are optional; omitting all returns every event in the account. Results are fetched in a single batched JMAP request using a result reference from ``CalendarEvent/query`` into ``CalendarEvent/get``. Args: calendar_id: Limit results to this calendar. start: Only events ending after this datetime (``YYYY-MM-DDTHH:MM:SS``). end: Only events starting before this datetime (``YYYY-MM-DDTHH:MM:SS``). text: Free-text search across title, description, locations, and participants. Returns: List of :class:`~caldav.jmap.objects.calendar_object.JMAPCalendarObject` instances. ``parent`` is ``None`` on these objects; use :meth:`JMAPCalendar.search` if you need ``parent`` set. )rirrrN)r)r(rirrrs r) search_eventszAsyncJMAPClient.search_events s50\\kCVZ\[[[[[[[[[r+cK|d{V}t|jg}||gd{V}|D]$\}}}|dkr|ddcS%t |jd)uyReturn the current CalendarEvent state string for use as a sync token. Calls ``CalendarEvent/get`` with an empty ID list — no event data is transferred, only the ``state`` field from the response. Returns: Opaque state string. Pass to :meth:`get_objects_by_sync_token` to retrieve only what changed since this point. Nrxrzstater~rC)r8rr`r]rRrrMr(rSrcrdrZr[res r)get_sync_tokenzAsyncJMAPClient.get_sync_token&s))++++++++w1r:::--//////// )2 2 2 %KA111 }}Wb111112'/:YZZZZr+ sync_tokenDtuple[list[JMAPCalendarObject], list[JMAPCalendarObject], list[str]]c^K|d{V}t|j|}||gd{V}g}g}g}|D];\}} } |dkr/t | \} } } }}}| rt |jdd<||z} | sgg|fSt|j| } || gd{V}i|D]8\}} } |dkr,t| D]}t|d|d <9fd |D}fd |D}|||fS) aFetch events changed since a previous sync token. Calls ``CalendarEvent/changes`` to discover which events were created, modified, or destroyed since ``sync_token`` was issued. Created and modified events are returned as :class:`~caldav.jmap.objects.calendar_object.JMAPCalendarObject` instances; destroyed events are returned as IDs (the objects no longer exist on the server). Args: sync_token: A state string previously returned by :meth:`get_sync_token` or by a prior call to this method. Returns: A 3-tuple ``(added, modified, deleted)``: - ``added``: objects for newly created events (``parent`` is ``None``). - ``modified``: objects for updated events (``parent`` is ``None``). - ``deleted``: Event IDs that were destroyed. Raises: JMAPMethodError: If the server reports ``hasMoreChanges: true``. NzCalendarEvent/changeszCalendarEvent/changes response was truncated by the server (hasMoreChanges=true). Call get_sync_token() to obtain a fresh baseline and re-sync.serverPartialFailrIrxrzr|roc(g|]}|v|Sr&r&ri events_by_ids r)rz=AsyncJMAPClient.get_objects_by_sync_token..ts'KKKQl9J9Ja9J9J9Jr+c(g|]}|v|Sr&r&rs r)rz=AsyncJMAPClient.get_objects_by_sync_token..us'NNNAy/I/I,1aA}},,))'=3JKKK 2 '/:YZZZZr+rcK|||d{VD]$}|jd|kr|cS%|d{V}t |jd|)N)rir}rz#No calendar object found with UID: rC)rrWrRr8rrM)r(rrir}objrSs r)_get_object_by_uidz"AsyncJMAPClient._get_object_by_uids+fMMMMMMMM  Cx||E""c)) *))++++++++(Sc(S(S    r+ list[dict]cK|d{V}t|j}||gtd{V}|D]\}}}|dkrt |cSgS)zFetch all task lists for the authenticated account. Returns: List of raw JMAP TaskList dicts as returned by the server. Nr;z TaskList/get)r8rr`r]rrrs r)get_task_listszAsyncJMAPClient.get_task_listss ))++++++++"7#566--k-BBBBBBBB )2 6 6 %KAn,,*955555- r+ task_list_idtitlecK|d{V}dttj||dddd}||t |jd|i}||gtd{V}|D]R\}} } |dkrFt| \} } } } } } d| vr| || d| dd cSSt|j d ) a,Create a task in a task list. Args: task_list_id: The JMAP task list ID to create the task in. title: Task title (maps to VTODO ``SUMMARY``). **kwargs: Optional JMAP Task fields using wire names: ``description``, ``due``, ``start``, ``timeZone``, ``estimatedDuration``, ``percentComplete``, ``progress``, ``priority``. Returns: The server-assigned JMAP task ID. Raises: JMAPMethodError: If the server rejects the create request. NTaskrz needs-action)z@typer taskListIdrpercentCompleteprogresspriorityrmrTask/setroNo Task/set responserC) r8rjuuiduuid4updaterr`r]rrrqrrM) r(rrkwargsrS task_dictrcrdrZr[rersrts r) create_taskzAsyncJMAPClient.create_tasksD ))++++++++tz||$$& &      $W%7'99MNN--k-BBBBBBBB )2 . . %KAj((3A)3L3L0A{Aqk))))';w3GHHHw'---- ) '/:PQQQQr+task_iddictcpK|d{V}t|j|g}||gtd{V}|D]H\}}}|dkr<|dg}|st |jd|d|d cSIt |jd ) zFetch a task by ID. Args: task_id: The JMAP task ID to retrieve. Returns: Raw JMAP Task dict as returned by the server. Raises: JMAPMethodError: If the task is not found. NrxrzTask/getr=zTask not found: r{rIrzNo Task/get responserC)r8rr`r]rrRrrM) r(rrSrcrdrZr[rers r)get_taskzAsyncJMAPClient.get_tasks))++++++++g0wi@@@--k-BBBBBBBB )2 %KAj((! fb11)#O;';;#- Qx)'/:PQQQQr+rcjK|d{V}t|j||i}||gtd{V}|D]E\}}}|dkr9t |\}}}}} }|| vr||| |dSFt|jd)aUpdate a task with a partial patch. Args: task_id: The JMAP task ID to update. patch: Partial patch dict mapping property names to new values. Raises: JMAPMethodError: If the server rejects the update. NrrrrC) r8rr`r]rrrqrrM) r(rrrSrcrdrZr[rers r) update_taskzAsyncJMAPClient.update_tasks))++++++++$W%7'59IJJ--k-BBBBBBBB )2   %KAj((-;I-F-F*1aKk))))';w3GHHH ) '/:PQQQQr+chK|d{V}t|j|g}||gtd{V}|D]E\}}}|dkr9t |\}}}}}}||vr||||dSFt|jd)zDelete a task. Args: task_id: The JMAP task ID to delete. Raises: JMAPMethodError: If the server rejects the delete. NrrrrC) r8rr`r]rrrqrrM) r(rrSrcrdrZr[rers r) delete_taskzAsyncJMAPClient.delete_tasks))++++++++%g&87)DD--k-BBBBBBBB )2   %KAj((/=i/H/H,1aA}m++))'=3IJJJ ) '/:PQQQQr+)r#r")r#r,)r#rr%)r9r:r;r<r#r=)r#r^)rirjrkrjr#rj)rvrjr#r)rvrjrkrjr#r,)NNNNN) rirrrrrrrr}rr#r)NNNN) rirrrrrrrr#r)r#rj)rrjr#r)rvrjr#r,)NN)rrjrirr}rr#r)r#r)rrjrrjr#rj)rrjr#r)rrjrrr#r,)rrjr#r,)__name__ __module__ __qualname____doc__r*r1r8r]rhrurrrrrrrrrrrrrr&r+r)r"r"-s*####7 7 7 7 7 r([[[[@[[[[@[[[[6#' &* .#'  \\\\\4[[[[$>*>*>*>*@[[[[.W[       %R%R%R%RNRRRR:RRRR.RRRRRRr+r").r __future__rloggingrniquestsrcaldav.jmap._methods.calendarrrcaldav.jmap._methods.eventrrr r r r r rcaldav.jmap._methods.taskrrrrrrrcaldav.jmap.clientrrrcaldav.jmap.convertrcaldav.jmap.errorrrcaldav.jmap.objects.calendarr#caldav.jmap.objects.calendar_objectrcaldav.jmap.sessionrr getLoggerrKr"r&r+r)rs#""""" !!!!!!PPPPPPPP                    LKKKKKKKKK------<<<<<<<<555555BBBBBB<<<<<<<<g &&lRlRlRlRlRolRlRlRlRlRr+