jz#ddlmZddlZddlZddlmZmZmZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZeje5ddlmZdddn #1swxYwYe rddlmZdd lmZdd lmZm Z m!Z!d d d dd d dEdZ"dFdZ#dGd Z$dGd!Z%e d"d d#dHd*Z&e d"d+dId.Z&d/d/d#dJd0Z&d/d+dKd2Z'd d d3dLd6Z(dMd9Z)dNd;Z*d d<dOd?Z+dPdBZ,dQdCZ-dRdDZ.dS)S) annotationsN) CollectionIterableMapping) TYPE_CHECKINGAnyLiteraloverload) functions)qualified_type_name) ComputeError)Expr)PyExpr)ColumnNameOrSelectorIntoExprPolarsDataTypeF) str_as_litlist_as_series structifydtyperequire_selectorinputrrboolrrrPolarsDataType | Nonerreturnrct|tjr|}|rt|}nt|tr|st j|}nz|r"dt|d}t|t|tr+|r)t j tj ||}nt j ||}|j S)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)rrrrrrexprmsgs ]/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/_utils/parse/expr.pyparse_into_expressionr-sD%!! -  /(..D E3   - -uU||  -M!4U!;!;MMMCC.. t $ $ - -55))777DD5e,,,D <r*rc6|j}|jrf |j}t j||}n$#t$rt j|}YnwxYw|SN)meta undo_aliaseshas_multiple_outputs output_namer#structaliasr )r*unaliased_expr expr_names r,r!r!JsY++--N//11= = --//I8N++11)<>DDD " KsA55BBinputsIntoExpr | Iterable[IntoExpr] named_inputs list[PyExpr]ctttjdd}t ||}|r&t ||}|||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_STRUCTIFYrrrintosenvironget_parse_positional_inputs_parse_named_inputsextendr9r;rexprs named_exprss r,parse_into_list_of_expressionsrKVsm&S(?CCDDEEI $Vy A A AE"),)LLL  [!!! Lr.ctttjdd}t ||d}|r&t ||}|||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 r>rT)rrequire_selectorsr?r@rHs r,0parse_into_list_of_expressions_require_selectorsrNqsz&S(?CCDDEEI $   E "),)LLL  [!!! Lr..)strictraise_if_not_selectorirrOrPLiteral[False] pl.SelectorcdSr0rQrOrPs r,parse_into_selectorrWs #r.rO Literal[True]pl.Selector | NonecdSr0rUrVs r,rWrWs r.Tc\t|tr#tj|g|dSt|tjr|St|tjr|jS|r"dt|d}t|dS)NTnamesrOexpand_patternsrr) rr"r Selector_by_namerr1 as_selectorr r%)rQrOrPr+s r,rWrWs !S {### $   Ar{ # # Arw  v!!### E033EEEnn 4r.7ColumnNameOrSelector | Collection[ColumnNameOrSelector]ct|trt|tsd|D}tj||d}t |t |kr|St |dkrddlm}| }|D]}|t||z}|St||S)Nc<g|]}t|t|SrU)rr").0rQs r, z,parse_list_into_selector..s'FFFA:a3E3EFaFFFr.Tr]rrX) rrr"r r`ralenpolars.selectors selectorsemptyrW)r9rOcolumnsselectorcsrQs r,parse_list_into_selectorros &*%%:j.E.E:FFFFF;'' (   w<<3v;; & &O w<<1   ) ) ) ) ) )xxzzH > >A +Af=== =HH"6&9999r.)rMr0tuple[IntoExpr, ...] | tuple[Iterable[IntoExpr]]rMcBt|}fd|DS)Nc4g|]}t|S))rr)r-)rferMrs r,rgz,_parse_positional_inputs..sC      .      r.)_parse_inputs_as_iterable)r9rMr inputs_iters `` r,rErEsE ,F33K          r.&tuple[Any, ...] | tuple[Iterable[Any]] Iterable[Any]c|sgSt|dkr,t|dtrd}t|t|dkrt |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()))rhrrr% _is_iterable)r9r+s r,rtrts   6{{aJvay':: e nn 6{{aL33ay Mr.rc|t|to't|tttjf Sr0)rrr"bytesr r()rs r,rzrzs8 eX & & z UBI&00,r.r?dict[str, IntoExpr]Iterable[PyExpr]c#K|D]+\}}t|||V,dS)Nr?)itemsr-r6)r;rnamers r,rFrFs_$))++LL e#EY???EEdKKKKKKLLr. predicates constraintsct|}|r$t|}||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 )rE_parse_constraintsrG_combine_predicates)rrall_predicatesconstraint_predicatess r,,parse_predicates_constraints_into_expressionr sJ*.j99N5 2; ? ?3444 ~ . ..r.c#K|D]3\}}tj||jV4dSr0)rr#r$eqr))rrvalues r,rr(sU"((**,, eeDkknnU##+++++,,r.c|sd}t|t|dkr|dStj|S)Nz5at least one predicate or constraint must be providedryr)r%rhplrall_horizontal)rr+s r,rr-sG Enn :!!}  j ) ))r.)rrrrrrrrrrrrrr)r*rrr)r9r:r;rrr<)rQrrOrrPrRrrS)rQrrOrrPrYrrZ)rQrrOrrPrrrZ)r9rcrOrrrS)r9rprMrrrrr<)r9rvrrw)rrrr)r;r}rrrr~)rr:rrrr)rr}rr~)rr<rr)/ __future__r contextlibrBcollections.abcrrrtypingrrr r polars._reexport _reexportr polarsr r#polars._utils.variousr polars.exceptionsr suppress ImportError polars._plr_plrrrrpolars._typingrrrr-r!rKrNrWrorErtrzrFrrrrUr.r,rsQ"""""" 9999999999888888888888!!!!!!555555******Z%%N""""""MMMMMMMMMM  #'"111111h    6> ,1    "& 2::::::<$ "2=BLLLLLL////<,,,, ******s A  A$'A$