Source code for caldav.collection

"""
I'm trying to be consistent with the terminology in the RFCs:

CalendarSet is a collection of Calendars
Calendar is a collection of CalendarObjectResources
Principal is not a collection, but holds a CalendarSet.

There are also some Mailbox classes to deal with RFC6638.

A SynchronizableCalendarObjectCollection contains a local copy of objects from a calendar on the server.
"""
import logging
import sys
import uuid
import warnings
from datetime import datetime
from typing import Any
from typing import List
from typing import Optional
from typing import Tuple
from typing import TYPE_CHECKING
from typing import TypeVar
from typing import Union
from urllib.parse import ParseResult
from urllib.parse import quote
from urllib.parse import SplitResult
from urllib.parse import unquote

import icalendar

try:
    from typing import ClassVar, Optional, Union, Type

    TimeStamp = Optional[Union[date, datetime]]
except:
    pass

if TYPE_CHECKING:
    from icalendar import vCalAddress

    from .davclient import DAVClient

if sys.version_info < (3, 9):
    from typing import Callable, Container, Iterable, Iterator, Sequence

    from typing_extensions import DefaultDict, Literal
else:
    from collections import defaultdict as DefaultDict
    from collections.abc import Callable, Container, Iterable, Iterator, Sequence
    from typing import Literal

if sys.version_info < (3, 11):
    from typing_extensions import Self
else:
    from typing import Self

from .calendarobjectresource import CalendarObjectResource
from .calendarobjectresource import Event
from .calendarobjectresource import FreeBusy
from .calendarobjectresource import Journal
from .calendarobjectresource import Todo
from .davobject import DAVObject
from .elements.cdav import CalendarData
from .elements import cdav
from .elements import dav
from .lib import error
from .lib import vcal
from .lib.python_utilities import to_wire
from .lib.url import URL

_CC = TypeVar("_CC", bound="CalendarObjectResource")
log = logging.getLogger("caldav")


