-Ph$UddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZmZddlmZmZmZmZerAddlmZmZdd lmZmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*m+Z+m,Z,e dZ-dDdZ.dEdZ/dFdZ0dGdZ1dHd$Z2dId+Z3Gd,d-eZ4dJd/Z5Gd0d1eZ6Gd2d3Z7dKd8Z8dLd=Z9dMd?Z:dNdBZ;dCS)O) annotations)Enumauto)chain) TYPE_CHECKINGAnyLiteralSequenceTypeVarcast)is_compliant_expr)is_narwhals_seriesis_numpy_array)InvalidOperationErrorLengthChangingExprErrorMultiOutputExpressionError ShapeError)NeverTypeIs) CompliantExprCompliantFrameT) AliasNamesCompliantExprAnyCompliantFrameAnyCompliantNamespaceAnyEagerNamespaceAny EvalNamesExprSeries)IntoExprNonNestedLiteral_1DArrayTobjrreturn TypeIs[Expr]c.ddlm}t||S)'Check whether `obj` is a Narwhals Expr.rr) narwhals.exprr isinstance)r&rs \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/narwhals/_expression_parsing.pyis_exprr.&s$"""""" c4  TypeIs[Series[Any]]c.ddlm}t||S)r*rr )narwhals.seriesr!r,)r&r!s r- is_seriesr3-s$&&&&&& c6 " ""r/exprs#CompliantExpr[CompliantFrameT, Any]EvalNames[CompliantFrameT]ctds(dtdd}t|d fd }|S) Nrz3Safety assertion failed, expected expression, got: z. Please report a bug.dfrr' Sequence[str]cJd|ddSNr)_evaluate_output_names)r8r4s r-evaluate_output_namesz.evaluate_output_names=s$Qx..r222A266r/)r8rr'r9)r typeAssertionError)r4msgr>s` r-combine_evaluate_output_namesrB4sg U1X & &"jDqNNjjjS!!!777777 ! r/rAliasNames | Nonec2djdSdfd }|S)Nrnamesr9r'cJd|ddSr;_alias_output_names)rEr4s r-alias_output_namesz6combine_alias_output_names..alias_output_namesIs$Qx++E222A266r/)rEr9r'r9rG)r4rIs` r-combine_alias_output_namesrJCs= Qx#+t777777 r/plxrother&IntoExpr | NonNestedLiteral | _1DArray str_as_litbool#CompliantExprAny | NonNestedLiteralct|r||St|tr|s||St |r|jSt|r>td|}|j ||S|S)Nr)context) r._to_compliant_exprr,strcolr_compliant_series_to_exprrr _series from_numpy)rKrLrNnss r-extract_compliantr[Os u~~-'',,,%jwwu~~%  2&//111eC %s + +z$$UB$77@@BBB Lr/exprr8rexcluder9#tuple[Sequence[str], Sequence[str]]c||}|j|n||}rM|jJ|jjr&t fdt ||D\}}||fS)Nc&g|] \}}|v ||fSra).0xaliasr]s r- z5evaluate_output_names_and_aliases..ns6 5''J'''r/)r=rH _metadataexpansion_kindis_multi_unnamedzip)r\r8r] output_namesaliasess ` r-!evaluate_output_names_and_aliasesrlas..r22L  # +   % %l 3 3  ~))) > ( 9 9 ; ; $'$' g$>$>% !L'   r/cdeZdZdZeZ eZ eZ eZ eZ eZ eZ eZ eZ eZ eZ ed dZed dZeddZedd Zd S)ExprKindz6Describe which kind of expression we are dealing with.r'rOc6|tjtjhvSN)rnLITERAL AGGREGATIONselfs r-is_scalar_likezExprKind.is_scalar_likes((*>???r/c6|tjtjhvSrp)rnORDERABLE_WINDOWORDERABLE_AGGREGATIONrss r-is_orderable_windowzExprKind.is_orderable_windows183QRRRr/r&rc|j}|jr tjS|jr tjS|jr tjStjSrp) rf is_literalrnrqrurris_elementwise ELEMENTWISEUNKNOWN)clsr&metas r- from_exprzExprKind.from_exprsM} ? $# #   (' '   (' 'r/rMrNct|r||St|s&t|st |t r|s t jSt jSrp) r.rrrr,rTrnr}rq)rr&rNs r-from_into_exprzExprKind.from_into_exprsv 3<< &==%% % s # # (c"" (3$$ (.8 ( ' 'r/Nr'rO)r&rr'rn)r&rMrNrOr'rn)__name__ __module__ __qualname____doc__rrqrrrxr}rwUNORDERABLE_WINDOW FILTRATIONORDERABLE_FILTRATIONNARYOVERr~propertyrury classmethodrrrar/r-rnrnws>@@dffG$&&KY DFFX$&&KgtvvHLJG466= 466D? 466D5dffGL @@@X@SSSXS   [     [    r/rn7TypeIs[Literal[ExprKind.LITERAL, ExprKind.AGGREGATION]]c|jSrpru)r&s r-rurus   r/cleZdZdZeZ eZ eZ d dZd dZ d dZ d S) ExpansionKindz8Describe what kind of expansion the expression performs.r'rOc|tjuSrp)r MULTI_UNNAMEDrss r-rhzExpansionKind.is_multi_unnameds}222r/c6|tjtjhvSrp)r MULTI_NAMEDrrss r-is_multi_outputzExpansionKind.is_multi_outputs 1=3NOOOr/rL$Literal[ExpansionKind.MULTI_UNNAMED]c|tjur|tjur tjSd|d|d}t|)Nz+Unsupported ExpansionKind combination, got z and z, please report a bug.)rrr@)rtrLrAs r-__and__zExpansionKind.__and__sL =. . .5M - ! q O   M+!      r/c |jrd}t||js|jrd}t|t |jt jd|jddddS)NrrTFr) rrr|rrrgrnrrrs r-with_partitioned_overz"ExprMetadata.with_partitioned_over|s   -2C',, ,   -$"4 -K (,, ,   M 0!      r/c |jrd}t|t|jtj|j|jddddS)NECan't apply filtration (e.g. `drop_nulls`) to scalar-like expression.Fr)rurrrgrnrrrrs r-with_filtrationzExprMetadata.with_filtrations[   -YC',, ,    ( 0"      r/c |jrd}t|t|jtj|j|jdzddddS)Nrr<Fr)rurrrgrnrrrrs r-with_orderable_filtrationz&ExprMetadata.with_orderable_filtrations`   -YC',, ,    )( 014"      r/cRttjtjdddS)NFT)r|rru)rrrrnrrrar/r- aggregationzExprMetadata.aggregations.    "     r/cTttjtjddddS)NFT)r|rr{ru)rrrrnrqrar/r-literalzExprMetadata.literals1    "     r/cJttjtjSrp)rrrrnr}rar/r-selector_singlezExprMetadata.selector_singlesM0(2FGGGr/cJttjtjSrp)rrrrnr}rar/r-selector_multi_namedz!ExprMetadata.selector_multi_namedsM5x7KLLLr/cJttjtjSrp)rrrrnr}rar/r-selector_multi_unnamedz#ExprMetadata.selector_multi_unnamedsM79MNNNr/lhsrrhsr"c*t||dddS)NTFrNallow_multi_outputto_single_outputcombine_metadata)rrrs r-from_binary_opzExprMetadata.from_binary_ops' %RW    r/r4ct|ddddS)NFTrr)rr4s r-from_horizontal_opzExprMetadata.from_horizontal_ops# uPT    r/N)rgrrrnrrOrrrrOr|rOrurOr{rOr'r)rrrrr'r)r'rTr)r'r)rrrr"r'r)r4r"r'r)rrr __slots__rrrrrrrrrrrrrr staticmethodrrrrrrrrrar/r-rrs9 I " !%#$ ++++++2     EEEXE                             6    *             \    \ HHH\HMMM\MOOO\O   [    [   r/rrIntoExpr | object | Nonerrc d}tj}d}d}d}d} d} d} t|D]\} } t| tr|rt | rd}d} d} 2t | r| j}|j r0|j}| dkr|sd}t||s| dkr|}n||z}|j rd}||j z }|j rd}|jsd} |jsd} |jsd} |jr|dz }|dkrd}t%||r|rd}t'|t)|t*j|||| | | S) aCombine metadata from `args`. Arguments: args: Arguments, maybe expressions, literals, or Series. str_as_lit: Whether to interpret strings as literals or as column names. allow_multi_output: Whether to allow multi-output inputs. to_single_output: Whether the result is always single-output, regardless of the inputs (e.g. `nw.sum_horizontal`). rFTz]Multi-output expressions (e.g. nw.col('a', 'b'), nw.all()) are not supported in this context.r<zXLength-changing expressions can only be used in isolation, or followed by an aggregationzVCannot combine length-changing expressions with length-preserving ones or aggregationsr)rr enumerater,rTr3r.rfrgrrrrrr|rur{rrrrrnr)rNrrr n_filtrationsresult_expansion_kindresult_has_windowsresult_n_orderable_opsresult_preserves_lengthresult_is_not_elementwiseresult_is_not_scalar_likeresult_is_not_literaliargmetadatargrAs r-rrsM)0# % %!D//"#"#3 sC ! #! # #! #&* #(, %$( ! ! S\\ #}H&6688 W!)!8q55!35=5S999'WAvv0>--0E0V-# *%)" "h&> > "( /*.'* 1,0)* 1,0)& -(,%% #" qh%c***=foo  &.04444,,    r/r" function_namerTrctddlmtfd|Dsd|d}t|dS)Nrr c3|K|]6}t|r |jjpt|tfV7dSrp)r.rfrr,rT)rbrcr!s r- z4check_expressions_preserve_length..;sZ   4 4UAV}9U9Ur/zBExpressions which aggregate or change length cannot be passed to 'z'.)r2r!allr)rrrAr!s @r-!check_expressions_preserve_lengthr5sy'&&&&&   eS`dddoo r/kwargscxt||}td|DS)Nc3LK|]}t|o |jjV dSrp)r.rfru)rbrcs r-rz,all_exprs_are_scalar_like..Hs2HHQwqzz8ak8HHHHHHr/)rvaluesr)rrr4s r-all_exprs_are_scalar_likerCs8 $  ( (E HH%HHH H HHr/function comparandscfd|D}fd|D}td|Dfdt||D}||S)Nc3<K|]}t|VdS)rNN)r[)rb comparandrKrNs r-rz(apply_n_ary_operation..QsF  #yZ@@@r/cHg|]}t|S)r)rnr)rbrrNs r-rez)apply_n_ary_operation..Us<      jAA   r/c3&K|] }|j V dSrpr)rbkinds r-rz(apply_n_ary_operation..Zs(>>++>>>>>>r/c3K|]>\}}r3t|r$t|r||n|V?dSrp)r ru broadcast)rbcompliant_exprrrs r-rz(apply_n_ary_operation..[sx !ND  *>:: ?Md?S?S   &&& r/)anyri)rKrrNrcompliant_exprskindsrs` ` @r-apply_n_ary_operationr Ks #O    #   E >>>>>>>I%($?$? O 8_ %%r/N)r&rr'r()r&rr'r0)r4r5r'r6)r4rr'rC)rKrrLrMrNrOr'rP)r\rr8rr]r9r'r^)r&rnr'r) rrrNrOrrOrrOr'r)rr"rrTr'r)rr"rr"r'rO) rKrrrrrMrNrOr'r)< __future__renumrr itertoolsrtypingrrr r r r narwhals._utilsr narwhals.dependenciesrrnarwhals.exceptionsrrrrtyping_extensionsrrnarwhals._compliantrrnarwhals._compliant.typingrrrrrrr+rr2r!narwhals.typingr"r#r$r%r.r3rBrJr[rlrnrurrrrrr rar/r-rs1#"""""GGGGGGGGGGGGGGGG------DDDDDDDD////////BBBBBBBB#"""""&&&&&&DDDDDDDDDD A!!!!#### ! ! ! !    $!!!!,C C C C C tC C C L """""D"""6A A A A A A A A HPPPPf    IIII&&&&&&r/