iRUddlmZddlmZddlmZerddlmZddlm Z m Z m Z e e zZ de d<Gdd ZGd d Zy) ) annotations) TYPE_CHECKINGN) TypeAlias) DataFrame LazyFrameSeriesr FrameOrSeriesceZdZUdZded<ded<ded<ded<ded<d ed <dd Zedd Zedd ZddZ ddZ ddZ ddZ y) PolarsSlicez} Apply Python slice object to Polars DataFrame or Series. Has full support for negative indexing and/or stride. intstopstartstride slice_lengthbool is_unboundedr objc||_yNrselfrs U/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/_utils/slice.py__init__zPolarsSlice.__init__ cz|j}t|tjr|S|j S)z.Return lazy variant back to its original type.)collect isinstanceplr to_series)lazyoriginalframes r _as_originalzPolarsSlice._as_original s. "8R\\:uQ@QQrct|tjr|jj S|j S)z2Make lazy to ensure efficient/consistent handling.)rr rto_framer"rs r_lazifyzPolarsSlice._lazify&s1)33 (Bs||~""$R Rrc|j|j|jj|jS)z&Logic for slices with positive stride.)slicerr gather_everyrrs r_slice_positivezPolarsSlice._slice_positive+s.yyT%6%67DDT[[QQrct|j}|j|jdz|jj }|dkDr|j |S|S)z&Logic for slices with negative stride.)absrr*r rreverser+)rrr lazyslices r_slice_negativezPolarsSlice._slice_negative0sST[[!IIdii!mT->->?GGI 281*y%%f-L9Lrct|j}t|j|j|j j |\}}}|dk\r|dkxr||k\|_n|dk(xr||dz k\|_|jjrd|_ nB|jr||_ n.||k(s|dkDr||kDs |dkr||krdn t||z |_ |||c|_|_|_ y)zENormalise slice bounds, identify unbounded and/or zero-length slices.r.rN) lenrr*rr stepindicesris_emptyrr/r)rsobj_lenrr rs r _slice_setupzPolarsSlice._slice_setup6sdhh-#AGGQVVQVV<DDWMtV Q;!&! B$'/D !% G%7Q;2FD  88    !D     'D  d] ut| ut|  &  .3D&* DIt{rcH|j||jdk(r|jjS|jrQ|j dvrC|j dkr|jj S|jjS|jdk\rN|jdk\r?|j dk(r0|jj|j|jS|j dkr8|jdk(r)|jj|jdzdS|j|j}|j dkDr|j|n|j|}|j||jS)zFApply a slice operation, taking advantage of any potential fast paths.r)r4r.r.)r;rrclearrrr0clonerr r*r(r,r2r%)rr9lazyobjsliceds rapplyzPolarsSlice.applySsE !    !88>># #   4;;'#9*.++/488##% P@P P ZZ1_aDKK14D88>>$**d.?.?@ @ [[1_!2!2a!788>>$))a-3 3ll488,G;;?$$W-))'2  $$VTXX6 6rN)rr returnNone)r"rr#r rBr )rr rBr)rrrBr)r9r*rBrC)r9r*rBr ) __name__ __module__ __qualname____doc____annotations__r staticmethodr%r(r,r2r;rArrr r sr I J K RR SSR M A:7rr c,eZdZUdZded<ddZddZy) LazyPolarsSlicez Apply python slice object to Polars LazyFrame. Only slices with efficient computation paths that map directly to existing lazy methods are supported. rrc||_yrrrs rrzLazyPolarsSlice.__init__yrrc|jxsd}|jxsd}|jr|jdkr d}t||dkrG|dkDs |j6||jk7r'|dcxkDr|kDrnn |j d}t||dkDr|j||jk\s?|dkrT|jH|j<|j|jcxk\rdk\rnn|jj S|j|j|dk(r|jj S|dkDr|jj|S|dk(r|jjS|dkr|jjjt|S|dcxk\r|kDronnl|j`|jj|jdzj}t|dk(r|S|jt|S|dk(rP|jxsddk\r=|jj|j}|dk(r|S|j|S|dkrM|jA|dkDr<|jjt|}|dk(r|S|j|S|dkDrl|j|jdk\rQ|jdn|j|z }|jj||}|dk(r|S|j|Sd|d}t|) a Apply a slice operation. Note that LazyFrame is designed primarily for efficient computation and does not know its own length so, unlike DataFrame, certain slice patterns (such as those requiring negative stop/step) may not be supported. rr.z.negative stop is not supported for lazy slicesNz?negative stride is not supported in conjunction with start+stopr4zthe given slice zv is not supported by lazy computation Consider a more efficient approach, or construct explicitly with other methods.) rr6r ValueErrorrr=r>r+r0r/headtailr*)rr9rr6msgrrs rrAzLazyPolarsSlice.apply|s 1vv{ 66affqjBCS/ ! !8aff&8uA$$W o% 1H!&&,!&& 1H$);!''@VUV@V88>># #WW_qyxx~~''xx,,T22xx''))xx'')66s4yAAa $ 166>((--! ,446Ct9>3 K0@0@T0K KaZQVV[qQ.((--'C193 A3+;+;D+A AQY166>dQh((--E +C193 A3+;+;D+A AQYAFFNaffk$%FFN4!&&5.L((.. 5C193 A3+;+;D+A Aqe$b b orN)rrrBrC)r9r*rBr)rDrErFrGrHrrArJrrrLrLos N]rrL) __future__rtypingrpolars._reexport _reexportr rpolarsrrrr rHr rLrJrrrXsB"  33(61M91]7]7@jjr