o .jx@sdZddlmZddlZddlmZmZddlmZddl Z er$ddl Z GdddeZ Gdd d e Z Gd d d e Z Gd d d e ZGddde ZdS)a Data state management for CalendarObjectResource. This module implements the Strategy/State pattern for managing different representations of calendar data (raw string, icalendar object, vobject object). See https://github.com/python-caldav/caldav/issues/613 for design discussion. TODO: verify that we have sufficient test coverage - both through unit tests and integration tests ) annotationsN)ABCabstractmethod) TYPE_CHECKINGc@sXeZdZdZedddZedddZedd d Zdd dZdddZ dddZ dS) DataStatezAbstract base class for calendar data states. Each concrete state represents a different "source of truth" for the calendar data. The state provides access to all representations, but only one is authoritative at any time. returnstrcCdS)zGet raw iCalendar string representation. This may involve serialization if the current state holds a parsed object. Nselfr r E/home/thesage/.local/lib/python3.10/site-packages/caldav/datastate.pyget_data!zDataState.get_dataicalendar.CalendarcCr )zGet a fresh copy of the icalendar object. This is safe for read-only access - modifications won't affect the stored data. Nr r r r r get_icalendar_copy*rzDataState.get_icalendar_copyvobject.base.ComponentcCr )zGet a fresh copy of the vobject object. This is safe for read-only access - modifications won't affect the stored data. Nr r r r r get_vobject_copy3rzDataState.get_vobject_copy str | NonecCs:|}|jD]}|jdvrd|vrt|dSqdS)zExtract UID without full parsing if possible. Default implementation parses the data, but subclasses can optimize. VEVENTVTODOVJOURNALFREEBUSYUIDN)r subcomponentsnamerr calcompr r r get_uid<s  zDataState.get_uidcCs,|}|jD] }|jdvr|jSqdS)zGet the component type (VEVENT, VTODO, VJOURNAL, FREEBUSY) without full parsing. Default implementation parses the data, but subclasses can optimize. rN)rrrrr r r get_component_typeGs    zDataState.get_component_typeboolcCr )z!Check if this state has any data.Tr r r r r has_dataRszDataState.has_dataNrrrrrrrrrr") __name__ __module__ __qualname____doc__rrrrr r!r#r r r r rs      rc@sLeZdZdZdddZdddZdd d Zdd dZdddZdddZ dS) NoDataStatezNull Object pattern - no data loaded yet. This state is used when a CalendarObjectResource is created without any initial data. It provides empty/default values for all accessors. rrcCr )Nr r r r r r^zNoDataState.get_datarcCstSN) icalendarCalendarr r r r raszNoDataState.get_icalendar_copyrcCsddl}|SNr)vobject iCalendarr r4r r r rdszNoDataState.get_vobject_copyrcCdSr0r r r r r r ir/zNoDataState.get_uidcCr7r0r r r r r r!lr/zNoDataState.get_component_typer"cCr )NFr r r r r r#or/zNoDataState.has_dataNr$r%r&r'r() r)r*r+r,rrrr r!r#r r r r r-Ws     r-csTeZdZdZdddZdddZdd d Zdd dZdfdd ZdddZ Z S) RawDataStatezState when raw string data is the source of truth. This is the most common initial state when data is loaded from a CalDAV server. datarcC ||_dSr0_data)r r9r r r __init__z zRawDataState.__init__rcC|jSr0r;r r r r r}szRawDataState.get_datarcCstj|jSr0)r1r2 from_icalr<r r r r rszRawDataState.get_icalendar_copyrcCsddl}||jSr3)r4readOner<r6r r r rs zRawDataState.get_vobject_copyrcs.td|jtj}|r|dStS)Nz ^UID:(.+)$)researchr< MULTILINEgroupstripsuperr )r match __class__r r r s zRawDataState.get_uidcCs<d|jvrdSd|jvrdSd|jvrdSd|jvrdSdS) Nz BEGIN:VEVENTrz BEGIN:VTODOrzBEGIN:VJOURNALrzBEGIN:FREEBUSY VFREEBUSYr;r r r r r!s    zRawDataState.get_component_type)r9rr$r%r&r') r)r*r+r,r=rrrr r! __classcell__r r rJr r8ss    r8c@sVeZdZdZdddZddd Zdd d Zdd d ZdddZdddZ dddZ dS)IcalendarStatezState when icalendar object is the source of truth. This state is entered when: - User calls edit_icalendar_instance() - User sets icalendar_instance property - User modifies the icalendar object calendarrcCr:r0 _calendar)r rOr r r r=r>zIcalendarState.__init__rrcCs|jdS)Nzutf-8)rQto_icaldecoder r r r rzIcalendarState.get_datacCtj|Sr0r1r2r@rr r r r rsz!IcalendarState.get_icalendar_copycCr?)z|Returns THE icalendar object (not a copy). This is the authoritative object - modifications will be saved. rPr r r r get_authoritative_icalendarz*IcalendarState.get_authoritative_icalendarrcCddl}||Sr3r4rArr6r r r rzIcalendarState.get_vobject_copyrcCs4|jjD]}|jdvrd|vrt|dSqdS)NrrrrLr)rQrrrr rr r r r s zIcalendarState.get_uidcCs&|jjD] }|jdvr|jSqdS)Nr\)rQrrr]r r r r!s   z!IcalendarState.get_component_typeN)rOrr$r%r&r') r)r*r+r,r=rrrWrr r!r r r r rNs      rNc@sVeZdZdZdddZddd Zdd d Zdd dZdddZdddZ dddZ dS) VobjectStatezState when vobject object is the source of truth. This state is entered when: - User calls edit_vobject_instance() - User sets vobject_instance property - User modifies the vobject object vobjrcCr:r0_vobject)r r_r r r r=r>zVobjectState.__init__rrcCs |jSr0)ra serializer r r r rr>zVobjectState.get_datarcCrUr0rVr r r r rrTzVobjectState.get_icalendar_copycCrYr3rZr6r r r rr[zVobjectState.get_vobject_copycCr?)zzReturns THE vobject object (not a copy). This is the authoritative object - modifications will be saved. r`r r r r get_authoritative_vobjectrXz&VobjectState.get_authoritative_vobjectrcCsz?t|jdrt|jjjjWSt|jdrt|jjjjWSt|jdr.t|jjjjWSt|jdr=t|jjjjWSWdSt yIYdSw)Nveventvtodovjournal vfreebusy) hasattrrarrduidvaluererfrgAttributeErrorr r r r r s     zVobjectState.get_uidcCsDt|jdrdSt|jdrdSt|jdrdSt|jdr dSdS) NrdrrerrfrrgrL)rhrar r r r r!s    zVobjectState.get_component_typeN)r_rr$r%r&r') r)r*r+r,r=rrrrcr r!r r r r r^s      r^)r, __future__rrCabcrrtypingrr1r4rr-r8rNr^r r r r s  >*,