ic(UddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZmZmZmZmZddlZdd lmZdd 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 l/m0Z0m1Z1er]ddl m2Z2m3Z3m4Z4m5Z5ddlm6Z6m7Z7m8Z8ddlm9Z9m:Z:ddl;mZidfd?Zj dNdddd@ dgdAZkelfdhdBZm didCZnd[dDZoddE djdFZpdkdGZqd.dddHdI dldJZrddKdmdLZsdndMZty)o) annotationsN)Counter) Collection GeneratorIterable MappingViewSequenceSized)Enum)BytesIO)Path) TYPE_CHECKINGAnyLiteralTypeVaroverload) functions)_check_for_numpyimport_optional subprocess)numpy)BooleanDateDatetimeDecimalDurationInt64StringTime) FLOAT_DTYPESINTEGER_DTYPES)CallableIteratorMutableMapping Reversible) ParamSpecProtocol TypeGuard) DataFrameExpr)PolarsDataTypeSizeUnit) )TypeIsPTceZdZddZy)IdentityFunctioncyN)selffns W/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/_utils/various.py__call__zIdentityFunction.__call__BsN)r8Callable[P, T]returnr<)__name__ __module__ __qualname__r:r6r;r9r3r3@sHr;r3r- zlist[Reversible[Any]]_viewsc#DK|]}tt|ywr5)typereversed).0views r9 rIHs"KD4#7"Ks cXt|trt|jS|Sr5) isinstancedictlistitems) null_valuess r9_process_null_valuesrPKs(+t$K%%'((r;ct|ttfxrt|t xs7t|txs%t j dk\xrt|tS)NrA)rKrrr rsys version_info_reverse_mapping_views)vals r9 _is_generatorrVTsV C)X. / N 38N4N U c; ' U    ' SJs.^s2z!V$2s)all)rUr[s `r9_is_iterable_ofr]\s 2c2 22r;F) 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)rKstrrnpndarray issubdtypedtypestr_plSeriesrbytesr r]r rUr^r_s r9is_path_or_str_sequencerkasEjc2 # :c2::#>}}SYY00 JsBII6yyBII%% sE "" . sH % . C$ -r;)r_c<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.) rrKrbrcrebool_rgrhrr r]boolrUr_s r9is_bool_sequencerpwshC!<yyBHH$$ JsBII6yyBJJ&& c8 $ Cd)CCr;cXt|rHt|tjr.tj|j tj S|r4t|tjr|j jSt|txrt|tS)z;Check whether the given sequence is a sequence of integers.) rrKrbrcrdreintegerrgrh is_integerr r]intros r9is_int_sequencerusnC!<}}SYY 33 JsBII6yy##%% c8 $ Bc)BBr;ct|xrt|tjxs;t||rtj t fnt xrt|t S)zBCheck whether the given input is a numpy array or python sequence.)rrKrbrcrgrhr raros r9 is_sequencerwsO S ! Ajbjj&A 3H-XN %3$ $r;c|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) rKrarrbrcrdrerfrgrhrr r]rjs r9is_str_sequencerysEjc2 # :c2::#>}}SYY00 JsBII6yyBII%% c8 $ Bc)BBr;c^ddlm}t||xr|jj S)z9Indicate if the given object is a basic/unaliased column.r)r*) polars.exprr*rKmeta is_column)objr*s r9r}r}s$ c4 9SXX%7%7%99r;cR|%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_stacklevelr~s r9warn_null_comparisonrs$ { k &( r;cV|xst}|jr9tj|j|j |j |d}nFtj|j|j |j dj|}|j|S)z/Fast conversion of the given range to a Series.T)startendstepreeager)rrrr) rrsF int_rangerstoprcastalias)namerngreranges r9range_to_seriesrs NUE  ))T  )) $u+  ;;t r;cXt|j|j|jS)z.Return the given range as an equivalent slice.)slicerrr)rs r9range_to_slicers CHHchh //r;cn ddlm}d|jvry y#t$rYyt$rYywxYw)Nr get_ipython IPKernelAppFT)IPythonrconfig ImportErrorAttributeErrorrs r9 _in_notebookrsF'  4 4 4 5   s 444cJ ddl}|jS#t$rYywxYw)NrF)marimorunning_in_notebookr)mos r9_in_marimo_notebookrs+%%'' s  ""ch t|ttfrdSt|S#t$rYywxYw)zLReturn length of (non-string/dict) sequence; returns None for non-sequences.N)rKrarLlen TypeErrorrs r9arrlenrs5!#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 r9normalize_filepathrsY 77  d #D GGNN4 GGMM$ &tho>$$ Kr;cht|tr|jd}td|DS)zASimple version parser; split into a tuple of ints for comparison..c 3nK|]-}ttjddt|/yw)z\DN)rtresubra)rGvs r9rIz parse_version..s&AARVVE2s1v./As35)rKrasplittuple)versions r9 parse_versionrs+'3--$ AA AAr;ctt}|j}|Dcgc]}||vr||r|c}Scc}w)zMReturn unique list of sequence values, maintaining their order of appearance.)setadd)valuesseenadd_rs r9ordered_uniquers2UD 88D bri>kb kilobytesi>mb megabytesi>gb gigabytesi@>tb terabyteslz9`unit` must be one of {'b', 'kb', 'mb', 'gb', 'tb'}, got ) ValueError)rrrs r9rrsq ~ $ $Dy $ $G| $ $G| $ $G|KD8Tor;c  |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(ywr5)rtstrip)rGvaluerduration_scalings r9rIzH_cast_repr_strings_with_schema..str_duration_..Zs2E4E -djjl;;s,/z([+-]?\d+)(\D+))sumrfindall)tdrs r9 str_duration_z5_cast_repr_strings_with_schema..str_duration_Vs<z  #%::.@"#E r;Nz[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 | Noner= int | Noner6))is_emptyschemarrrrN base_typer time_unitrcolrareplacewhen len_bytesthen otherwiserstrptimegetattrdtreplace_time_zonerrrr map_elementsrrrreplace_strictr! replace_allrgr rcontains to_lowercaseis_in concat_str with_columns)dfrtprspecial_floatsns_secr cast_colsctp_baser int_string integer_partfractional_partrs @r9_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!%%(,,..|SABY (,,88RH+&)T&\T"X'! *ryy|# uuQx}}R0 ! A@1D,5?2?? 'Y '<"  , K s%%99N)r)rr*ztype[NS]r=r$)r>r?r@r+r6r;r9r&r&s#    r;r&BUILDING_SPHINX_DOCSceZdZdZddZy) _NoDefault NO_DEFAULTcy)Nz r6)r7s r9__repr__z_NoDefault.__repr__sr;N)r=ra)r>r?r@ no_defaultr1r6r;r9r.r.sJr;r.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) rar rg__file__parentinspect currentframegetfile startswithrf_codef_back)pkg_dirframenfnamequalnames r9rrs$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. )messagecategoryrNr6rrr)rCrDkwargss r9 issue_warningrGs' MM(7HLRr;)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|Sr5rY)r~of_types r9 matches_typez'_get_stack_locals..matches_type%sc7+ +r;r<rr~rr=rn)rKrarRmaxsizer7 isfunctionrrrr8rrMf_localsrNrFrr<) rMrHrIrJobjectsexamined_framesrN stack_frame local_itemsrr~s ` r9_get_stack_localsrWsY.GO%;;'" gz *GnG ,  eS !HEE3'JE&&(K+x6Ko8{3399; *>2::->::>>.1X=>::>>&)-== 5ZZ^^, - 4 5 r;) 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)rrNr6rGps r9rIz$parse_percentiles..s2Q ! 2sz-`percentiles` must all be in the range [0, 1]c3,K|] }|dks |yw?Nr6rss r9rIz$parse_percentiles..sBa!c'B c3,K|] }|dk\s |ywrvr6rss r9rIz$parse_percentiles..s'Ka!s('Krxrrw)rKfloatr\rsorted) percentilesrprsub_50_percentilesat_or_above_50_percentiless r9parse_percentilesrxs+u%"m   2k2 2=oB;BB!''K;'K!K &*DQ*G3*N&)%G,F%G" =  ="< ==r;c<d}tjd|dd|S)z1Escape a string for use in a Polars (Rust) regex.z\\?()|\[\]{}^$#&~.+*-z([z])z\\\1)rr)rre_rust_metacharss r9 re_escapers(1 66B(),gq 99r;)g0@g(@)show output_path raw_outputfigsizecB|r|S|t|jds&tstsdtj vrdnd} t jddd|zg|j}|rt|j||sytrd d l m}m} | ||Str$d dl} | j#|j%Stj j'ddx} td dl} | j+d 5} | j-|| j/| j1d | j2} t j4| ddddyt7dddd dlm}d dlm}|jA||jCtE|}|jGd|jI||jKy#ttf$rd }t|dwxYw#1swYyxYw)Nz.svgPOLARS_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)SVGdisplay)suffixz%file%T)shell matplotlibrz"should be installed to show graphs) err_prefix err_suffix)roff)&raendswithrrrrmr check_outputencoderFileNotFoundErrorr write_bytesIPython.displayrrrHtmldecoderntempfileNamedTemporaryFilewriteflushrrrunrmatplotlib.imageimagematplotlib.pyplotpyplotfigureimreadr axisimshowr)rrrrr output_typegraphrrrrcmdrfilempimgpltimgs r9display_dot_graphrs   #K(8(A(A&(I >  "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 ,G4AH4HH)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)rKrEr?r> __class__r:)r~rmodulers r9qualified_type_namersq#t||))}}%%  Z 6#4#4Y#? XQtf r;ct|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)rKrErr)currentotherrs r9require_same_typers] eT'] +JwU 4T'(;G(D'GH&u-0 2 n 5U +r;r5)rOz+None | str | Sequence[str] | dict[str, str]r=z2None | str | Sequence[str] | list[tuple[str, str]])rUzobject | Iterator[T]r=zTypeIs[Iterator[T]])rUzIterable[object]r[ztype | tuple[type, ...]r=rn)rUobjectr^rnr_rnr=zTypeGuard[Sequence[str | Path]])rUrr_rnr=zTypeGuard[Sequence[bool]])rUrr_rnr=zTypeGuard[Sequence[int]])rUrr_rnr=zTypeGuard[Sequence[Any]])rUrr^rnr_rnr=zTypeGuard[Sequence[str]]rO)r~rr=None)rrarrrezPolarsDataType | Noner=z pl.Series)rrr=r)r=rn)r~rr=r)rz str | Pathrrnr=ra)rzSequence[str | int]r=ztuple[int, ...])rz Sequence[Any]r=z list[Any])rz Iterable[str]r=z list[str])rrtrr,r=z int | float)rr*rr,r=r*)rz int | Exprrr,r=zint | float | Expr)rr)rz dict[str, PolarsDataType | None]r=r))r=rt)rCrarD type[Warning]rFrr=r) rMz6type | Collection[type] | Callable[[Any], bool] | NonerHzstr | Collection[str] | NonerIrrJrr=zdict[str, Any])rrarDrr=r) rzbool | Sequence[bool]r\rtr]rar^rar=zSequence[bool])r|zSequence[float] | float | Nonerprnr=zSequence[float])rrar=ra) rrarrnrzstr | Path | Nonerrnrztuple[float, float]r=r)r~rrrnr=ra)rrrrr=r)u __future__rr7rrrRr collectionsrcollections.abcrrrrr r enumr ior pathlibr typingrrrrrpolarsrgrrpolars._dependenciesrrrrrbpolars.datatypesrrrrrrrrpolars.datatypes.groupr r!r"r#r$r%r&r'r(r)r*polars._typingr+r,rSr/typing_extensionsr0r1r3keysrrNrC__annotations__rrTrPrVr]rkrprurwryr}rrrrrrrrrrrr#r$propertyr&getenvr,r.r2r/ NoDefaultrrGrWrrYr_rorrrrrr6r;r9rs" !NN,   @ 65&7 7"!,#A AI8I w%'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:%)#/ F F F# F  F ! FFR=B8r;