.i <ddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZmZmZddlmZddlmZmZddlmZdd lmZmZmZmZmZdd lm Z er0dd l!m"Z"dd l#m$Z$dd l m Z mZddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.Gdde eefe eefZ/y)) annotationsN) TYPE_CHECKINGAnyCallableLiteralProtocol)LazyExpr) AliasNames EvalNames EvalSeries NativeExprTWindowFunction) WindowInputs)combine_alias_output_namescombine_evaluate_output_names) SQLLazyFrameT)ImplementationVersion extend_boolis_pyspark_pre_4not_implemented)InvalidOperationError)Sequence)Self)r r)SQLExprDateTimeNamesSpace)SQLExprStringNamespace) SQLNamespace)ModeKeepStrategy PythonLiteral RankMethodceZdZUded<ded<ded<ded<d ed <d ed < dmej d dndZdodZ dpdZ dqdZ drdZ dsdZ dm dtdZ dudZ dvdZdwdZedxdZdydZdzdZd{dZd|dZd}dZd}dZd~d Z dm dd!Z dd d d" dd#Z dd$Z dm dd%Zedd&Zedd'Ze dd(Z dd)Z!dd*Z"dd+Z#dd,Z$dd-Z%dd.Z&dd/Z'dd0Z(dd1Z)dd2Z*dd3Z+dd4Z,dd5Z-dd6Z.dd7Z/dd8Z0dd9Z1dd:Z2dd;Z3dd<Z4dd=Z5dd>Z6dd?Z7dd@Z8ddAZ9ddBZ:ddCZ;ddDZddGZ?ddHZ@ddIZAddJZBddKZCddLZDddMZEddNZFddOZGddPZHddQZIddRZJddSZKddTZLddUZMddVZNddWZOddXZPddYZQddZZRdd[ZSdd\ZTdd]ZU dd^ZV dd_ZWdd`ZXddaZYddbZZddcZ[ddddZ\dddeZ]ddfZ^ddgZ_ddhZ` ddiZaddjZbeddkZceddlZdeeZfeeZgeeZhy )SQLExpr&EvalSeries[SQLLazyFrameT, NativeExprT]_callEvalNames[SQLLazyFrameT]_evaluate_output_namesAliasNames | None_alias_output_namesr_versionr_implementation1WindowFunction[SQLLazyFrameT, NativeExprT] | None_window_functionN)implementationcyN)selfcallwindow_functionevaluate_output_namesalias_output_namesversionr-s O/opt/house_gps_flutter/.venv/lib/python3.12/site-packages/narwhals/_sql/expr.py__init__zSQLExpr.__init__1sc$|j|Sr/)r$)r1dfs r7__call__zSQLExpr.__call__<szz"~r9cyr/r0r1s r7__narwhals_namespace__zSQLExpr.__narwhals_namespace__?s>Ar9c dfd }|S)Nc|}jDcic]\}}||j|}}}|Dcgc] }|fi| c}Scc}}wcc}wr/)items_evaluate_single_output_expr) r;native_series_listkeyvalueother_native_series native_seriesr2expressifiable_argsr1s r7funcz.SQLExpr._callable_to_eval_series..funcFs|!%b #6";";"=#CR44U;;# # &8!]:&9:  #s AA)r;rreturnzlist[NativeExprT]r0)r1r2rIrJs``` r7_callable_to_eval_seriesz SQLExpr._callable_to_eval_seriesCs  r9c ( dfd }|S)Nc  j||}jDcic]\}}||j||}}}|Dcgc] }|fi| c}Scc}}wcc}wr/)r3rB_evaluate_window_expr) r; window_inputsrDrErFrGrHr2rIr1s r7window_fz4SQLExpr._push_down_window_function..window_fVs"&!5!5b-!H #6";";"=#CR--e]CC# # &8!]:&9:  #s A A$r;rrPWindowInputs[NativeExprT]rKSequence[NativeExprT]r0)r1r2rIrQs``` r7_push_down_window_functionz"SQLExpr._push_down_window_functionSs'  .G  " $r9c|j|j||j|j|j|j SNr4r5r6r-) __class__r$r&r(r)r*)r1r3s r7_with_window_functionzSQLExpr._with_window_functionjsF~~ JJ "&"="=#77MM//   r9c |j|j|fi|||j|j|j|j SrW)rYrLr&r(r)r*)r1r2 window_funcrIs r7_with_callablezSQLExpr._with_callablevsU~~ )D ) )$ F2E F "&"="=#77MM//   r9c |j|j|fi||j|fi||j|j|j |j SrWrYrLrUr&r(r)r*)r1r2rIs r7_with_elementwisezSQLExpr._with_elementwisesh~~ )D ) )$ F2E F +D + +D H4G H"&"="=#77MM//   r9c|j|j|||j|||j|j|j |j S)NotherrXr_)r1oprcs r7 _with_binaryzSQLExpr._with_binarys`~~  ) )"E ) :  + +Be + <"&"="=#77MM//   r9c|jdn nfd}t||j|j|j||j |j S)Nc |Sr/r0) output_namescurrent_alias_output_namesrJs r7z2SQLExpr._with_alias_output_names..sd+El+S&Tr9rX)r(typer$r,r&r)r*)r1rJr5ris ` @r7_with_alias_output_namesz SQLExpr._with_alias_output_namessq%)%=%="| *1T tDz JJ  ! !"&"="=1MM//   r9c< dfd }jxs|S)Nc|jrJ|Dcgc]}j||j c}Scc}wr/)order_by_window_expression partition_byr;inputsexprr1s r7default_window_funcz4SQLExpr.window_function..default_window_funcsG &&OSTVxGK''f.A.AB s#Ar;rrsrSrKrT)r,)r1rus` r7r3zSQLExpr.window_functions4  '@  " $$;(;;r9cD|jj|g|Sr/)r? _function)r1nameargss r7rxzSQLExpr._functions#6t**,66tCdCCr9c@|jj|Sr/)r?_lit)r1rFs r7r|z SQLExpr._lits**,11%88r9c<|jj|Sr/)r? _coalesce)r1rts r7r~zSQLExpr._coalesces6t**,66==r9cyr/r0r>s r7 _count_starzSQLExpr._count_starr9cyr/r0r1rtros r7_firstzSQLExpr._firstrr9cyr/r0rs r7_lastz SQLExpr._lastrr9cyr/r0)r1rt ignore_nullss r7 _any_valuezSQLExpr._any_valuerr9cD|jj|||Sr/)r?_when)r1 conditionrF otherwises r7rz SQLExpr._whens" **,229eYOOr9 descending nulls_lastcyr/r0)r1rtrqro rows_startrows_endrrs r7rpzSQLExpr._window_expressionsr9c( dfd }|S)Nc,tt|j}|Dcgc]^}jj d|j j ||j |j||d`c}Scc}w)Nisnullrrrr)rlenrorrxrprq)r;rsflagsrt func_namereverser1s r7rJz&SQLExpr._cum_window_func..funcs V__)=>E!H  ^^Hd33++y$7++#(#(!" ,    sA#Brvr0)r1rrrJs``` r7_cum_window_funczSQLExpr._cum_window_funcs'  '@  " & r9c gd |r|dz dz}|dz dz}||z | n|dz d d  fd }|S)N)summeanstdvarrc dvr}n\dk(rdk(rd}nOdvrdk(rd}nCdk(rdk(rd}n6dk(rdk(rd }n)d vrd d }t|d dd }t||j|j d} |Dcgc]h} j j j d|fi| j k\ j j ||fi|jc}Scc}w)N>rrrrvar_poprvar_sampr stddev_pop stddev_samp>rrz;Only ddof=0 and ddof=1 are currently supported for rolling_.z,Only the following functions are supported: z. Got: )rqrorrcount) ValueErrorrqrorrprxr|) r;rsfunc_msg window_kwargsrtddofendr min_samplesr1startsupported_funcss r7rJz*SQLExpr._rolling_window_func..funcsMO+&e# !e# "e# $e# %n,ST]S^^_` o%D_DUU]^g]hhij o% & 3 3"OO# "M!H  +D++w59Fyy-.,D++DNN5$,GY=Y    s A-C:rvr0) r1r window_sizerrcenterhalf remainderrJrrrs `` `` @@@r7_rolling_window_funczSQLExpr._rolling_window_funcsv8 !O)D$qA-IY&'EC!Ao&EC" " '@"  "" " H r9c6|jjSr/)r*_backend_versionr>s r7rzSQLExpr._backend_version,s##4466r9cyr/r0)clsrtrys r7 _alias_nativezSQLExpr._alias_native0sLOr9cdfd } dfd }d}|||tt|j|jS)Nc \Dcgc]}||D]}|c}}gScc}}wr/r0)r;rteexprsrJs r7r2z5SQLExpr._from_elementwise_horizontal_op..call7s/@tBx@!!@!@AB B@s( crDcgc]}|j||D]}|}}}|gScc}}wr/)r3)r;rPrtrlstrrJs r7r3z@SQLExpr._from_elementwise_horizontal_op..window_function:sC"'X0D0DR0WX11X1XCXI; Ys 3r)r3r4r5r6r-r;rrKrTrR)rrr)r*)rrJrr2r3contexts `` r7_from_elementwise_horizontal_opz'SQLExpr._from_elementwise_horizontal_op3sf C  .G  "  ( +"?"G95A$$"22   r9cJ|jjjS)a Return `True` for multi-output aggregations without names. For example, column `'a'` only appears in the output as a grouping key: df.group_by('a').agg(nw.all().sum()) It does not get included in: nw.all().sum(). ) _metadataexpansion_kindis_multi_unnamedr>s r7_is_multi_output_unnamedz SQLExpr._is_multi_output_unnamedJs~~,,==??r9c(|jd|S)Nc$|j|Sr/)__eq__rtrcs r7rjz SQLExpr.__eq__..YT[[5Gr9rer1rcs r7rzSQLExpr.__eq__X  !GOOr9c(|jd|S)Nc$|j|Sr/)__ne__rs r7rjz SQLExpr.__ne__..\rr9rrs r7rzSQLExpr.__ne__[rr9c(|jd|S)Nc$|j|Sr/)__add__rs r7rjz!SQLExpr.__add__.._T\\%5Hr9rrs r7rzSQLExpr.__add__^  !H%PPr9c(|jd|S)Nc$|j|Sr/)__sub__rs r7rjz!SQLExpr.__sub__..brr9rrs r7rzSQLExpr.__sub__arr9cF|jd|jdS)Nc ||z Sr/r0rs r7rjz"SQLExpr.__rsub__..e UT\r9literalrealiasrs r7__rsub__zSQLExpr.__rsub__d"  !A5IOOPYZZr9c(|jd|S)Nc$|j|Sr/)__mul__rs r7rjz!SQLExpr.__mul__..hrr9rrs r7rzSQLExpr.__mul__grr9c(|jd|S)Nc$|j|Sr/) __truediv__rs r7rjz%SQLExpr.__truediv__..ksT5E5Ee5Lr9rrs r7rzSQLExpr.__truediv__js  !LeTTr9cF|jd|jdS)Nc ||z Sr/r0rs r7rjz&SQLExpr.__rtruediv__..nrr9rrrs r7 __rtruediv__zSQLExpr.__rtruediv__mrr9c(|jd|S)Nc$|j|Sr/)__pow__rs r7rjz!SQLExpr.__pow__..qrr9rrs r7rzSQLExpr.__pow__prr9cF|jd|jdS)Nc ||zSr/r0rs r7rjz"SQLExpr.__rpow__..ts UD[r9rrrs r7__rpow__zSQLExpr.__rpow__ss!  !@%HNNyYYr9c(|jd|S)Nc$|j|Sr/)__mod__rs r7rjz!SQLExpr.__mod__..wrr9rrs r7rzSQLExpr.__mod__vrr9cF|jd|jdS)Nc ||zSr/r0rs r7rjz"SQLExpr.__rmod__..zrr9rrrs r7__rmod__zSQLExpr.__rmod__yrr9c(|jd|S)Nc$|j|Sr/)__ge__rs r7rjz SQLExpr.__ge__..}rr9rrs r7rzSQLExpr.__ge__|rr9c(|jd|S)Nc$|j|Sr/)__gt__rs r7rjz SQLExpr.__gt__..rr9rrs r7rzSQLExpr.__gt__rr9c(|jd|S)Nc$|j|Sr/)__le__rs r7rjz SQLExpr.__le__..rr9rrs r7rzSQLExpr.__le__rr9c(|jd|S)Nc$|j|Sr/)__lt__rs r7rjz SQLExpr.__lt__..rr9rrs r7rzSQLExpr.__lt__rr9c(|jd|S)Nc$|j|Sr/)__and__rs r7rjz!SQLExpr.__and__..rr9rrs r7r zSQLExpr.__and__rr9c(|jd|S)Nc$|j|Sr/)__or__rs r7rjz SQLExpr.__or__..rr9rrs r7r zSQLExpr.__or__rr9c6dfd }j||S)Ncj|jdk7tj||jdSNrrr|rdfloordivrtrcr1s r7rJz"SQLExpr.__floordiv__..funcs;::1%r{{4'?4 r9rbrtr rcr rKr rr1rcrJs` r7 __floordiv__zSQLExpr.__floordiv__s     U 33r9cTdfd }j||jdS)Ncj|jdk7tj||jdSrrrs r7rJz#SQLExpr.__rfloordiv__..funcs;:: ! $bkk%&> $ r9rbrrrrs` r7 __rfloordiv__zSQLExpr.__rfloordiv__s+    U 399)DDr9cLdfd } dfd }j||S)Ncfjjd|jdSNbool_andTr~rxr|rtr1s r7fzSQLExpr.all..fs'>>$..T"BDIIdOT Tr9c |Dcgc]M}jjjd||jj dOc}Scc}wrr~rprxrqr|rrs r7rQzSQLExpr.all..window_fse!H ++z48&:M:MIIdO  AA!rtr rKr rvr]r1r rQs` r7allz SQLExpr.all; U  '@  " ""1h//r9cLdfd } dfd }j||S)Ncfjjd|jdSNbool_orFrrs r7r zSQLExpr.any..fs(>>$..D"A499UCST Tr9c |Dcgc]M}jjjd||jj dOc}Scc}wr+r"rrs r7rQzSQLExpr.any..window_fsf!H ++y$79L9LIIe$  r#r$rvr%r&s` r7anyz SQLExpr.anyr(r9c,jfdS)Nc(jd|S)Nmaxrxrs r7rjzSQLExpr.max..ud0Kr9r%r>s`r7r1z SQLExpr.max""#KLLr9c,jfdS)Nc(jd|S)Nrr2rs r7rjzSQLExpr.mean..vt0Lr9r%r>s`r7rz SQLExpr.means""#LMMr9c6dfd }j|S)Nc Fdk(rjd|Sdk(rjd|Sjd|}tjjd|jdtj|dtj|z S)Nrrrrrsqrtrxrdmulsubrt n_samplesrr1s r7rJzSQLExpr.std..funcsqy~~lD99qy~~mT::w5I66}d3vy!(<yRV@W'WX r9r$r%r1rrJs`` r7rz SQLExpr.std ""4((r9c6dfd }j|S)Nc&dk(rjd|Sdk(rjd|Sjd|}tjjd|tj|dtj|z S)Nrrrrrr;r>s r7rJzSQLExpr.var..funcsqy~~i66qy~~j$77w5I66z40y!$rvvi'>> r9r$r%r@s`` r7rz SQLExpr.varrAr9c,jfdS)Nc(jd|S)Nmedianr2rs r7rjz SQLExpr.median..sx0Nr9r%r>s`r7rFzSQLExpr.medians""#NOOr9c6dfd }j|S)Nchjjd|j|S)Nisnan)rrxr|)rtr1rFs r7 _fill_nanz#SQLExpr.fill_nan.._fill_nans*::dnnWd;TYYu=MtT Tr9r$r`)r1rFrJs`` r7fill_nanzSQLExpr.fill_nans U%%i00r9c,jfdS)Nc(jd|S)Nminr2rs r7rjzSQLExpr.min..r3r9r%r>s`r7rOz SQLExpr.minr4r9c,jfdS)Nc(jd|S)Nrr2rs r7rjzSQLExpr.count..sw0Mr9r%r>s`r7rz SQLExpr.counts""#MNNr9cLdfd } dfd }j||S)Ncfjjd|jdSNrrrrs r7r zSQLExpr.sum..fs'>>$.."=tyy|L Lr9c |jrJ|Dcgc]M}jjjd||jj dOc}Scc}wrT)ror~rprxrqr|rrs r7rQzSQLExpr.sum..window_fst &&!H ++ud3V5H5HIIaL  sAA/r$rvr%r&s` r7rz SQLExpr.sums; M  '@  " ""1h//r9cjjjdjdcdfd } dfd }j||S)Nrrc |tjd|djd|SNcount_distinctr1r)rdaddr)rtFoner1zeros r7rJzSQLExpr.n_unique..funcs=66"D)%Ah$5sDAB r9c|Dcgc]`}tjd||jdjd||jbc}Scc}wrX)rdrZrqr)r;rsrtr[Wr\r1r]s r7rQz"SQLExpr.n_unique..window_fs{!H a($/1D1DE%Ah,=sD!IJ++   sA%A4r$rv)rxrpr|r])r1rJrQr[r_r\r]s` @@@@r7n_uniquezSQLExpr.n_unique ss NN  # #IIaL$))A, c    '@  "  ""422r9c,jfdS)Nc(jd|S)Nabsr2rs r7rjzSQLExpr.abs..'4>>%3Nr9rKr>s`r7rcz SQLExpr.abs&%%&NOOr9cH dfd }j|||S)NcLjdjd|||S)Ngreatestleastr2)rt lower_bound upper_boundr1s r7_clipzSQLExpr.clip.._clip*s*>>DNN7D+F  r9)rjrk)rtr rjr rkr rKr rK)r1rjrkrls` r7clipz SQLExpr.clip)sI  ,7 FQ   %% { &  r9c6dfd }j||S)Nc*jd||S)Nrhr2)rtrjr1s r7rlz!SQLExpr.clip_lower.._clip6s>>*dK@ @r9)rj)rtr rjr rKr rK)r1rjrls` r7 clip_lowerzSQLExpr.clip_lower5s! A%%e%EEr9c6dfd }j||S)Nc*jd||S)Nrir2)rtrkr1s r7rlz!SQLExpr.clip_upper.._clip<s>>'4= =r9)rk)rtr rkr rKr rK)r1rkrls` r7 clip_upperzSQLExpr.clip_upper;s  >%%e%EEr9c,jfdS)Nc(jd|S)Nrr2rs r7rjz!SQLExpr.is_null..Bs4>>(D3Qr9rKr>s`r7is_nullzSQLExpr.is_nullAs%%&QRRr9ctj}|r|nj|jfdS)Nc*jd|S)Nroundr2)rtround_decimalsr1s r7rjzSQLExpr.round..Js~Fr9)rr*r|r`)r1decimals_is_pre4rzs` @r7ryz SQLExpr.roundDs=$D$8$89%-499X3F%% F  r9c,jfdS)Nc(jd|S)Nfloorr2rs r7rjzSQLExpr.floor..Ns4>>'43Pr9rKr>s`r7rz SQLExpr.floorMs%%&PQQr9c,jfdS)Nc(jd|S)Nceilr2rs r7rjzSQLExpr.ceil..Qs4>>&$3Or9rKr>s`r7rz SQLExpr.ceilPs%%&OPPr9c,jfdS)Nc(jd|S)Nsinr2rs r7rjzSQLExpr.sin..Trdr9rKr>s`r7rz SQLExpr.sinSrer9c,jfdS)Nc(jd|S)Ncosr2rs r7rjzSQLExpr.cos..Wrdr9rKr>s`r7rz SQLExpr.cosVrer9c2dfd }j|S)Ncj|jdkjtdjd|S)Nrnanr:)rr|floatrxrs r7_sqrtzSQLExpr.sqrt.._sqrtZsA::tyy|#TYYuU|%Z r9r$rK)r1rs` r7r:z SQLExpr.sqrtYs  %%e,,r9c,jfdS)Nc(jd|S)Nexpr2rs r7rjzSQLExpr.exp..brdr9rKr>s`r7rz SQLExpr.exparer9c6dfd }j|S)Ncnj}j|jdkjtdj|jdk(jtdt j |d||djS)Nrrz-inflog)rxrr|rrdtruediv)rtr[baser1s r7_logzSQLExpr.log.._logesA::tyy|# %,' DIIaL(IIeFm,JJq~q $/HI r9r$rK)r1rrs`` r7rz SQLExpr.logds %%d++r9cF|j|jd|S)NrrrZrr1rs r7cum_sumzSQLExpr.cum_sumt#))$*?*?w*?*WXXr9cF|j|jd|S)Nr1rrrs r7cum_maxzSQLExpr.cum_maxwrr9cF|j|jd|S)NrOrrrs r7cum_minzSQLExpr.cum_minzrr9cB dfd }j|S)Nc tt|j}|Dcgc]=}jj d||j |j||d?c}Scc}w)Nrrr)rrrorprxrq)r;rsrrtrr1s r7rJzSQLExpr.cum_count..func~sy V__)=>E!H ''NN7D1''OO$$ (   sAA0rvrZ)r1rrJs`` r7 cum_countzSQLExpr.cum_count}s3  '@  "  ))$//r9cF|j|jd|S)Nproductrrrs r7cum_prodzSQLExpr.cum_prods())  ! !)W ! =  r9cJ|j|jd|||S)NrrrZrr1rrrs r7 rolling_sumzSQLExpr.rolling_sums,))  % %e[+f % U  r9cJ|j|jd|||S)Nrrrrs r7 rolling_meanzSQLExpr.rolling_means,))  % %fk;v % V  r9c L|j|jd||||S)Nrrrrr1rrrrs r7 rolling_varzSQLExpr.rolling_var5))  % %{Kd6 &   r9c L|j|jd||||S)Nrrrrs r7 rolling_stdzSQLExpr.rolling_stdrr9c> dfd }j|S)Nc j}j}|Dcgc];}tj|||d||j|j =c}Scc}wNlag)rxrprdr=rqro)r;rsr[windowrtr1s r7rJzSQLExpr.diff..funcscA,,F!HtVAeTNF4G4GYZ sAA'rvrr1rJs` r7diffz SQLExpr.diffs3  '@  " ))$//r9cB dfd }j|S)Nc |Dcgc]:}jjd||j|j<c}Scc}wr)rprxrqro)r;rsrtnr1s r7rJzSQLExpr.shift..funcsT!H ''NN5$2F4G4G s?Arvr)r1rrJs`` r7shiftz SQLExpr.shifts3  '@  " ))$//r9c> dfd }j|S)Nc|Dcgc]O}jjdg|j||jj dk(Qc}Scc}w)N row_numberr)rprxrqror|rrs r7rJz'SQLExpr.is_first_distinct..funcsn!H ''NN<00f))040OO 99Q<   sAA#rvrrs` r7is_first_distinctzSQLExpr.is_first_distincts3  '@  " ))$//r9c> dfd }j|S)Nc tdt|j}|Dcgc]R}jj dg|j ||j||j dk(Tc}Scc}w)NTrrr)rrrorprxrqr|)r;rsrrtr1s r7rJz&SQLExpr.is_last_distinct..funcs c&//&:;E!H ''NN<00f))040OO$$ (99Q<    sABrvrrs` r7is_last_distinctzSQLExpr.is_last_distincts3  '@  "  ))$//r9cRdfd } dfd }j||S)NcHs d}t|j|gSNz#Expected `order_by` to be specified)rrrtrror1s r7r zSQLExpr.first..fs,;+C004;;t/h/ /r9c r|jr d}t|s|js d}t||Dcgc]=}jj|g|jxs|j?c}Scc}w)Nz4Can't specify both `order_by` in `over` and `first`.z4Must specify `order_by` either in `over` or `first`.)rorrprrqr;rsrrtror1s r7rQzSQLExpr.first..window_fsFOOL+C00FOOL+C00 !H ''DKKE(C8EvGZGZ ABr$rvr%r1ror rQs`` r7firstz SQLExpr.firsts: 0   '@  "  ""1h//r9cRdfd } dfd }j||S)NcHs d}t|j|gSr)rrrs r7r zSQLExpr.last..f s,;+C004::d.X. .r9c r|jr d}t|s|js d}t||Dcgc]=}jj|g|jxs|j?c}Scc}w)Nz3Can't specify both `order_by` in `over` and `last`.z3Must specify `order_by` either in `over` or `last`.)rorrprrqrs r7rQzSQLExpr.last..window_fsFOOK+C00FOOK+C00 !H ''DJJtDv'B(DfFYFY rr$rvr%rs`` r7lastz SQLExpr.last s: /   '@  "  ""1h//r9c0jfdS)Nc*j|S)N)r)r)rtrr1s r7rjz#SQLExpr.any_value..%sLIr9r%)r1rs``r7 any_valuezSQLExpr.any_value#s"" I  r9cdvrjdn(dk(rjdnjd d d fd d fd } d fd }j||S) N>r1rOaveragerankdense dense_rankrc : j}||f|gdgd}dg||i} j} j} dk(rHtjtj | fi|||fi| j d}ny dk(rktj | fi|tjtj||fi| j d j d}n | fi|} j|d||S) NT)rqrorrrqr1rrg@r) rrprxrdr=rZr|rr) rtrqr count_exprrcount_window_kwargsrr[ rank_exprrJmethodr1s r7_rankzSQLExpr.rank.._rank0s- ))+J ,!G)l#f -M 4BCX\CXSWCX2Y ,,FAFFFFt5}5zA-@AIIaL  9$FF41=1JJvjH4GH$))TU,W # #49=9 ::q400)< ._unpartitioned_rankTs*5 5r9c|jr d}t||Dcgc]}||jc}Scc}w)Nz[`rank` followed by `over` with `order_by` specified is not supported for SQL-like backends.r)roNotImplementedErrorrq)r;rsrrtrrr1s r7_partitioned_rankz'SQLExpr.rank.._partitioned_rankWsQs)#..!HdF//JG sAr0)rtr rqSequence[str | NativeExprT]rboolrKr r$rv)rxr])r1rrrrrrJs``` @@r7rz SQLExpr.rank(s . .>>&)D w >>,/D>>,/D9;" =" =5" = " =  " =H 6  '@  " ""#68IJJr9c dfd dfd } dfd }j|j|S)Ncpjj|g|jdk(S)Nr)rprr|)rtrqr1s r7 _is_uniquez%SQLExpr.is_unique.._is_uniquees>**  "T$9L$91 r9c|Sr/r0)rtrs r7_unpartitioned_is_uniquez3SQLExpr.is_unique.._unpartitioned_is_uniquelsd# #r9cx|jrJ|Dcgc]}|g|jc}Scc}wr/)rorq)r;rsrtrr1s r7_partitioned_is_uniquez1SQLExpr.is_unique.._partitioned_is_uniqueos: &&GKBxPtJt:f&9&9:P PPs7)rtr rqzstr | NativeExprTrKr r$rv)r]rZ)r1rrrs` @r7 is_uniquezSQLExpr.is_uniquedsj  .?    $ Q Q'@ Q " Q ""#;<RR "  r9cdfd }j|jjjjS)Nc<j|tSr/)r3r)r;rorqr1s r7rJzSQLExpr.over..func}s''Lx,PQ Qr9rXr)rYr&r(r)r*)r1rqrorJs``` r7overz SQLExpr.overzsF R~~ "&"="=#77MM//   r9cr|dk7rd|djd}t|jfdS)Nr.z`Expr.mode(keep='z#')` is not implemented for backend z3 Hint: Use `nw.col(...).mode(keep='any')` instead.c(jd|S)Nmoder2rs r7rjzSQLExpr.mode..r7r9)r*rr])r1keeprs` r7rz SQLExpr.modesP 5=#D6)LTMaMaLbcDD &c* *""#LMMr9cyr/r0r>s r7strz SQLExpr.strs36r9cyr/r0r>s r7dtz SQLExpr.dts58r9r/)r2r#r3r+r4r%r5r'r6rr-rrKNoner)rKz3SQLNamespace[SQLLazyFrameT, Self, Any, NativeExprT])r2Callable[..., NativeExprT]rIrrKr#)r2rrIrrK*WindowFunction[SQLLazyFrameT, NativeExprT])r3rrKr)r2rr\r+rIrrKr)r2rrIrrKr)rdrrcrrKr)rJr'rKr)rKr)ryrrzzNativeExprT | PythonLiteralrKr )rFrrKr r$)rKr )rtr rorrKr )rtr rrrKr )rr rFr rzNativeExprT | NonerKr )r0r0NN)rtr rqrrorr int | NonerrrSequence[bool] | NonerrrKr )rz'Literal['sum', 'max', 'min', 'product']rrrKr) rz$Literal['sum', 'mean', 'std', 'var']rintrrrrrrrKr)rKztuple[int, ...])rtr ryrrKr )rJz*Callable[[list[NativeExprT]], NativeExprT]rrrKr)rKr)rcrrKr)rKr)rrrKr)rFz float | NonerKr)rjrrkrrKr)rjrrKr)rkrrKr)r{rrKr)rrrKr)rrrKr)rrrrrrrKr) rrrrrrrrrKr)rrrKrr)ro Sequence[str]rKr)rrrKr)rr rrrKr)rqrrorrKr)rrrKr)rKzSQLExprStringNamespace[Self])rKzSQLExprDateTimeNamesSpace[Self])i__name__ __module__ __qualname____annotations__rDUCKDBr8r<r?rLrUrZr]r`rerlpropertyr3rxr|r~rrrrrrprrr classmethodrrrrrrrrrrrrrrrrrrrr r rrr'r.r1rrrrFrLrOrrr`rcrmrprsrvryrrrrr:rrrrrrrrrrrrrrrrrrrrrrrrr drop_nullsfilteruniquer0r9r7r"r")s 1144**##GG NR *8)>)> 4 K  8  .  '   B <B.JN / .JN 3.  I    JN ( G $     .  JN      $ < <D9>.KJV )- PPP& P  P5702!%# -1,0  2 .     * *   @NR 3: 7777 7  77 47r77OO = GK   , @PPQQ[QU[QZQ[PPPPQP4E0&0&MN ) )P1 MO0(38P  F F S RQPP-P , YYY0&     03 =A IL    03 =A IL   0 00"0&0202 :Kx ,  7  CP    N66 88 "J  F  Fr9r")0 __future__roperatorrdtypingrrrrrnarwhals._compliant.exprr narwhals._compliant.typingr r r r rnarwhals._compliant.windowrnarwhals._expression_parsingrrnarwhals._sql.typingrnarwhals._utilsrrrrrnarwhals.exceptionsrcollections.abcrtyping_extensionsrnarwhals._sql.expr_dtrnarwhals._sql.expr_strrnarwhals._sql.namespacernarwhals.typingrrr r"r0r9r7r s{"BB-4/6(&E?=4KKr h}k12H]K=W4Xr r9