U j$@szddlmZWnek r,ddlZYnXddlmZddlZddlZddlmZz ddl Z Wnek rzdZ YnXddl Z ddl Z ddl Z ddl mZddlmZddlmZmZdd lmZdd lmZe dk Zere je d gZd d gZdgZn gZgZgZGdddZeej Z!e!dZ"e!dZ#e!dZ$e!dZ%ej&ddddZ'ej&ddddZ(ej&ddddZ)ej&ddddZ*ej&ee+d ej,e-d gdddZej&edd dd gdd d!Z.d"Z/d#d$e 0e"1e#1e$1DZ2ej&e2dd%d&Z3d'd$e"1DZ4ej&e4dd(d)Z5ej&d*d+Z6ej&d,d-Z7ej&d.d/Z8ej&d0d1Z9ej&d2d3Z:e:Z;ej&d4d5Zej&ddeeddfd:d;d<Zej&ed=gdd>d?Z?ej&e@d:d@dAZAej&e@d:dBdCZBej&e@d:dDdEZCej&e@d:dFdGZDdHdIZEGdJdKdKej+ZFdLdMZGdNdOZHej&dPdQZIej&ejJdejKe@dRdSdTZLdS)U)zoneinfoN) Generator) timezone_ids)Path)tz)Calendar Component)TZPtzpUTCcCst||SN)pytztimezoneZlocalizedttznamer#rrc@sReZdZdZedddZddZddZd d Zd d Z d dZ e ddZ dS) DataSourcezFA collection of parsed ICS elements (e.g calendars, timezones, events))data_source_foldercCs||_||_dSr)_parser_data_source_folder)selfrparserrrr__init__/szDataSource.__init__cCsdd|jDS)z(Return all the files that could be used.cSs g|]}|jdkr|jqS).ics)suffixlowerstem).0prrr 5sz#DataSource.keys..)riterdirrrrrkeys3szDataSource.keysc Cs|dr|}|dd}n|d}|j|}|sDt|d|d}|}W5QRX||}t|ts||_ ||_ ||j |<|S)z;Parse a file and return the result stored in the attribute.rNz does not exist.rb) endswithris_fileAttributeErroropenreadr isinstancelistraw_ics source_file__dict__)r attributer3 source_pathfr2sourcerrr __getitem__;s      zDataSource.__getitem__cCs"|dr|dd}||kS)zkey in self.keys()rNr))r+r(rkeyrrr __contains__Ns  zDataSource.__contains__cCs||Srrr:rrr __getattr__TszDataSource.__getattr__cCs t|jSr)reprr4r'rrr__repr__WszDataSource.__repr__csjfddS)z'Return a list of all components parsed.csj|ddS)NT)multiple)r)datar'rrr^rz%DataSource.multiple..) __class__rr'rr'rr@Zs zDataSource.multipleN) __name__ __module__ __qualname____doc__rrr(r9r<r=r?propertyr@rrrrr,sr calendars timezoneseventsalarmsmoduleZscopecCstttjjSr)rCALENDARS_FOLDER icalendarr from_icalr rrrrHiscCstttjjSr)rTIMEZONES_FOLDERrOZTimezonerPr rrrrInscCstttjjSr)r EVENTS_FOLDERrOEventrPr rrrrJsscCstttjjSr)r ALARMS_FOLDERrOZAlarmrPr rrrrKxs)paramscCs|jSrparamrequestr rrrutc}srZcCs|jt|dSN)tzinfo)replacergettzrrrrrrcCs|jt|dSr[)r]rZoneInforrrrrrcCs|jSrrVrXrrr in_timezonesr`)zbig_bad_calendar.icszissue_104_broken_calendar.icszsmall_bad_calendar.icsz multiple_calendar_components.icszpr_480_summary_with_colon.icszparsing_error_in_UTC_offset.icszparsing_error.icscCsg|]}|jtkr|jqSr)nameICS_FILES_EXCLUDE)r#filerrrr%s r%cCsH|j}td||||fD]}||kr||Sqtd|ddS)zAn example ICS file.z example file:zCould not find file .N)rWprint ValueError)r rHrIrJrYics_filerArrrrgs  rgcCsg|]}dt|kr|qS)z fuzz-testcase)str)r#r;rrrr%s cCs|jS)zClusterfuzz calendars.rVrYrrrfuzz_v1_calendarsrjccs(dtjjjd<dVtjjjddS)zMap x_sometime to timetimez X-SOMETIMEN)rOcalZ types_factory types_mappoprrrr x_sometimesrocCstS)zReturn a new component factory.)rOZComponentFactoryrrrrfactorysrpccsdtj_dVdtj_dS)NTF)rOZ vUTCOffsetZignore_exceptionsrrrrvUTCOffset_ignore_exceptionssrqcCst}d|_|S)zReturn an event component.VEVENTr rar crrrevent_componentsrvcCs t}|S)Return an empty component.)r rtrrrrusrucCst}d|_|S)rwZ VCALENDARrsrtrrrcalendar_componentsrxcCs>tdd}d|_|jdddd|jdd dd|||S) zBReturn an event with some values and add it to calendar_component.zA brief history of time)summaryrrZdtendZ20000102T000000r)encodeZdtstartZ20000101T000000)r raaddZ add_component)rurxerrrfilled_event_components   r}cCst}d|d<|S)NzChair, Table, "Room: 42" resources)rrtrrrcalendar_with_resourcessr)returnccst|tVtdS)zThe timezone provider.N)_tzpZuseZ use_defaulttzp_namerrrr s r rcCs t|jS)zThis is annother timezone provider. The purpose here is to cross test: pytz <-> zoneinfo. tzp as parameter makes sure we test the cross product. )r rWrXrrr other_tzpsrcCs|s t|S)z+Skip tests that are not running under pytz.)Z uses_pytzAssertionError)r rrrr pytz_onlys rcCs|s t|S)z/Skip tests that are not running under zoneinfo.)Z uses_zoneinfor)r rYrrrr zoneinfo_onlys rcCs|dks t|S)zDo not run tests with pytz.rrrrrrno_pytzs rcCs|dks t|S)zDo not run tests with zoneinfo.rrrrrr no_zoneinfos rcCsd|jkrtdg}d|jkr$dg}d|jkr2t}d|jkrNd|jkrNtddD]&}d||jkrR||krR||qR|jd|dd d S) aParametrize without skipping: tzp_name will be parametrized according to the use of - pytz_only - zoneinfo_only - no_pytz - no_zoneinfo See https://docs.pytest.org/en/6.2.x/example/parametrize.html#deferring-the-setup-of-parametrized-resources rrrrz,Use pytz_only or zoneinfo_only but not both!)rrZno_rLrMN)Z fixturenamesPYTZ_TZPrremoveZ parametrize)ZmetafuncZ tzp_namesrarrrpytest_generate_tests#s     rc@seZdZdZddZdS)DoctestZoneInfoz+Constent ZoneInfo representation for tests.cCsd|jdS)Nz ZoneInfo(key=))r;r'rrrr?AszDoctestZoneInfo.__repr__N)rCrDrErFr?rrrrr>srcCs8t|tr|d}tt|dddddS)z doctest printzUTF-8z   N)r0bytesdecodererhstripr])objrrr doctest_printEs  rcOs|dkr tSt|f||S)zIReplace the import mechanism to skip the whole doctest if we import pytz.r)r __import__)raargskwrrrdoctest_importLsrcsj|tjdt|tdtddlm}tj ddd|tdfd d ||d t d d t iS)z,Modify the environment to make doctests run.rr_r)ZONEINFOz$d755cef5-2311-46ed-a0e1-6733c9e15c63)versionZuuid4csSrrruidrrrZrz!env_for_doctest..rZr re) setitemsysmodulesrsetattrrZicalendar.timezone.zoneinforuuidUUIDr_r)Z monkeypatchrrrrenv_for_doctestSs r)rYrcCs|jS)zxReturn a timezone id to be used with pytz or zoneinfo. This goes through all the different timezones possible. rVrirrrtzid`sr)MZ backportsr ImportErrortypingrZpytestrOrr itertoolsrrpathlibrZdateutilrZ icalendar.calrr Zicalendar.timezoner r rZHAS_PYTZrZrZPYTZ_UTCZPYTZ_IN_TIMEZONErr__file__parentZHERErNrQrRrTZfixturerHrIrJrKr_r r^r`rbchainr&Z ICS_FILESrgZFUZZ_V1rjrorprqrvrucomprxr}rrrhrrrrrrrrrZTZIDSZFixtureRequestrrrrrs        6         "