o .j^`@s`dZddlmZddlZddlZz ddlZddlmZWne y/ddlZddl mZYnwddl 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/ddl0m1Z1e2dZ3e"e!gZ4e"e#gZ5GdddZ6Gddde6Z7dS)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.jmapc@sDeZdZ     dd d dZd!ddZd"ddZed#ddZdS)$_JMAPClientBaseNurlstrusername str | Nonepassword auth_typetimeoutintreturnNonecCs@||_||_||_||_d|_|dur||_dS|||_dSN)r$r&r(r*_session_cache_auth _build_auth)selfr$r&r(authr)r*r4G/home/thesage/.local/lib/python3.10/site-packages/caldav/jmap/client.py__init__;s  z_JMAPClientBase.__init__cCs|}|dur|jr d}n |jrd}nt|jdd|dkr1|jr#|js*t|jddt|j|jS|dkrD|js?t|jddt|jSt|jd|d 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!)r2r)effective_typer4r4r5r1Os8    z_JMAPClientBase._build_authsessionrerrdictcCst|jd||ddd)Nz set failed: type serverErrorr$r: error_type)rapi_urlget)r2r<r=r4r4r5_raise_set_errorys  z _JMAPClientBase._raise_set_error account_id calendar_idstartendtext list[tuple]cCsvi}|dur |g|d<|dur||d<|dur||d<|dur#||d<t||p(dd}d|dd d d d d f}||gS)zPReturn a batched [CalendarEvent/query, CalendarEvent/get] call list for _search.N inCalendarsafterbeforerJ)filter_conditionCalendarEvent/getz ev-query-0zCalendarEvent/queryz/ids)resultOfnamepath) accountIdz#idszev-get-1)r)rFrGrHrIrJ filter_dict query_callget_callr4r4r5_build_event_search_callss(   z)_JMAPClientBase._build_event_search_calls)NNNNr#) r$r%r&r'r(r'r)r'r*r+r,r-)r)r')r<rr=r>r,r-) rFr%rGr'rHr'rIr'rJr'r,rK)__name__ __module__ __qualname__r6r1rE staticmethodrXr4r4r4r5r":s  *r"c@seZdZdZdGddZdHddZdId d ZdJdKddZdLddZdMddZ dNddZ dOdd Z dPdQd(d)Z dRdSd*d+Z dTd,d-ZdUd0d1ZdVd2d3Z dWdXd5d6ZdYd8d9ZdZd     zJMAPClient._requestlist[JMAPCalendar]c Cs`|}t|j}||g}|D]\}}}|dkr-t|}|D]}||_d|_q |SqgS)zFetch all calendars for the authenticated account. Returns: List of :class:`~caldav.jmap.objects.calendar.JMAPCalendar` objects. z Calendar/getF)rerrFrr_client _is_async) r2r<call responsesr}r~_ calendarscalr4r4r5 get_calendarss  zJMAPClient.get_calendarsrGr%ical_strc Cs|}t||d}t|jd|i}||g}|D]2\}}} |dkrLt|\} } } } } } d| vr9||| dd| vrDt|jdd| ddSqt|jdd)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. )rGnew-0CalendarEvent/setz:CalendarEvent/set response missing created entry for new-0r9idNo CalendarEvent/set response) rerr rFrrrErrC) r2rGrr<jscalrrr}r~rcreated not_createdr4r4r5 create_events"   zJMAPClient.create_eventevent_idrc Cs|}t|j|gd}||g}|D]$\}}}|dkr8t|}|s.t|jd|ddt|dddSqt|jd d ) 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. idsrPzEvent not found: notFoundrArNrzparentNo CalendarEvent/get responser9)rerrFrr rrCr) r2rr<rrr}r~ritemsr4r4r5 get_event/s  zJMAPClient.get_eventc Cs|}t|}|ddt|j||i}||g}|D]"\}}} |dkr@t|\} } } } } } || vr=||| |dSqt|j dd)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. uidNrrr9) rerpopr rFrrrErrC) r2rrr<patchrrr}r~r not_updatedr4r4r5 update_eventOs   zJMAPClient.update_eventr'rHrIrJrJMAPCalendar | Nonelist[JMAPCalendarObject]c s\|}||j||||}||}|D]\} } } | dkr+fddt| DSqgS)NrPcsg|]}t|dqS)rr).0itemrr4r5 vs z&JMAPClient._search..)rerXrFrr ) r2rGrHrIrJrr<callsrr}r~rr4rr5_searchhs   zJMAPClient._searchcCs|j||||dS)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. )rGrHrIrJ)r)r2rGrHrIrJr4r4r5 search_events}szJMAPClient.search_eventscCsX|}t|jgd}||g}|D]\}}}|dkr$|ddSqt|jdd)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. rrPstaterr9)rerrFrrDrrCr2r<rrr}r~rr4r4r5get_sync_tokens  zJMAPClient.get_sync_token sync_tokenDtuple[list[JMAPCalendarObject], list[JMAPCalendarObject], list[str]]cs|}t|j|}||g}g}g}g}|D]\}} } |dkr5t| \} } } }}}| r5t|jdddq||} | sAgg|fSt|j| d} || g}i|D]\}} } |dkrlt| D] }t |dd|d <q_qRfd d |D}fd d |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.serverPartialFailrArrPNrrcg|] }|vr|qSr4r4ri events_by_idr4r5rz8JMAPClient.get_objects_by_sync_token..crr4r4rrr4r5rr) rerrFrr rrCrr r)r2rr< changes_callr created_ids updated_ids destroyedr}r~rhas_more fetch_idsrW get_responsesraddedmodifiedr4rr5get_objects_by_sync_tokens<       z$JMAPClient.get_objects_by_sync_tokenc Csz|}t|j|g}||g}|D]"\}}}|dkr5t|\}}}}}}||vr2||||dSqt|jdd)zDelete a calendar event. Args: event_id: The JMAP event ID to delete. Raises: JMAPMethodError: If the server rejects the delete. rNrr9)rer rFrrrErrC) r2rr<rrr}r~r not_destroyedr4r4r5 delete_events  zJMAPClient.delete_eventrcCsD|j||dD]}|jd|kr|Sqt|jd|d)N)rGrrz#No calendar object found with UID: r9)rrzrDrrerC)r2rrGrobjr4r4r5_get_object_by_uidszJMAPClient._get_object_by_uid list[dict]cCsJ|}t|j}|j|gtd}|D]\}}}|dkr"t|SqgS)zFetch all task lists for the authenticated account. Returns: List of raw JMAP TaskList dicts as returned by the server. rgz TaskList/get)rerrFr _TASK_USINGrrr4r4r5get_task_lists s  zJMAPClient.get_task_lists task_list_idtitlec Ks|}dtt||dddd}||t|jd|i}|j|gtd}|D]'\}} } |dkrPt | \} } } } } } d| vrH| || d| ddSq)t |j d 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. Taskrz needs-action)z@typer taskListIdrpercentCompleteprogresspriorityrrTask/setrNo Task/set responser9) rer%uuiduuid4updaterrFrrrrErrC) r2rrkwargsr< task_dictrrr}r~rrrr4r4r5 create_tasks(  zJMAPClient.create_tasktask_idr>c Cs|}t|j|gd}|j|gtd}|D]"\}}}|dkr8|dg}|s2t|jd|dd|dSqt|jd d ) 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. rrzTask/getrizTask not found: rrArzNo Task/get responser9)rerrFrrrDrrC) r2rr<rrr}r~rrr4r4r5get_taskDs    zJMAPClient.get_taskrc Cs|}t|j||i}|j|gtd}|D]"\}}}|dkr8t|\}}}}} }|| vr5||| |dSqt|jdd)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. rrNrr9) rerrFrrrrErrC) r2rrr<rrr}r~rrr4r4r5 update_taskas zJMAPClient.update_taskc Cs~|}t|j|g}|j|gtd}|D]"\}}}|dkr7t|\}}}}}}||vr4||||dSqt|jdd)zDelete a task. Args: task_id: The JMAP task ID to delete. Raises: JMAPMethodError: If the server rejects the delete. rrNrr9) rerrFrrrrErrC) r2rr<rrr}r~rrr4r4r5 delete_taskxs zJMAPClient.delete_task)r,r])r,r-)r,rr.)rfrKrgrhr,ri)r,r)rGr%rr%r,r%)rr%r,r)rr%rr%r,r-)NNNNN) rGr'rHr'rIr'rJr'rrr,r)NNNN) rGr'rHr'rIr'rJr'r,r)r,r%)rr%r,r)rr%r,r-)NN)rr%rGr'rrr,r)r,r)rr%rr%r,r%)rr%r,r>)rr%rr>r,r-)rr%r,r-)rYrZr[__doc__r_rdrerrrrrrrrrrrrrrrrr4r4r4r5r]s>     9      @  ' r])8r __future__rloggingrniquestsrt niquests.authr ImportError requests.authcaldav.jmap._methods.calendarrrcaldav.jmap._methods.eventrrrr r r r r rcaldav.jmap._methods.taskrrrrrrrcaldav.jmap.constantsrrrcaldav.jmap.convertrcaldav.jmap.errorrrcaldav.jmap.objects.calendarr#caldav.jmap.objects.calendar_objectrcaldav.jmap.sessionrr caldav.requestsr! getLoggerrqrprr"r]r4r4r4r5s4  ,$      h