}iNddlmZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z er ddlmZddlmZ dd lZn#e$rZd Zeeed Z[wwxYwejZeed ZGd d e ejZddZdgZd S)) annotations) TYPE_CHECKINGLiteral)DeferredTimeZonenarwhals_to_native_dtype) LazyFrame) from_native)Version)CompliantLazyFrameAny) IntoSchemaNzY`narwhals.sql` requires DuckDB to be installed. Hint: run `pip install -U narwhals[sql]`z;select value from duckdb_settings() where name = 'TimeZone'c2eZdZdZdfd Zd d ddZxZS)SQLTablez/A LazyFrame with an additional `to_sql` method.dfr level&Literal['full', 'interchange', 'lazy']returnNonecNt||dS)Nr)super__init__)selfrr __class__s @/home/jrussi/.local/lib/python3.11/site-packages/narwhals/sql.pyrzSQLTable.__init__ s( 5)))))F)prettyrboolstrc|}|s|S ddl}n$#t$r}d}t ||d}~wwxYw||ddS)a4Convert to SQL query. Arguments: pretty: Whether to pretty-print SQL query. If `True`, requires `sqlparse` to be installed. Examples: >>> import narwhals as nw >>> from narwhals.sql import table >>> schema = {"date": nw.Date, "price": nw.Int64, "symbol": nw.String} >>> assets = table("assets", schema) >>> result = assets.filter(nw.col("price") > 100) >>> print(result.to_sql()) SELECT * FROM main.assets WHERE (price > 100) rNzr`SQLTable.to_sql` with `pretty=True`requires `sqlparse` to be installed. Hint: run `pip install -U narwhals[sql]`Tupper)reindent keyword_case) to_native sql_querysqlparse ImportErrorModuleNotFoundErrorformat)rrr$r%_excmsgs rto_sqlzSQLTable.to_sql%s NN$$..00    5 OOOO 5 5 5;  &c** 4  5y4gNNNs1 AA  A)rr rrrr)rrrr)__name__ __module__ __qualname____doc__rr+ __classcell__)rs@rrrsp99****** (-OOOOOOOOOOrrnamerschemar rcfd|D}dd|D}td|d|dt t|}t |j|jS)uGenerate standalone LazyFrame which you can use to generate SQL. Note that this requires DuckDB to be installed. Parameters: name: Table name. schema: Table schema. Examples: >>> import narwhals as nw >>> from narwhals.sql import table >>> schema = {"date": nw.Date, "price": nw.List(nw.Int64), "symbol": nw.String} >>> table("t", schema) ┌────────────────────────────┐ | Narwhals LazyFrame | |----------------------------| |┌──────┬─────────┬─────────┐| |│ date │ price │ symbol │| |│ date │ int64[] │ varchar │| |└──────┴─────────┴─────────┘| | 0 rows | └────────────────────────────┘ cVi|]&\}}|t|tjt'S)rr MAINTZ.0coldtypes r ztable..\s= C %eW\2 > >rz, c3,K|]\}}d|d|VdS)"z" Nr5r8s r ztable..`s7SSjc5)3))%))SSSSSSrz CREATE TABLE "z " (z ); r) itemsjoinCONNsqlr tabler_compliant_frame_level)r1r2column_mappingdtypeslfs rrDrDDs0 ,,..NYYSSNrVs""""""))))))))MMMMMMMM((((((******""""""+@@@@@@******-MMMM--- 3  c " ", -v~HH JKK $O$O$O$O$Oy01$O$O$ON":":":":J )s=AAA