o LhC @sUdZddlmZddlZddlZddlmZmZm Z m Z er-ddl m Z m Z ddlmZdgZedZd ed< ed Zd ed <ed Zd ed<egdZegdZdddddddZded<GdddZdS)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)rrrrr r z#Mapping[IntervalUnit, TimedeltaKwd]UNIT_TO_TIMEDELTAc@sTeZdZdddZed d dddZedddZedddZe dddZ dS)Intervalmultipleintunitr returnNonecCs||_||_dSN)r%r')selfr%r'r,S/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/narwhals/_duration.py__init__2s zInterval.__init__)r rrr) unsupportedr/Container[IntervalUnit] dt.timedeltacCs@|j|vrd|jd}t|t|j}tjdi||jiS)NzCreating timedelta with z unit is not supported.r,)r'NotImplementedErrorr#dt timedeltar%)r+r/msgkwdr,r,r- to_timedelta6s  zInterval.to_timedeltaeverystrcCs||\}}|dkr|tvrdtd|d}t||dkr1|tvr1dtd|d}t||dkrC|dkrCd |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)clsr8r%r'r5r,r,r-parse@s  zInterval.parsecCs|||Sr*)r;)r?r8r,r,r-parse_no_constraintsPszInterval.parse_no_constraintstuple[int, IntervalUnit]cCsNt|}rt|d}td|d}||fSd|dttd}t|)Nr%r r'zInvalid `every` string: zB. Expected string of kind , where 'unit' is one of: r:)rmatchr&rrr r=)r8rCr%r'r5r,r,r-r;Ts zInterval._parseN)r%r&r'r r(r))r/r0r(r1)r8r9r(r$)r8r9r(rB) __name__ __module__ __qualname__r. frozensetr7 classmethodr@rA staticmethodr;r,r,r,r-r$1s   r$)__doc__ __future__rdatetimer3retypingrrrrcollections.abcrrtyping_extensionsr __all__r __annotations__rcompilerrGr<r>r#r$r,r,r,r-s6