o 8Di@sddlmZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZddlmZdd lmZmZ      d$d%d"d#ZdS)&) annotationsN)exp)Dialect DialectType)isolate_table_selects)normalize_identifiers)pushdown_cte_alias_columnsqualify_columnsquote_identifiersvalidate_qualify_columns)qualify_tables)Schema ensure_schemaTF expressionexp.Expressiondialectrdbt.Optional[str]catalogschemat.Optional[dict | Schema]expand_alias_refsbool expand_stars infer_schemat.Optional[bool]isolate_tablesr allow_partial_qualificationr r identifyinfer_csv_schemasreturncCst||d}t||||||d}t||d}|rt||d}t|jr(t|}| r4t|||||| d}| r=t ||| d}| rCt ||S)a| Rewrite sqlglot AST to have normalized and qualified tables and columns. This step is necessary for all further SQLGlot optimizations. Example: >>> import sqlglot >>> schema = {"tbl": {"col": "INT"}} >>> expression = sqlglot.parse_one("SELECT col FROM tbl") >>> qualify(expression, schema=schema).sql() 'SELECT "tbl"."col" AS "col" FROM "tbl" AS "tbl"' Args: expression: Expression to qualify. db: Default database name for tables. catalog: Default catalog name for tables. schema: Schema to infer column names and types. expand_alias_refs: Whether to expand references to aliases. expand_stars: Whether to expand star queries. This is a necessary step for most of the optimizer's rules to work; do not set to False unless you know what you're doing! infer_schema: Whether to infer the schema if missing. isolate_tables: Whether to isolate table selects. qualify_columns: Whether to qualify columns. allow_partial_qualification: Whether to allow partial qualification. validate_qualify_columns: Whether to validate columns. quote_identifiers: Whether to run the quote_identifiers step. This step is necessary to ensure correctness for case sensitive queries. But this flag is provided in case this step is performed at a later time. identify: If True, quote all identifiers, else only necessary ones. infer_csv_schemas: Whether to scan READ_CSV calls in order to infer the CSVs' schemas. Returns: The qualified expression. )r)rrrrr)r)rrrr)rr) rr rrr get_or_raisePREFER_CTE_ALIAS_COLUMNpushdown_cte_alias_columns_funcqualify_columns_funcquote_identifiers_funcvalidate_qualify_columns_func)rrrrrrrrrr rr r rrr'R/var/www/Datamplify/venv/lib/python3.10/site-packages/sqlglot/optimizer/qualify.pyqualifys8 4    r))NNNNTTNFTFTTTF) rrrrrrrrrrrrrrrrrrr rrrr rr rrrrrr r) __future__rtypingtsqlglotrsqlglot.dialects.dialectrr'sqlglot.optimizer.isolate_table_selectsr'sqlglot.optimizer.normalize_identifiersr!sqlglot.optimizer.qualify_columnsrr#r r$r r%r r& sqlglot.optimizer.qualify_tablesr sqlglot.schemar rr)r'r'r'r(s0