!g ,dZddlmZmZmZmZddlZddlZddlZ ddl m Z ddddddddd Z gd Z eed d e Zeeeeed d eZej)eegd edZdZdZdZddZddZdZddZy)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 cPt|}|dzdk(xr|dzdk7xs|dzdk(S)Nrdi)int)years [/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/tsa/base/datetools.py_is_leapr(;s3 t9D !8q= AdSjAo@qAc tjtjz}tjt||rX|j ddj jd\}}t|j\}}t|}ntjt||rm|j ddj jd\}} t| j\}}t|}t|rC|dk(r>|dz }n8tjt||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 MMBJJ &E yyWe,ooc"%++-33C8!$QWWY/ s1v :w .ooc"%++-33C8!"1779- s1v A;5A: 1HC :w . s7|7-f--   T5# ..r)c Ztjtjz}|j}tjt ||rd}d}nZtjt ||rd}d}n:tjt||rd}|dz }|r|dz }d}ntd|ztt|jd d j|\}}|C|j}tt|jd d j|\}} n |s td |||zz}||z|dz z} tjt|dz||D cgc] } t!| } } t!|g|dz|z z| z} | t!|g| zz} |dk7rtj"tj$d|dz||z dz j'd } tj(tj$||dzj'd | f} tj(| tj$d| dzj'd f} t+| | D cgc]"\} }d j-| |t/|g$}} }|S| }|Scc} wcc}} w) 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,z&length must be provided if end is NoneS2)r0r1r2r6r3r:r4r< ValueErrorrr%r5r7nprepeatrstrtilearangeastyper_zipjoinr)startendlengthrA annual_freqr7yr1offset1yr2offset2yryearsoffsetijdate_arr_ranges r'date_range_strrb[sq$ MMBJJ &E KKME yyUE*  :ue ,  :ue ,     4KC1E9::U]]3r288?@LC iikCS!4!:!:5!AB WEF FFk));&'A+6!yya)={K LSW LE L#hLK!Og5 6 >E c#hL7* *E |1kAo6c A FMMdSryy+/:AA$GOPvryyGaK8??EEF&)%&8:"a''1eU1X"67::   M:s (J"2'J'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_strres  U ##r)c2t|||}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. )rbre)rTrUrVrds r'dates_from_rangergs4 5#v .E %  r))N)NN)__doc__statsmodels.compat.pythonrrrrr=r0numpyrKpandasrr8_mdays_months_with_daysdictrRmaprMr;updater<r4r:r(rErbrergr)r'rrs@?    :"v.SS".0ABC T#F0234      B /69x$&!r)