o 'i1+@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_filterrN/home/ubuntu/transcripts/venv/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|dkrY||} t| tsJd|dt| | d} t| t sJd|dt| | d} t| ts Jd|dt| | d} t| ts"Jd|dt| d| vrP| d} t| d}| dkrd>tj|d?}d2|d@S|dAkr|dA}t|t sJd|dBt|t|}dCdDdE|D}d2|dFS|dGks|dHkr|dH|dGdI}tt|SdJ|dK|}t |dL|dM||}t |)NaRecursively 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 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: 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) Int8Int16Int32Int64UInt8UInt16UInt32UInt64Float32Float64r?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,r1intlistr+nextiter startswithlenrdatetime fromtimestamptimezoneutcmin timedeltatimefloatdatebytesjoingetrConstantExpression) r node_type bin_expr_treelhsrrhscol_nameval_tree int_literal func_treeinputs input_tree func_dictfuncarg_sqlmsg scalar_treerCrD decimal_valuemicros dt_timestamp nanosecondsrWrXdt_timedays_since_epochrv 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| |}|durO| }n| |}t |j dD]}|durn|durnt| |VqZt|VqZdS)N,)rxmapr1projectlimit contextlibsuppressAssertionErrorr*rfilterfetch_arrow_readerrkread_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__rrrnr typingdecimalrpolarsrpolars.io.pluginsrr TYPE_CHECKINGcollections.abcrtyping_extensionsDictrUnionrhAnyr__annotations__rr,r1rrrrrrs&   &    *