o >@sfdZddlmZddlmZddlmZddlZgdZddZd d Zdd dZ Gddde Z dS)zyISO 8601 duration/period support. https://en.wikipedia.org/wiki/ISO_8601#Durations https://tools.ietf.org/html/rfc3339 )absolute_import)division)unicode_literalsN) rrrrrrrrrcCs$|ddkp|ddko|ddkS)zReturns True if year is a leap year. Cheaper than `import calendar` because its the only thing needed here. Args: year: The 4 digit year. Returns: True if year is a leap year. ird)yearr r B/tmp/google-cloud-sdk/lib/googlecloudsdk/core/util/iso_duration.py IsLeapYear s$ r cCs&t|d|dkrt|rdSdS)zReturns the number of days in the given month and calendar year. Args: year: The 4 digit calendar year. month: The month number 1..12. Returns: The number of days in the given month and calendar year. r)_ISO_DAYS_IN_MONTHr )r monthr r r DaysInCalendarMonth.s rcCsddj|d}|t|}|r|d}|dr|dd}|r.|dkr0|||dSdSdS)aAppends a formatted number + suffix to result. Trailing "0" and "." are stripped. If no digits remain then nothing is appended to result. Args: result: The formatted number, if any is appended to this list. number: The int or float to format. suffix: A suffix string to append to the number. precision: Format the last duration part with precision digits after the decimal point. Trailing "0" and "." are always stripped. z{{0:.{precision}f}} precision0.N)formatfloatrstripendswithappend)resultnumbersuffixrfmtsr r r _FormatNumber=s     r$c@seZdZdZeZdZdZdZdZ dZ dZ ee Z e e Z e eZee Z  dd d Zd d ZddZdddZdddZddZdS)Durationa8The parts of an ISO 8601 duration plus microseconds. Durations using only hours, miniutes, seconds and microseconds are exact. calendar=True allows the constructor to use duration units larger than hours. These durations will be inexact across daylight savings time and leap year boundaries, but will be "calendar" correct. For example: 2015-02-14 + P1Y => 2016-02-14 2015-02-14 + P365D => 2016-02-14 2016-02-14 + P1Y => 2017-02-14 2016-02-14 + P366D => 2017-02-14 2016-03-13T01:00:00 + P1D => 2016-03-14T01:00:00 2016-03-13T01:00:00 + PT23H => 2016-03-14T01:00:00 2016-03-13T01:00:00 + PT24H => 2016-03-14T03:00:00 delta durations (initialized from datetime.timedelta) are calendar=False. Parsed durations containing duration units larger than hours are calendar=True. gvq v@@B< rNFc CsX||_||_||_||_||_||_||_d|_|r#|j|7_| |_| dS)Nr) yearsmonthsdayshoursminutesseconds microseconds total_secondscalendar _Normalize) selfr*r+r,r-r.r/r0deltar2r r r __init__xs zDuration.__init__cCsdd}||j\|_}|r|j|j|7_||j\|_}|r1|jt|j||j7_||j\|_}|rE|j|j|7_||j\|_}|rY|j|j |7_||j\|_}|rm|j |j |7_ ||j \|_ }|rt|j ||_ d|_t|j |j }|j t||j 8_ |j|j |j 7_|j |7_ t|j |j }|j ||j 8_ |j|j 7_|j|7_t|j|j }|j||j 8_|j|j|j 7_|j|7_|js |js|js|jrd|_n |j|j|j7_dSt|j|j}|j||j8_|j|j|j7_|j|7_|jt|jdkrN|jt|jd8_|jd7_n|jt|jd krm|jt|jd7_|jd8_t|j|j}|jt||j8_|j|j|j7_|j|7_t|j|j}|j||j8_|j|j|j7_|j|7_|j|j|j7_t|jd|j |j |_dS)zNormalizes duration values to integers in ISO 8601 ranges. Normalization makes formatted durations aesthetically pleasing. For example, P2H30M0.5S instead of P9000.5S. It also determines if the duration is exact or a calendar duration. cSs$t|}tt|d|d}||fS)Nr )intround)f int_valuefractionr r r _Percolatesz'Duration._Normalize.._PercolategTNrr)r*r,_DAYS_PER_YEARr+r7_MONTHS_PER_YEARr-_HOURS_PER_DAYr._MINUTES_PER_HOURr/_SECONDS_PER_MINUTE_MICROSECONDS_PER_SECONDr0r1r2_SECONDS_PER_HOUR_SECONDS_PER_DAY_SECONDS_PER_MONTH_SECONDS_PER_YEARr8)r4r<r;carryr r r r3sx   zDuration._Normalizec CsX|}d}d}|dr|dd}d}n |dr!|dd}d}|dr/|dd}nd}|g}t|D]\}}|rF||q8|d ksN|d krT|d q8|d krc|r^td d}}q8t|dkrttd ||dtd |} |g}|dkr|j | 7_ q8|dkr|j | d7_ q8|dkr|j | 7_ q8|dvrt||dkr||ddkr|dkrd} n |dkrd} nd} |j | | 7_ n>|dkr|sd}|j | 7_ q8d}|dkr|j| 7_q8|dkr|j| 7_q8|dkr |j | 7_ q8td|t|dkr&|ddkr&td||S)a%Parses an ISO 8601 duration from string and returns a Duration object. If P is omitted then T is implied (M == minutes). Args: string: The ISO 8601 duration string to parse. Raises: ValueError: For invalid duration syntax. Returns: A Duration object. F-rN+rPTr,Tz1A duration may contain at most one 'T' separator.z0Duration unit '{}' must be preceded by a number.YWD)MUNrSrQirRr&iʚ;Hz$Unknown character '{0}' in duration.z+-P0z+Duration must end with time part character.)upper startswith enumerateisdigitr ValueErrorlenrrjoinr*r,r/r+r-r.lstripr3) r4stringr# t_separator t_impliedsignamounticr nr r r Parsesv          (  "zDuration.ParsercCs|dkrd}t|j}d}d}g}|jdkr|d|d||krJ|jrJd}t|j}|||j8}||dkrB|||j7}t||ddd||7}||krx|jrxd}t|j}|||j8}||dkrp|||j7}t||ddd||7}||kr|jrd}t|j}|||j 8}||dkr|||j 7}t||d dd|d ||7}||kr|j rd}t|j }|||j 8}||dkr|||j 7}t||d dd||7}||kr |j r d}t|j }|||j 8}||dkr|||j 7}t||ddd||7}||kr2|js|jr2|d7}t|t|jt|j|jd |d|d d kr?|dd }|d dkrK|dd|S)aReturns an ISO 8601 string representation of the duration. The Duration format is: "[-]P[nY][nM][nD][T[nH][nM][n[.m]S]]". The 0 duration is "P0". Otherwise at least one part will always be displayed. Negative durations are prefixed by "-". "T" disambiguates months "P2M" to the left of "T" and minutes "PT5M" to the right. Args: parts: Format at most this many duration parts starting with largest non-zero part, 0 for all parts. Zero-valued parts in the count are not shown. precision: Format the last duration part with precision digits after the decimal point. Trailing "0" and "." are always stripped. Returns: An ISO 8601 string representation of the duration. rrOrHrJrrMrrQrPrLrUrTrNrr)absr1rr*rFr$r+rEr,rDr-rCr.rAr/r0rBr])r4partsrr1countshownrrfr r r Format;s                 zDuration.FormatcCs,|dur||_|j|7_||S)zAdds a datetime.timdelta to the duration. Args: delta: A datetime.timedelta object to add. calendar: Use duration units larger than hours if True. Returns: The modified Duration (self). N)r2r/r1r3)r4r5r2r r r AddTimeDeltas  zDuration.AddTimeDeltac Cs4|j|j|j|j|j|j|jf\}}}}}}}||j7}||jkr+||j8}|d7}n |dkr8||j7}|d8}||j 7}||j krL||j 8}|d7}n |dkrY||j 7}|d8}||j 7}||j krm||j 8}|d7}n |dkrz||j 7}|d8}t ||j|j} ||j| |j7}|dkr||j7}| d8} || 7}||j7}||jkr||j8}|d7}n |dkr||j7}|d8}||j7}||j7}|dkr|dkr|d8}|dkr|j}|d8}|t||7}|dksn t||} || krn|| 8}|d7}||jkr d}|d7}qt||||||||jS)zReturns a copy of the datetime object dt relative to the duration. Args: dt: The datetime object to add the duration to. Returns: The a copy of datetime object dt relative to the duration. rr) microsecondsecondminutehourdayrr r0rBr/rAr.r@r7r-r?r+r>r*r,rdatetimetzinfo) r4dtrnrorprqrrrr rG days_in_monthr r r GetRelativeDateTimesx                          zDuration.GetRelativeDateTime) rrrrrrrNF)rr)N)__name__ __module__ __qualname____doc__r_DAYS_IN_MONTHr=rBrAr@r?r>rCrDrFrEr6r3rgrlrmrwr r r r r%Ts* ` T ^ r%)rr) r{ __future__rrrrsrr rr$objectr%r r r r s