[docs] class CalendarSet(DAVObject): """ A CalendarSet is a set of calendars. """
[docs] def calendars(self) -> List["Calendar"]: """ List all calendar collections in this set. Returns: * [Calendar(), ...] """ cals = [] data = self.children(cdav.Calendar.tag) for c_url, c_type, c_name in data: try: cal_id = c_url.split("/")[-2] if not cal_id: continue except: log.error(f"Calendar {c_name} has unexpected url {c_url}") cal_id = None cals.append( Calendar(self.client, id=cal_id, url=c_url, parent=self, name=c_name) ) return cals
[docs] def make_calendar( self, name: Optional[str] = None, cal_id: Optional[str] = None, supported_calendar_component_set: Optional[Any] = None, method=None, ) -> "Calendar": """ Utility method for creating a new calendar. Args: name: the display name of the new calendar cal_id: the uuid of the new calendar supported_calendar_component_set: what kind of objects (EVENT, VTODO, VFREEBUSY, VJOURNAL) the calendar should handle. Should be set to ['VTODO'] when creating a task list in Zimbra - in most other cases the default will be OK. Returns: Calendar(...)-object """ return Calendar( self.client, name=name, parent=self, id=cal_id, supported_calendar_component_set=supported_calendar_component_set, ).save(method=method)
[docs] def calendar( self, name: Optional[str] = None, cal_id: Optional[str] = None ) -> "Calendar": """ The calendar method will return a calendar object. If it gets a cal_id but no name, it will not initiate any communication with the server Args: name: return the calendar with this display name cal_id: return the calendar with this calendar id or URL Returns: Calendar(...)-object """ if name and not cal_id: for calendar in self.calendars(): display_name = calendar.get_display_name() if display_name == name: return calendar if name and not cal_id: raise error.NotFoundError( "No calendar with name %s found under %s" % (name, self.url) ) if not cal_id and not name: cals = self.calendars() if not cals: raise error.NotFoundError("no calendars found") return cals[0] if self.client is None: raise ValueError("Unexpected value None for self.client") if cal_id is None: raise ValueError("Unexpected value None for cal_id") if str(URL.objectify(cal_id).canonical()).startswith( str(self.client.url.canonical()) ): url = self.client.url.join(cal_id) elif isinstance(cal_id, URL) or ( isinstance(cal_id, str) and (cal_id.startswith("https://") or cal_id.startswith("http://")) ): if self.url is None: raise ValueError("Unexpected value None for self.url") url = self.url.join(cal_id) else: if self.url is None: raise ValueError("Unexpected value None for self.url") if cal_id is None: raise ValueError("Unexpected value None for cal_id") url = self.url.join(quote(cal_id) + "/") return Calendar(self.client, name=name, parent=self, url=url, id=cal_id)
[docs] class Principal(DAVObject): """ This class represents a DAV Principal. It doesn't do much, except keep track of the URLs for the calendar-home-set, etc. A principal MUST have a non-empty DAV:displayname property (defined in Section 13.2 of [RFC2518]), and a DAV:resourcetype property (defined in Section 13.9 of [RFC2518]). Additionally, a principal MUST report the DAV:principal XML element in the value of the DAV:resourcetype property. (TODO: the resourcetype is actually never checked, and the DisplayName is not stored anywhere) """ def __init__( self, client: Optional["DAVClient"] = None, url: Union[str, ParseResult, SplitResult, URL, None] = None, calendar_home_set: URL = None, **kwargs, ## to be passed to super.__init__ ) -> None: """ Returns a Principal. End-users usually shouldn't need to construct Principal-objects directly. Use davclient.principal() to get the principal object of the logged-in user and davclient.principals() to get other principals. Args: client: a DAVClient() object url: The URL, if known. calendar_home_set: the calendar home set, if known If url is not given, deduct principal path as well as calendar home set path from doing propfinds. """ self._calendar_home_set = calendar_home_set super(Principal, self).__init__(client=client, url=url, **kwargs) if url is None: if self.client is None: raise ValueError("Unexpected value None for self.client") self.url = self.client.url cup = self.get_property(dav.CurrentUserPrincipal()) if cup is None: log.warning("calendar server lacking a feature:") log.warning("current-user-principal property not found") log.warning("assuming %s is the principal URL" % self.client.url) self.url = self.client.url.join(URL.objectify(cup))
[docs] def make_calendar( self, name: Optional[str] = None, cal_id: Optional[str] = None, supported_calendar_component_set: Optional[Any] = None, method=None, ) -> "Calendar": """ Convenience method, bypasses the self.calendar_home_set object. See CalendarSet.make_calendar for details. """ return self.calendar_home_set.make_calendar( name, cal_id, supported_calendar_component_set=supported_calendar_component_set, method=method, )
[docs] def calendar( self, name: Optional[str] = None, cal_id: Optional[str] = None, cal_url: Optional[str] = None, ) -> "Calendar": """ The calendar method will return a calendar object. It will not initiate any communication with the server. """ if not cal_url: return self.calendar_home_set.calendar(name, cal_id) else: if self.client is None: raise ValueError("Unexpected value None for self.client") return Calendar(self.client, url=self.client.url.join(cal_url))
[docs] def get_vcal_address(self) -> "vCalAddress": """ Returns the principal, as an icalendar.vCalAddress object """ from icalendar import vCalAddress, vText cn = self.get_display_name() ids = self.calendar_user_address_set() cutype = self.get_property(cdav.CalendarUserType()) ret = vCalAddress(ids[0]) ret.params["cn"] = vText(cn) ret.params["cutype"] = vText(cutype) return ret
@property def calendar_home_set(self): if not self._calendar_home_set: calendar_home_set_url = self.get_property(cdav.CalendarHomeSet()) ## owncloud returns /remote.php/dav/calendars/tobixen@e.email/ ## in that case the @ should be quoted. Perhaps other ## implementations return already quoted URLs. Hacky workaround: if ( calendar_home_set_url is not None and "@" in calendar_home_set_url and "://" not in calendar_home_set_url ): calendar_home_set_url = quote(calendar_home_set_url) self.calendar_home_set = calendar_home_set_url return self._calendar_home_set @calendar_home_set.setter def calendar_home_set(self, url) -> None: if isinstance(url, CalendarSet): self._calendar_home_set = url return sanitized_url = URL.objectify(url) ## TODO: sanitized_url should never be None, this needs more ## research. added here as it solves real-world issues, ref ## https://github.com/python-caldav/caldav/pull/56 if sanitized_url is not None: if ( sanitized_url.hostname and sanitized_url.hostname != self.client.url.hostname ): # icloud (and others?) having a load balanced system, # where each principal resides on one named host ## TODO: ## Here be dragons. sanitized_url will be the root ## of all future objects derived from client. Changing ## the client.url root by doing a principal.calendars() ## is an unacceptable side effect and may be a cause of ## incompatibilities with icloud. Do more research! self.client.url = sanitized_url self._calendar_home_set = CalendarSet( self.client, self.client.url.join(sanitized_url) )
[docs] def calendars(self) -> List["Calendar"]: """ Return the principials calendars """ return self.calendar_home_set.calendars()
[docs] def freebusy_request(self, dtstart, dtend, attendees): """Sends a freebusy-request for some attendee to the server as per RFC6638 """ freebusy_ical = icalendar.Calendar() freebusy_ical.add("prodid", "-//tobixen/python-caldav//EN") freebusy_ical.add("version", "2.0") freebusy_ical.add("method", "REQUEST") uid = uuid.uuid1() freebusy_comp = icalendar.FreeBusy() freebusy_comp.add("uid", uid) freebusy_comp.add("dtstamp", datetime.now()) freebusy_comp.add("dtstart", dtstart) freebusy_comp.add("dtend", dtend) freebusy_ical.add_component(freebusy_comp) outbox = self.schedule_outbox() caldavobj = FreeBusy(data=freebusy_ical, parent=outbox) caldavobj.add_organizer() for attendee in attendees: caldavobj.add_attendee(attendee, no_default_parameters=True) response = self.client.post( outbox.url, caldavobj.data, headers={"Content-Type": "text/calendar; charset=utf-8"}, ) return response.find_objects_and_props()
[docs] def calendar_user_address_set(self) -> List[Optional[str]]: """ defined in RFC6638 """ _addresses: Optional[_Element] = self.get_property( cdav.CalendarUserAddressSet(), parse_props=False ) if _addresses is None: raise error.NotFoundError("No calendar user addresses given from server") assert not [x for x in _addresses if x.tag != dav.Href().tag] addresses = list(_addresses) ## possibly the preferred attribute is iCloud-specific. ## TODO: do more research on that addresses.sort(key=lambda x: -int(x.get("preferred", 0))) return [x.text for x in addresses]
[docs] def schedule_inbox(self) -> "ScheduleInbox": """ Returns the schedule inbox, as defined in RFC6638 """ return ScheduleInbox(principal=self)
[docs] def schedule_outbox(self) -> "ScheduleOutbox": """ Returns the schedule outbox, as defined in RFC6638 """ return ScheduleOutbox(principal=self)
[docs] class Calendar(DAVObject): """ The `Calendar` object is used to represent a calendar collection. Refer to the RFC for details: https://tools.ietf.org/html/rfc4791#section-5.3.1 """ def _create( self, name=None, id=None, supported_calendar_component_set=None, method=None ) -> None: """ Create a new calendar with display name `name` in `parent`. """ if id is None: id = str(uuid.uuid1()) self.id = id if method is None: if self.client: supported = self.client.features.is_supported( "create-calendar", return_type=dict ) if supported["support"] not in ("full", "fragile", "quirk"): raise error.MkcalendarError( "Creation of calendars (allegedly) not supported on this server" ) if ( supported["support"] == "quirk" and supported["behaviour"] == "mkcol-required" ): method = "mkcol" else: method = "mkcalendar" else: method = "mkcalendar" path = self.parent.url.join(id + "/") self.url = path # TODO: mkcalendar seems to ignore the body on most servers? # at least the name doesn't get set this way. # zimbra gives 500 (!) if body is omitted ... prop = dav.Prop() if name: display_name = dav.DisplayName(name) prop += [ display_name, ] if supported_calendar_component_set: sccs = cdav.SupportedCalendarComponentSet() for scc in supported_calendar_component_set: sccs += cdav.Comp(scc) prop += sccs if method == "mkcol": from caldav.lib.debug import printxml prop += dav.ResourceType() + [dav.Collection(), cdav.Calendar()] set = dav.Set() + prop mkcol = (dav.Mkcol() if method == "mkcol" else cdav.Mkcalendar()) + set r = self._query( root=mkcol, query_method=method, url=path, expected_return_value=201 ) # COMPATIBILITY ISSUE # name should already be set, but we've seen caldav servers failing # on setting the DisplayName on calendar creation # (DAViCal, Zimbra, ...). Doing an attempt on explicitly setting the # display name using PROPPATCH. if name: try: self.set_properties([display_name]) except Exception as e: ## TODO: investigate. Those asserts break. try: current_display_name = self.get_display_name() error.assert_(current_display_name == name) except: log.warning( "calendar server does not support display name on calendar? Ignoring", exc_info=True, )
[docs] def get_supported_components(self) -> List[Any]: """ returns a list of component types supported by the calendar, in string format (typically ['VJOURNAL', 'VTODO', 'VEVENT']) """ if self.url is None: raise ValueError("Unexpected value None for self.url") props = [cdav.SupportedCalendarComponentSet()] response = self.get_properties(props, parse_response_xml=False) response_list = response.find_objects_and_props() prop = response_list[unquote(self.url.path)][ cdav.SupportedCalendarComponentSet().tag ] return [supported.get("name") for supported in prop]
[docs] def save_with_invites(self, ical: str, attendees, **attendeeoptions) -> None: """ sends a schedule request to the server. Equivalent with save_event, save_todo, etc, but the attendees will be added to the ical object before sending it to the server. """ ## TODO: consolidate together with save_* obj = self._calendar_comp_class_by_data(ical)(data=ical, client=self.client) obj.parent = self obj.add_organizer() for attendee in attendees: obj.add_attendee(attendee, **attendeeoptions) obj.id = obj.icalendar_instance.walk("vevent")[0]["uid"] obj.save() return obj
def _use_or_create_ics(self, ical, objtype, **ical_data): if ical_data or ( (isinstance(ical, str) or isinstance(ical, bytes)) and b"BEGIN:VCALENDAR" not in to_wire(ical) ): ## TODO: the ical_fragment code is not much tested if ical and "ical_fragment" not in ical_data: ical_data["ical_fragment"] = ical return vcal.create_ical(objtype=objtype, **ical_data) return ical
[docs] def save_object( self, ## TODO: this should be made optional. The class may be given in the ical object. ## TODO: also, accept a string. objclass: Type[DAVObject], ## TODO: ical may also be a vobject or icalendar instance ical: Optional[str] = None, no_overwrite: bool = False, no_create: bool = False, **ical_data, ) -> "CalendarResourceObject": """Add a new event to the calendar, with the given ical. Args: objclass: Event, Journal or Todo ical: ical object (text, icalendar or vobject instance) no_overwrite: existing calendar objects should not be overwritten no_create: don't create a new object, existing calendar objects should be updated dt_start: properties to be inserted into the icalendar object , dt_end: properties to be inserted into the icalendar object summary: properties to be inserted into the icalendar object alarm_trigger: when given, one alarm will be added alarm_action: when given, one alarm will be added alarm_attach: when given, one alarm will be added Note that the list of parameters going into the icalendar object and alamrs is not complete. Refer to the RFC or the icalendar library for a full list of properties. """ o = objclass( self.client, data=self._use_or_create_ics( ical, objtype=f"V{objclass.__name__.upper()}", **ical_data ), parent=self, ) o = o.save(no_overwrite=no_overwrite, no_create=no_create) ## TODO: Saving nothing is currently giving an object with None as URL. ## This should probably be changed in some future version to raise an error ## See also CalendarObjectResource.save() if o.url is not None: o._handle_reverse_relations(fix=True) return o
## TODO: maybe we should deprecate those three
[docs] def save_event(self, *largs, **kwargs) -> "Event": """ Returns ``self.save_object(Event, ...)`` - see :class:`save_object` """ return self.save_object(Event, *largs, **kwargs)
[docs] def save_todo(self, *largs, **kwargs) -> "Todo": """ Returns ``self.save_object(Todo, ...)`` - so see :class:`save_object` """ return self.save_object(Todo, *largs, **kwargs)
[docs] def save_journal(self, *largs, **kwargs) -> "Journal": """ Returns ``self.save_object(Journal, ...)`` - so see :class:`save_object` """ return self.save_object(Journal, *largs, **kwargs)
## legacy aliases ## TODO: should be deprecated ## TODO: think more through this - is `save_foo` better than `add_foo`? ## `save_foo` should not be used for updating existing content on the ## calendar! add_object = save_object add_event = save_event add_todo = save_todo add_journal = save_journal
[docs] def save(self, method=None): """ The save method for a calendar is only used to create it, for now. We know we have to create it when we don't have a url. Returns: * self """ if self.url is None: self._create( id=self.id, name=self.name, method=method, **self.extra_init_options ) return self
# def data2object_class def _multiget( self, event_urls: Iterable[URL], raise_notfound: bool = False ) -> Iterable[str]: """ get multiple events' data. TODO: Does it overlap the _request_report_build_resultlist method """ if self.url is None: raise ValueError("Unexpected value None for self.url") rv = [] prop = dav.Prop() + cdav.CalendarData() root = ( cdav.CalendarMultiGet() + prop + [dav.Href(value=u.path) for u in event_urls] ) response = self._query(root, 1, "report") results = response.expand_simple_props([cdav.CalendarData()]) if raise_notfound: for href in response.statuses: status = response.statuses[href] if status and "404" in status: raise error.NotFoundError(f"Status {status} in {href}") for r in results: yield (r, results[r][cdav.CalendarData.tag]) ## Replace the last lines with
[docs] def multiget( self, event_urls: Iterable[URL], raise_notfound: bool = False ) -> Iterable[_CC]: """ get multiple events' data TODO: Does it overlap the _request_report_build_resultlist method? @author mtorange@gmail.com (refactored by Tobias) """ results = self._multiget(event_urls, raise_notfound=raise_notfound) for url, data in results: yield self._calendar_comp_class_by_data(data)( self.client, url=self.url.join(url), data=data, parent=self, )
[docs] def calendar_multiget(self, *largs, **kwargs): """ get multiple events' data @author mtorange@gmail.com (refactored by Tobias) This is for backward compatibility. It may be removed in 3.0 or later release. """ return list(self.multiget(*largs, **kwargs))
## TODO: this logic has been partly duplicated in calendar_multiget, but ## the code there is much more readable and condensed than this. ## Can code below be refactored? def _request_report_build_resultlist( self, xml, comp_class=None, props=None, no_calendardata=False ): """ Takes some input XML, does a report query on a calendar object and returns the resource objects found. """ matches = [] if props is None: props_ = [cdav.CalendarData()] else: props_ = [cdav.CalendarData()] + props response = self._query(xml, 1, "report") results = response.expand_simple_props(props_) for r in results: pdata = results[r] if cdav.CalendarData.tag in pdata: cdata = pdata.pop(cdav.CalendarData.tag) comp_class_ = ( self._calendar_comp_class_by_data(cdata) if comp_class is None else comp_class ) else: cdata = None if comp_class_ is None: ## no CalendarData fetched - which is normal i.e. when doing a sync-token report and only asking for the URLs comp_class_ = CalendarObjectResource url = URL(r) if url.hostname is None: # Quote when result is not a full URL url = quote(r) ## icloud hack - icloud returns the calendar URL as well as the calendar item URLs if self.url.join(url) == self.url: continue matches.append( comp_class_( self.client, url=self.url.join(url), data=cdata, parent=self, props=pdata, ) ) return (response, matches)
[docs] def search( self, xml=None, comp_class: Optional[_CC] = None, todo: Optional[bool] = None, include_completed: bool = None, sort_keys: Sequence[str] = (), sort_reverse: bool = False, expand: bool = False, server_expand: bool = False, split_expanded: bool = True, props: Optional[List[cdav.CalendarData]] = None, **kwargs, ) -> List[_CC]: """Sends a search request towards the server, processes the results if needed and returns the objects found. Caveat: The searching is done on the server side, the RFC is not very crystal clear on many of the corner cases, and servers often behave differently when presented with a search request. There is planned work to work around server incompatibilities on the client side, but as for now complicated searches will give different results on different servers. ``todo`` - searches explicitly for todo. Unless ``include_completed`` is specified, there is some special logic ensuring only pending tasks is returned. There is corresponding ``event`` and ``journal`` bools to specify that the search should be only for events or journals. When neither are set, one should expect to get all objects returned - but quite some calendar servers will return nothing. This will be solved client-side in the future, as for 2.0 it's recommended to search separately for tasks, events and journals to ensure consistent behaviour across different calendar servers and providers. ``sort_keys`` refers to (case-insensitive) properties in the icalendar object, ``sort_reverse`` can also be given. The sorting will be done client-side. Use ``start`` and ``end`` for time-range searches. Open-ended searches are supported (i.e. "everything in the future"), but it's recommended to use closed ranges (i.e. have an "event horizon" of a year and ask for "everything from now and one year ahead") and get the data expanded. With the boolean ``expand`` set, you don't have to think too much about recurrences - they will be expanded, and with the (default) ``split_expanded`` set, each recurrence will be returned as a separate list object (otherwise all recurrences will be put into one ``VCALENDAR`` and returned as one ``Event``). This makes it safe to use the ``event.component`` property. The non-expanded resultset may include events where the timespan doesn't match the date interval you searched for, as well as items with multiple components ("special" recurrences), meaning you may need logic on the client side to handle the recurrences. *Only time range searches over closed time intervals may be expanded*. As for 2.0, the expand-logic is by default done on the client-side, for consistent results across various server incompabilities. However, you may force server-side expansion by setting ``server_expand=True`` Text attribute search parameters can be given to query the "properties" in the calendar data: category, uid, summary, comment, description, location, status. According to the RFC, a substring search should be done. You may use no_category, no_summary, etc to search for objects that are missing those attributes. Negated text matches are not supported yet. For power-users, those parameters are also supported: * ``xml`` - use this search query, and ignore other filter parameters * ``comp_class`` - alternative to the ``event``, ``todo`` or ``journal`` booleans described above. * ``filters`` - other kind of filters (in lxml tree format) """ if expand not in (True, False): warnings.warn( "in cal.search(), expand should be a bool", DeprecationWarning, stacklevel=2, ) if expand == "client": expand = True if expand == "server": server_expand = True expand = False if expand or server_expand: if not kwargs.get("start") or not kwargs.get("end"): raise error.ReportError("can't expand without a date range") ## special compatibility-case when searching for pending todos if todo and not include_completed: ## There are two ways to get the pending tasks - we can ## ask the server to filter them out, or we can do it ## client side. ## If the server does not support combined searches, then it's ## safest to do it client-side. ## There is a special case (observed with radicale as of ## 2025-11) where future recurrences of a task does not ## match when doing a server-side filtering, so for this ## case we also do client-side filtering (but the ## "feature" ## search.recurrences.includes-implicit.todo.pending will ## not be supported if the feature ## "search.recurrences.includes-implicit.todo" is not ## supported ... hence the weird or below) ## To be completely sure to get all pending tasks, for all ## server implementations and for all valid icalendar ## objects, we send three different searches to the ## server. This is probably bloated, and may in many ## cases be more expensive than to ask for all tasks. At ## the other hand, for a well-used and well-handled old ## todo-list, there may be a small set of pending tasks ## and heaps of done tasks. ## TODO: consider if not ignore_completed3 is sufficient, ## then the recursive part of the query here is moot, and ## we wouldn't waste so much time on repeated queries if self.client.features.is_supported("search.combined-is-logical-and") and ( not self.client.features.is_supported( "search.recurrences.includes-implicit.todo" ) or self.client.features.is_supported( "search.recurrences.includes-implicit.todo.pending" ) ): matches = ( self.search( todo=True, ignore_completed1=True, include_completed=True, **kwargs, ) + self.search( todo=True, ignore_completed2=True, include_completed=True, **kwargs, ) + self.search( todo=True, ignore_completed3=True, include_completed=True, **kwargs, ) ) else: matches = self.search(todo=True, include_completed=True, **kwargs) objects = [] match_set = set() for item in matches: if item.url not in match_set: match_set.add(item.url) ## Client-side filtering is probably cheap, so we'll do it ## even when it shouldn't be needed. ## (can we assert all tasks have a valid STATUS field?) if any( x.get("STATUS") not in ("COMPLETED", "CANCELLED") for x in item.icalendar_instance.subcomponents ): objects.append(item) else: if not xml: if server_expand: kwargs["expand"] = True (xml, comp_class) = self.build_search_xml_query( comp_class=comp_class, todo=todo, props=props, **kwargs ) elif kwargs: raise error.ConsistencyError( "Inconsistent usage parameters: xml together with other search options" ) ## For some of the workarounds below, we will do a recursive search, with all ## those arguments: kwargs2 = { "include_completed": include_completed, "sort_reverse": sort_reverse, "expand": expand, "server_expand": server_expand, "split_expanded": split_expanded, "props": props, } if not comp_class and not self.client.features.is_supported( "search.comp-type-optional" ): if kwargs2["include_completed"] is None: kwargs2["include_completed"] = True objects = ( self.search(event=True, **kwargs2, **kwargs) + self.search(todo=True, **kwargs2, **kwargs) + self.search(journal=True, **kwargs2, **kwargs) ) self.sort_objects(objects, sort_keys, sort_reverse) return objects try: (response, objects) = self._request_report_build_resultlist( xml, comp_class, props=props ) except error.ReportError as err: ## This is only for backward compatibility. The logic is even flawed. ## But it does partially fix https://github.com/python-caldav/caldav/issues/401 if ( self.client.features.backward_compatibility_mode and not comp_class and not "400" in err.reason ): return self.search( sort_keys=sort_keys, sort_reverse=sort_reverse, *kwargs2, **kwargs, ) raise obj2 = [] for o in objects: ## This would not be needed if the servers would follow the standard ... ## TODO: use self.calendar_multiget - see https://github.com/python-caldav/caldav/issues/487 try: o.load(only_if_unloaded=True) obj2.append(o) except: logging.error( "Server does not want to reveal details about the calendar object", exc_info=True, ) pass objects = obj2 ## Google sometimes returns empty objects objects = [o for o in objects if o.has_component()] if expand: ## expand can only be used together with start and end (and not ## with xml). Error checking has already been done in ## build_search_xml_query above. start = kwargs["start"] end = kwargs["end"] ## Verify that any recurring objects returned are already expanded for o in objects: component = o.icalendar_component if component is None: continue recurrence_properties = ["exdate", "exrule", "rdate", "rrule"] if any(key in component for key in recurrence_properties): o.expand_rrule(start, end, include_completed=include_completed) ## An expanded recurring object comes as one Event() with ## icalendar data containing multiple objects. The caller may ## expect multiple Event()s. This code splits events into ## separate objects: if (expand or server_expand) and split_expanded: objects_ = objects objects = [] for o in objects_: objects.extend(o.split_expanded()) ## partial workaround for https://github.com/python-caldav/caldav/issues/201 for obj in objects: try: obj.load(only_if_unloaded=True) except: pass self.sort_objects(objects, sort_keys, sort_reverse) return objects
def sort_objects(self, objects, sort_keys, sort_reverse): def sort_key_func(x): ret = [] comp = x.icalendar_component defaults = { ## TODO: all possible non-string sort attributes needs to be listed here, otherwise we will get type errors when comparing objects with the property defined vs undefined (or maybe we should make an "undefined" object that always will compare below any other type? Perhaps there exists such an object already?) "due": "2050-01-01", "dtstart": "1970-01-01", "priority": 0, "status": { "VTODO": "NEEDS-ACTION", "VJOURNAL": "FINAL", "VEVENT": "TENTATIVE", }[comp.name], "category": "", ## Usage of strftime is a simple way to ensure there won't be ## problems if comparing dates with timestamps "isnt_overdue": not ( "due" in comp and comp["due"].dt.strftime("%F%H%M%S") < datetime.now().strftime("%F%H%M%S") ), "hasnt_started": ( "dtstart" in comp and comp["dtstart"].dt.strftime("%F%H%M%S") > datetime.now().strftime("%F%H%M%S") ), } for sort_key in sort_keys: val = comp.get(sort_key, None) if val is None: ret.append(defaults.get(sort_key.lower(), "")) continue if hasattr(val, "dt"): val = val.dt elif hasattr(val, "cats"): val = ",".join(val.cats) if hasattr(val, "strftime"): ret.append(val.strftime("%F%H%M%S")) else: ret.append(val) return ret if sort_keys: ## ref https://github.com/python-caldav/caldav/issues/448 - allow strings instead of a sequence here if isinstance(sort_keys, str): sort_keys = (sort_keys,) objects.sort(key=sort_key_func, reverse=sort_reverse)
[docs] def build_search_xml_query( self, comp_class=None, todo=None, ignore_completed1=None, ignore_completed2=None, ignore_completed3=None, event=None, journal=None, filters=None, expand=None, start=None, end=None, props=None, alarm_start=None, alarm_end=None, **kwargs, ): """This method will produce a caldav search query as an etree object. It is primarily to be used from the search method. See the documentation for the search method for more information. """ # those xml elements are weird. (a+b)+c != a+(b+c). First makes b and c as list members of a, second makes c an element in b which is an element of a. # First objective is to let this take over all xml search query building and see that the current tests pass. # ref https://www.ietf.org/rfc/rfc4791.txt, section 7.8.9 for how to build a todo-query # We'll play with it and don't mind it's getting ugly and don't mind that the test coverage is lacking. # we'll refactor and create some unit tests later, as well as ftests for complicated queries. # build the request data = cdav.CalendarData() if expand: if not start or not end: raise error.ReportError("can't expand without a date range") data += cdav.Expand(start, end) if props is None: props_ = [data] else: props_ = [data] + props prop = dav.Prop() + props_ vcalendar = cdav.CompFilter("VCALENDAR") comp_filter = None filters = filters or [] vNotCompleted = cdav.TextMatch("COMPLETED", negate=True) vNotCancelled = cdav.TextMatch("CANCELLED", negate=True) vNeedsAction = cdav.TextMatch("NEEDS-ACTION") vStatusNotCompleted = cdav.PropFilter("STATUS") + vNotCompleted vStatusNotCancelled = cdav.PropFilter("STATUS") + vNotCancelled vStatusNeedsAction = cdav.PropFilter("STATUS") + vNeedsAction vStatusNotDefined = cdav.PropFilter("STATUS") + cdav.NotDefined() vNoCompleteDate = cdav.PropFilter("COMPLETED") + cdav.NotDefined() if ignore_completed1: ## This query is quite much in line with https://tools.ietf.org/html/rfc4791#section-7.8.9 filters.extend([vNoCompleteDate, vStatusNotCompleted, vStatusNotCancelled]) elif ignore_completed2: ## some server implementations (i.e. NextCloud ## and Baikal) will yield "false" on a negated TextMatch ## if the field is not defined. Hence, for those ## implementations we need to turn back and ask again ## ... do you have any VTODOs for us where the STATUS ## field is not defined? (ref ## https://github.com/python-caldav/caldav/issues/14) filters.extend([vNoCompleteDate, vStatusNotDefined]) elif ignore_completed3: ## ... and considering recurring tasks we really need to ## look a third time as well, this time for any task with ## the NEEDS-ACTION status set (do we need the first go? ## NEEDS-ACTION or no status set should cover them all?) filters.extend([vStatusNeedsAction]) if start or end: filters.append(cdav.TimeRange(start, end)) if alarm_start or alarm_end: filters.append( cdav.CompFilter("VALARM") + cdav.TimeRange(alarm_start, alarm_end) ) ## Deal with event, todo, journal or comp_class for flagged, comp_name, comp_class_ in ( (event, "VEVENT", Event), (todo, "VTODO", Todo), (journal, "VJOURNAL", Journal), ): if flagged is not None: if not flagged: raise NotImplementedError( f"Negated search for {comp_name} not supported yet" ) if flagged: ## event/journal/todo is set, we adjust comp_class accordingly if comp_class is not None and comp_class is not comp_class: raise error.ConsistencyError( f"inconsistent search parameters - comp_class = {comp_class}, want {comp_class_}" ) comp_class = comp_class_ if comp_class == comp_class_: comp_filter = cdav.CompFilter(comp_name) if comp_class and not comp_filter: raise error.ConsistencyError( f"unsupported comp class {comp_class} for search" ) for other in kwargs: find_not_defined = other.startswith("no_") find_defined = other.startswith("has_") if find_not_defined: other = other[3:] if find_defined: other = other[4:] if other in ( "uid", "summary", "comment", "class_", "class", "category", "description", "location", "status", "due", "dtstamp", "dtstart", "dtend", "duration", "priority", ): ## category and class_ is special if other.endswith("category"): ## TODO: we probably need to do client side filtering. I would ## expect --category='e' to fetch anything having the category e, ## but not including all other categories containing the letter e. ## As I read the caldav standard, the latter will be yielded. target = other.replace("category", "categories") elif other == "class_": target = "class" else: target = other if find_not_defined: match = cdav.NotDefined() elif find_defined: raise NotImplementedError( "Seems not to be supported by the CalDAV protocol? or we can negate? not supported yet, in any case" ) else: match = cdav.TextMatch(kwargs[other]) filters.append(cdav.PropFilter(target.upper()) + match) else: raise NotImplementedError("searching for %s not supported yet" % other) if comp_filter and filters: comp_filter += filters vcalendar += comp_filter elif comp_filter: vcalendar += comp_filter elif filters: vcalendar += filters filter = cdav.Filter() + vcalendar root = cdav.CalendarQuery() + [prop, filter] return (root, comp_class)
[docs] def freebusy_request(self, start: datetime, end: datetime) -> "FreeBusy": """ Search the calendar, but return only the free/busy information. Args: start : defaults to datetime.today(). end : same as above. Returns: [FreeBusy(), ...] """ root = cdav.FreeBusyQuery() + [cdav.TimeRange(start, end)] response = self._query(root, 1, "report") return FreeBusy(self, response.raw)
[docs] def todos( self, sort_keys: Sequence[str] = ("due", "priority"), include_completed: bool = False, sort_key: Optional[str] = None, ) -> List["Todo"]: """ Fetches a list of todo events (this is a wrapper around search) Args: sort_keys: use this field in the VTODO for sorting (iterable of lower case string, i.e. ('priority','due')). include_completed: boolean - by default, only pending tasks are listed sort_key: DEPRECATED, for backwards compatibility with version 0.4. """ if sort_key: sort_keys = (sort_key,) return self.search( todo=True, include_completed=include_completed, sort_keys=sort_keys )
def _calendar_comp_class_by_data(self, data): """ takes some data, either as icalendar text or icalender object (TODO: consider vobject) and returns the appropriate CalendarResourceObject child class. """ if data is None: ## no data received - we'd need to load it before we can know what ## class it really is. Assign the base class as for now. return CalendarObjectResource if hasattr(data, "split"): for line in data.split("\n"): line = line.strip() if line == "BEGIN:VEVENT": return Event if line == "BEGIN:VTODO": return Todo if line == "BEGIN:VJOURNAL": return Journal if line == "BEGIN:VFREEBUSY": return FreeBusy elif hasattr(data, "subcomponents"): if not len(data.subcomponents): return CalendarObjectResource ical2caldav = { icalendar.Event: Event, icalendar.Todo: Todo, icalendar.Journal: Journal, icalendar.FreeBusy: FreeBusy, } for sc in data.subcomponents: if sc.__class__ in ical2caldav: return ical2caldav[sc.__class__] return CalendarObjectResource
[docs] def event_by_url(self, href, data: Optional[Any] = None) -> "Event": """ Returns the event with the given URL """ return Event(url=href, data=data, parent=self).load()
[docs] def object_by_uid( self, uid: str, comp_filter: Optional[cdav.CompFilter] = None, comp_class: Optional["CalendarObjectResource"] = None, ) -> "Event": """ Get one event from the calendar. Args: uid: the event uid comp_class: filter by component type (Event, Todo, Journal) comp_filter: for backward compatibility Returns: Event() or None """ if comp_filter: assert not comp_class if hasattr(comp_filter, "attributes"): if comp_filter.attributes is None: raise ValueError( "Unexpected None value for variable comp_filter.attributes" ) comp_filter = comp_filter.attributes["name"] if comp_filter == "VTODO": comp_class = Todo elif comp_filter == "VJOURNAL": comp_class = Journal elif comp_filter == "VEVENT": comp_class = Event else: raise error.ConsistencyError("Wrong compfilter") query = cdav.TextMatch(uid) query = cdav.PropFilter("UID") + query root, comp_class = self.build_search_xml_query( comp_class=comp_class, filters=[query] ) try: items_found: List[Event] = self.search(root) if not items_found: raise error.NotFoundError("%s not found on server" % uid) except Exception as err: if comp_filter is not None: raise logging.warning( "Error %s from server when doing an object_by_uid(%s). search without compfilter set is not compatible with all server implementations, trying event_by_uid + todo_by_uid + journal_by_uid instead" % (str(err), uid) ) items_found = [] for compfilter in ("VTODO", "VEVENT", "VJOURNAL"): try: items_found.append( self.object_by_uid(uid, cdav.CompFilter(compfilter)) ) except error.NotFoundError: pass if len(items_found) >= 1: if len(items_found) > 1: logging.error( "multiple items found with same UID. Returning the first one" ) return items_found[0] # Ref Lucas Verney, we've actually done a substring search, if the # uid given in the query is short (i.e. just "0") we're likely to # get false positives back from the server, we need to do an extra # check that the uid is correct items_found2 = [] for item in items_found: ## In v0.10.0 we used regexps here - it's probably more optimized, ## but at one point it broke due to an extra CR in the data. ## Usage of the icalendar library increases readability and ## reliability if item.icalendar_component: item_uid = item.icalendar_component.get("UID", None) if item_uid and item_uid == uid: items_found2.append(item) if not items_found2: raise error.NotFoundError("%s not found on server" % uid) error.assert_(len(items_found2) == 1) return items_found2[0]
[docs] def todo_by_uid(self, uid: str) -> "CalendarObjectResource": """ Returns the task with the given uid (wraps around :class:`object_by_uid`) """ return self.object_by_uid(uid, comp_filter=cdav.CompFilter("VTODO"))
[docs] def event_by_uid(self, uid: str) -> "CalendarObjectResource": """ Returns the event with the given uid (wraps around :class:`object_by_uid`) """ return self.object_by_uid(uid, comp_filter=cdav.CompFilter("VEVENT"))
[docs] def journal_by_uid(self, uid: str) -> "CalendarObjectResource": """ Returns the journal with the given uid (wraps around :class:`object_by_uid`) """ return self.object_by_uid(uid, comp_filter=cdav.CompFilter("VJOURNAL"))
# alias for backward compatibility event = event_by_uid
[docs] def events(self) -> List["Event"]: """ List all events from the calendar. Returns: * [Event(), ...] """ return self.search(comp_class=Event)
[docs] def objects_by_sync_token( self, sync_token: Optional[Any] = None, load_objects: bool = False ) -> "SynchronizableCalendarObjectCollection": """objects_by_sync_token aka objects Do a sync-collection report, ref RFC 6578 and https://github.com/python-caldav/caldav/issues/87 This method will return all objects in the calendar if no sync_token is passed (the method should then be referred to as "objects"), or if the sync_token is unknown to the server. If a sync-token known by the server is passed, it will return objects that are added, deleted or modified since last time the sync-token was set. If load_objects is set to True, the objects will be loaded - otherwise empty CalendarObjectResource objects will be returned. This method will return a SynchronizableCalendarObjectCollection object, which is an iterable. """ cmd = dav.SyncCollection() token = dav.SyncToken(value=sync_token) level = dav.SyncLevel(value="1") props = dav.Prop() + dav.GetEtag() root = cmd + [level, token, props] (response, objects) = self._request_report_build_resultlist( root, props=[dav.GetEtag()], no_calendardata=True ) ## TODO: look more into this, I think sync_token should be directly available through response object try: sync_token = response.sync_token except: sync_token = response.tree.findall(".//" + dav.SyncToken.tag)[0].text ## this is not quite right - the etag we've fetched can already be outdated if load_objects: for obj in objects: try: obj.load() except error.NotFoundError: ## The object was deleted pass return SynchronizableCalendarObjectCollection( calendar=self, objects=objects, sync_token=sync_token )
objects = objects_by_sync_token
[docs] def journals(self) -> List["Journal"]: """ List all journals from the calendar. Returns: * [Journal(), ...] """ return self.search(comp_class=Journal)
[docs] class ScheduleMailbox(Calendar): """ RFC6638 defines an inbox and an outbox for handling event scheduling. TODO: As ScheduleMailboxes works a bit like calendars, I've chosen to inheritate the Calendar class, but this is a bit incorrect, a ScheduleMailbox is a collection, but not really a calendar. We should create a common base class for ScheduleMailbox and Calendar eventually. """ def __init__( self, client: Optional["DAVClient"] = None, principal: Optional[Principal] = None, url: Union[str, ParseResult, SplitResult, URL, None] = None, ) -> None: """ Will locate the mbox if no url is given """ super(ScheduleMailbox, self).__init__(client=client, url=url) self._items = None if not client and principal: self.client = principal.client if not principal and client: if self.client is None: raise ValueError("Unexpected value None for self.client") principal = self.client.principal if url is not None: if client is None: raise ValueError("Unexpected value None for client") self.url = client.url.join(URL.objectify(url)) else: if principal is None: raise ValueError("Unexpected value None for principal") if self.client is None: raise ValueError("Unexpected value None for self.client") self.url = principal.url try: # we ignore the type here as this is defined in sub-classes only; require more changes to # properly fix in a future revision self.url = self.client.url.join(URL(self.get_property(self.findprop()))) # type: ignore except: logging.error("something bad happened", exc_info=True) error.assert_(self.client.check_scheduling_support()) self.url = None # we ignore the type here as this is defined in sub-classes only; require more changes to # properly fix in a future revision raise error.NotFoundError( "principal has no %s. %s" % (str(self.findprop()), error.ERR_FRAGMENT) # type: ignore )
[docs] def get_items(self): """ TODO: work in progress TODO: perhaps this belongs to the super class? """ if not self._items: try: self._items = self.objects(load_objects=True) except: logging.debug( "caldav server does not seem to support a sync-token REPORT query on a scheduling mailbox" ) error.assert_("google" in str(self.url)) self._items = [ CalendarObjectResource(url=x[0], client=self.client) for x in self.children() ] for x in self._items: x.load() else: try: self._items.sync() except: self._items = [ CalendarObjectResource(url=x[0], client=self.client) for x in self.children() ] for x in self._items: x.load() return self._items
## TODO: work in progress # def get_invites(): # for item in self.get_items(): # if item.vobject_instance.vevent.
[docs] class ScheduleInbox(ScheduleMailbox): findprop = cdav.ScheduleInboxURL
[docs] class ScheduleOutbox(ScheduleMailbox): findprop = cdav.ScheduleOutboxURL
[docs] class SynchronizableCalendarObjectCollection: """ This class may hold a cached snapshot of a calendar, and changes in the calendar can easily be copied over through the sync method. To create a SynchronizableCalendarObjectCollection object, use calendar.objects(load_objects=True) """ def __init__(self, calendar, objects, sync_token) -> None: self.calendar = calendar self.sync_token = sync_token self.objects = objects self._objects_by_url = None def __iter__(self) -> Iterator[Any]: return self.objects.__iter__() def __len__(self) -> int: return len(self.objects)
[docs] def objects_by_url(self): """ returns a dict of the contents of the SynchronizableCalendarObjectCollection, URLs -> objects. """ if self._objects_by_url is None: self._objects_by_url = {} for obj in self: self._objects_by_url[obj.url.canonical()] = obj return self._objects_by_url
[docs] def sync(self) -> Tuple[Any, Any]: """ This method will contact the caldav server, request all changes from it, and sync up the collection """ updated_objs = [] deleted_objs = [] updates = self.calendar.objects_by_sync_token( self.sync_token, load_objects=False ) obu = self.objects_by_url() for obj in updates: obj.url = obj.url.canonical() if ( obj.url in obu and dav.GetEtag.tag in obu[obj.url].props and dav.GetEtag.tag in obj.props ): if obu[obj.url].props[dav.GetEtag.tag] == obj.props[dav.GetEtag.tag]: continue obu[obj.url] = obj try: obj.load() updated_objs.append(obj) except error.NotFoundError: deleted_objs.append(obj) obu.pop(obj.url) self.objects = obu.values() self.sync_token = updates.sync_token return (updated_objs, deleted_objs)