o .jQ@sdZddlmZddlZddlZddlmZddlmZm 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*e+dZ,GdddeZ-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.jmapc@seZdZdZdGddZdHddZdId d ZdJdKddZdLddZdMddZ dNddZ dOdd Z dPdQd(d)Z dRdSd*d+Z dTd,d-ZdUd0d1ZdVd2d3Z dWdXd5d6ZdYd8d9ZdZd error_type)r3rr8logdebugapi_urllenrpostr2r/ status_coderraise_for_statusr;getr)r%r4r6sessionpayloadhttpresponsedatamethod_responsesresp method_name resp_argscall_idrDr#r#r&_requestQsD(     zAsyncJMAPClient._requestlist[JMAPCalendar]c sn|IdH}t|j}||gIdH}|D]\}}}|dkr4t|}|D]}||_d|_q'|SqgS)zFetch all calendars for the authenticated account. Returns: List of :class:`~caldav.jmap.objects.calendar.JMAPCalendar` objects. Nz Calendar/getT)r3r account_idrWr_client _is_async) r%rMcall responsesrTrU_ calendarscalr#r#r& get_calendarss zAsyncJMAPClient.get_calendars calendar_idstrical_strc s|IdH}t||d}t|jd|i}||gIdH}|D]2\}}} |dkrSt|\} } } } } } d| vr@||| dd| vrKt|jdd| ddSq!t|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. N)rbnew-0CalendarEvent/setz:CalendarEvent/set response missing created entry for new-0r=idNo CalendarEvent/set response) r3rrrYrWr _raise_set_errorrrG) r%rbrdrMjscalr\r]rTrUr^created not_createdr#r#r& create_events$  zAsyncJMAPClient.create_eventevent_idrc s|IdH}t|j|gd}||gIdH}|D]$\}}}|dkr?t|}|s5t|jd|ddt|dddSqt|jd d ) 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: notFoundrCrrQparentNo CalendarEvent/get responser=)r3rrYrWr rrGr) r%rnrMr\r]rTrUr^itemsr#r#r& get_events  zAsyncJMAPClient.get_eventc s|IdH}t|}|ddt|j||i}||gIdH}|D]"\}}} |dkrGt|\} } } } } } || vrD||| |dSq%t|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. Nuidrfrhr=) r3rpopr rYrWr rirrG) r%rnrdrMpatchr\r]rTrUr^ not_updatedr#r#r& update_events  zAsyncJMAPClient.update_event str | NonestartendtextrtJMAPCalendar | Nonelist[JMAPCalendarObject]c sj|IdH}||j||||}||IdH}|D]\} } } | dkr2fddt| DSqgS)Nrqcsg|]}t|dqS)rsr).0itemrtr#r& s z+AsyncJMAPClient._search..)r3_build_event_search_callsrYrWr ) r%rbr~rrrtrMcallsr]rTrUr^r#rr&_searchs  zAsyncJMAPClient._searchcs|j||||dIdHS)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. )rbr~rrN)r)r%rbr~rrr#r#r& search_events szAsyncJMAPClient.search_eventscsf|IdH}t|jgd}||gIdH}|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. Nrorqstaterur=)r3rrYrWrLrrGr%rMr\r]rTrUr^r#r#r&get_sync_token&s zAsyncJMAPClient.get_sync_token sync_tokenDtuple[list[JMAPCalendarObject], list[JMAPCalendarObject], list[str]]cs|IdH}t|j|}||gIdH}g}g}g}|D]\}} } |dkr.crr#r#rrr#r&rur) r3rrYrWr rrGrr r)r%rrM changes_callr] created_ids updated_ids destroyedrTrUr^has_more fetch_idsget_call get_responsesraddedmodifiedr#rr&get_objects_by_sync_token8s>     z)AsyncJMAPClient.get_objects_by_sync_tokenc s|IdH}t|j|g}||gIdH}|D]"\}}}|dkrt|\}}}}}}||vr;||||dSqt|jdd)zDelete a task. Args: task_id: The JMAP task ID to delete. Raises: JMAPMethodError: If the server rejects the delete. Nrrrr=) r3rrYrWrrrirrG) r%rrMr\r]rTrUr^rr#r#r& delete_tasks zAsyncJMAPClient.delete_task)r!r )r!r))r!rr")r4r5r6r7r!r8)r!rX)rbrcrdrcr!rc)rnrcr!r)rnrcrdrcr!r))NNNNN) rbr}r~r}rr}rr}rtrr!r)NNNN) rbr}r~r}rr}rr}r!r)r!rc)rrcr!r)rnrcr!r))NN)rxrcrbr}rtrr!r)r!r)rrcrrcr!rc)rrcr!r)rrcrzrr!r))rrcr!r))__name__ __module__ __qualname____doc__r'r-r3rWrarmrwr|rrrrrrrrrrrr#r#r#r&r -s>     9      @  ' r ).r __future__rloggingrniquestsrcaldav.jmap._methods.calendarrrcaldav.jmap._methods.eventrrrr r r r r caldav.jmap._methods.taskrrrrrrrcaldav.jmap.clientrrrcaldav.jmap.convertrcaldav.jmap.errorrrcaldav.jmap.objects.calendarr#caldav.jmap.objects.calendar_objectrcaldav.jmap.sessionrr getLoggerrEr r#r#r#r&s   ($