hhC 0UdZddlmZddlZddlZddlmZmZm Z m Z erddl m Z m Z ddlmZdgZedZd ed< ed Zd ed <ejd Zd ed<egdZegdZdddddddZded<GddZdS)z;Tools for working with the Polars duration string language.) annotationsN) TYPE_CHECKINGLiteralcastget_args) ContainerMapping) TypeAlias IntervalUnit) nsusmssmhdmoqyr )dayshoursminutesseconds milliseconds microseconds TimedeltaKwdz7^(?P-?\d+)(?Pns|us|ms|mo|m|s|h|d|q|y)\Zzre.Pattern[str]PATTERN_INTERVAL) )rrr!rrrrrr)rrrrrr z#Mapping[IntervalUnit, TimedeltaKwd]UNIT_TO_TIMEDELTAc~eZdZddZedd dd ZeddZeddZe ddZ dS)Intervalmultipleintunitr returnNonec"||_||_dSN)r'r))selfr'r)s I/var/www/histauto/venv/lib/python3.11/site-packages/narwhals/_duration.py__init__zInterval.__init__2s% "& )r rrr) unsupportedr2Container[IntervalUnit] dt.timedeltac|j|vrd|jd}t|t|j}tjdi||jiS)NzCreating timedelta with z unit is not supported.)r)NotImplementedErrorr$dt timedeltar')r.r2msgkwds r/ to_timedeltazInterval.to_timedelta6sZ 9 # #OTYOOOC%c** * *|33sDM2333r1everystrc<||\}}|dkr&|tvrdtd|d}t||dkr&|tvrdtd|d}t||dkr|dkrd |d}t||||S) Nrz:Only the following multiples are supported for 'mo' unit: z. Got: .rz9Only the following multiples are supported for 'q' unit: rrz:Only multiple 1 is currently supported for 'y' unit. Got: )_parseMONTH_MULTIPLES ValueErrorQUARTER_MULTIPLES)clsr=r'r)r:s r/parsezInterval.parse@sE**$ 4<, where 'unit' is one of: r@)rmatchr(rrr rC)r=rKr'r)r:s r/rAzInterval._parseTs$**511 15 "5,--Hf 66DT> ! Au A A'/ '='= A A A oor1N)r'r(r)r r*r+)r2r3r*r4)r=r>r*r&)r=r>r*rI) __name__ __module__ __qualname__r0 frozensetr< classmethodrFrH staticmethodrAr6r1r/r&r&1s'''' 9B BX8Y8Y444444 # # #[ #'''['   \   r1r&)__doc__ __future__rdatetimer8retypingrrrrcollections.abcrr typing_extensionsr __all__r __annotations__rcompilerrOrBrDr$r&r6r1r/r\sAAA"""""" 999999999999,22222222++++++  !"VW WWWW "I %/BJ>%%)///00Iiii((       ::----------r1