ˀh/ddlmZddlZddlmZmZddlmZmZddl m Z ddl m ZddlmZej"e5ddlm Zddderddl mZddlmZmZdd lmZd d d dd  dd Zdd Zd d ddZd d ddZ ddZddZd d ddZ ddZ!d dZ"d!dZ#y#1swY}xYw)") annotationsN)IterableMapping) TYPE_CHECKINGAny) functions) ComputeError)Expr)IntoExprPolarsDataType)PyExprF) 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 t/var/www/html/wine-match-dev/backend/winematch-backend/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 )runaliased_expr expr_names rrrBsYY++-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 rparse_into_list_of_expressionsr6Ns0, %V{ CE),+N  [! Lr!r.cXt|}|Dcgc]}t||c}Scc}wNr.)_parse_inputs_as_iterabler )r2r inputs_iteres rr/r/ls* ,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()))lenrr TypeError _is_iterable)r2msgs rr9r9usf   6{aJvay': e n 6{aL3ay Mr!crt|txr&t|tttj f Sr#)rrrbytesrr)rs rr@r@s2 eX & z UBII&0,r!c#xK|jD]#\}}t||j|%ywr8)itemsr r))r3rnamers rr0r0s>$))+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 )r/_parse_constraintsr1_combine_predicates) predicates constraintsall_predicatesconstraint_predicatess r,parse_predicates_constraints_into_expressionrNs6*.j9N 2; ?34 ~ ..r!c#K|jD]5\}}tj|j|j7ywr#)rErreqr)rKrFvalues rrHrHs?"((*, eeeDknnU#+++,sA A cp|s d}t|t|dk(r|dStj|S)Nz5at least one predicate or constraint must be providedr=r)r?r>plrall_horizontal)rJrAs rrIrIs; En :!!}   j ))r!) rr rboolrrUrrUrzPolarsDataType | Nonereturnr )rr rVr )r2IntoExpr | Iterable[IntoExpr]r,rUr3r rV list[PyExpr])r2z0tuple[IntoExpr, ...] | tuple[Iterable[IntoExpr]]rrUrVrX)r2z&tuple[Any, ...] | tuple[Iterable[Any]]rVz Iterable[Any])rzAny | Iterable[Any]rVrU)r3dict[str, IntoExpr]rrUrVIterable[PyExpr])rJrWrKrrVr )rKrYrVrZ)rJrXrVr )$ __future__r contextlibcollections.abcrrtypingrrpolars._reexport _reexportrpolarsrrpolars.exceptionsr suppress ImportError polars.polarsrSr polars._typingr r r r rr6r/r9r@r0rNrHrIr!rrhsg"-%!*Z%  7$  #' + ++ +  + ! + +\  * BP <PP P 22=BL%L59LL/.// /<, *g  s CC