o .jy! @sJdZddlZddlmZddlZddlZddlmZmZddl Z ddl Z ddl m Z ddl Z dZddZGd d d Zd d Zejd e jde jfddZGdddeZ  d&de jd e e jdede jfddZe e jde ddde jde ddde e e jddd d!deded"efd#d$Z gd%Z!dS)'z7Bring calendars using X-WR-TIMEZONE into RFC 5545 form.N)BytesIO) vDDDTypes vDDDLists)Optionalz X-WR-TIMEZONEcCs(t|t|kotddt||DS)z6Return wether all contents of two lists are identical.css|] \}}||uVqdSN).0e1e2rrB/home/thesage/.local/lib/python3.10/site-packages/x_wr_timezone.py szlist_is..)lenallzip)l1l2rrr list_iss(rc@sxeZdZdZgdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZdS)CalendarWalkerzsI walk along the components and values of an icalendar object. The idea is the same as a visitor pattern. )DTSTARTDTENDRDATEz RECURRENCE-IDEXDATEcCsz|D]\}}|||ur||||Sqt|jt|ks#Jt||jD]\}}||ur:||||Sq)|S)z~Check if an icalendar Component has changed and copy it if it has. atributes and subcomponents are put into the copy.)itemscopy_componentr subcomponentsr)self component attributesrkeyvaluenew_subcomponentold_subcomponentrrr copy_if_changed(s zCalendarWalker.copy_if_changedcCsL|}|D]\}}|||<qt|jdksJ|D]}||q|S)zACreate a copy of the component with attributes and subcomponents.r)copyrr r add_component)rrrrrr subcomponentrrr r5s  zCalendarWalker.copy_componentcCs@g}|jD]}t|tjjr||}||q||i|S)zCWalk along the calendar and return the changed or identical object.)r isinstance icalendarcalEvent walk_eventappendr")rcalendarrr%rrr walk?s    zCalendarWalker.walkcCs@i}|jD]}||}|dur||||<q||||jS)z@Walk along the event and return the changed or identical object.N)VALUE_ATTRIBUTESget walk_valuer"r)reventrnamerrrr r*Hs  zCalendarWalker.walk_eventcC|S)z.Default method for walking along a value type.r)rrrrr walk_value_defaultQz!CalendarWalker.walk_value_defaultcCs$dt|j}t|||j}||S)zWalk along a value type. walk_value_)type__name__getattrr4)rrr2r-rrr r0UszCalendarWalker.walk_valuecCs"tt|j|}t||r|S|S)zWalk through a list of values.)listmapr0r)rlvrrr walk_value_list[s zCalendarWalker.walk_value_listcs8dd|jD}fdd|D}t||r|St|S)NcSsg|]}|jqSr)dt)rdddrrr csz7CalendarWalker.walk_value_vDDDLists..csg|]}|qSr)r0)rr?rrr rAds)dtsrr)rr<rCnew_dtsrrBr walk_value_vDDDListsbs  z#CalendarWalker.walk_value_vDDDListscCs"||j}||jur |St|S)z"Walk along an icalendar value type)r0r?r)rrr?rrr walk_value_vDDDTypesis  z#CalendarWalker.walk_value_vDDDTypescCr3)&Walk along a datetime.datetime object.rrr?rrr walk_value_datetimepr5z"CalendarWalker.walk_value_datetimecCs |durdS|dkS)z0Return whether the time zone is a UTC time zone.NFUTC)tznameupperrHrrr is_UTCts zCalendarWalker.is_UTCcCs |duSr)rKrHrrr is_Floatingzs zCalendarWalker.is_FloatingN)r8 __module__ __qualname____doc__r.r"rr-r*r4r0r>rErFrIrMrNrrrr r s     rcCs t|dS)zWhether the time zone requires localize() and normalize(). pytz requires these funtions to be used in order to correctly use the time zones after operations. localize)hasattrtzinforrr is_pytz~s rVtimezonereturncCs tj|S)zZReturn a timezone component for the tzid and cache it. The result is cached. )r'Timezone from_tzinfo)rWrrr get_timezone_components r[c@s eZdZdZddZddZdS)UTCChangingWalkerz/Changes the UTC time zone into a new time zone.cCs ||_dS)z-Initialize the walker with the new time zone.N) new_timezone)rrWrrr __init__s zUTCChangingWalker.__init__cCsH||r ||jS||r"t|jr|j|S|j|jdS|S)rGrT)rM astimezoner]rNrVrRreplacerHrrr rIs     z%UTCChangingWalker.walk_value_datetimeN)r8rOrPrQr^rIrrrr r\s r\Fr,add_timezone_componentcCs|dur |td}|durt|tjstt|}|}~|durDt|}| |}|rD| }|j dd|_ |}|j dt ||S)aMake a calendar that might use X-WR-TIMEZONE compatible with RFC 5545. Arguments: calendar: is an icalendar.Calendar object. It does not need to have the X-WR-TIMEZONE property but if it has, calendar will be converted to conform to RFC 5545. timezone: an optional timezone argument if you want to override the existence of the actual X-WR-TIMEZONE property of the calendar. This can be a string like "Europe/Berlin" or "UTC" or a pytz.timezone or any other timezone accepted by the datetime module. add_timezone_component: whether to add a VTIMEZONE component to the result. Nr)r/ X_WR_TIMEZONEr&datetimerUzoneinfoZoneInfostrr\r-r#rinsertr[)r,rWraresultwalkernew_calrrr to_standards  rkin_filerb-)r7defaultout_filewbz--add-timezone/--no-timezoneTz(Add a VTIMEZONE component to the result.)rohelp add_timezonecCs.tj|}t||d}||dS)ax-wr-timezone converts ICSfiles with X-WR-TIMEZONE to use RFC 5545 instead. Convert input: cat in.ics | x-wr-timezone > out.ics wget -O- https://example.org/in.ics | x-wr-timezone > out.ics curl https://example.org/in.ics | x-wr-timezone > out.ics Convert files: x-wr-timezone in.ics out.ics By default, x-wr-timezone will add a VTIMEZONE component to the result. Use --no-vtimezone to remove it. (Added in v2.0.0) Get help: x-wr-timezone --help For bug reports, code and questions, visit the projet page: https://github.com/niccokunzmann/x-wr-timezone License: LPGLv3+ )rar)r'Calendar from_icalreadrkwriteto_ical)rlrprsr,rjrrr mains ry)ryrkr\rrbrr[)NF)"rQ functoolsiorsysrdicalendar.proprrrcr'typingrclickrbrrrVcacherUrYr[r\rtboolrkcommandargumentFileversion_option help_optionoptionry__all__rrrr sH   ^  $"