o 8Di @sdZddlmZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZed Zejrlddl m!Z!ddl"m#Z#ddlm$Z$    ddddZ%dS) z5 .. include:: ../../posts/python_sql_engine.md ---- ) annotationsN)exp) ExecuteError)PythonExecutor)Table ensure_tables) dict_depth)optimize)annotate_types)Plan) ensure_schemaflatten_schema nested_get nested_setsqlglot) DialectType) Expression)Schemasqlstr | Expressionschemat.Optional[t.Dict | Schema]readrdialecttablest.Optional[t.Dict]returnrcCsT|p|}t||d}|sTi}t|jt|jd}|D]9}t|jgt||R}|dus.J|jD]!} |d| } tt | |dj pGt | j } t |g|| | q1qt ||d}|jrg|j|jkrgtdt} t||dd|d} tdt| td | jdd t| }td |t} t|d |}td t| |S)aO Run a sql query against data. Args: sql: a sql statement. schema: database schema. This can either be an instance of `Schema` or a mapping in one of the following forms: 1. {table: {col: type}} 2. {db: {table: {col: type}}} 3. {catalog: {db: {table: {col: type}}}} read: the SQL dialect to apply during parsing (eg. "spark", "hive", "presto", "mysql"). dialect: the SQL dialect (alias for read). tables: additional tables to register. Returns: Simple columnar data structure. )r)depthNrz1Tables must support the same table args as schemaT)leave_tables_isolatedinfer_csv_schemasrzOptimization finished: %fzOptimized SQL: %s)prettyzLogical Plan: %s)rzQuery finished: %f)rr mappingrrzipcolumnsr rconverttype__name__rr supported_table_argsrtimer loggerdebugrr rexecute)rrrrrtables_flattened_tableskeystablecolumnvalue column_typenow expressionplanresultr7R/var/www/Datamplify/venv/lib/python3.10/site-packages/sqlglot/executor/__init__.pyr+s:       r+)NNNN) rrrrrrrrrrrr)&__doc__ __future__rloggingr(typingtrrsqlglot.errorsrsqlglot.executor.pythonrsqlglot.executor.tablerrsqlglot.helperrsqlglot.optimizerr sqlglot.optimizer.annotate_typesr sqlglot.plannerr sqlglot.schemar r rr getLoggerr) TYPE_CHECKINGsqlglot.dialects.dialectrsqlglot.expressionsrrr+r7r7r7r8s0