iz#\ddlmZddlZddlZddlmZmZmZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZej.e5ddlmZddde rddlmZdd lmZdd lmZm Z m!Z!d d d dd d  d"d Z"d#dZ# d$dZ$ d$dZ%e dd d d%dZ&e dd d&dZ&ddd d'dZ&dd d(dZ'd d d d)dZ( d*dZ)d+dZ*d d d,dZ+ d-dZ,d.d Z-d/d!Z.y#1swYxYw)0) annotationsN) CollectionIterableMapping) TYPE_CHECKINGAnyLiteraloverload) functions)qualified_type_name) ComputeError)Expr)PyExpr)ColumnNameOrSelectorIntoExprPolarsDataTypeF) str_as_litlist_as_series structifydtyperequire_selectorct|tjr|}|r t|}|jSt|tr#|s!t j |}|jS|rdt|d}t|t|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. require_selector Require that the input is a valid selector (eg: column name or selector). Returns ------- PyExpr cannot turn  into selector)r) isinstanceplr_structify_expressionstrFcolr TypeErrorlistlitSeries_pyexpr)inputrrrrrexprmsgs Z/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/_utils/parse/expr.pyparse_into_expressionr*sD%! (.D << E3  uuU| <<  !4U!; >nMCC. t $555)7D <<55e,D <<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)rrJsYY++-N//1 = --/I88N+11)D K  "sA88BBctttjj dd}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. Returns ------- list of PyExpr POLARS_AUTO_STRUCTIFYrrboolintosenvironget_parse_positional_inputs_parse_named_inputsextendinputs named_inputsrexprs named_exprss r)parse_into_list_of_expressionsrGVsO&S(?CDEI $Vy AE),)L  [! Lr+ctttjj dd}t ||d}|rt ||}|j||S)a Parse multiple inputs (required to be valid selectors) 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. Returns ------- list of PyExpr r7rT)rrequire_selectorsr8r9rBs r)0parse_into_list_of_expressions_require_selectorsrJqsX&S(?CDEI $ E ),)L  [! Lr+.)strictraise_if_not_selectorcyr-irKrLs r)parse_into_selectorrQs r+rKcyr-rNrOs r)rQrQs r+TcBt|tr#tjj |g|dSt|tjr|St|tj r|j jS|rdt|d}t|y)NTnamesrKexpand_patternsrr) rrrSelector_by_namerr. as_selectorr r!)rPrKrLr(s r)rQrQs !S{{### $  Ar{{ # Arww vv!!## 036nEn r+ct|trt|ts|Dcgc]}t|ts|}}tjj ||d}t |t |k(r|St |dk(rddlm}|j}|D]}|t||z}|St||Scc}w)NTrUrrR) rrrrrXrYlenpolars.selectors selectorsemptyrQ)rCrKrPcolumnsselectorcss r)parse_list_into_selectorrcs &*%j.E)/FA:a3EaFF;;'' (  w<3v; &O w<1  )xxzH >A +Af= =H >"6&99%Gs CC)rIrcZt|}|Dcgc]}t|||c}Scc}w)N)rr)_parse_inputs_as_iterabler*)rCrIr inputs_iteres r)r?r?s@ ,F3K     .   s(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()))r\rrr! _is_iterable)rCr(s r)reresf   6{aJvay': e n 6{aL3ay Mr+crt|txr&t|tttj f Sr-)rrrbytesrr$)r&s r)rjrjs2 eX & z UBII&0,r+r8c#xK|jD]#\}}t||j|%yw)Nr8)itemsr*r3)rDrnamer&s r)r@r@s>$))+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_constraintsrA_combine_predicates) predicates constraintsall_predicatesconstraint_predicatess r),parse_predicates_constraints_into_expressionrw s6*.j9N 2; ?34 ~ ..r+c#K|jD]5\}}tj|j|j7ywr-)rnrr eqr%)rtrovalues r)rqrq(s?"((*, eeeDknnU#+++,sA A cp|s d}t|t|dk(r|dStj|S)Nz5at least one predicate or constraint must be providedrir)r!r\plrall_horizontal)rsr(s r)rrrr-s; En :!!}   j ))r+)r&rrr:rr:rr:rzPolarsDataType | Nonerr:returnr)r'rr~r)rCIntoExpr | Iterable[IntoExpr]rDrr~ list[PyExpr])rPrrKr:rLzLiteral[False]r~ pl.Selector)rPrrKr:rLz Literal[True]r~pl.Selector | None)rPrrKr:rLr:r~r)rCz7ColumnNameOrSelector | Collection[ColumnNameOrSelector]rKr:r~r)rCz0tuple[IntoExpr, ...] | tuple[Iterable[IntoExpr]]rIr:rr:r~r)rCz&tuple[Any, ...] | tuple[Iterable[Any]]r~z Iterable[Any])r&rr~r:)rDdict[str, IntoExpr]rr:r~Iterable[PyExpr])rsrrtrr~r)rtrr~r)rsrr~r)/ __future__r contextlibr<collections.abcrrrtypingrrr r polars._reexport _reexportrpolarsr rpolars._utils.variousr polars.exceptionsr suppress ImportError polars._plr_plrr|rrpolars._typingrrrr*rrGrJrQrcr?rerjr@rwrqrrrNr+r)rsu" 9988!5*Z%"MM  #'"1 11 1  1 ! 11 1h  *6 *> ,1  *      )    "&     2: C: : :<$  <   " 22=BL%L59LL/.// /<, *A s D""D+