.hb>UddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZmZmZmZmZmZddlZdd lmZ dd l!m"Z"m#Z#m$Z$dd l!m%Z&dd l'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/dd l0m1Z1m2Z2erfddl m3Z3m4Z4m5Z5ddlm6Z6m7Z7ddl8m9Z9m:Z:ejvdk\rddlmZ>m?Z?nddl=m>Z>m?Z?e>dZ@edZAejvdk\rCijijijgZEdeFd<eGdeEDZH dM dNdZIdOdZJdPdZKddd dQd ZLdd! dRd"ZMdd! dSd#ZNdd! dTd$ZOddd dUd%ZPdVd&ZQdWd'ZR dM dXd(ZSdYd)ZTdZd*ZUdZd+ZVd[d,ZWd-d.d\d/ZXd]d0ZYd^d1ZZd_d2Z[ed`d3Z\edad4Z\dbd5Z\ dcd6Z]ed7Z^Gd8d9e_Z`ejd:ZbGd;dZg dMdddd? dfd@ZheifdgdAZj dhdBZkdZdCZlddD didEZmdjdFZnd-dddGdH dkdIZoddJdldKZpdmdLZqy)n) annotationsN)Counter) Collection GeneratorIterable MappingViewSequenceSized)Enum)BytesIO)Path) TYPE_CHECKINGAnyCallableLiteralTypeVaroverload) functions)_check_for_numpyimport_optional subprocess)numpy)BooleanDateDatetimeDecimalDurationInt64StringTime) FLOAT_DTYPESINTEGER_DTYPES)IteratorMutableMapping Reversible) DataFrameExpr)PolarsDataTypeSizeUnit) )TypeIs)r* ) ParamSpec TypeGuardPTr* zlist[Reversible[Any]]_viewsc#DK|]}tt|ywN)typereversed).0views S/var/www/html/ai-service/venv/lib/python3.12/site-packages/polars/_utils/various.py r<Cs"KD4#7"Ks cXt|trt|jS|Sr6) isinstancedictlistitems) null_valuess r;_process_null_valuesrCFs(+t$K%%'((ct|ttfxrt|t xs7t|txs%t j dk\xrt|tS)Nr2)r>rrr rsys version_info_reverse_mapping_views)vals r; _is_generatorrJOsV C)X. / N 38N4N U c; ' U    ' SJs)r9xeltypes r;r<z"_is_iterable_of..Ys2z!V$2s)all)rIrOs `r;_is_iterable_ofrQWs 2c2 22rDF) allow_strinclude_seriesc|durt|tryt|rHt|tjr.tj |j tjS|r7t|tjr|j tjk(St|t xr(t|txrt|ttfS)z Check that `val` is a sequence of strings or paths. Note that a single string is a sequence of strings by definition, use `allow_str=False` to return False on a single string. F)r>strrnpndarray issubdtypedtypestr_plSeriesrbytesr rQr rIrRrSs r;is_path_or_str_sequencer_\sEjc2 # :c2::#>}}SYY00 JsBII6yyBII%% sE "" . sH % . C$ -rD)rSc<t|r7t|tjr|jtj k(S|r7t|t jr|jt jk(St|txrt|tS)z;Check whether the given sequence is a sequence of booleans.) rr>rVrWrYbool_r[r\rr rQboolrIrSs r;is_bool_sequencerdrshC!<yyBHH$$ JsBII6yyBJJ&& c8 $ Cd)CCrDcXt|rHt|tjr.tj|j tj S|r4t|tjr|j jSt|txrt|tS)z;Check whether the given sequence is a sequence of integers.) rr>rVrWrXrYintegerr[r\ is_integerr rQintrcs r;is_int_sequenceri}snC!<}}SYY 33 JsBII6yy##%% c8 $ Bc)BBrDct|xrt|tjxs;t||rtj t fnt xrt|t S)zBCheck whether the given input is a numpy array or python sequence.)rr>rVrWr[r\r rUrcs r; is_sequencerksO S ! Ajbjj&A 3H-XN %3$ $rDc|durt|tryt|rHt|tjr.tj |j tjS|r7t|tjr|j tjk(St|txrt|tS)z Check that `val` is a sequence of strings. Note that a single string is a sequence of strings by definition, use `allow_str=False` to return False on a single string. F) r>rUrrVrWrXrYrZr[r\rr rQr^s r;is_str_sequencermsEjc2 # :c2::#>}}SYY00 JsBII6yyBII%% c8 $ Bc)BBrDc^ddlm}t||xr|jj S)z9Indicate if the given object is a basic/unaliased column.r)r') polars.exprr'r>meta is_column)objr's r;rqrqs$ c4 9SXX%7%7%99rDcR|%tjdttyy)z6Warn for possibly unintentional comparisons with None.Nz]Comparisons with None always result in null. Consider using `.is_null()` or `.is_not_null()`.) stacklevel)warningswarn UserWarningfind_stacklevelrrs r;warn_null_comparisonrzs$ { k &( rDcV|xst}|jr9tj|j|j |j |d}nFtj|j|j |j dj|}|j|S)z/Fast conversion of the given range to a Series.T)startendsteprYeager)r|r}r~r) rrgF int_ranger|stopr~castalias)namerngrYranges r;range_to_seriesrs NUE  ))T  )) $u+  ;;t rDcXt|j|j|jS)z.Return the given range as an equivalent slice.)slicer|rr~)rs r;range_to_slicers CHHchh //rDcn ddlm}d|jvry y#t$rYyt$rYywxYw)Nr get_ipython IPKernelAppFT)IPythonrconfig ImportErrorAttributeErrorrs r; _in_notebookrsF'  4 4 4 5   s 444cJ ddl}|jS#t$rYywxYw)NrF)marimorunning_in_notebookr)mos r;_in_marimo_notebookrs+%%'' s  ""ch t|ttfrdSt|S#t$rYywxYw)zLReturn length of (non-string/dict) sequence; returns None for non-sequences.N)r>rUr?len TypeErrorrys r;arrlenrs5!#T{3tASA s% % 11T)check_not_directoryctjj|}|rOtjj|r0tjj |rd|d}t ||S)z>Create a string path, expanding the home directory if present.zexpected a file path; z is a directory)ospath expanduserexistsisdirIsADirectoryError)rrmsgs r;normalize_filepathrsY 77  d #D GGNN4 GGMM$ &tho>$$ KrDcht|tr|jd}td|DS)zASimple version parser; split into a tuple of ints for comparison..c 3nK|]-}ttjddt|/yw)z\DN)rhresubrU)r9vs r;r<z parse_version..s&AARVVE2s1v./As35)r>rUsplittuple)versions r; parse_versionrs+'3--$ AA AArDctt}|j}|Dcgc]}||vr||r|c}Scc}w)zMReturn unique list of sequence values, maintaining their order of appearance.)setadd)valuesseenadd_rs r;ordered_uniquers2UD 88D br]>kb kilobytesi>mb megabytesi>gb gigabytesi@>tb terabyteslz9`unit` must be one of {'b', 'kb', 'mb', 'gb', 'tb'}, got ) ValueError)rrrs r;rrsq ~ $ $Dy $ $G| $ $G| $ $G|KD8TorDc  |js8|jjD]}|tk7s d|}t |hd}d}dddd||dz|dzdz|dzd z|dzd zd zd d)fd }i}|j D]\}}| |j tk(r t|j} tj|jjdd} tj| jjdk(j| dzj!| dzjj#ddjj%| d||<t'|dd ||j(j+|j,||<,|t.k(r4tj|jj%|d||<i|t0k(rtjtj|jjdk(jtj|dzj!tj|dzjj#ddjj%|d||<8|t2k(rVtj|j5|t6j9t3dj9|||<|t:k(r,tj|j=ddd||<|t>vrtj|jjAd d} tCj| jjdkDj| j9|||<V|tDvs|j tFk(r+tj|jjd!d"} tj|jjd!d#} tCjtj|jjId$tj|jjKjM|zjtCjtj|jjdkDjtj|jtj|jjId%jtj|jjd&d'j!tCjN| jjAd d| d'(j9tj9|||<||j|k7stj|j9|||<|r|jPd*i|S|S)+a Utility function to cast table repr/string values into frame-native types. Parameters ---------- df Dataframe containing string-repr column data. schema DataFrame schema containing the desired end-state types. Notes ----- Table repr strings are less strict (or different) than equivalent CSV data, so need special handling; as this function is only used for reprs, parsing is flexible. z6DataFrame should contain only String repr data; found >+inf-infinfnaniʚ;rii@B<i) nsusuµsmssmhdwc\|dStfdtjd|DS)Nc3bK|]&\}}t||jz(ywr6)rhstrip)r9valuerduration_scalings r;r<zH_cast_repr_strings_with_schema..str_duration_..Us2E4E -djjl;;s,/z([+-]?\d+)(\D+))sumrfindall)tdrs r; str_duration_z5_cast_repr_strings_with_schema..str_duration_Qs<z  #%::.@"#E rDNz[A-Z ]+$rz .000000000 000000000rz%Y-%m-%d %H:%M:%S.%9f time_zonez%Y-%m-%dz %H:%M:%S.%9f) return_dtyperTF)truefalsez[^\d+-]z ^(.*)\D(\d*)$z$1z$2z ^[+-]?\d*$z[eE]z[^eE\d]r) separator)r str | Nonereturn int | Noner))is_emptyschemarrrrA base_typer time_unitrcolrUreplacewhen len_bytesthen otherwiserstrptimegetattrdtreplace_time_zonerrr r map_elementsrrrreplace_strictr" replace_allr[r!rcontains to_lowercaseis_in concat_str with_columns)dfrtprspecial_floatsns_secr cast_colsctp_baser int_string integer_partfractional_partrs @r;_cast_repr_strings_with_schemar's& ;;=))""$ %BV|NrfUn$ % 4NF  b[ b[2  e^b e^b 1 $  I@12 >||~)"2<<0EE!HLL((b9FF155??,23T!l*+Yq;/Sq"S'+BC ! 2{D1=#,Q?DDQUU1XNOT!%%(,,//78T!%%(,,..z3?@Y (,,88RH+&)T&\T"X'! *ryy|# uuQx}}R0 ! A@1D,5?2?? 'Y '<"rr)selfinstanceclss r;__get__zsphinx_accessor.__get__sL  99&x5 ;>  , K s%%99N)r rr!ztype[NS]rr)__name__ __module__ __qualname__r"rrDr;rrs#    rDrBUILDING_SPHINX_DOCSceZdZdZddZy) _NoDefault NO_DEFAULTcy)Nz r)rs r;__repr__z_NoDefault.__repr__srDN)rrU)r#r$r% no_defaultr+rrDr;r(r(sJrDr(cptttjj}t j }d} |rit j|}|j|s*t|jddx}r#|jdr|j}|dz }n ~|S|ri~|S#~wxYw)z Find the first place in the stack that is not inside Polars. Taken from: https://github.com/pandas-dev/pandas/blob/ab89c53f48df67709a533b6a95ce3d911871a0a8/pandas/util/_exceptions.py#L30-L51 r co_qualnameNzsingledispatch.r) rUr r[__file__parentinspect currentframegetfile startswithrf_codef_back)pkg_dirframenfnamequalnames r;rxrxs$r{{#**+G  "E AOOE*E($U\\=$GGG''(9: Q  H#  H sA$B2,B22B5c Ftjd||td|y)a  Issue a warning. Parameters ---------- message The message associated with the warning. category The warning category. **kwargs Additional arguments for `warnings.warn`. Note that the `stacklevel` is determined automatically. )messagecategoryrtNrrurvrx)r=r>kwargss r; issue_warningrAs' MM(7HLRrD)named n_objectsn_framesci}d}t|tr|f}|tj}t j r}n!tt r tdfd }|/t|tr|f}nt|ts t|}t j}t|dd} |r||krt|jj}t|D]7\} } | |vs || |vs || s| || <|%t||k\s4|c~S|j }|dz }|r||kr~|S#~wxYw)a@ Retrieve f_locals from all (or the last 'n') stack frames from the calling location. Parameters ---------- of_type Only return objects of this type; can be a single class, tuple of classes, or a callable that returns True/False if the object being tested is considered a match. n_objects If specified, return only the most recent `n` matching objects. n_frames If specified, look at objects in the last `n` stack frames only. named If specified, only return objects matching the given name(s). rNct|Sr6rM)rrof_types r; matches_typez'_get_stack_locals..matches_type sc7+ +rDr6rrrrrrb)r>rUrFmaxsizer1 isfunctionrrrr2rr@f_localsrAr8rr6) rGrBrCrDobjectsexamined_framesrH stack_frame local_itemsrrrs ` r;_get_stack_localsrQsY.GO%;;'" gz *GnG ,  eS !HEE3'JE&&(K+x6Ko8{3399;rbrr)rn_match value_name match_namerrs r; extend_boolrYMsf#-UD"9eWw uF#f+j\S[M:--7LG9A G o MrDcttjdrtjjxrtjdk7xsndt j vxsZdt j vxsFt j jddk(xs"t j jddk(xs"t j jd d k(Sy ) z Determine (within reason) if we are in an interactive terminal that supports color. Note: this is not exhaustive, but it covers a lot (most?) of the common cases. isattywin32ANSICON WT_SESSION TERM_PROGRAMvscodeTERMzxterm-256colorPYCHARM_HOSTED1F)hasattrrFstdoutr[platformrenvirongetrrDr; in_terminal_that_supports_colourri^s szz8$ JJ     '> *>2::->::>>.1X=>::>>&)-== 5ZZ^^, - 4 5 rD) inject_medianct|tr|g}n|g}td|Ds d}t|t d|D}t d|D}|r|r|ddk7rdg|}g||S)z Transforms raw percentiles into our preferred format, adding the 50th percentile. Raises a ValueError if the percentile sequence is invalid (e.g. outside the range [0, 1]) c3<K|]}d|cxkxrdkncyw)rrNrr9ps r;r<z$parse_percentiles..s2Q ! 2sz-`percentiles` must all be in the range [0, 1]c3,K|] }|dks |yw?Nrrms r;r<z$parse_percentiles..sBa!c'B c3,K|] }|dk\s |ywrprrms r;r<z$parse_percentiles..s'Ka!s('Krrrrq)r>floatrPrsorted) percentilesrjrsub_50_percentilesat_or_above_50_percentiless r;parse_percentilesryss+u%"m   2k2 2=oB;BB!''K;'K!K &*DQ*G3*N&)%G,F%G" =  ="< ==rDc<d}tjd|dd|S)z1Escape a string for use in a Polars (Rust) regex.z\\?()|\[\]{}^$#&~.+*-z([z])z\\\1)rr)rre_rust_metacharss r; re_escaper|s(1 66B(),gq 99rD)g0@g(@)show output_path raw_outputfigsizec |r|Ststsdtjvrdnd} t j ddd|zg|j }|rt|j||sytrd d l m }m } | ||Str$d dl} | j|j!Stjj#ddx} td dl} | j'd 5} | j)|| j+| j-d | j.} t j0| ddddyt3dddd dlm}d dlm}|j=||j?tA|}|jCd|jE||jGy#ttf$rd}t|dwxYw#1swYyxYw)NPOLARS_DOT_SVG_VIEWERsvgpngdotz -Nshape=boxz-T)inputzythe graphviz `dot` binary should be on your PATH.(If not installed you can download here: https://graphviz.org/download/)r)SVGdisplayz.svg)suffixz%file%T)shell matplotlibrz"should be installed to show graphs) err_prefix err_suffix)roff)$rrrrgr check_outputencoderFileNotFoundErrorr write_bytesIPython.displayrrrHtmldecoderhtempfileNamedTemporaryFilewriteflushrrrunrmatplotlib.imageimagematplotlib.pyplotpyplotfigureimreadr axisimshowr})rr}r~rr output_typegraphrrrrcmdrfilempimgpltimgs r;display_dot_graphrs  >  "bjj 0   )'' M4+#5 6oo>O  [%%e, ~0s5z""  ww%,,.)++::>>"94@ @C M ,,F,; 0t 5! kk(DII6s$/  0  ; )' 7 #ll75>*  3  [ * +) W #D( )2 0 s,GAG9G69H)qualify_polarsct|tr|j}|j}n,|jj}|jj}|r|dk(s|s|j dr|S|d|S)a Return the module-qualified name of the given object as a string. Parameters ---------- obj The object to get the qualified name for. qualify_polars If False (default), omit the module path for our own (Polars) objects. builtinszpolars.r)r>r7r$r# __class__r4)rrrmodulers r;qualified_type_namersq#t||))}}%%  Z 6#4#4Y#? XQtf rDct|t|s;t|t|s%dt|dt|}t|yy)aB Raise an error if the two arguments are not of the same type. The check will not raise an error if one object is of a subclass of the other. Parameters ---------- current The object the type of which is being checked against. other An object that has to be of the same type. zexpected `other` to be a z, not N)r>r7rr)currentotherrs r;require_same_typers] eT'] +JwU 4T'(;G(D'GH&u-0 2 n 5U +rDr6)rBz+None | str | Sequence[str] | dict[str, str]rz2None | str | Sequence[str] | list[tuple[str, str]])rIzobject | Iterator[T]rzTypeIs[Iterator[T]])rIzIterable[object]rOztype | tuple[type, ...]rrb)rIobjectrRrbrSrbrzTypeGuard[Sequence[str | Path]])rIrrSrbrzTypeGuard[Sequence[bool]])rIrrSrbrzTypeGuard[Sequence[int]])rIrrSrbrzTypeGuard[Sequence[Any]])rIrrRrbrSrbrzTypeGuard[Sequence[str]]rI)rrrrNone)rrUrrrYzPolarsDataType | Nonerz pl.Series)rrrr)rrb)rrrrr)rz str | PathrrbrrU)rzSequence[str | int]rztuple[int, ...])rz Sequence[Any]rz list[Any])rz Iterable[str]rz list[str])rrhrr)rz int | float)rr'rr)rr')rz int | Exprrr)rzint | float | Expr)rr&rz dict[str, PolarsDataType | None]rr&)rrh)r=rUr> type[Warning]r@rrr) rGz6type | Collection[type] | Callable[[Any], bool] | NonerBzstr | Collection[str] | NonerCrrDrrzdict[str, Any])rrUr>rrr) rzbool | Sequence[bool]rVrhrWrUrXrUrzSequence[bool])rvzSequence[float] | float | NonerjrbrzSequence[float])rrUrrU) rrUr}rbr~zstr | Path | Nonerrbrztuple[float, float]rr)rrrrrbrrU)rrrrrr)r __future__rr1rrrFru collectionsrcollections.abcrrrrr r enumr ior pathlibr typingrrrrrrpolarsr[rrpolars._dependenciesrrrrrVpolars.datatypesrrrrrrrr polars.datatypes.groupr!r"r#r$r%r&r'polars._typingr(r)rGr,typing_extensionsr.r/r0r1keysrrAr4__annotations__rrHrCrJrQr_rdrirkrmrqrzrrrrrrrrrrrrpropertyrgetenvr&r(r, NoDefaultrxrArQrwrSrYriryr|rrrrrDr;rs" !NN,   @DD&7 7"!, 7"//:#A Aw%'WWY RXXZ$HF !H""KF"KK@D<73 ',E #=A$.,1D D$(DD,1C C$(CC,1 $(',EC C#C=ACC$: ;?    "7   0 IM B= < < 6 6"w=w=;w=w=z T] h !ryy!78 " " J)) *  B(GKF+/  F CF (F F  F  FT6A     ",KP>/>CG>>8:%)#/ E E E# E  E ! EEP=B8rD