o i!@sddlmZddlZddlmZmZmZddlmZm Z m Z m Z ddl m ZddlmZddlmZddlmZeeddlmZWdn1sNwYerkddlmZdd lmZdd lmZmZm Z d d d dd d dTddZ!dUddZ"d d ddVd%d&Z#e d'd d(dWd/d0Z$e d'd1dXd4d0Z$d5d5d(dYd6d0Z$d5d1dZd8d9Z%d d d:d[d=d>Z&d\dAdBZ'd]dDdEZ(d dFd^dIdJZ)d_dNdOZ*d`dPdQZ+dadRdSZ,dS)b) annotationsN) CollectionIterableMapping) TYPE_CHECKINGAnyLiteraloverload) functions)qualified_type_name) ComputeError)Expr)PyExpr)ColumnNameOrSelectorIntoExprPolarsDataTypeF) str_as_litlist_as_series structifydtyperequire_selectorinputrrboolrrrPolarsDataType | NonerreturnrcCst|tjr|}|rt|}|j St|tr |s t|}|j S|r.dt|d}t|t|t rB|rBtj t ||d}|j Stj ||d}|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 TypeErrorlistZlitSeries_pyexpr)rrrrrrexprmsgr)O/home/app/Keep/.python/lib/python3.10/site-packages/polars/_utils/parse/expr.pyparse_into_expressions "   r+r'r cCsV|j}|jr)z|j}Wnty t|}Y|Swt||}|SN)metaZ undo_aliasesZhas_multiple_outputsZ output_namer r!structalias)r'Zunaliased_exprZ expr_namer)r)r*rIs    r) __structify__require_selectorsinputsIntoExpr | Iterable[IntoExpr]r0r1 named_inputs list[PyExpr]cOs,t|||d}|rt||d}|||S)av 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. __require_selectors Require that all inputs are valid selectors (eg: column names or selector expressions), disallowing literals. Returns ------- list of PyExpr require_selectorsrr)_parse_positional_inputs_parse_named_inputsextend)r0r1r2r4exprsZ named_exprsr)r)r*parse_into_list_of_expressionsUs  r=.)strictraise_if_not_selectorirr>r?Literal[False] pl.SelectorcCdSr,r)r@r>r?r)r)r*parse_into_selectorzrEr> Literal[True]pl.Selector | NonecCrCr,r)rDr)r)r*rErFTcCsdt|trtjj|g|ddSt|tjr|St|tjr"|jS|r0dt|d}t |dS)NTnamesr>Zexpand_patternsrr) rr rSelector_by_namer r-Z as_selectorr r#)r@r>r?r(r)r)r*rEs    7ColumnNameOrSelector | Collection[ColumnNameOrSelector]cCst|trCt|tsCdd|D}tjj||dd}t|t|kr$|St|dkr4ddlm}| }|D] }|t ||dO}q6|St ||dS)NcSsg|] }t|tr|qSr))rr ).0r@r)r)r* sz,parse_list_into_selector..TrJrrG) rrr rrLrMlenZpolars.selectors selectorsemptyrE)r2r>columnsselectorcsr@r)r)r*parse_list_into_selectors    rWr60tuple[IntoExpr, ...] | tuple[Iterable[IntoExpr]]r7cst|}fdd|DS)Ncsg|] }t|dqS))rr)r+)rOer6r)r*rPsz,_parse_positional_inputs..)_parse_inputs_as_iterable)r2r7rZ inputs_iterr)r6r*r9s r9&tuple[Any, ...] | tuple[Iterable[Any]] Iterable[Any]cCsR|sgSt|dkrt|dtrd}t|t|dkr't|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()))rQrrr# _is_iterable)r2r(r)r)r*rZsrZAny | Iterable[Any]cCst|tot|tttjf Sr,)rrr bytesrr%)rr)r)r*r^s  r^r8dict[str, IntoExpr]Iterable[PyExpr]ccs,|D]\}}t||d|VqdS)Nr8)itemsr+r/)r4rnamerr)r)r*r:sr: predicates constraintsrcOs&t|}|rt|}||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 )r9_parse_constraintsr;_combine_predicates)rerfZall_predicatesZconstraint_predicatesr)r)r*,parse_predicates_constraints_into_expressions  riccs,|D]\}}t||jVqdSr,)rcr!r"eqr&)rfrdvaluer)r)r*rgsrgcCs.|sd}t|t|dkr|dSt|S)Nz5at least one predicate or constraint must be providedr]r)r#rQplrZall_horizontal)rer(r)r)r*rhs   rh)rrrrrrrrrrrrrr)r'r rr ) r2r3r0rr1rr4rrr5)r@rr>rr?rArrB)r@rr>rr?rHrrI)r@rr>rr?rrrI)r2rNr>rrrB)r2rXr7rrrrr5)r2r[rr\)rr_rr)r4rarrrrb)rer3rfrrr)rfrarrb)rer5rr)- __future__r contextlibcollections.abcrrrtypingrrrr Zpolars._reexportZ _reexportrZpolarsr r!Zpolars._utils.variousr Zpolars.exceptionsr suppress ImportErrorZ polars._plrZ_plrrlr rZpolars._typingrrrr+rr=rErWr9rZr^r:rirgrhr)r)r)r*s^         4%