.h;ddlmZddlZddlmZmZmZddlmZm Z ddl m Z ddl mZddlmZddlmZej(e5ddlmZddderddl mZdd lmZdd lmZmZmZd d d dd  dd ZddZ d d ddZ!dd d dZ"dd d!dZ#d d d"dZ$ d#dZ%d$dZ&d d d%dZ' d&dZ(d'dZ)d(dZ*y#1swYxYw))) annotationsN) CollectionIterableMapping) TYPE_CHECKINGAny) functions)qualified_type_name) ComputeError)Expr)PyExpr)ColumnNameOrSelectorIntoExprPolarsDataTypeF) str_as_litlist_as_series structifydtypect|tjr|}|r t|}|jSt|tr#|s!t j |}|jSt|tr8|r6t jtj||}|jSt j||}|jS)a Parse a single input into an expression. Parameters ---------- input The input to be parsed as an expression. str_as_lit Interpret string input as a string literal. If set to `False` (default), strings are parsed as column names. list_as_series Interpret list input as a Series literal. If set to `False` (default), lists are parsed as list literals. structify Convert multi-column expressions to a single struct expression. dtype If the input is expected to resolve to a literal with a known dtype, pass this to the `lit` constructor. Returns ------- PyExpr )r) isinstanceplr _structify_expressionstrFcollistlitSeries_pyexpr)inputrrrrexprs V/var/www/html/ai-service/venv/lib/python3.12/site-packages/polars/_utils/parse/expr.pyparse_into_expressionr#s>%! (.D << E3  uuU| << E4 ^uuRYYu%U3 <<uuU%( <<c:|jj}|jjrA |jj}t j |j |}|S|S#t$rt j |}Y|SwxYwN)meta undo_aliaseshas_multiple_outputs output_namerstructaliasr )r!unaliased_expr expr_names r"rrCsYY++-N//1 = --/I88N+11)D K  "sA88BB) __structifyc`t||}|rt||}|j||S)a Parse multiple inputs into a list of expressions. Parameters ---------- *inputs Inputs to be parsed as expressions, specified as positional arguments. **named_inputs Additional inputs to be parsed as expressions, specified as keyword arguments. The expressions will be renamed to the keyword used. __structify Convert multi-column expressions to a single struct expression. Returns ------- list of PyExpr r)_parse_positional_inputs_parse_named_inputsextend)r/inputs named_inputsexprs named_exprss r"parse_into_list_of_expressionsr9Os0, %V{ CE),+N  [! Lr$Tstrictc*t|trddlm}|j |g|St|t j r|St|t jr|jjSdt|d}t|)Nr require_allz cannot turn z into selector) rrpolars.selectors selectorsby_namerSelectorr r' as_selectorr TypeError)ir;csmsgs r"parse_into_selectorrHmsy !S%zz1#6z22 Ar{{ # Arww vv!!##036nEnr$cZt|trt|tsddlm}t t d|}|j||}t|t|k(r|St|dk(r|j}|D]}|t||z}|St||S)Nrc"t|tSr&)rr)rEs r"z*parse_list_into_selector..s 1c(:r$r=r:) rrrr?r@rfilterrAlenemptyrH)r5r;rFcolumnsselectorrEs r"parse_list_into_selectorrQs &*%j.E%v:FCD::g6:: w<3v; &O w<1 xxzH >A +Af= =H >"6&99r$r1cXt|}|Dcgc]}t||c}Scc}wNr1)_parse_inputs_as_iterabler#)r5r inputs_iteres r"r2r2s* ,F3KCN Oa !!y 9 OO Os'c|sgSt|dk(r t|dtr d}t|t|dk(rt |dr|dS|S)Nru8Cannot pass a dictionary as a single positional argument. If you merely want the *keys*, use: • df.method(*your_dict.keys()) If you need the key value pairs, use one of: • unpack as keywords: df.method(**your_dict) • build expressions: df.method(expr.alias(k) for k, expr in your_dict.items()))rMrrrD _is_iterable)r5rGs r"rTrTsf   6{aJvay': e n 6{aL3ay Mr$crt|txr&t|tttj f Sr&)rrrbytesrr)r s r"rYrYs2 eX & z UBII&0,r$c#xK|jD]#\}}t||j|%ywrS)itemsr#r,)r6rnamer s r"r3r3s>$))+L e#EY?EEdKKLs8:cjt|}|rt|}|j|t|S)a Parse predicates and constraints into a single expression. The result is an AND-reduction of all inputs. Parameters ---------- *predicates Predicates to be parsed, specified as positional arguments. **constraints Constraints to be parsed, specified as keyword arguments. These will be converted to predicates of the form "keyword equals input value". Returns ------- PyExpr )r2_parse_constraintsr4_combine_predicates) predicates constraintsall_predicatesconstraint_predicatess r",parse_predicates_constraints_into_expressionrfs6*.j9N 2; ?34 ~ ..r$c#K|jD]5\}}tj|j|j7ywr&)r]rreqr)rcr^values r"r`r`s?"((*, eeeDknnU#+++,sA A cp|s d}t|t|dk(r|dStj|S)Nz5at least one predicate or constraint must be providedrXr)rDrMplrall_horizontal)rbrGs r"raras; En :!!}   j ))r$) r rrboolrrmrrmrzPolarsDataType | Nonereturnr )r!r rnr )r5IntoExpr | Iterable[IntoExpr]r/rmr6rrn list[PyExpr])rErr;rmrn pl.Selector)r5z7ColumnNameOrSelector | Collection[ColumnNameOrSelector]r;rmrnrq)r5z0tuple[IntoExpr, ...] | tuple[Iterable[IntoExpr]]rrmrnrp)r5z&tuple[Any, ...] | tuple[Iterable[Any]]rnz Iterable[Any])r zAny | Iterable[Any]rnrm)r6dict[str, IntoExpr]rrmrnIterable[PyExpr])rbrorcrrnr )rcrrrnrs)rbrprnr )+ __future__r contextlibcollections.abcrrrtypingrrpolars._reexport _reexportrpolarsr rpolars._utils.variousr polars.exceptionsr suppress ImportError polars._plr_plrrkr r polars._typingrrrr#rr9rHrQr2rTrYr3rfr`rar$r"rs"99%!5*Z%"MM  #' + ++ +  + ! + +\  * B  *: C: : :8P <PP P 22=BL%L59LL/.// /<, *}s C$$C-