o 8Di@sdZddlmZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZmZmZddlmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z)m*Z*m+Z+m,Z,m-Z-dd l.m/Z/dd l0m1Z1dd l2m3Z3m4Z4dd l5m6Z6m7Z7m8Z8ej9rdd l:m;Z;ddl mz ddl?m@Z@mAZAWneBye>CdYnwdZD d1d2ddZE d1d3ddZFejGd4d"d#ZHejGd5d%d#ZH   d6d7d'd#ZH   ( d8d9d/d0ZIdS):z! .. include:: ../README.md ---- ) annotationsN) expressions)DialectDialects)diff) ErrorLevel ParseError TokenErrorUnsupportedError) Expressionalias_and_casecastcolumn conditiondeleteexcept_from_funcinsert intersect maybe_parsemergenot_or_selectsubquerytable_ to_column to_identifierto_tableunion) Generator)Parser) MappingSchemaSchema)Token Tokenizer TokenType)E) DialectTypesqlglot) __version____version_tuple__zUUnable to set __version__, run `pip install -e .` or `python setup.py develop` first.Fsqlstrreadr+dialectreturn t.List[Token]cCst|p||S)a, Tokenizes the given SQL string. Args: sql: the SQL code string to tokenize. read: the SQL dialect to apply during tokenizing (eg. "spark", "hive", "presto", "mysql"). dialect: the SQL dialect (alias for read). Returns: The resulting list of tokens. )r get_or_raisetokenize)r/r1r2r7I/var/www/Datamplify/venv/lib/python3.10/site-packages/sqlglot/__init__.pyr6Hs r6t.List[t.Optional[Expression]]cKst|p|j|fi|S)a Parses the given SQL string into a collection of syntax trees, one per parsed SQL statement. Args: sql: the SQL code string to parse. read: the SQL dialect to apply during parsing (eg. "spark", "hive", "presto", "mysql"). dialect: the SQL dialect (alias for read). **opts: other `sqlglot.parser.Parser` options. Returns: The resulting syntax tree collection. rr5parse)r/r1r2optsr7r7r8r;Wsr;into t.Type[E]r*cKdSNr7)r/r=r<r7r7r8 parse_oneirAr cKr?r@r7)r/r<r7r7r8rAmrBt.Optional[exp.IntoType]cKsnt|p|}|r|j||fi|}n |j|fi|}|D]}|s+td|d|Std|d)a Parses the given SQL string and returns a syntax tree for the first parsed SQL statement. Args: sql: the SQL code string to parse. read: the SQL dialect to apply during parsing (eg. "spark", "hive", "presto", "mysql"). dialect: the SQL dialect (alias for read) into: the SQLGlot Expression to parse into. **opts: other `sqlglot.parser.Parser` options. Returns: The syntax tree for the first parsed statement. zNo expression was parsed from '')rr5 parse_intor;r)r/r1r2r=r<result expressionr7r7r8rAqsTwriteidentitybool error_levelt.Optional[ErrorLevel] t.List[str]c s@|r dur|nntfddt|||dDS)af Parses the given SQL string in accordance with the source dialect and returns a list of SQL strings transformed to conform to the target dialect. Each string in the returned list represents a single transformed SQL statement. Args: sql: the SQL code string to transpile. read: the source dialect used to parse the input string (eg. "spark", "hive", "presto", "mysql"). write: the target dialect into which the input should be transformed (eg. "spark", "hive", "presto", "mysql"). identity: if set to `True` and if the target dialect is not specified the source dialect will be used as both: the source and the target dialect. error_level: the desired error level of the parser. **opts: other `sqlglot.generator.Generator` options. Returns: The list of transpiled SQL statements. Ncs*g|]}|rj|fddindqS)copyF)generate).0rGr<rHr7r8 sztranspile..)rKr:)r/r1rHrIrKr<r7rRr8 transpiles    rT)NN)r/r0r1r+r2r+r3r4)r/r0r1r+r2r+r3r9)r/r0r=r>r3r*)r/r0r3r )NNN) r/r0r1r+r2r+r=rCr3r )NNTN) r/r0r1r+rHr+rIrJrKrLr3rM)J__doc__ __future__rloggingtypingtr,rexpsqlglot.dialects.dialectrr sqlglot.diffrsqlglot.errorsrrr r sqlglot.expressionsr r aliasr rrrrrrrrrrrrrrrrrtablerr r!r"sqlglot.generatorr#sqlglot.parserr$sqlglot.schemar%r&sqlglot.tokensr'r(r) TYPE_CHECKINGsqlglot._typingr*r+ getLoggerloggersqlglot._versionr-r. ImportErrorerrorprettyr6r;overloadrArTr7r7r7r8sT   h           &