M/Ph fdZddlmZmZmZmZddlZddlZddlZ ddl m Z ddddddddd Z gd Z eed d e Zeeeeed d eZeeegd edZdZdZdZddZddZdZddZdS)z Tools for working with dates )asstrlmaplrangelzipN) to_datetime))) r  r )1234IIIIIIIV) r r r r r r r r r r r  ) rrrrVVIVIIVIIIIXXXIXIIz ^\d?\d?\d?\d$z ^ # beginning of string \d?\d?\d?\d # match any number 1-9999, includes leading zeros (:?q) # use q or a : as a separator ([1-4]|(I{1,3}V?)) # match 1-4 or I-IV roman numerals $ # end of string a= ^ # beginning of string \d?\d?\d?\d # match any number 1-9999, includes leading zeros (:?m) # use m or a : as a separator (([1-9][0-2]?)|(I?XI{0,2}|I?VI{0,3}|I{1,3})) # match 1-12 or # I-XII roman numerals $ # end of string cVt|}|dzdko|dzdkp|dzdkS)Nrdi)int)years ^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/tsa/base/datetools.py_is_leapr(;s5 t99D !8q= AdSjAo@qAc :tjtjz}tjt||rp|ddd\}}t| \}}t|}ntjt||r|ddd\}} t| \}}t|}t|r |dkr|dz }n=tjt||rd\}}t|}n t|fi|St!j|||S)a$ Uses dateutil.parser.parse, but also handles monthly dates of the form 1999m4, 1999:m4, 1999:mIV, 1999mIV and the same for quarterly data with q instead of m. It is not case sensitive. The default for annual data is the end of the year, which also differs from dateutil. :qmrr )re IGNORECASEVERBOSEsearch _q_patternreplacelowersplit_quarter_to_dayupperr% _m_pattern _month_to_dayr( _y_patternrdatetime) timestr parserinfokwargsflagsyr-monthdayr&r.s r' date_parserrE@s[ MBJ &E yWe,,.ooc"%%++--33C88!$QWWYY/ s1vv :w . . .ooc"%%++--33C88!"17799- s1vv A;; 5A:: 1HC :w . .. s7||7--f---  T5# . ..r)cZ tjtjz}|}tjt ||rd}d n^tjt ||rd}d n>tjt||rd}|dz }|r|dz }d ntd|ztt| d d  \}}|T|}tt| d d  \}}n$|std |||zz}||z|dz z}d tjt|dz||D} t!|g|dz|z z| z} | t!|g|zz} dkrtjtjd|dz||z dz d} tjtj||dzd| f} tj| tjd|dzdf} fdt+| | D} n| } | S)a Returns a list of abbreviated date strings. Parameters ---------- start : str The first abbreviated date, for instance, '1965q1' or '1965m1' end : str, optional The last abbreviated date if length is None. length : int, optional The length of the returned array of end is None. Returns ------- date_range : list List of strings rr.r#r-ra1azDate %s not understoodr+r,Nz&length must be provided if end is Nonec,g|]}t|S)str).0yrs r' z"date_range_str..s L L LSWW L L Lr)S2c `g|]*\}}d|t|g+S)r,)joinr)rLijr7s r'rNz"date_range_str..sF:::"a''1eU1XX"677:::r))r0r1r2r6r3r:r4r< ValueErrorrr%r5r7nprepeatrrKtilearangeastyper_zip) startendlengthrA annual_freqyr1offset1yr2offset2yearsoffsetdate_arr_ranger7s @r'date_range_strrg[s$ MBJ &E KKMME yUE** ;  :ue , , ;  :ue , ,;      4KC1E9:::U]]3r2288??@@LC iikkCS!4!4!:!:5!A!ABB WW GEFF FFk));&'A+6 L Lrya)=)={KK L L LE#hhLK!Og5 6 >E c#hhL7* *E ||1kAo66c A FFMMdSSry+/::AA$GGOPvryGaK88??EEEF::::&)%&8&8::: r)c,tt|S)ak Turns a sequence of date strings and returns a list of datetime. Parameters ---------- dates : array_like A sequence of abbreviated dates as string. For instance, '1996m1' or '1996Q1'. The datetime dates are at the end of the period. Returns ------- date_list : ndarray A list of datetime types. )rrE)datess r'dates_from_strrjs  U # ##r)cBt|||}t|S)aX Turns a sequence of date strings and returns a list of datetime. Parameters ---------- start : str The first abbreviated date, for instance, '1965q1' or '1965m1' end : str, optional The last abbreviated date if length is None. length : int, optional The length of the returned array of end is None. Examples -------- >>> import statsmodels.api as sm >>> import pandas as pd >>> nobs = 50 >>> dates = pd.date_range('1960m1', length=nobs) Returns ------- date_list : ndarray A list of datetime types. )rgrj)r\r]r^ris r'dates_from_rangerls#4 5#v . .E %  r))N)NN)__doc__statsmodels.compat.pythonrrrrr=r0numpyrUpandasrr8_mdays_months_with_daysdictr[maprKr;updater<r4r:r(rErgrjrlrJr)r'rvs@???????????     : 9 9D"v..SSS"..0ABBCC TT##FFF02233444      BBB ////69999x$$$&!!!!!!r)