o i0@sUddlmZddlZddlZddlZddlZddlmZddlZ ddl m Z ddl Z ej r8ddlmZddlZejeejeedejffZded<dd d Zd ddZd!ddZd"ddZd#ddZdS)$) annotationsN)Decimalregister_io_source)Iterator_ExpressionTreeztyping_extensions.TypeAlias predicatepl.Exprreturnduckdb.Expression | NonecCs>t|jjdd}z t|}t|WStyYdSw)aConvert a Polars predicate expression to a DuckDB-compatible SQL expression. Parameters: predicate (pl.Expr): A Polars expression (e.g., col("foo") > 5) Returns: SQLExpression: A DuckDB SQL expression string equivalent. None: If conversion fails. Example: >>> _predicate_to_expression(pl.col("foo") > 5) SQLExpression("(foo > 5)") json)formatN)r loadsmeta serialize_pl_tree_to_sqlduckdb SQLExpression Exception)rtree sql_filterrD/home/ubuntu/.local/lib/python3.10/site-packages/duckdb/polars_io.py_predicate_to_expressions  ropstrc Cs4zddddddddd |WStyt|w) z}Map Polars binary operation strings to SQL equivalents. Example: >>> _pl_operation_to_sql("Eq") '=' z>==%ANDOR)LtLtEqGtGtEqEqModulusAndOr)KeyErrorNotImplementedError)rrrr_pl_operation_to_sql0s  r, identifiercCs|dd}d|dS)zEscape SQL identifiers by doubling any double quotes and wrapping in double quotes. Example: >>> _escape_sql_identifier('column"name') '"column""name"' "z"")replace)r-escapedrrr_escape_sql_identifierFs  r1rc$Cs&|\}|dkrq||}t|tsJd|dt||d|d|d}}}t|ts=Jd|dt|t|tsNJd|d t|t|ts_Jd |dt|d t|d t|d t|d S|dkr||}t|tsJd|d t|t|S|dvr||}t|tsJd|dt|t|S|dkr||}t|ttfsJdt|t|S|dkr||} t| t ttfsJdt| t| S|dkrw||} t| tsJd|dt| | d} t| t sJd|dt| | d} t| ts*Jd|dt| | d} t| ts@Jd|dt| d| vrn| d}t| d}|dkrZd |dS|dkred |dSd |}t |d!| }t ||d"kr||}t|tsJd|dt|| d#}|d$kr| d%i}t|trd&|vrd'}t |n|d(krd)|}t ||d%}t|tsJd|dt|t|S|d*kr||}t|tsJd|dt|d+|vr d,|vr t|d+}|d,}n t t|}|}t|ts(Jd|d-t|t|ts:Jd|d.t||d/sE|d0kr|d0}t|t s[Jd|d1t|d2t|krid3kswnJd|d1t|d4tt|dtd5|d6S|d7s|d8kr|d8}t|t sJd|d9t|tjj|dd:tjjd;}d<|d=S|d>vrt||S|d?kr|d?}t|tsJd|d@t||dA}|dAdB}tjjtj||dC}d<|dDS|dEkr/|dE}t|t tfsJd|dFt|tdGdHdHtj|dI}d<|dJS|dKkr^|dK} t| t sJJd|dLt| t| }!dMdNdO|!D}"d<|"dPS|dQksh|dRkry| dR| dQdS}#tt|#SdT|dU|}t |dV|dW||}t |)XaRecursively convert a Polars expression tree (as JSON) to a SQL string. Parameters: tree (dict): JSON-deserialized expression tree from Polars Returns: str: SQL expression string Example: Input tree: { "BinaryExpr": { "left": { "Column": "foo" }, "op": "Gt", "right": { "Literal": { "Int": 5 } } } } Output: "(foo > 5)" BinaryExprzA z should be a dict but got leftrrightz LHS of a z The op of a z should be a str but got z RHS of a ( )ColumnzThe col name of a )LiteralDynIntz4The value of an Int should be an int or str but got Floatz;The value of a Float should be a float, int or str but got Functioninputz. should have a list of dicts as input but got rfunctionz. should have a function dict as input but got BooleanIsNullz IS NULL) IsNotNullz IS NOT NULL)z Boolean function not supported: zUnsupported function type: CastoptionsStrictexprr9z;Strict cast on non-literal expression cannot be pushed down NonStrictz9Only NonStrict/Strict casts can be safely unwrapped, got Scalardtypevaluez" should have a str dtype but got z" should have a dict value but got z {'Decimal'rz. should be a two or three member list but got z member list z {'Datetime'Datetimez% should be a one member list but got i@B)tz'z '::TIMESTAMP) Int8Int16Int32Int64Int128UInt8UInt16UInt32UInt64UInt128Float32Float64r@Timez should be an int but got iʚ;i)seconds microsecondsz'::TIMEDatez should be a number but got i)daysz'::DATEBinaryz should be a list but got css|] }d|dVqdS)z\x02xNr).0brrr sz"_pl_tree_to_sql..z'::BLOBString StringOwnedNzUnsupported scalar type z , with value z Node type: z is not implemented. )keys isinstancedicttyperrr,r1intfloatlistr+getnextiter startswithlenrdatetime fromtimestamptimezoneutcmin timedeltatimedatebytesjoinrConstantExpression)$r node_type bin_expr_treelhsrrhscol_nameval_tree int_literal float_literal func_treeinputs input_tree func_dictfuncarg_sqlmsg cast_treerD cast_expr scalar_treerIrJ decimal_valuemicros dt_timestamp nanosecondsr_r`dt_timedays_since_epochr bin_value binary_datar0 string_valrrrrQs """"$""   "$         $     $ $  $$  $    $     $ rrelationduckdb.DuckDBPyRelationschemapl.schema.Schema pl.LazyFramecsdfd d }t||d S)zA polars IO plugin for DuckDB. with_columnslist[str] | Nonerpl.Expr | Nonen_rows int | None batch_sizer Iterator[pl.DataFrame]c 3sd}}|durdtt|}||}|dur||}|dur=ttt t |}Wdn1s8wY|durF| |}|durN| n| |}t |j dD]}|durm|durmt| |VqYt|VqYdS)N,)rmapr1projectlimit contextlibsuppressAssertionErrorr*rfilterto_arrow_readerruread_next_batchpl from_arrow) rrrrduck_predicaterelation_finalcolsresults record_batchrrrsource_generators(    z'duckdb_source..source_generator)rN) rrrrrrrrr rr)rrrrrr duckdb_sources r)rr r r )rrr r)r-rr r)rrr r)rrrrr r) __future__rrrxr typingdecimalrpolarsrpolars.io.pluginsrr TYPE_CHECKINGcollections.abcrtyping_extensionsDictrUnionrpAnyr__annotations__rr,r1rrrrrrs&   &    G