o .jy@ @s4dZddlZddlmZmZmZddlmZddlZddlm Z ddl m Z m Z m Z mZddlmZmZejdgd Zd e d e fd d ZeddZeddZeddZddZddZddZeddZejdgdedddd fded!d"d#d fded!d"d#d$d%d fded!d"d#d$d&d fded!d"d#d$d'd'd fded(ddd)fded(ddddd)fded(dddd&d)fded(ddddd)fded(ddddd)fd*edddd+fd*ed,d-d#ddd+fd*ed,d-d#ddd+fd*ed,d-d#d.dd+fd*ed,d-d#d.d'd'd+fd*ed,d-d#d-dd/fd*ed,d-d#d-ddd/fd*ed,d-d#d0dd/fd*ed,d%d1ddd/fd*ed,d%d1ddd/fd*ed,d%d1d.dd/fd*ed,d%d1d.d2d/fd*ed,d%d1d.d'd'd/fd*ed,d%d1d-dd+fd*ed,d%d1d-ddd+fd*ed,d%d1d0dd+fd3edddd4fd3ed,d5d-ddd6fd3ed,d5d-ddd6fd3ed,d5d-dd'd'd6fd3ed,d5d-d.ddd4fd3ed,d5d-d-dd4fd3ed7d-d8ddd4fd3ed7d-d8dd'd'd4fd3ed7d-d8d.dd4fd3ed7d-d8d-dd6fd3ed7d-d8d-ddd6fd3ed7d-d8d0dd6fd9d:Zejd;gd<d=d>Zd?ed@e dAefdBdCZ gdDZ!ejd?e!dEdFZ"dGdHZ#dIdJZ$dKdLZ%dMdNZ&dOdPZ'dQdRZ(dSdTZ)ejdUd3dVgdWdXZ*dYdZZ+ejd[gd\d]d^Z,d_d`Z-dadbZ.dcddZ/dedfZ0ejde1dgdhhdidjZ2dS)kaRGenerate VTIMEZONE components from actual timezone information. When we generate VTIMEZONE from actual tzinfo instances of - dateutil - zoneinfo - pytz Then, we cannot assume that the future information stays the same but we should be able to create tests that work for the past. See https://github.com/collective/icalendar/issues/722 N)datedatetime timedelta)findall)gettz)Calendar ComponentEventTimezone)tzid_from_tzinfotzids_from_tzinfotzid) Europe/BerlinAsia/SingaporeAmerica/New_Yorkc1c2c sd|}|}tfdd|D}d}t||ddD]8\}}t|dkp5t|dk}t|d|d|t||d |d|t|||krUd nd |||k7}q&|rhJ|d d S)z!Print the diff of two components. c3s$|] }t|krt|VqdSN)len).0lMLf/home/thesage/.local/lib/python3.10/site-packages/icalendar/tests/test_issue_722_generate_vtimezone.py 's"z*assert_components_equal..rF)strictz  z ->z diff!z lines differN)to_icaldecode splitlinesmaxziprprint) rrll1ll2paddiffl1l2arrrassert_components_equal"s   r-cCs&|r td||}|dusJt|}d|d<|}t|d}|}t|d}t||t||dt|j |j krLdksOJJdt|j |j kr`dkscJJt |t |ksmJ| | ks{J|j |j ksJ|j |j ksJ||ksJdS)zqtzinfo -> VTIMEZONE -> tzinfo -> VTIMEZONE We can assume that both generated VTIMEZONEs are equivalent. zGpytz will not converge on the first run. This is problematic. PYTZ-TODONztest-generatedTZID) uses_pytzpytestskiptimezoner from_tzinfoto_tzr-rstandarddaylightdictr r!)tzpr tzinfo1 generated1tzinfo2 generated2tzinfo3 generated3rrrtest_conversion_converges6s,       ((rAcCsRtj||tdddd}|tj||tdddd}t||||ks'JdS)a We want to make sure that we get the same info for all timezone implementations. We assume that - the timezone implementations have the same info within the days we test - the timezone transitions times do not change because they are before last_date ) last_dateN)r from_tzidr use_zoneinfor-)r r:tz1tz2rrr both_tzps_generate_the_same_infoWs   rIcCs$t||}|d|ksJdS)zCheck the TZID.r.N)r r5r4)r r:tzrrrtest_tzid_matchesgsrKcCsttd}|jgks Jt|jdksJ|jd}|djtdks%J|djtdks0J|ddks8JdS)z1We do not need to convert UTC but it should work.UTCrCr TZOFFSETFROM TZOFFSETTOTZNAMEN)r rEr8rr7tdr)r:utcr7rrrtest_do_not_convert_utcns  rRcCstd}|jD]}t|j|d|dj|djt|qt|j dks,Jt|jdvs7Jd|j d}|jd}|dd ksIJ|dd ksQJ|djt dd ks]J|djt d d ksiJ|djt d d ksuJ|djt dd ksJd S)z+Test the Europe/Berlin timezone conversion.rrOrMrNrC)rCr/zWe start in winterCESTCET)hoursr/N) r rEr7r%namerPr r!rr8r)r:rJxdststarrrtest_berlin_timeys     r[csptdddtdddfdd}tjdd}|j|jD]}||j|dgD]}||q.q!dS) NirCrBcs |kr ksJJdSrr)dt first_daterDrrchecks z(test_range_is_not_crossed..checkr)rDr^RDATE)rr rEr7r8DTSTARTget)r_rJsubrdaterr]rtest_range_is_not_crosseds    recCsL||}|dus Jt|}|}t|t|ksJ||ks$JdS)zVWhen we get the timezone again, usually, we should use the one of the library/tzp.N)r4r r5r6type)r r:r;r<r=rrrtest_use_the_original_timezones   rg)r r\tznamerirCz+0730i  ";iz+08rrUrBr0r/rTrEST EDTi c CsL|d|d|d}|||}t||}|j|dd}|j|d}t||rv||}|t ddd d d t dd d d ddt dd d d d t dd d d d t ddd dddt ddd ddt ddd dd t ddd d d fvrvt d| | kr|ksJ|J|| | ksJ|||ksJ|dS)zWe should make sure than the datetimes with the generated timezones work as expected: They have the right UTC offset, dst and tzname. z: z (expected in )F) lookup_tzidtzinforBrlrprCrrsr0rnr/rqz(We do not know how to do this. PYTZ-TODON)localizer r5r4r6replacer%r1 normalizerr2r3rhrY utcoffset) r:r r\rhmessage expected_dt componentgenerated_tzinfo generated_dtrrr,test_check_datetimes_around_transition_timess,4    ,ruid)rrCr/r0cCs>|j}|j|}t|d}|j}||ksJ|ddS)aWhen going from Daylight Saving Time to Standard Time, times can be ambiguous. For example, at 3:00 AM, the time falls back to 2:00 AM, repeating a full hour of times from 2:00 AM to 3:00 AM on the same date. By the RFC 5545, we cannot accommodate this case. All datetimes should be BEFORE the transition if ambiguous. However, dateutil can create a timezone that allows the event to be after this ambiguous time span, of course. Each event has its timezone saved in it. zX-TZNAMESUMMARYN)'issue_722_timezone_transition_ambiguityeventsstrstartrh) calendarsr:rcaleventexpected_tzname actual_tznamerrr3test_dateutil_timezone_when_time_is_going_backwardss    rqueryrreturnc Cs<z t|d|i}Wt|Sty}zt||d}~ww)zThe tzid from the query.rN)eval Exception ValueErrorr )rrryerrr query_tzids r)zcal.events[0].start.tzinfozcal.events[0].end.tzinfozcal.todos[0].end.tzinfoz.cal.events[0].get('RDATE').dts[0].dt[0].tzinfoz,cal.events[1].get('RECURRENCE-ID').dt.tzinfoz.cal.events[2].get('RDATE')[0].dts[0].dt.tzinfoz.cal.events[2].get('RDATE')[1].dts[0].dt.tzinfocCs(|j}t||}|}||vsJdS)z*Add the missing timezones to the calendar.N)issue_722_missing_timezonesrget_missing_tzids)rrrr tzsrrr%test_add_missing_timezones_to_examples rcCsN|j}t}tD]}t||}t|d|||qt|ttks%JdS)zGWe make sure each query tests a unique place to find for the algorithm.z->N)rsetqueriesrr%addr)rrtzidsrr rrrtest_queries_yield_unique_tzids%s   rcCsN|j}|j}ttd|}||ksJdt|ttks%JddS)z)Make sure all tzids are actually queried.zTZID=([a-zA-Z_/+-]+)z&We find all tzids and they are unique.zWe query all the tzids.N)rraw_icsr!rrget_used_tzidsrr)rrrawidsrrr"test_we_do_not_miss_to_add_a_query0s  rcCs*|j}d|vs Jd|vsJdS)z6If we have an unknown tzid with no timezone component. CUSTOM_tzidN)"issue_722_missing_VTIMEZONE_customrrrrrrrtest_unknown_tzid9rcCs*|j}d|vs Jd|vsJdS)z3Check the custom timezone component is not missing.custom_America/New_YorkN)america_new_yorkrrrrrr&test_custom_timezone_is_found_and_used@rrcCst}|tks JdS)z"Check that no timezone is missing.N)rrr)rrrrtest_not_missing_anythingGsrcCs*|j}d|vs Jd|vsJdS)z UTC should not be found missing.rLN)rrrrrrrtest_utc_is_not_missingMrrcCsJ|j}|j|jd|dhksJdt|jdjjj vs#JdS)zdateutil is an example of a timezone that has no tzid. In this test we make sure that the timezone is said to be missing. rrdateutilN) r subcomponentsremove timezonesrreprrrry __class__)rno_pytzrrrr/test_dateutil_timezone_is_not_found_with_tznameTs rrhzArctic/LongyearbyencCslt}t}tdddt|d|_tt|jj||| |hks'J| | t ks4JdS)zdateutil is an example of a timezone that has no tzid. In this test we make sure that the timezone is matched by its tzname() in the timezone in the STANDARD and DAYLIGHT components. rBrsrirxN) rr rrrr%dirry add_componentradd_missing_timezonesr)rhrrrrr-test_dateutil_timezone_is_matched_with_tzname_s rcCsdS)zpWe find and add a dateutil timezone. This is important as we use those in the zoneinfo implementation. Nr)rrrr$test_dateutil_timezone_is_also_addedqsrcalendar)examplertimezone_same_startperiod_with_timezonecCsT||}|jdd}t|ksJ|t|ks!J|j|ks(JdS)z4Check that these calendars have no timezone missing.N)rrrr)rrrrrrrtest_timezone_is_not_missingxs  rcCsR|j}t|jdks J|t|jttksJdt|dks'JdS)zAdd all timezones specified.rzall timezones are knownN)rrrrrrrrrr test_add_missing_known_timezoness rcCsZ|j}t|jdks J|dhksJ||jgks"Jd|dhks+JdS)z(I cannot add a timezone that is unknown.rrzwe cannot add this timezoneN)rrrrrrrrr test_cannot_add_unknown_timezones rcCs:tttdWddS1swYdS)Nz invalid/tzid)r2raisesrr rErrrr2test_cannot_create_a_timezone_from_an_invalid_tzids  "rcCstddS)zWhen we add the timezones, we should check the calendar to see if all dates really occur in the span we use. We should also consider a huge default range. todoN)r2r3rrrr*test_dates_before_and_after_are_consideredsrFactory localtimecCs(t|}|dus|t|vsJdSdS)zxdateutil and others were badly supported. But if we know their shortcodes, we should be able to identify them. N)rr )r rJrrr'test_we_can_identify_dateutil_timezoness r)3__doc__zoneinforrrrerr2 dateutil.tzr icalendarrrr r icalendar.timezoner r mark parametrizerr-rArIrKrRr[rergrrrrrrrrrrrrrrrrrrrravailable_timezonesrrrrrs                 #$%&')*+ 0