i^`dZddlmZddlZddlZ ddlZddlmZn#e $r ddlZddl mZYnwxYwddl m Z m Z ddlmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZmZddl m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'm(Z(dd l)m*Z*dd l+m,Z,dd l-m.Z.m/Z/ddl0m1Z1ej2dZ3e"e!gZ4e"e#gZ5GddZ6Gdde6Z7dS)a  Synchronous JMAP client. Wraps session establishment, HTTP communication, and method dispatching into a single object with a clean public API. Auth note: JMAP has no 401-challenge-retry dance (unlike CalDAV). Credentials are sent upfront on every request. A 401/403 is a hard failure. ) annotationsN) HTTPBasicAuth)build_calendar_getparse_calendar_get) build_event_changesbuild_event_getbuild_event_querybuild_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)CALENDAR_CAPABILITYCORE_CAPABILITYTASK_CAPABILITY) ical_to_jscal) JMAPAuthErrorJMAPMethodError) JMAPCalendarJMAPCalendarObject)Session fetch_session)HTTPBearerAuthz caldav.jmapcJeZdZ ddd ZddZddZeddZdS) _JMAPClientBaseNurlstrusername str | Nonepassword auth_typetimeoutintreturnNonec||_||_||_||_d|_| ||_dS|||_dSN)r&r(r*r,_session_cache_auth _build_auth)selfr&r(r*authr+r,s M/root/projects/butler/venv/lib/python3.11/site-packages/caldav/jmap/client.py__init__z_JMAPClientBase.__init__;sR     .2  DJJJ)))44DJJJc|}|*|jrd}n |jrd}nt|jd|dkr>|jr|jst|jdt |j|jS|dkr1|jst|jdt |jSt|jd|d ) aSelect and construct the auth object. **The JMAP support is experimental, the API may change in minor-releases** JMAP supports Basic and Bearer auth; Digest is not supported. When ``auth_type`` is ``None`` the type is inferred from the credentials supplied: a username triggers Basic, a password alone triggers Bearer, and neither raises :class:`JMAPAuthError`. NbasicbearerzDNo credentials provided. Supply username+password or a bearer token.r&reasonz/Basic auth requires both username and password.z?Bearer auth requires a token supplied as the password argument.zUnsupported auth_type z. Use 'basic' or 'bearer'.)r(r*rr&rr")r5r+effective_types r7r4z_JMAPClientBase._build_authOs#  !} !( !)#a W $ $=   #L! >> > x ' '= #\"$-00 0H\\\\ r9sessionr errdictc^t|jd||dd)Nz set failed: type serverErrorr&r> error_type)rapi_urlget)r5r@rAs r7_raise_set_errorz _JMAPClientBase._raise_set_errorys:'#''wwv}55    r9 account_id calendar_idstartendtext list[tuple]ci}||g|d<|||d<|||d<|||d<t||pd}d|dd d d d d f}||gS)zPReturn a batched [CalendarEvent/query, CalendarEvent/get] call list for _search.N inCalendarsafterbeforerO)filter_conditionCalendarEvent/getz ev-query-0zCalendarEvent/queryz/ids)resultOfnamepath) accountIdz#idszev-get-1)r )rKrLrMrNrO filter_dict query_callget_calls r7_build_event_search_callsz)_JMAPClientBase._build_event_search_callss  "*5K &  #(K ?$'K !  "&K &zKDWSWXXX ' ,1"     H%%r9)NNNNr%) r&r'r(r)r*r)r+r)r,r-r.r/)r+r))r@r rArBr.r/) rKr'rLr)rMr)rNr)rOr)r.rP)__name__ __module__ __qualname__r8r4rJ staticmethodr^r9r7r$r$:s $#  $55555(((((T    &&&\&&&r9r$ceZdZdZd4dZd5dZd6dZd7d8dZd9dZd:dZ d;dZ dd Z d?d@d!Z dAd"ZdBd%ZdCd&Z dDdEd(ZdFd*ZdGd-ZdHd0ZdId2ZdJd3ZdS)K JMAPClientaSynchronous JMAP client for calendar operations. Usage:: from caldav.jmap import get_jmap_client client = get_jmap_client(url="https://jmap.example.com/.well-known/jmap", username="alice", password="secret") calendars = 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 requests-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. r.c|Sr1rcr5s r7 __enter__zJMAPClient.__enter__s r9r/cdSr1rc)r5exc_typeexc_valexc_tbs r7__exit__zJMAPClient.__exit__str9r cj|j&t|j|j|j|_|jS)z5Return the cached Session, fetching it on first call.N)r6r,)r2r!r&r3r,rgs r7 _get_sessionzJMAPClient._get_sessions3   &"/tzSWS_"`"`"`D ""r9N method_callsrPusinglist[str] | Nonelistcd|}||ntt|d}td|jt |tj|j||j ddd|j }|j dvrt|jd|j d  | |}|d g}|D]>}|\} } } | d kr0| d d} t!|jd| | ?|S)aPOST 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. requests.HTTPError: On other non-2xx HTTP responses. N)rq methodCallsz"JMAP POST to %s: %d method call(s)zapplication/json)z Content-TypeAccept)jsonr6headersr,)iizHTTP z from API endpointr=methodResponseserrorrDrEzMethod call failed: rF)ro_DEFAULT_USINGrslogdebugrHlenrequestspostr3r, status_coderraise_for_statusrwrIr) r5rprqr@payloadresponsedatamethod_responsesresp method_name resp_argscall_idrGs r7_requestzJMAPClient._requestst"##%%$/UU^ --   6\IZIZ[[[= O%7CUVVL      : - -OGx3GGG  !!###}}88$5r::$  D.2 +KGg%%&]]6=AA %=)==)& r9list[JMAPCalendar]c|}t|j}||g}|D]2\}}}|dkr&t |}|D]}||_d|_|cS3gS)zFetch all calendars for the authenticated account. Returns: List of :class:`~caldav.jmap.objects.calendar.JMAPCalendar` objects. z Calendar/getF)rorrKrr_client _is_async) r5r@call responsesrr_ calendarscals r7 get_calendarszJMAPClient.get_calendarss ##%%!'"455MM4&)) )2 ! ! %KAn,,.y99 $**C"&CK$)CMM    - r9rLr'ical_strc|}t||}t|jd|i}||g}|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. )rLnew-0CalendarEvent/setz:CalendarEvent/set response missing created entry for new-0r=idNo CalendarEvent/set response) rorr rKrrrJrrH) r5rLrr@jscalrrrrrcreated not_createds r7 create_eventzJMAPClient.create_events##%%hK@@@%g&87E:JKKMM4&)) )2 . . %KA1113B93M3M0A{Aqk))))';w3GHHH')))#O[w'----2'/:YZZZZr9event_idrcV|}t|j|g}||g}|D]P\}}}|dkrDt |}|st |jd|dt|ddcSQt |jd ) aFetch a calendar event by JMAP event ID. 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. idsrVzEvent not found: notFoundrFrNrparentNo CalendarEvent/get responser=)rorrKrrrrHr) r5rr@rrrrritemss r7 get_eventzJMAPClient.get_event/s##%%w1zBBBMM4&)) )2 F F %KA111' 22)#O=8==#- *uQxEEEEEE2'/:YZZZZr9c|}t|}|ddt|j||i}||g}|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. uidNrrr=) rorpopr rKrrrJrrH) r5rrr@patchrrrrr not_updateds r7 update_eventzJMAPClient.update_eventOs##%%h'' %%g&88U:KLLMM4&)) )2   %KA111-z&JMAPClient._search..vs6'D@@@r9)ror^rKrr) r5rLrMrNrOrr@callsrrrrs ` r7_searchzJMAPClient._searchhs##%%..w/A;PUWZ\`aaMM%(( )2   %KA111 / : :2  r9c4|||||S)aSearch 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 since no :class:`~caldav.jmap.objects.calendar.JMAPCalendar` is available at the client level; use :meth:`JMAPCalendar.search` if you need ``parent`` set. )rLrMrNrO)r)r5rLrMrNrOs r7 search_eventszJMAPClient.search_events}s4|| 5cPT|UUUr9c|}t|jg}||g}|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. rrVstaterr=)rorrKrrIrrHr5r@rrrrrs r7get_sync_tokenzJMAPClient.get_sync_tokens##%%w1r:::MM4&)) )2 2 2 %KA111 }}Wb111112'/:YZZZZr9 sync_tokenDtuple[list[JMAPCalendarObject], list[JMAPCalendarObject], list[str]]c6|}t|j|}||g}g}g}g}|D];\}} } |dkr/t | \} } } }}}| rt |jdd<||z} | sgg|fSt|j| } || g}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``. zCalendarEvent/changeszCalendarEvent/changes response was truncated by the server (hasMoreChanges=true). Call get_sync_token() to obtain a fresh baseline and re-sync.serverPartialFailrFrrVNrrc(g|]}|v|Srcrcri events_by_ids r7rz8JMAPClient.get_objects_by_sync_token..s'KKKQl9J9Ja9J9J9Jr9c(g|]}|v|Srcrcrs r7rz8JMAPClient.get_objects_by_sync_token..s'NNNA|}t|j|g}||g}|D]E\}}}|dkr9t |\}}}}}}||vr||||dSFt |jd)zDelete a calendar event. Args: event_id: The JMAP event ID to delete. Raises: JMAPMethodError: If the server rejects the delete. rNrr=)ror rKrrrJrrH) r5rr@rrrrr not_destroyeds r7 delete_eventzJMAPClient.delete_events##%%&w'9H:FFMM4&)) )2   %KA111/>y/I/I,1aA}},,))'=3JKKK 2 '/:YZZZZr9rc|||D]$}|jd|kr|cS%t|jd|)N)rLrrz#No calendar object found with UID: r=)rrrIrrorH)r5rrLrobjs r7_get_object_by_uidzJMAPClient._get_object_by_uids<*>*>*>*@[[[[.W[       %R%R%R%RNRRRR:RRRR.RRRRRRr9re)8r __future__rloggingrniquestsr niquests.authr ImportError requests.authcaldav.jmap._methods.calendarrrcaldav.jmap._methods.eventrrr r r r r rrcaldav.jmap._methods.taskrrrrrrrcaldav.jmap.constantsrrrcaldav.jmap.convertrcaldav.jmap.errorrrcaldav.jmap.objects.calendarr#caldav.jmap.objects.calendar_objectrcaldav.jmap.sessionr r!caldav.requestsr" getLoggerr|r{rr$rercr9r7rs#""""" ,+++++++,,,OOO++++++++,QPPPPPPP                      XWWWWWWWWW------<<<<<<<<555555BBBBBB66666666******g &&!#670 e&e&e&e&e&e&e&e&PjRjRjRjRjRjRjRjRjRjRs //