o
    g2                    @  s	  U d Z ddlmZ ddlZddlmZ ddlZddlm	Z	 ddl
Z
ddlZddlZddlZddlmZ ddlZddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z  ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddl&m'Z' ddl&m(Z( ddl&m)Z) ddl&m*Z* dd l&m+Z+ dd!l&m,Z, dd"l&m-Z- dd#l&m.Z. dd$l&m/Z/ dd%l&m0Z0 dd&l&m1Z2 dd'l3m4Z4 dd(l3m5Z5 dd)l'm6Z6 dd*l'm7Z7 dd+l'm8Z8 dd,l'm9Z9 dd-l'm:Z: dd.l'm;Z; dd/l*m<Z< dd0l*m=Z= dd1l.m>Z> dd2l0m?Z? dd3l@mAZA dd4lBmCZC dd5lBmDZD d6d7l&mEZE d6d&l&m1Z1 d6d8l1mFZF d6d9lGmHZH d6d:lGmIZI d6d;lGmJZJ ej$rEdd<lKmLZL dd=l'mMZM dd>l'mNZN dd?lOmPZP dd@lQmRZR ddAlSmTZT ddBlSmUZU ddClSmVZV ddDl*mWZW ddEl*mXZX ddFl*mYZY ddGl*mZZZ ddHl*m[Z[ ddIl+m\Z\ ddJl.m]Z] ddKl/m^Z^ ddLl/m_Z_ ddMl/m`Z` ddNl/maZa ddOl/mbZb ddPl/mcZc ddQl/mdZd ddRl/meZe ddSl@mfZf d6dTlgmhZh d6dUlimjZj d6dVlimkZk d6dWlimlZl d6dXlimmZm d6dYlimnZn d6dZlimoZo d6d[limpZp d6d\limqZq d6d]limrZr ed^esf Zth d_Zuevd`ejwZxevdaejwZydbdc ezdddD {degZ|evdfejwZ}evdfejwZ~evdgejwZevdhejZevdiejZdjZedkdldmdndodpZi e,jdqe,jdre,jdse,jdte,jdue,jdve,jdwe,jdxe,jdye,jdze,jd{e,jd|e,jd}e,jd~e,jde,jde,jdi e,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jde,jdiZe+jde+jde+jde+jde+jde+jde+jde+jde+jde+jde+jde+jde+jde+jdiZded< ddddddddddddddddZe/jjde/jjde/jjde/jjde/jjde/jjdiZG dd deZG ddÄ deIZdZded< dZded< d6Zded< dZded< G dd΄ deJZG ddЄ deddҍZG ddԄ deZG ddք deZG dd؄ deZG ddڄ deFZG dd܄ de	ZG ddބ de	ZeeՃ\ZZZZG dd dedddgZG dd dZG dd de1jރZG dd de-je e*jZG dd de-je e*jZG dd de݃ZG dd deZG dd de݃ZG dd de߃ZG dd deZG dd deIZG dd deIZG dd dZdS )a]  Base SQL and DDL compiler implementations.

Classes provided include:

:class:`.compiler.SQLCompiler` - renders SQL
strings

:class:`.compiler.DDLCompiler` - renders DDL
(data definition language) strings

:class:`.compiler.GenericTypeCompiler` - renders
type specification strings.

To generate user-defined SQL strings, see
:doc:`/ext/compiler`.

    )annotationsN)IntEnum)perf_counter)Any)Callable)cast)ClassVar)Dict)	FrozenSet)Iterable)Iterator)List)Mapping)MutableMapping)
NamedTuple)NoReturn)Optional)Pattern)Sequence)Set)Tuple)Type)TYPE_CHECKING)Union   )base)	coercions)crud)elements)	functions)	operators)roles)schema)
selectable)sqltypes)utilis_column_element)is_dml)	_de_clone)_from_objects)
_NONE_NAME) _SentinelDefaultCharacterization)
Executable)NO_ARG)ClauseElement)quoted_name)Column)	TupleType)
TypeEngine)prefix_anon_map)	Visitable   )exc)FastIntFlag)Literal)Protocol)	TypedDict)_AnnotationDict)_AmbiguousTableNameMap)CompileState)CacheKey)ExecutableDDLElement)Insert)
UpdateBase)
ValuesBase)_truncated_label)BindParameter)ColumnClause)ColumnElement)Label)Function)Table)AliasedReturnsRows)CompoundSelectState)CTE)
FromClause)NamedFromClause)ReturnsRows)Select)SelectState)_BindProcessorType)CursorResultMetaData)_CoreSingleExecuteParams)_DBAPIAnyExecuteParams)_DBAPIMultiExecuteParams)_DBAPISingleExecuteParams)_ExecuteOptions)_GenericSetInputSizesType)_MutableCoreSingleExecuteParams)Dialect)SchemaTranslateMapTyperN   >^   asdoinisonortoallandanyascendfornewnotoffoldsetbothcaser   descelsefromfullintojoinleftlikenullonlysomethentrueuserwhenarraycheckcrossfalsegrantgroupilikeinnerlimitorderouterrighttableunionusingwherebinarycolumncreateexceptfreezehavingisnulloffsetselectuniqueanalyseanalyzebetweencollatedefaultforeignleadingnaturalnotnullplacingprimarysimilarverbosedistinctoverlapstrailing	initially	intersect	localtime	symmetric
asymmetric
constraint
deferrable
referencescurrent_datecurrent_rolecurrent_timecurrent_usersession_userauthorizationlocaltimestampcurrent_timestampz^[A-Z0-9_$]+$z^[A-Z0-9_ $]+$c                 C  s   h | ]}t |qS  )str).0xr   r   Y/var/www/html/ecg_monitoring/venv/lib/python3.10/site-packages/sqlalchemy/sql/compiler.py	<setcomp>       r   
   $z5^(?:RESTRICT|CASCADE|SET NULL|NO ACTION|SET DEFAULT)$z^(?:DEFERRED|IMMEDIATE)$z%(?<![:\w\$\x5c]):([\w\$]+)(?![:\w\$])z\x5c(:[\w\$]*)(?![:\w\$])z%%(%(name)s)s?z%%sz:[_POSITION]z$[_POSITION]z	:%(name)s)pyformatqmarkformatnumericnumeric_dollarnamedz AND z OR z + z * z -  % -z < z <= z != z > z >= z = z IS DISTINCT FROM z IS NOT DISTINCT FROM z || z MATCH z NOT MATCH z IN z NOT IN , z FROM  AS z IS z IS NOT z	 COLLATE zEXISTS 	DISTINCT zNOT zANY zALL z DESCz ASCz NULLS FIRSTz NULLS LASTz ^ z | z & ~z << z >> coalesceCURRENT_DATECURRENT_TIMECURRENT_TIMESTAMPCURRENT_USER	LOCALTIMELOCALTIMESTAMPrandomsysdateSESSION_USERUSERCUBEROLLUPzGROUPING SETSzDict[Type[Function[Any]], str]	FUNCTIONSmonthdayyearsecondhourdoyminutequarterdowweekepochmillisecondsmicrosecondstimezone_hourtimezone_minute)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   UNIONz	UNION ALLEXCEPTz
EXCEPT ALL	INTERSECTzINTERSECT ALLc                   @  s8   e Zd ZU dZded< 	 ded< 	 ded< 	 ded< d	S )
ResultColumnsEntrya  Tracks a column expression that is expected to be represented
    in the result rows for this statement.

    This normally refers to the columns clause of a SELECT statement
    but may also refer to a RETURNING clause, as well as for dialect-specific
    emulations.

    r   keynamenameTuple[Any, ...]objectsTypeEngine[Any]typeN__name__
__module____qualname____doc____annotations__r   r   r   r   r   [  s   
 	r   c                   @  s   e Zd Zdd
dZdS )_ResultMapAppenderr   r   r   r   Sequence[Any]type_r   returnNonec                 C     d S Nr   selfr   r   r   r   r   r   r   __call__{     z_ResultMapAppender.__call__N)
r   r   r   r   r   r   r   r   r  r  r   r   r   r  r   r   r   r   r   z      r   z
Literal[0]RM_RENDERED_NAMEz
Literal[1]RM_NAMEz
Literal[2]
RM_OBJECTS   z
Literal[3]RM_TYPEc                   @  s&   e Zd ZU ded< ded< ded< dS )_BaseCompilerStackEntryzSet[FromClause]asfrom_fromscorrelate_fromsrP   r#   Nr   r   r   r   r   r   r   r   r    s   
 r  c                   @  s6   e Zd ZU ded< ded< ded< ded< ded	< d
S )_CompilerStackEntryr>   compile_stateboolneed_result_map_for_nestedneed_result_map_for_compoundrP   select_0Select[Any]insert_from_selectNr  r   r   r   r   r    s   
 r  F)totalc                   @  s`   e Zd ZU dZded< 	 ded< 	 ded< 	 ded	< 	 d
ed< 	 edddZedddZdS )ExpandedStatea  represents state to use when producing "expanded" and
    "post compile" bound parameters for a statement.

    "expanded" parameters are parameters that are generated at
    statement execution time to suit a number of parameters passed, the most
    prominent example being the individual elements inside of an IN expression.

    "post compile" parameters are parameters where the SQL literal value
    will be rendered into the SQL statement at execution time, rather than
    being passed as separate parameters to the driver.

    To create an :class:`.ExpandedState` instance, use the
    :meth:`.SQLCompiler.construct_expanded_state` method on any
    :class:`.SQLCompiler` instance.

    r   	statementrU   
parameters%Mapping[str, _BindProcessorType[Any]]
processorsOptional[Sequence[str]]positiontupzMapping[str, List[str]]parameter_expansionr  r   c                   s,    j du r
tdt fdd j D S )zrTuple of positional parameters, for statements that were compiled
        using a positional paramstyle.

        Nz.statement does not use a positional paramstylec                 3  s    | ]} j | V  qd S r  r  r   keyr  r   r   	<genexpr>  s    z6ExpandedState.positional_parameters.<locals>.<genexpr>)r#  r7   InvalidRequestErrortupler(  r   r(  r   positional_parameters  s
   
z#ExpandedState.positional_parametersc                 C     | j S )z.synonym for :attr:`.ExpandedState.parameters`.r%  r(  r   r   r   additional_parameters  s   z#ExpandedState.additional_parametersN)r  r   )r  rU   )r   r   r   r   r   propertyr,  r.  r   r   r   r   r    s    
 	r  c                   @  s   e Zd ZU dZded< 	 ded< 	 ded< 	 ded	< 	 d
Zded< 	 d
Zded< 	 dZded< 	 dZded< 	 dZ	ded< 	 d
Z
ded< 	 d
Zded< dS )_InsertManyValuesa  represents state to use for executing an "insertmanyvalues" statement.

    The primary consumers of this object are the
    :meth:`.SQLCompiler._deliver_insertmanyvalues_batches` and
    :meth:`.DefaultDialect._deliver_insertmanyvalues_batches` methods.

    .. versionadded:: 2.0

    r  is_default_exprr   single_values_exprList[crud._CrudParamElementStr]insert_crud_paramsintnum_positional_params_countedFsort_by_parameter_orderincludes_upsert_behaviorsNOptional[Sequence[Column[Any]]]sentinel_columnsr   num_sentinel_columnsr"  sentinel_param_keysimplicit_sentinelembed_values_counter)r   r   r   r   r   r7  r8  r:  r;  r<  r=  r>  r   r   r   r   r0    s0   
 

r0  c                   @  sb   e Zd ZU dZded< ded< ded< ded	< d
ed< ded< ded< ded< ded< ded< dS )_InsertManyValuesBatchas  represents an individual batch SQL statement for insertmanyvalues.

    This is passed through the
    :meth:`.SQLCompiler._deliver_insertmanyvalues_batches` and
    :meth:`.DefaultDialect._deliver_insertmanyvalues_batches` methods out
    to the :class:`.Connection` within the
    :meth:`.Connection._exec_insertmany_context` method.

    .. versionadded:: 2.0.10

    r   replaced_statementrV   replaced_parameters#Optional[_GenericSetInputSizesType]processed_setinputsizesz#Sequence[_DBAPISingleExecuteParams]batchzSequence[Tuple[Any, ...]]sentinel_valuesr5  current_batch_sizebatchnumtotal_batchesr  rows_sortedis_downgradedNr   r   r   r   r   r?  N  s   
 r?  c                   @  s<   e Zd ZdZdZdZdZdZeeB eB ZeeB Z	dZ
dZdS )	InsertmanyvaluesSentinelOptszcbitflag enum indicating styles of PK defaults
    which can work as implicit sentinel columns

    r   r6            @   N)r   r   r   r   NOT_SUPPORTEDAUTOINCREMENTIDENTITYSEQUENCEANY_AUTOINCREMENT_SUPPORTED_OR_NOTUSE_INSERT_FROM_SELECTRENDER_SELECT_COL_CASTSr   r   r   r   rK  g  s    rK  c                   @  s   e Zd ZdZ	 dZ	 dZdS )CompilerStater   r   r6   N)r   r   r   	COMPILINGSTRING_APPLIEDNO_STATEMENTr   r   r   r   rX  y  s    rX  c                   @  s*   e Zd ZdZdZ	 dZ	 dZ	 eeB ZdS )Lintingzrepresent preferences for the 'SQL linting' feature.

    this feature currently includes support for flagging cartesian products
    in SQL statements.

    r   r   r6   N)r   r   r   r   
NO_LINTINGCOLLECT_CARTESIAN_PRODUCTSWARN_LINTINGFROM_LINTINGr   r   r   r   r\    s    r\  c                   @  s$   e Zd ZdZdddZd	ddZdS )

FromLinterzKrepresents current state for the "cartesian product" detection
    feature.Nc                   s   | j }|sdS t| j}t|}|d ur|}|| n| }t|g}|rQ|rQ|  |   fdd|D }|	 fdd|D  |
| |rQ|s*|rW||fS dS )NNNc                   s   h | ]} |v r|qS r   r   r   edgenoder   r   r         z"FromLinter.lint.<locals>.<setcomp>c                 3  s     | ]}||    V  qd S r  )indexrc  re  r   r   r)    s    z"FromLinter.lint.<locals>.<genexpr>)fromsro   edgesremovepopcollectionsdequepopleftdiscard
extendleftdifference_update)r  startri  rj  the_rest
start_withstack	to_remover   re  r   lint  s(   


zFromLinter.lintSELECTc                   s`      \}}|r,|}|r.d}d fdd|D }|j|| j| d}t| d S d S d S )Nz{stmt_type} statement has a cartesian product between FROM element(s) {froms} and FROM element "{start}".  Apply join condition(s) between each element to resolve.r   c                 3  s"    | ]}d  j |  d V  qdS )"N)ri  )r   from_r(  r   r   r)    s    
z"FromLinter.warn.<locals>.<genexpr>)	stmt_typeri  rs  )rx  rw   r   ri  r%   warn)r  r|  rt  ru  ri  template	froms_strmessager   r(  r   r}    s"   zFromLinter.warnr  )ry  )r   r   r   r   rx  r}  r   r   r   r   ra    s    
%ra  ri  rj  c                      s@  e Zd ZU dZdZded< 	 dZded< 	 ded	< 	 d
Zd
ZdZ	ded< dZ
ded< dZded< ejZded< 	 ded< ejZded< dZded< 	 dZded< 	 dZded< 	 ded< 	 dd
e fdEd$d%ZdF fd(d)Zed*d+ Zd,d- Zd.d/ Zed0d1 ZdGd6d7ZdHd8d9Z			:dIdJdAdBZedCdD Z  Z S )KCompileda  Represent a compiled SQL or DDL expression.

    The ``__str__`` method of the ``Compiled`` object should produce
    the actual text of the statement.  ``Compiled`` objects are
    specific to their underlying database dialect, and also may
    or may not be specific to the columns referenced within a
    particular set of bind parameters.  In no case should the
    ``Compiled`` object be dependent on the actual values of those
    bind parameters, even though it may reference those values as
    defaults.
    NOptional[ClauseElement]r   r   stringrX  stateFzOptional[CursorResultMetaData]_cached_metadataz"Optional[List[ResultColumnsEntry]]_result_columns Optional[SchemaTranslateMapType]schema_translate_maprY   execution_optionsIdentifierPreparerpreparerr<   _annotationszOptional[CompileState]r  dml_compile_stateOptional[CacheKey]	cache_keyfloat	_gen_timedialectr\   render_schema_translater  compile_kwargsMapping[str, Any]c                 C  s   || _ | j j| _|r|| _| j|| _|durRtj| _|| _|j	| _
|j| _| j
r7tr3t|ts3J |j| _| j| jfi || _|rM| j| j|| _tj| _ntj| _t | _dS )a  Construct a new :class:`.Compiled` object.

        :param dialect: :class:`.Dialect` to compile against.

        :param statement: :class:`_expression.ClauseElement` to be compiled.

        :param schema_translate_map: dictionary of schema names to be
         translated when forming the resultant SQL

         .. seealso::

            :ref:`schema_translating`

        :param compile_kwargs: additional kwargs that will be
         passed to the initial call to :meth:`.Compiled.process`.


        N)r  identifier_preparerr  r  _with_schema_translaterX  rY  r  r  supports_executioncan_executer  r   
isinstancer-   _execution_optionsr  processr  _render_schema_translatesrZ  r[  r   r  r  r  r  r  r  r  r   r   r   __init__;  s0   

zCompiled.__init__r  r  c                   s   |    t  S r  )_init_compiler_clssuper__init_subclass__cls	__class__r   r   r  s  s   
zCompiled.__init_subclass__c                 C  r  r  r   r  r   r   r   r  w     zCompiled._init_compiler_clsc                 C  s    | j r
|| ||S t| jr  )r  _execute_compiledr7   ObjectNotExecutableErrorr  )r  
connectiondistilled_paramsr  r   r   r   _execute_on_connection{  s
   zCompiled._execute_on_connectionc                 K  s   t | t||r  r7   UnsupportedCompilationErrorr   r  elementerrkwr   r   r   visit_unsupported_compilation     z&Compiled.visit_unsupported_compilationc                 C     t  )zReturn a Compiled that is capable of processing SQL expressions.

        If this compiler is one, it would likely just return 'self'.

        NotImplementedErrorr(  r   r   r   sql_compiler  s   zCompiled.sql_compilerobjr5   kwargsr   c                 K  s   |j | fi |S r  _compiler_dispatch)r  r  r  r   r   r   r    r  zCompiled.processc                 C  s   | j tju r	| jS dS )z3Return the string text of the generated SQL or DDL.r  )r  rX  rZ  r  r(  r   r   r   __str__  s   zCompiled.__str__Tparams"Optional[_CoreSingleExecuteParams]extracted_parameters&Optional[Sequence[BindParameter[Any]]]escape_names)Optional[_MutableCoreSingleExecuteParams]c                 C  r  )zReturn the bind params for this compiled object.

        :param params: a dict of string/object pairs whose values will
                       override bind values compiled in to the
                       statement.
        r  r  r  r  r  r   r   r   construct_params  s   zCompiled.construct_paramsc                 C  s   |   S )z0Return the bind params for this compiled object.r  r(  r   r   r   r    s   zCompiled.params)
r  r\   r  r  r  r  r  r  r  r  )r  r  )r  r5   r  r   r  r   )r  r   NNTr  r  r  r  r  r  r  r  )!r   r   r   r   r  r   r  is_sqlis_ddlr  r  r  r%   
EMPTY_DICTr  r  r  r  r  immutabledictr  r  classmethodr  r  r  r/  r  r  r  r  r  __classcell__r   r   r  r   r    sX   
 8



	

r  c                   @  s2   e Zd ZdZdZdddZdddZdddZdS )TypeCompilerz2Produces DDL specification for TypeEngine objects.z	visit_\w+r  r\   c                 C  s
   || _ d S r  )r  )r  r  r   r   r   r       
zTypeCompiler.__init__r   r   r  r   r  r   c                 K  s4   |j r| jj|j v r|j | jj }|j| fi |S r  )_variant_mappingr  r   r  r  r   r  r   r   r   r    s
   zTypeCompiler.processr  r  	Exceptionr   c                 K  s   t | ||r  )r7   r  r  r   r   r   r    s   z*TypeCompiler.visit_unsupported_compilationN)r  r\   )r   r   r  r   r  r   )r  r   r  r  r  r   r  r   )r   r   r   r   ensure_kwargr  r  r  r   r   r   r   r    s    

r  c                   @  sB   e Zd ZdZdZdZdddZedd Zed	d
 Z	dd Z
dS )_CompileLabelz;lightweight label object which acts as an expression.Label.labelr  r   
_alt_namesr   c                 C  s   || _ || _|f| | _d S r  r  )r  colr   	alt_namesr   r   r   r    s   z_CompileLabel.__init__c                 C     | j jS r  r  	proxy_setr(  r   r   r   r       z_CompileLabel.proxy_setc                 C  r  r  r  r   r(  r   r   r   r     r  z_CompileLabel.typec                 K     | S r  r   r  r  r   r   r   
self_group     z_CompileLabel.self_groupN)r   )r   r   r   r   __visit_name__	__slots__r  r/  r  r   r  r   r   r   r   r    s    


r  c                   @  sH   e Zd ZdZdZdZdd Zedd Zedd	 Z	d
d Z
dd ZdS )ilike_case_insensitivea  produce a wrapping element for a case-insensitive portion of
    an ILIKE construct.

    The construct usually renders the ``lower()`` function, but on
    PostgreSQL will pass silently with the assumption that "ILIKE"
    is being used.

    .. versionadded:: 2.0

    ilike_case_insensitive_operandr  
comparatorc                 C  s   || _ |j| _d S r  r  )r  r  r   r   r   r    s   zilike_case_insensitive.__init__c                 C  r  r  r  r(  r   r   r   r    r  z ilike_case_insensitive.proxy_setc                 C  r  r  r  r(  r   r   r   r     r  zilike_case_insensitive.typec                 K  r  r  r   r  r   r   r   r     r  z!ilike_case_insensitive.self_groupc                 C  s   t | j|S r  )r  r  _with_binary_element_type)r  r   r   r   r   r    s   
z0ilike_case_insensitive._with_binary_element_typeN)r   r   r   r   r  r  r  r/  r  r   r  r  r   r   r   r   r    s    

r  c                   @  s  e Zd ZU dZeZedddddddddZde	d	< 	 d
e	d< de	d< dZ
eZdZde	d< dZde	d< dZde	d< 	 de	d< 	 dZde	d< 	 dZde	d< 	 dZde	d< 	 dZde	d< de	d< 	 de	d < 	 d!e	d"< 	 dZde	d#< 	 dZde	d$< 	 dZde	d%< 	 d&e	d'< 	 d&e	d(< 	 d&e	d)< 	 d*e	d+< 	 dZde	d,< 	 dZde	d-< 	 dZde	d.< 	 dZde	d/< 	 dZde	d0< 	 dZde	d1< 	 dZd2e	d3< 	 dZd4e	d5< 	 dZd6e	d7< dZ d8e	d9< dZ!d:e	d;< e" Z#d<e	d=< 	 e" Z$d<e	d>< 	 ej%Z&d?e	d@< 	 dZ'	 dZ(	 dZ)dAe	dB< 	 dZ*d6e	dC< 	 dZ+d6e	dD< dZ,d6e	dE< dZ-de	dF< dGe	dH< dIe	dJ< dKe	dL< de	dM< e./dNZ0e./dOZ1e./e1j2 dPe0j2 Z3e4dQdR Z5e4dSdT Z6ddde7dfddddeZ8e9ddgdhZ:ej;ddidjZ<e9dkdl Z=e9dmdn Z>e9dodp Z?ej@ddrdsZAejBddudvZCeDjEdwdx ZFdydz ZGd{d| ZHej@dd~dZIdd ZJe9dd ZK		ddddZL						ddddZMejBdd ZNe9dd ZO	ddddZPeQddd ZRde	d< ej@dddZSej@eQddd ZTej@eQddd ZUdd ZVdd ZWdddZXdd ZY	dddZZ	dddZ[					dddZ\dd Z]dd Z^				ddddĄZ_ddƄ Z`ddȄ Zaddʄ Zbdd̄ Zcdd΄ ZdddЄ Zeddd҄Zf	dddԄZgddք Zhdd؄ Ziddڄ Zjdd܄ Zkddބ Zldd Zmdd Zndd Zodd Zpdd Zqdd Zrdd Zsdd Ztdd Zudd Zvdd Zwdd Zx	ddddZydd Zzdd Z{dd  Z|	d ddZ}dd Z~dd Z	d!ddZd	d
 Zdd Zdd Zdd Zdd Zdd Zdd Zdd Z	dddZdd Z				d"ddZdd  Zd!d" Zd#d$ Zd%d& Zd'd( Z	dd)d*Zd+d, Zd-d. Zej@d/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Zd?d@ ZdAdB ZdCdD ZdEdF ZdGdH ZdIdJ ZdKdL ZdMdN ZdOdP ZdQdR ZdSdT ZdUdV ZdWdX ZdYdZ Zd[d\ Z					d#d]d^Zd_d` ZedfdadbZdcdd Zdedf Zd$djdkZd%dldmZ						d&d'dwdxZdydz Zdddddedfd(ddZdd Zdd Z								d)ddZdd Zdd ZdddZdd Zd ddZdd Zdd Zd*ddZ	dddZ							d+ddZĐdd ZŐdd ZƐdd Zǐdd ZȐdd Zɐde	d< ejːsede" fde" fgZ	dddZdZdce	d< 	 							d,ddZϐd-ddZАdd ZѐddÄ ZҐdĐdń Z		ddƐdǄZԐdȐdɄ ZՐdʐd˄ Z֐d̐d̈́ Zאdΐdτ ZؐdАdф Zِd.dאd؄Zڐdِdڄ Z			d/dېd܄Z							d0dݐdބZݐd dߐdZސdd ZeejejejejejejejejejejiZeejejiZd1ddZd2ddZ	d3ddZdd Zdd Zdd Zdd Zddd Zdd Zdd ZdddZdd Zd	d
 Zdd ZdS (4  SQLCompilerz~Default implementation of :class:`.Compiled`.

    Compiles :class:`_expression.ClauseElement` objects into SQL strings.

    PAZC_)%():.[] zClassVar[Mapping[str, str]]bindname_escape_characterszClassVar[Pattern[str]]_bind_translate_re_bind_translate_charsTFr  isdeleteisinsertisupdatezOptional[List[Column[Any]]]	postfetchr   zSequence[Column[Any]]insert_prefetchupdate_prefetchN&Optional[Sequence[ColumnElement[Any]]]implicit_returningisplaintextzDict[str, BindParameter[Any]]bindszDict[BindParameter[Any], str]
bind_nameszList[_CompilerStackEntry]rv  returning_precedes_values'render_table_with_column_in_update_fromansi_bind_rulesr   bindtemplatecompilation_bindtemplate_numeric_binds_identifier_charzList[ResultColumnsEntry]r  _textual_ordered_columns_ad_hoc_textual_ordered_columns_loose_column_name_matching_numeric_binds_render_postcompilezOptional[ExpandedState]_post_compile_expanded_stateOptional[str]_pre_expanded_stringOptional[List[str]]_pre_expanded_positiontupzOptional[_InsertManyValues]_insertmanyvaluesz!Optional[crud._CrudParamSequence]_insert_crud_paramszFrozenSet[BindParameter[Any]]literal_execute_paramspost_compile_paramszutil.immutabledict[str, str]escaped_bind_nameszbOptional[Tuple[Dict[BindParameter[Any], List[BindParameter[Any]]], Dict[str, BindParameter[Any]]]]_cache_key_bind_matchr#  _values_bindparam_visited_bindparaminlinez"Optional[MutableMapping[CTE, str]]cteszDict[Tuple[int, str], CTE]ctes_by_level_namez/Dict[CTE, Tuple[int, str, selectable._CTEOpts]]level_name_by_ctectes_recursivez"__\[POSTCOMPILE_(\S+?)(~~.+?~~)?\]z%\(([^)]+?)\)s|c                 C  s   |    d S r  )_init_bind_translater  r   r   r   r  2  s   zSQLCompiler._init_compiler_clsc                 C  s2   t d| j}t d| d| _| j| _d S )Nr  r  r  )reescaperw   r  compiler  r  )r  regr   r   r   r'  6  s   z SQLCompiler._init_bind_translater  r\   r  r  r  r  column_keysr"  for_executemanylintingr\  _supporting_againstOptional[SQLCompiler]r  r   c                 K  s  || _ || _|rdd |d D }	dd |d D }
|
|	f| _|| _|| _i | _t | _g | _	g | _
|j| _| jrS|jd | _}|rO|jdkrLdnd| _t| _nt|j | _d	| _|jpa|j| _t | _i | _i | _tj| ||fi | | js| js| jrtrt |t!sJ | js| jrtrt |t"sJ |j#rd
| _$n| jr| jr| j%j&r|j'rd
| _$t|j | _(|r| j)*dd |j)+ D  | j,t-j.u r| jr| jr| /  n| 0  | j1r| j2dd
d}| j3|d
d d	S d	S d	S )a  Construct a new :class:`.SQLCompiler` object.

        :param dialect: :class:`.Dialect` to be used

        :param statement: :class:`_expression.ClauseElement` to be compiled

        :param column_keys:  a list of column names to be compiled into an
         INSERT or UPDATE statement.

        :param for_executemany: whether INSERT / UPDATE statements should
         expect that they are to be invoked in an "executemany" style,
         which may impact how the statement will be expected to return the
         values of defaults and autoincrement / sequences and similar.
         Depending on the backend and driver in use, support for retrieving
         these values may be disabled which means SQL expressions may
         be rendered inline, RETURNING may not be rendered, etc.

        :param kwargs: additional keyword arguments to be consumed by the
         superclass.

        c                 S  s   i | ]}|j |qS r   r'  r   br   r   r   
<dictcomp>a  r   z(SQLCompiler.__init__.<locals>.<dictcomp>r   c                 S  s   i | ]}||gqS r   r   r2  r   r   r   r4  b  r   r   r   r   r  NTc                 S  s   i | ]\}}|d vr||qS )>   r  r  r  
positionalr  r  r  r   r   kvr   r   r   r4    s    Fr  _no_postcompile)_populate_self)4r,  r  r  r-  r.  r  r%   column_dictr  rv  r  r5  
paramstyle
startswithr  r  _pyformat_templater  BIND_TEMPLATESr"  label_lengthmax_identifier_lengthr4   anon_maptruncated_names_truncated_countersr  r  r  r  r  r   r  rB   rC   _inliner!  r  insert_executemany_returning_return_defaultsr  __dict__updateitemsr  rX  rZ  _process_numeric_process_positionalr  r  #_process_parameters_for_postcompile)r  r  r  r  r,  r-  r.  r/  r  cksmckbmnbr  r   r   r   r  <  s    




zSQLCompiler.__init__r  c                 C  s   | j du rdS | j jS )a  When an INSERT is compiled with a single set of parameters inside
        a VALUES expression, the string is assigned here, where it can be
        used for insert batching schemes to rewrite the VALUES expression.

        .. versionadded:: 1.3.8

        .. versionchanged:: 2.0 This collection is no longer used by
           SQLAlchemy's built-in dialects, in favor of the currently
           internal ``_insertmanyvalues`` collection that is used only by
           :class:`.SQLCompiler`.

        N)r  r2  r(  r   r   r   insert_single_values_expr  s   
z%SQLCompiler.insert_single_values_exprc                 C  s6   | j r| j S | jdurt| jrdd | jjD S dS )a  The effective "returning" columns for INSERT, UPDATE or DELETE.

        This is either the so-called "implicit returning" columns which are
        calculated by the compiler on the fly, or those present based on what's
        present in ``self.statement._returning`` (expanded into individual
        columns using the ``._all_selected_columns`` attribute) i.e. those set
        explicitly using the :meth:`.UpdateBase.returning` method.

        .. versionadded:: 2.0

        Nc                 S  s   g | ]}t |r|qS r   r&   r   cr   r   r   
<listcomp>  s    z3SQLCompiler.effective_returning.<locals>.<listcomp>)r  r  r(   _all_selected_columnsr(  r   r   r   effective_returning  s   zSQLCompiler.effective_returningc                 C  r-  )zVbackwards compatibility; returns the
        effective_returning collection.

        )rW  r(  r   r   r   	returning  s   zSQLCompiler.returningc              
   C  s4   z| j d d W S  ty } ztd|d}~ww )a  Return the current 'executable' that is being compiled.

        This is currently the :class:`_sql.Select`, :class:`_sql.Insert`,
        :class:`_sql.Update`, :class:`_sql.Delete`,
        :class:`_sql.CompoundSelect` object that is being compiled.
        Specifically it's assigned to the ``self.stack`` list of elements.

        When a statement like the above is being compiled, it normally
        is also assigned to the ``.statement`` attribute of the
        :class:`_sql.Compiler` object.   However, all SQL constructs are
        ultimately nestable, and this attribute should never be consulted
        by a ``visit_`` method, as it is not guaranteed to be assigned
        nor guaranteed to correspond to the current statement being compiled.

        .. versionadded:: 1.3.21

            For compatibility with previous versions, use the following
            recipe::

                statement = getattr(self, "current_executable", False)
                if statement is False:
                    statement = self.stack[-1]["selectable"]

            For versions 1.4 and above, ensure only .current_executable
            is used; the format of "self.stack" may change.


        r#   z$Compiler does not have a stack entryN)rv  
IndexError)r  ier   r   r   current_executable  s   
zSQLCompiler.current_executablec                 C  s   t | jt | j S r  )listr  r  r(  r   r   r   prefetch%  s   zSQLCompiler.prefetchDict[Any, Any]c                 C  s   i S r  r   r(  r   r   r   _global_attributes)  r  zSQLCompiler._global_attributesMutableMapping[CTE, str]c                 C  s$   t  }|| _i | _i | _d| _|S )zInitialize collections related to CTEs only if
        a CTE is located, to save on the overhead of
        these collections otherwise.

        F)r%   OrderedDictr"  r#  r$  r%  )r  r"  r   r   r   _init_cte_state-  s   	zSQLCompiler._init_cte_statec              	   c  s    | j | j}}g d| _ | _z'| jr| jd }d|d< nd}| j | jfV  W |r.|d ||| _ | _dS |r>|d ||| _ | _w )z;special API to support the use case of 'nested result sets'FrY  Tr  N)r  r  rv  rl  )r  result_columnsordered_columnsentryr   r   r   _nested_resultE  s*   



zSQLCompiler._nested_resultc                   s  j rJ jtju sJ jrJ jjdkrdn
jjdks#J dg dfd	d
 tj	 j
_
jr\dd j D tjtksQJ fddD _ n_ jrg tj	 jj} fddjjD }jj||d_d S d S )Nr   %sr   r   mre.Match[str]r  r   c                   s6   |  d}|r|  S |  d |  dS )Nr   r6   r   )r   append)ri  normal_bind)placeholder	positionsr   r   find_positionj  s   


z6SQLCompiler._process_positional.<locals>.find_positionc                 S     i | ]\}}||qS r   r   r6  r   r   r   r4  y      z3SQLCompiler._process_positional.<locals>.<dictcomp>c                   s   g | ]}  ||qS r   getr   r   )reverse_escaper   r   rU  {  s    z3SQLCompiler._process_positional.<locals>.<listcomp>c              
     s4   g | ]}|d  |d t j |d |d fqS )r   r   r6   r  )r(  sub_positional_patternr   r8  )ro  r  r   r   rU    s    r2  r4  )ri  rj  r  r   )r#  r  rX  rZ  r  r  r=  r(  rv  rw  r  r  rK  lenr  r2  r4  _replace)r  r2  r4  r   )ro  rm  rn  ru  r  r   rM  ]  sD   






zSQLCompiler._process_positionalc           	        sZ  j sJ jtju sJ d}i  jr-jd ur-tfddj	 D j	 }nj	 }|D ]*}| v r;q4j
| }|jv sJ|jv rOd  |< q4j | }|d7 }| |< q4|_t _jrt }fdd  D  t |ksJ j fddj_jrjj  }dd	 jjD }jj||d
_d S d S )Nr   c                 3  s    | ]
}| j vr|V  qd S r  )r  rt  r(  r   r   r)    s    
z/SQLCompiler._process_numeric.<locals>.<genexpr>c                   s    i | ]\}} j |||qS r   )r  rs  )r   r   posr(  r   r   r4        z0SQLCompiler._process_numeric.<locals>.<dictcomp>c                       |  d S Nr   r   ri  	param_posr   r   <lambda>      z.SQLCompiler._process_numeric.<locals>.<lambda>c                 S  s$   g | ]}|d  |d d|d fqS )r   r   rh  r  r   rx  r   r   r   rU    s    z0SQLCompiler._process_numeric.<locals>.<listcomp>ry  )r  r  rX  rZ  r  r  	itertoolschainr  valuesr  r  r  r  next_numeric_posr]  r#  r  rz  rK  _pyformat_patternrv  r  r2  r4  r{  )	r  numr   	bind_namebindph
len_beforer2  r4  r   )r  r  r   rL    s\   


	






zSQLCompiler._process_numericVMutableMapping[str, Union[_BindProcessorType[Any], Sequence[_BindProcessorType[Any]]]]c                   s   dd  fdd j D D S )Nc                 S  s   i | ]\}}|d ur||qS r  r   r   r'  valuer   r   r   r4    s
    z0SQLCompiler._bind_processors.<locals>.<dictcomp>c                 3  sP    | ]#} j | |jjs|j jnt fd dtt|jjD fV  qdS )c                 3  s    | ]	}|  jV  qd S r  )_cached_bind_processorr  )r   	elem_typer(  r   r   r)    
    

z9SQLCompiler._bind_processors.<locals>.<genexpr>.<genexpr>N)	r  r   _is_tuple_typer  r  r+  r   r2   types)r   	bindparamr(  r   r   r)    s    
z/SQLCompiler._bind_processors.<locals>.<genexpr>)r  r(  r   r(  r   _bind_processors  s
   
zSQLCompiler._bind_processorsc                 C  s   t | jdkS r  )rz  rv  r(  r   r   r   is_subquery     zSQLCompiler.is_subqueryc                 C  r  r  r   r(  r   r   r   r    r  zSQLCompiler.sql_compilerr  r  r  r  c                 C  s   | j ||dd}| |S )aR  Return a new :class:`.ExpandedState` for a given parameter set.

        For queries that use "expanding" or other late-rendered parameters,
        this method will provide for both the finalized SQL string as well
        as the parameters that would be used for a particular parameter set.

        .. versionadded:: 2.0.0rc1

        Tr9  )r  rN  )r  r  r  r  r   r   r   construct_expanded_state  s   z$SQLCompiler.construct_expanded_stater  r  _group_numberOptional[int]_checkr:  r[   c                   s
  | j r|s| jdusJ |st| jjS td|ot| j}|rK| jdu r,t	d| jd }| j
}	|	dus:J |	\ }
 fddt||D }nd}|ri }| j D ]]\}}|rc| j||n|}|j|v rr||j ||< qV||v r}|| ||< qV|r|jr|rtjd|j|f dd	tjd
|j dd	|r|||}n|}|jr|j||< qV|j||< qV|S i }| j D ]E\}}|r| j||n|}|r|jr|rtjd|j|f dd	tjd
|j dd	|r|||}n|}|jr|j||< q|j||< q|S )z5return a dictionary of bind parameter keys and valuesNzcan't construct new parameters when render_postcompile is used; the statement is hard-linked to the original parameters.  Use construct_expanded_state to generate a new statement and parameters.zcThis compiled object has no original cache key; can't pass extracted_parameters to construct_paramsr   c                   s$   i | ]\}} | D ]}||q
qS r   r   )r   r3  	extractedr  rP  r   r   r4  G  s    z0SQLCompiler.construct_params.<locals>.<dictcomp>z@A value is required for bind parameter %r, in parameter group %dcd3x)codez)A value is required for bind parameter %r)r  r  dictr  r7   r*  r  r  r  CompileErrorr  zipr  rK  rs  r'  requiredcallableeffective_valuer  )r  r  r  r  r  r  r:  has_escaped_namesorig_extracted
ckbm_tupler  resolved_extractedpdr  r   escaped_namevalue_paramr   r  r   r    s   






zSQLCompiler.construct_paramsc                   s   | j jjj  fddi }| j}| jD ]#}||v r$q|jjr9fddtt	|jj
D ||< q|j||< q|S )Nc                   s@   |   }|d urd u s|vrd u s|v r|S d S r  )_unwrapped_dialect_implget_dbapi_type)typdbtype)dbapir  exclude_typesinclude_typesr   r   lookup_type  s   z<SQLCompiler._get_set_input_sizes_lookup.<locals>.lookup_typec                      g | ]} |qS r   r   )r   r  )r  r   r   rU    s    z;SQLCompiler._get_set_input_sizes_lookup.<locals>.<listcomp>)r  include_set_input_sizesexclude_set_input_sizesr  r  r  r   r  r   r2   r  )r  
inputsizesr  r  r   )r  r  r  r  r  r   _get_set_input_sizes_lookup  s    

z'SQLCompiler._get_set_input_sizes_lookupc                 C  s   | j ddS )zReturn the bind param dictionary embedded into this
        compiled object, for those values that are present.

        .. seealso::

            :ref:`faq_sql_expression_string` - includes a usage example for
            debugging use cases.

        F)r  r  r(  r   r   r   r    s   zSQLCompiler.paramsr  r;  c                   s  i }j }|du rj}jrg }j}|du rj}nd }}j}td|td|i }i i }	d}
jrD|durD|}jrCg }
nj	 }j
}|D ] |rX|  n }j  }|jv ru|vrtj|||d|< qN|jv r|v r|	| }d}n| }|||}|\}}||	|< ||< |js|| |jjr|dusJ | fddt|dD  n| fdd|D  |
dur|
d	d |D  n|dur|d
d |D  dd |D | < qN|dur|  qNfdd}tj||}|
dur5|dusJ fddt|
jD jfdd|}||
 t|||||}|r\|_ |_|j_jrVt |jpTdnd_|_!|S )a  handle special post compile parameters.

        These include:

        * "expanding" parameters -typically IN tuples that are rendered
          on a per-parameter basis for an otherwise fixed SQL statement string.

        * literal_binds compiled with the literal_execute flag.  Used for
          things like SQL Server "TOP N" where the driver does not accommodate
          N as a bound parameter.

        Nr   z/Mapping[str, Sequence[_BindProcessorType[Any]]]render_literal_valuec                 3  sb    | ],\}}t |d D ]"\}} v r-  |d   durd ||f   |d   fV  qqdS )r   N%s_%s_%s	enumerate)r   ituple_elementjr  )r   tuple_processorsr   r   r)  =  s    zBSQLCompiler._process_parameters_for_postcompile.<locals>.<genexpr>r   c                 3  s(    | ]\}} v r|  fV  qd S r  r   )r   r'  r  )r   single_processorsr   r   r)  H  s    
c                 s      | ]\}}|V  qd S r  r   r   r   r  r   r   r   r)  N  s    
c                 s  r  r  r   r  r   r   r   r)  U  s    c                 S  s   g | ]\}}|qS r   r   )r   
expand_keyr  r   r   r   rU  V  s    zCSQLCompiler._process_parameters_for_postcompile.<locals>.<listcomp>c                   sb   |  d}| }|  dr/|  dd}|d |d  d fdd|dD }|S )Nr   r6   ~~r  r   c                 3  s    | ]
}d  |f V  qdS z%s%s%sNr   )r   expbe_leftbe_rightr   r   r)  e  
    
z]SQLCompiler._process_parameters_for_postcompile.<locals>.process_expanding.<locals>.<genexpr>)r   splitrw   )ri  r'  exprtok)replacement_expressionsr  r   process_expanding\  s   

zJSQLCompiler._process_parameters_for_postcompile.<locals>.process_expandingc                   s    i | ]\}}| j  | qS r   r  )r   r  r'  r(  r   r   r4  q  r}  zCSQLCompiler._process_parameters_for_postcompile.<locals>.<dictcomp>c                   r~  r  r  r  r  r   r   r  y  r  zASQLCompiler._process_parameters_for_postcompile.<locals>.<lambda>r   )"r  r  r5  r  r#  r  r   r  r  r  r  rs  r  r  render_literal_bindparamrl  r  $_literal_execute_expanding_parameterliteral_executerJ  r   r  r  extendrk  r(  rv  _post_compile_patternr  r  r  r  r]  r  )r  r  r;  expanded_parameterspre_expanded_stringnew_positiontuppre_expanded_positiontupr!  new_processorsto_update_setsnumeric_positiontupnamesebnr  	parameter	to_updater  leep_resreplacement_exprr  r  expanded_stater   )r   r  r  r  r  r  r   rN    s   	













z/SQLCompiler._process_parameters_for_postcompilezsqlalchemy.engine.cursorc                 C  s   t jj}|j| jS )z(utility method used for unit tests only.)r%   	preloadedengine_cursorrT   _create_description_match_mapr  )r  cursorr   r   r   _create_result_map  s   zSQLCompiler._create_result_map_BindNameForColProtocol_get_bind_name_for_colCallable[[Any], str]c                 C  s
   | j }|S r  )r  )r  getterr   r   r   _within_exec_param_key_getter  s   z)SQLCompiler._within_exec_param_key_getterzsqlalchemy.engine.resultc                   s   t jj}| j| jd usJ | jj}trt|tsJ |j	}fdd|j
D d |j  d urK j| jd  | jv rJfdd}|nd |dd |j
D  fdd}|S )Nc                   s"   g | ]}t d  |d|fqS )rs  N)operatormethodcallerr   r  )param_key_getterr   r   rU    s    zKSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>.<listcomp>c                   s   |  | }|d ur|S | S r  rr  )	lastrowidr  param_value)autoinc_keyr   r   _autoinc_getter  s   zPSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>._autoinc_getterc                 S     g | ]}|j qS r   r1  r  r   r   r   rU        c                   sJ   dur   du rfddD S  fddD S )a  given cursor.lastrowid value and the parameters used for INSERT,
            return a "row" that represents the primary key, either by
            using the "lastrowid" or by extracting values from the parameters
            that were sent along with the INSERT.

            Nc                 3  s    | ]	\}}| V  qd S r  r   r   r  r  r%  r   r   r)        zWSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>.get.<locals>.<genexpr>c                 3  s:    | ]\}}| u rd urnn|V  qd S r  r   r  )autoinc_colautoinc_getterr  r  r   r   r)    s    


r   r  r  )r  r  getterslastrowid_processorrow_fnr  r   rs    s   
zDSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>.get)r%   r  engine_resultr  r  r  r   r  rA   r   primary_key_autoincrement_columnr   _cached_result_processorr  r  result_tuple)r  resultr  r   r  rs  r   )r  r  r  r  r  r  r  r   +_inserted_primary_key_from_lastrowid_getter  s2   

z7SQLCompiler._inserted_primary_key_from_lastrowid_getterc                   s   t jr
ddlm} ntjj}| jd usJ | jj}tr"t	|t
s"J | j|j}| j}|d us1J dd t|D tdfdd|jD  |dd |jD  fd	d
}|S )Nr6   )r  c                 S  rp  r   r   )r   idxr  r   r   r   r4  	  rq  zKSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.<dictcomp>z'List[Tuple[Callable[[Any], Any], bool]]c                   s<   g | ]}|v rt | d fn
t d |ddfqS )Trs  NF)r  
itemgetterr  r  )r  retr   r   rU  	  s    
zKSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.<listcomp>c                 S  r  r   r1  r  r   r   r   rU  	  r  c                   s    fddD S )Nc                 3  s(    | ]\}}|r|n| V  qd S r  r   )r   r  use_rowr  rowr   r   r)  "	  s
    
zWSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.get.<locals>.<genexpr>r   )r  r  )r  r  r  r   rs  !	  s   zDSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.get)typingr   enginer  r%   r  r  r  r  r  rA   r  r   r  r  r   r  r  )r  r  r  r   rX  rs  r   )r  r  r  r  r   +_inserted_primary_key_from_returning_getter  s*   z7SQLCompiler._inserted_primary_key_from_returning_getterc                 C     dS )zCalled when a SELECT statement has no froms, and no FROM clause is
        to be appended.

        Gives Oracle Database a chance to tack on a ``FROM DUAL`` to the string
        output.

        r  r   r(  r   r   r   default_from)	  s   zSQLCompiler.default_fromc                 K  s   |j j| fi |}|jD ]n}|| jvrq| j| }|j|j|j ddd}| j| }| | j|< | j|< || j|< | j|d || jv rN|  j|hO  _|| j	v r[|  j	|hO  _	| j
}|r{|\}	}
|jD ]}|j|
v rz|
|j }|	| | qgq|S )a  SQL compile the nested element of an _OverrideBinds with
        bindparams swapped out.

        The _OverrideBinds is not normally expected to be compiled; it
        is meant to be used when an already cached statement is to be used,
        the compilation was already performed, and only the bound params should
        be swapped in at execution time.

        However, there are test cases that exericise this object, and
        additionally the ORM subquery loader is known to feed in expressions
        which include this construct into new queries (discovered in #11173),
        so it has to do the right thing at compile time as well.

        TF)maintain_keyr  N)r  r  	translater  _with_valuer'  r  rl  r  r  r  _cloned_setrk  )r  override_bindsr  sqltextr7  bpnew_bpr   r  rP  rO  cbr   r   r   visit_override_binds3	  s6   
	

	







z SQLCompiler.visit_override_bindsc                 K  s   d|j j| fi | d S Nr  r  r  r  )r  groupingasfromr  r   r   r   visit_groupings	     zSQLCompiler.visit_groupingc                 K  s   d|j j| fi | d S r#  r$  )r  r%  r  r   r   r   visit_select_statement_groupingv	  r(  z+SQLCompiler.visit_select_statement_groupingc              
   K  s   | j rN| jjrNztd| j d d }W n ty% } ztd|d }~ww |j\}}}|r1|}	n|}	|jj	}
|
d urN|
j
|	v rN|
|	|
j
 rN|jj	|d< | j|jfd|i|S )N'Union[SelectState, CompoundSelectState]rY  r  ECan't resolve label reference for ORDER BY / GROUP BY / DISTINCT etc.render_label_as_labelwithin_columns_clause)rv  r  supports_simple_order_by_labelr   KeyErrorr7   r  _label_resolve_dictr  _order_by_label_elementr   shares_lineager  )r  r  r-  r  r  ke	with_cols
only_froms	only_colsresolve_dictorder_by_elemr   r   r   visit_label_referencey	  sL   

z!SQLCompiler.visit_label_referencec              
   K  s   | j s	| |jS ztd| j d d }W n ty2 } ztj|jdtj	|d W Y d }~nd }~ww |j
\}}}z|rB||j }	n||j }	W n tyf }
 ztj|jdtj	|
d W Y d }
~
d S d }
~
ww |	|d< | j|	fd|i|S )Nr*  rY  r  r+  )extraexc_clsr  r,  r-  )rv  r  _text_clauser   r/  r   _no_text_coercionr  r7   r  r0  )r  r  r-  r  r  r3  r4  r5  r6  r  r  r   r   r   visit_textual_label_reference	  sN   
z)SQLCompiler.visit_textual_label_referencec                 K  s   |o| }||u }	|	s|rt |jtjr| d|j}
n|j}
|rN|d ur5||
|j||
f|j | |j |jj| fddd|t	t
j  | j||
 S |	rW| j||
S |jj| fddi|S )NcolidentT)r-  within_label_clauser-  F)r  r   r   rD   _truncated_identifierr  r   r  r  	OPERATORSr    as_r  format_label)r  r  add_to_result_mapr@  r-  r,  result_map_targetsr  render_label_with_asrender_label_only	labelnamer   r   r   visit_label	  sJ   
zSQLCompiler.visit_labelc                 C  s
   t d)Nz:Cannot compile Column object until its 'name' is assigned.r7   r  r  r   r   r   r   _fallback_column_name
  s   z!SQLCompiler._fallback_column_namec                 K  s   |j }| j|fi |S r  )	_resolvedr  )r  r  r  sql_elementr   r   r   visit_lambda_element
  s   z SQLCompiler.visit_lambda_elementr   ColumnClause[Any]rE  Optional[_ResultMapAppender]include_tablerF  r   ambiguous_table_name_map Optional[_AmbiguousTableNameMap]c                 K  s6  |j  }}|d u r| |}|j}	|	st|tjr| d|}|d ur<|||jf| }
|jr4|
|jf7 }
||||
|j	 |	rD| 
|}n| j|}|j}|d u sV|rV|jsX|S | j|}|ri| j|d }nd}trtt|tstJ |j }|s|r||v r|| }t|tjr| d|}|| j| d | S )Nr?  r  r  alias)r   rM  
is_literalr  r   rD   rA  r'  	_tq_labelr   escape_literal_columnr  quoter   named_with_columnschema_for_objectquote_schemar   rO   )r  r   rE  rS  rF  rT  r  r   	orig_namerW  targetsr   effective_schemaschema_prefix	tablenamer   r   r   visit_column
  sD   
	
zSQLCompiler.visit_columnc                 K  s   | j |jS r  )r  format_collation	collationr  r  r  r   r   r   visit_collationJ
  r  zSQLCompiler.visit_collationc                 K     |j S r  r   )r  
fromclauser  r   r   r   visit_fromclauseM
     zSQLCompiler.visit_fromclausec                 K  rh  r  ri  )r  rh  r  r   r   r   visit_indexP
  rl  zSQLCompiler.visit_indexc                 K  s*   ||d< | j |d< | jjj|jfi |S )Ntype_expressionr  )r  r  type_compiler_instancer  r   )r  
typeclauser  r   r   r   visit_typeclauseS
  s   
zSQLCompiler.visit_typeclausec                 C     | j jr
|dd}|S Nr  %%r  _double_percentsreplacer  textr   r   r   post_process_textZ
     zSQLCompiler.post_process_textc                 C  rr  rs  ru  rx  r   r   r   rY  _
  r{  z!SQLCompiler.escape_literal_columnc              	     sR    fdd}j sd_|r|d d ftj tdd t|jS )Nc                   s>   |  d}|jv rjj| fi  S j|fi  S r  )r   _bindparamsr  bindparam_string)ri  r   r  r  
textclauser   r   do_bindparame
  s   

z2SQLCompiler.visit_textclause.<locals>.do_bindparamTc                 S  s
   |  dS r  r  r  r   r   r   r  w
  s   
 z.SQLCompiler.visit_textclause.<locals>.<lambda>)	rv  r  r$   NULLTYPEBIND_PARAMS_ESCrv  BIND_PARAMSrz  ry  )r  r  rE  r  r  r   r~  r   visit_textclaused
  s   zSQLCompiler.visit_textclausec                 K  s   | j  }|r	| jn| j d }t t |d}| j | |jr%| || |p6|dko0|ddp6|dd}|rY|j | _| _	|j oHt
|j| _|jD ]}	| j|	d| jd qM| j|jfi |}
| jrw|smt| j nd }| j|d	|
 }
| j d |
S )
NrY  r  r  r#   r   r  Fr  T)r-  rE  nesting_level)rv  _default_stack_entryro   rk  _independent_ctes_dispatch_independent_ctesrs  r5  r  r  r  column_argsr  r  _add_to_result_mapr  r"  rz  _render_cte_clauserl  )r  tafcompound_indexr&  r  toplevelrf  	new_entrypopulate_result_maprT  ry  r  r   r   r   visit_textual_select}
  sB   

	


z SQLCompiler.visit_textual_selectc                 K  r  NNULLr   r  r  r  r   r   r   
visit_null
  r  zSQLCompiler.visit_nullc                 K     | j jrdS dS )Nr~   1r  supports_native_booleanr  r   r   r   
visit_true
     zSQLCompiler.visit_truec                 K  r  )Nr   0r  r  r   r   r   visit_false
  r  zSQLCompiler.visit_falsec                   s$   | dd  fdd|D D S )Nc                 s      | ]}|r|V  qd S r  r   r   sr   r   r   r)  
      
z7SQLCompiler._generate_delimited_list.<locals>.<genexpr>c                 3  "    | ]}|j fi  V  qd S r  r  rS  r  r  r   r   r)  
       rw   )r  r   	separatorr  r   r  r   _generate_delimited_list
  s   
z$SQLCompiler._generate_delimited_listc                   sl   t jtjt jjt jj|\}}|dkr|d jfi  S t	tj }|
dd  fdd|D D S )Nr   r   c                 s  r  r  r   r  r   r   r   r)  
  r  z;SQLCompiler._generate_delimited_and_list.<locals>.<genexpr>c                 3  r  r  r  rS  r  r   r   r)  
  r  )r   BooleanClauseList_process_clauses_for_booleanr    and_True_
_singletonFalse_r  rB  rw   )r  clausesr  lccr  r   r  r   _generate_delimited_and_list
  s   

z(SQLCompiler._generate_delimited_and_listc                 K  s   d| j |fi | S N(%s))visit_clauselist)r  
clauselistr  r   r   r   visit_tuple
  s   zSQLCompiler.visit_tuplec                 K  s4   |j }|d u r
d}nt|j  }| j|j|fi |S Nr  )r  rB  r  r  )r  r  r  sepr   r   r   r  
  s
   
zSQLCompiler.visit_clauselistc              
   K  s|   |j }| |dd }|r|||fi |S zt| }W n ty. } zt| ||d }~ww d|d< | j|j|fi |S )Nexpression_clauselistT_in_operator_expression)r  _get_operator_dispatchrB  r/  r7   r  r  r  )r  r  r  	operator_dispopstringr  r   r   r   visit_expression_clauselist
  s$   z'SQLCompiler.visit_expression_clauselistc                 K  s   d}|j d ur||j j| fi |d 7 }|jD ]\}}|d|j| fi | d |j| fi | d 7 }q|jd urL|d|jj| fi | d 7 }|d7 }|S )NzCASE r  zWHEN z THEN zELSE END)r  r  whenselse_)r  clauser  r   condr  r   r   r   
visit_case
  s*   

zSQLCompiler.visit_casec                 K     |j j| fi |S r  )typed_expressionr  )r  type_coercer  r   r   r   visit_type_coerce
     zSQLCompiler.visit_type_coercec                 K  s^   |j j| fi |}td|}d|jj| fi ||r!|dn||r+|df S df S )Nz(.*)( COLLATE .*)zCAST(%s AS %s)%sr   r6   r  )rp  r  r(  matchr  r   )r  r   r  type_clauser  r   r   r   
visit_cast  s   zSQLCompiler.visit_castc              	   K  s   d|d t ju r
dn2|d t ju rdn)|d dk r,d| jt t|d fi |f nd| jt |d fi |f |d t ju rGdf S |d t ju rRdf S |d dk rmd| jt t|d fi |f f S d| jt |d fi |f f S )	Nz	%s AND %sr   zUNBOUNDED PRECEDINGzCURRENT ROWz%s PRECEDINGz%s FOLLOWINGr   zUNBOUNDED FOLLOWING)r   RANGE_UNBOUNDEDRANGE_CURRENTr  literalabs)r  range_r  r   r   r   _format_frame_clause  sJ   	!"z SQLCompiler._format_frame_clausec                   s   |j jfi  }|jrdj|jfi   }n|jr*dj|jfi   }nd }d|d fddd|jfd|jffD |rF|gng  f S )	NzRANGE BETWEEN %szROWS BETWEEN %sz%s OVER (%s)r  c                   s:   g | ]\}}|d urt |rd||jfi  f qS )Nz%s BY %s)rz  r  )r   wordr  r  r  r   r   rU  P  s    z*SQLCompiler.visit_over.<locals>.<listcomp>	PARTITIONORDER)r  r  r  r  rowsrw   partition_byorder_by)r  overr  ry  r  r   r  r   
visit_over@  s4   

	zSQLCompiler.visit_overc                 K  ,   d|j j| fi ||jj| fi |f S )Nz%s WITHIN GROUP (ORDER BY %s))r  r  r  )r  withingroupr  r   r   r   visit_withingroup]     zSQLCompiler.visit_withingroupc                 K  r  )Nz%s FILTER (WHERE %s))funcr  	criterion)r  
funcfilterr  r   r   r   visit_funcfilterc  r  zSQLCompiler.visit_funcfilterc                 K  s.   | j |j|j}d||jj| fi |f S )NzEXTRACT(%s FROM %s))extract_maprs  fieldr  r  )r  extractr  r  r   r   r   visit_extracti  s
   zSQLCompiler.visit_extractc                 K  s2   | j |jfi |}| j|fi |}d||f S )Nz(%s).%s)visit_functionfnrc  )r  r  r  compiled_fncompiled_colr   r   r   visit_scalar_function_columnp  s   z(SQLCompiler.visit_scalar_function_columnr  Function[Any]c                   s   |d ur||j |j |j f|j td|j   d }|r&||fi |}nJt| jd   r9|jr8 d7  n|j  j	
 sHt tjrNj	 n   d  d fdd|jD  g dj|fi |i }|jrw|d7 }|S )Nzvisit_%s_funcz%(expr)sr  c                   s4   g | ]}j |st tjrj |n|qS r   )r  _requires_quotes_illegal_charsr  r   r0   rZ  )r   r  r   r  r   r   rU    s    

z.SQLCompiler.visit_function.<locals>.<listcomp>r  z WITH ORDINALITY)r   r   getattrlowerr   rs  _deannotater  	_has_argsr  r  r  r   r0   rZ  rw   packagenamesfunction_argspec_with_ordinality)r  r  rE  r  r  ry  r   r  r   r  u  s<   

	zSQLCompiler.visit_functionc                 K  s   |  |jS r  )visit_sequencesequence)r  
next_valuer  r   r   r   visit_next_value_func     z!SQLCompiler.visit_next_value_funcc                 K     t d| jj )Nz2Dialect '%s' does not support sequence increments.r  r  r   r  r  r  r   r   r   r    
   zSQLCompiler.visit_sequencec                 K  r  r  )clause_exprr  )r  r  r  r   r   r   r    r  zSQLCompiler.function_argspecc                   sl  j  }|j|fi }|rjs|_|j}|rjnj d }|p-| o-|dd}	|dkr6||d< j |d |d |||	d |jrO| j	|j
 }
d	|
 d	  fd
dt|jD }dd< |j|fi tdd i7 }|j|fi 7 }|jr|j|fi 7 }jr|stj nd }j|dd| }j d |S )NrY  r  Fr   r  r  r  )r  r  r#   r  r  r  c                 3  s,    | ]\}}|j f |d V  qdS ))r&  r  Nr  )r   r  rT  r&  r  r  r   r   r)    s    
z4SQLCompiler.visit_compound_select.<locals>.<genexpr>rS  r&  Tr  include_following_stackr   )rv  _compile_state_factoryr  r  r  rs  rk  r  r  compound_keywordskeywordrw   r  selectsgroup_by_clauser  order_by_clause_has_row_limiting_clause_row_limit_clauser"  rz  r  rl  )r  csr&  r  r  r  r  compound_stmtrf  need_result_mapr  ry  r  r   r  r   visit_compound_select  sX   


	$z!SQLCompiler.visit_compound_selectc                 K  s.   |j d ur| j|fi |S | j|fi |S r  )_fetch_clausefetch_clauselimit_clause)r  r  r  r   r   r   r    s   
zSQLCompiler._row_limit_clausec                 C  s(   d|j ||r
d| ndf }t| |d S )Nzvisit_%s_%s%sr  r  )r   r  )r  r  
qualifier1
qualifier2attrnamer   r   r   r    s   z"SQLCompiler._get_operator_dispatchc                 K  s   |d ur||f7 }||d< ||d< |j r=|jrtd| |j dd}|r0|||j fi |S | j|t|j  fi |S |jra| |jdd}|rT|||jfi |S | j|t|j fi |S td)NrE  rF  zFUnary expression does not support operator and modifier simultaneouslyunaryr  modifierz,Unary expression has no operator or modifier)r  r  r7   r  r   _generate_generic_unary_operatorrB   _generate_generic_unary_modifier)r  r  rE  rF  r  r  r   r   r   visit_unary  sB   


zSQLCompiler.visit_unaryc                 K  s   | j jr,| j|jfi |d | jt|j|jjjt	j
u r!|jjnt	
 fi | S | j|jfi |d | j|jfi | S )N / )r  div_is_floordivr  rx   r   Castr   r   _type_affinityr$   Numericr  r   r  r  r   r   r   visit_truediv_binary$  s0   
z SQLCompiler.visit_truediv_binaryc                 K  st   | j jr"|jjjtju r"| j|jfi |d | j|jfi | S d| j|jfi |d | j|jfi |  S )Nr  z	FLOOR(%s))	r  r  r   r   r   r$   Integerr  rx   r"  r   r   r   visit_floordiv_binary?  s    z!SQLCompiler.visit_floordiv_binaryc                 K  s:   |j s| jjr| j|jfi |S d| j|jfi | S )Nz%s = 1_is_implicitly_booleanr  r  r  r  r  r  r  r  r   r   r   visit_is_true_unary_operatorP  s   z(SQLCompiler.visit_is_true_unary_operatorc                 K  s>   |j s| jjrd| j|jfi | S d| j|jfi | S )NNOT %sz%s = 0r&  r(  r   r   r   visit_is_false_unary_operatorY  s   z)SQLCompiler.visit_is_false_unary_operatorc                 K  s   d| j |tjd S )Nr*  )override_operator)visit_binaryr    match_opr"  r   r   r   visit_not_match_op_binaryb  s   z%SQLCompiler.visit_not_match_op_binaryc                 K  s   d| j |t| fi | S r  )_generate_generic_binaryrB  r"  r   r   r   visit_not_in_op_binaryg  s
   z"SQLCompiler.visit_not_in_op_binaryc                 K  sn   |t ju rt|dkrdddd |D  S dS |t ju r2t|dkr0dddd |D  S d	S | |S )
Nr   z(%s)) OR (1 = 1r   c                 s      | ]}d V  qdS r  Nr   r   r  r   r   r   r)  s      z6SQLCompiler.visit_empty_set_op_expr.<locals>.<genexpr>zNULL) OR (1 = 1z(%s)) AND (1 != 1c                 s  r2  r3  r   r4  r   r   r   r)  z  r5  zNULL) AND (1 != 1)r    	not_in_oprz  rw   in_opvisit_empty_set_expr)r  r   	expand_opr  r   r   r   visit_empty_set_op_expro  s   


z#SQLCompiler.visit_empty_set_op_exprc                 K  r  )Nz3Dialect '%s' does not support empty set expression.r  r  element_typesr  r   r   r   r8    r  z SQLCompiler.visit_empty_set_exprc           	        sb  j j}|s.|jr!jjrdndj jj }d|fS j gj}d|fS |jsE|jrgt	|d t
jrgt	|d ttfsg|jrLtdjjrRdnddfddt|D  }d|fS |rj}||}|rx|ds|J d	|dd
}|d |d  d fdd|D }d|fS dfdd|D }d|fS )NVALUES r  r   z?bind_expression() on TupleType not supported with literal_bindsr   c                 3  s:    | ]\}}d d fddt| jjD  V  qdS )r  r   c                 3  s     | ]\}}  ||V  qd S r  r  )r   r  
param_typer(  r   r   r)    s
    

z[SQLCompiler._literal_execute_expanding_parameter_literal_binds.<locals>.<genexpr>.<genexpr>N)rw   r  r   r  )r   r  r  r  r  r   r   r)    s    

zQSQLCompiler._literal_execute_expanding_parameter_literal_binds.<locals>.<genexpr>r6   )unexpected format for expanding parameterr  r   r  c                 3  s(    | ]}d   |jf V  qdS r  r  r   r   r  r  r  r  r  r   r   r)    s    
c                 3  s    | ]
} | jV  qd S r  rA  rB  r?  r   r   r)    r  r   )r   r  r  r  tuple_in_valuesr:  r  r9  _isnullr  collections_abcr   r   bytes_has_bind_expressionr  rw   r  r  searchr   r  )	r  r  r  bind_expression_templatetyp_dialect_implreplacement_expressionpost_compile_patternri  r  r   rC  r   2_literal_execute_expanding_parameter_literal_binds  s`   
=
6
'
z>SQLCompiler._literal_execute_expanding_parameter_literal_bindsc                   sd  j r	|S j}j|jrjnjjjr-j	r-fdd nfdd |sTg j
rGjjj}|fS jgj}|fS j
skjrt|d tjrt|d ttfsjrpJ fddt|dD |jrdnd	d
 fddt|D  }|fS fddt|dD d
 fddD }|fS )Nc                   s    j d| i S Nr   )render_bind_castr   ri  )bind_templater  r  rK  r   r   _render_bindtemplate  s
   
zNSQLCompiler._literal_execute_expanding_parameter.<locals>._render_bindtemplatec                   s    d| i S rO  r   ri  )rQ  r   r   rR    r  r   c                   s6   g | ]\}}t |d D ]\}}d ||f |fqqS )r   r  r  )r   r  r  r  r  ri  r   r   rU    s    zDSQLCompiler._literal_execute_expanding_parameter.<locals>.<listcomp>r   r=  r  r   c                 3  s:    | ]\ d d  fddtD  V  qdS )r  r   c                 3  s0    | ]\}} t  |  d  V  qdS )r   N)rz  )r   r  r  )rR  r  r  r  r   r   r)    s    
zMSQLCompiler._literal_execute_expanding_parameter.<locals>.<genexpr>.<genexpr>N)rw   r  )r   )rR  r  )r  r  r   r)  	  s    

zCSQLCompiler._literal_execute_expanding_parameter.<locals>.<genexpr>c                   s    g | ]\}}d  |f |fqS )z%s_%sr   )r   r  r  ri  r   r   rU    r}  c                 3  s    | ]	\}} |V  qd S r  r   r  )rR  r   r   r)    s    
)r  rN  r  r   r  r  r  r  _bind_typing_render_castsrP  r  r:  r  r9  rE  r  rF  r   r   rG  	_is_arrayr  rD  rw   )r  r   r  r  r  rL  r   )rR  rQ  r   r  r  r  rK  r   r    sd   	
+
'



z0SQLCompiler._literal_execute_expanding_parameterc              
   K  s  |r;t |jr;|d ur)|d }|jtt|jj	|g t|j
j	|g  n|jtt|jj	t|j
j	 | jrPt|jtjrPt|j
tjrPd|d< |pT|j}| |dd }	|	rg|	||fi |S zt| }
W n ty } zt| ||d }~ww | j||
f||d|S )Nenclosing_lateralTr  r   )from_linterlateral_from_linter)r    is_comparisonr  rj  rJ  r  productr)   rx   r*   r   r
  r  r   rE   r  rB  r/  r7   r  r0  )r  r   r,  eager_groupingrV  rW  r  rU  r  r  r  r  r   r   r   r-     sZ   	

	
zSQLCompiler.visit_binaryc                 K  s   | j |jfi |S r  )r  sql_functionr(  r   r   r   &visit_function_as_comparison_op_binaryX  r  z2SQLCompiler.visit_function_as_comparison_op_binaryc                 K  s`   | j jr| j|jfi |d | j|jfi | S | j|jfi |d | j|jfi | S )Nz %% r   )r  rv  r  rx   r   r"  r   r   r   visit_mod_binary[  s   zSQLCompiler.visit_mod_binaryc                 K  s.   |j |d< | j|d| |j d fi |S )NrZ  r  )rZ  r0  rY  r  r(  r   r   r   visit_custom_op_binaryi  s   
z"SQLCompiler.visit_custom_op_binaryc                 K  s    | j || |jd fi |S r  )r  rY  r  r(  r   r   r   visit_custom_op_unary_operatorq  
   z*SQLCompiler.visit_custom_op_unary_operatorc                 K  s    | j |d| |j fi |S r  )r  rY  r  r(  r   r   r   visit_custom_op_unary_modifierv  r`  z*SQLCompiler.visit_custom_op_unary_modifierc                 K  sf   | dd}d|d< |j|d< |jj| fd|i|| |jj| fd|i| }|r1|r1d| }|S )Nr  FT
_binary_oprZ  r  )rs  r  rx   r  r   )r  r   r  rZ  r  r  ry  r   r   r   r0  {  s0   

z$SQLCompiler._generate_generic_binaryc                 K  s   ||j j| fi | S r  r$  r  r  r  r  r   r   r   r       z,SQLCompiler._generate_generic_unary_operatorc                 K  s   |j j| fi || S r  r$  rc  r   r   r   r    rd  z,SQLCompiler._generate_generic_unary_modifierc                 C  s   t jdtjdS )Nz'%')r   )r   literal_columnr$   
STRINGTYPEr(  r   r   r   _like_percent_literal  s   z!SQLCompiler._like_percent_literalc                 K  s   d|j j| fi | dS )Nzlower(r  r$  rf  r   r   r   $visit_ilike_case_insensitive_operand  r(  z0SQLCompiler.visit_ilike_case_insensitive_operandc                 K  6   |  }| j}||j||_| j||fi |S r  _clonerg  concatr   visit_like_op_binaryr  r   r  r  percentr   r   r   visit_contains_op_binary     z$SQLCompiler.visit_contains_op_binaryc                 K  ri  r  rk  rg  rl  r   visit_not_like_op_binaryrn  r   r   r   visit_not_contains_op_binary  rq  z(SQLCompiler.visit_not_contains_op_binaryc                 K  F   |  }| j}t|j|_|t|j||_| j||fi |S r  rk  rg  r  rx   rl  r   visit_ilike_op_binaryrn  r   r   r   visit_icontains_op_binary     z%SQLCompiler.visit_icontains_op_binaryc                 K  ru  r  rk  rg  r  rx   rl  r   visit_not_ilike_op_binaryrn  r   r   r   visit_not_icontains_op_binary  ry  z)SQLCompiler.visit_not_icontains_op_binaryc                 K  0   |  }| j}||j|_| j||fi |S r  )rk  rg  _rconcatr   rm  rn  r   r   r   visit_startswith_op_binary     z&SQLCompiler.visit_startswith_op_binaryc                 K  r}  r  )rk  rg  r~  r   rs  rn  r   r   r   visit_not_startswith_op_binary  r  z*SQLCompiler.visit_not_startswith_op_binaryc                 K  @   |  }| j}t|j|_|t|j|_| j||fi |S r  )rk  rg  r  rx   r~  r   rw  rn  r   r   r   visit_istartswith_op_binary  
   z'SQLCompiler.visit_istartswith_op_binaryc                 K  r  r  )rk  rg  r  rx   r~  r   r{  rn  r   r   r   visit_not_istartswith_op_binary  r  z+SQLCompiler.visit_not_istartswith_op_binaryc                 K  r}  r  rj  rn  r   r   r   visit_endswith_op_binary  r  z$SQLCompiler.visit_endswith_op_binaryc                 K  r}  r  rr  rn  r   r   r   visit_not_endswith_op_binary  r  z(SQLCompiler.visit_not_endswith_op_binaryc                 K  r  r  rv  rn  r   r   r   visit_iendswith_op_binary  r  z%SQLCompiler.visit_iendswith_op_binaryc                 K  r  r  rz  rn  r   r   r   visit_not_iendswith_op_binary  r  z)SQLCompiler.visit_not_iendswith_op_binaryc                 K  Z   |j dd }d|jj| fi ||jj| fi |f |d ur*d| |tj  S d S )Nr)  z
%s LIKE %s ESCAPE r  	modifiersrs  rx   r  r   r  r$   rf  r  r   r  r  r)  r   r   r   rm    s   z SQLCompiler.visit_like_op_binaryc                 K  r  )Nr)  z%s NOT LIKE %sr  r  r  r  r   r   r   rs    s   z$SQLCompiler.visit_not_like_op_binaryc                 K  >   |t ju r| }t|j|_t|j|_| j||fi |S r  )r    ilike_oprk  r  rx   r   rm  r"  r   r   r   rw    
   
z!SQLCompiler.visit_ilike_op_binaryc                 K  r  r  )r    not_ilike_oprk  r  rx   r   rs  r"  r   r   r   r{    r  z%SQLCompiler.visit_not_ilike_op_binaryc                 K  *   |j dd}| j||rdndfi |S )Nr   Fz BETWEEN SYMMETRIC z	 BETWEEN r  rs  r0  r  r   r  r  r   r   r   r   visit_between_op_binary  s   z#SQLCompiler.visit_between_op_binaryc                 K  r  )Nr   Fz NOT BETWEEN SYMMETRIC z NOT BETWEEN r  r  r   r   r   visit_not_between_op_binary  s   
z'SQLCompiler.visit_not_between_op_binaryc                 K     t d| jj Nz/%s dialect does not support regular expressionsr7   r  r  r   r"  r   r   r   visit_regexp_match_op_binary&  
   z(SQLCompiler.visit_regexp_match_op_binaryc                 K  r  r  r  r"  r   r   r    visit_not_regexp_match_op_binary,  r  z,SQLCompiler.visit_not_regexp_match_op_binaryc                 K  r  )Nz;%s dialect does not support regular expression replacementsr  r"  r   r   r   visit_regexp_replace_op_binary2  r  z*SQLCompiler.visit_regexp_replace_op_binaryc                 K  s  |sW|j | j}|jrW||}	| j|	fd||o|j ||d|}
|jrUtd|
}|s3J dd|	d|	d|	df }
|rU| j
|fd|
d	|}d
| S |
S |si|pb|jpb|ob| j}|pg|j}nd}|r| j
|fddi|}|jrd
| }|S | |}|| jv r| j| }||ur|js|jr|j|js|j|jstd| |j|jkrtd|f |js|jr|jr|jrtdtd|j d|j d| | j|j< | j|< | j}|r	|\}}|jD ]}|j|v r||j }|| | q|jrd| _|r-|rd| _|r%|  j|hO  _n|  j|hO  _| j|f||j|j d|}|jrDd
| }|S )NT)skip_bind_expressionr-  literal_bindsr  render_postcompilez&^(.*)\(__\[POSTCOMPILE_(\S+?)\]\)(.*)$r@  z$(__[POSTCOMPILE_%s~~%s~~REPL~~%s~~])r6   r   r  )r-  rJ  r  Fr-  zIBind parameter '%s' conflicts with unique bind parameter of the same namea  Can't reuse bound parameter name '%s' in both 'expanding' (e.g. within an IN expression) and non-expanding contexts.  If this parameter is to receive a list/array value, set 'expanding=True' on it for expressions that aren't IN, otherwise use a different parameter name.a)  Encountered unsupported case when compiling an INSERT or UPDATE statement.  If this is a multi-table UPDATE statement, please provide string-named arguments to the values() method with distinct names; support for multi-table UPDATE statements that target multiple tables for UPDATE is very limitedzbindparam() name 'z' is reserved for automatic usage in the VALUES or SET clause of this insert/update statement.   Please use a name other than column name when using bindparam() with insert() or update() (for example, 'b_z').)post_compile	expandingbindparam_type)r   dialect_implr  rH  bind_expressionr  r  r(  r  r   r  r  r
  _truncate_bindparamr  r   r  intersectionr  r7   r  _is_crudr'  r  rk  
isoutparamhas_out_parametersr  r  r  r}  )r  r  r-  r  r  r  r  r  implr  wrappedri  r  r  r   existingr  rP  rO  r  r!  r   r   r   visit_bindparam8  s   

	


	



zSQLCompiler.visit_bindparamc                 C  r  r  r  )r  r   
dbapi_typer  r   r   r   rP    rl  zSQLCompiler.render_bind_castc           
      K  s   |t ur|}n/|jd u r3|jd u r3|dd }|r)|tjtjfvr)td|j	f | j
tjfi |S |j}|jrG| j}||||d\}}	|	S | ||jS )Nrb  zBound parameter '%s' rendering literal NULL in a SQL expression; comparisons to NULL should not use operators outside of 'is' or 'is not')rJ  )r.   r  r  rs  r    is_is_notr%   warn_limitedr'  r  r$   r  r  r  rN  r  r   )
r  r  r  rJ  r  r  opleepr  r  r   r   r   r    s(   
z$SQLCompiler.render_literal_bindparamc              
   C  s   |du r|j s| tj S || j}|r9z||W S  ty8 } zt	dt
| d| d|d}~ww t	dt
| d| )a  Render the value of a bind parameter as a quoted literal.

        This is used for statement sections that do not accept bind parameters
        on the target driver/database.

        This should be implemented by subclasses using the quoting services
        of the DBAPI.

        Nz Could not render literal value "z" with datatype z); see parent stack trace for more detail.z:No literal value renderer is available for literal value ")should_evaluate_noner  r   Null	_instance_cached_literal_processorr  r  r7   r  sql_util_repr_single_value)r  r  r   	processorer   r   r   r    s2   

z SQLCompiler.render_literal_valuec                 C  s@   || j v r
| j | S |j}t|tjr| d|}|| j |< |S )Nr  )r  r'  r  r   rD   rA  )r  r  r  r   r   r   r  &  s   


zSQLCompiler._truncate_bindparamident_classr   rD   c                 C  s   ||f| j v r| j ||f S || j}t|| jd krB| j|d}|dt| jd d d t|dd   }|d | j|< n|}|| j ||f< |S )N   r   r   r  r6   )	rD  	apply_maprC  rz  rA  rE  rs  maxhex)r  r  r   anonnamecounter	truncnamer   r   r   rA  3  s   z!SQLCompiler._truncated_identifierc                 C  s
   || j  S r  rC  r  r   r   r   r   
_anonymizeH  r  zSQLCompiler._anonymizer  r  escaped_fromr  Optional[TypeEngine[Any]]accumulate_bind_namesOptional[Set[str]]visited_bindparamr  c                   s  |d ur	| | |d ur|| |s) j|r) j fdd|}	|}|	}|r4 j||i _|rTd| }
|r>|
S |d urR| j}|j	rR 
|||
}
|
S  jtju rb jd|i }
n jd|i }
|d ur jjr| j}|j
r 
|||
}
|
S )Nc                   s    j | d S )Nr   )r  r   r  r(  r   r   r  e      z.SQLCompiler.bindparam_string.<locals>.<lambda>z__[POSTCOMPILE_%s]r   )addrk  r  rI  rv  r  r   r  r  render_literal_castrP  r  rX  rY  r  r  rS  )r  r   r  r  r  r  r  r  r  new_namer  	type_implr   r(  r   r}  K  sJ   


zSQLCompiler.bindparam_stringc                 C  sF   |  }|dd  t|j|jD ]\}}|j| fd|i| qd S )Ncte_opts)copyrl  r  r  _independent_ctes_optsr  )r  stmtr  local_kwcteoptr   r   r   r    s   z&SQLCompiler._dispatch_independent_ctesr  rM   r&  ashint	fromhintsOptional[_FromHintsType]visiting_cteOptional[CTE]rV  Optional[FromLinter]r  selectable._CTEOptsc                   s     }	|	 ju sJ ||d< |j}
t|
tjr d|
}
d}d}| }|jp+|j}| j	v rx j	| \}}}||
ks?J ||
f} j
| }|jrw|jrStd||
f}|r^t jnd}||
f }} j
|= | j
|< ||f  j	|< n|rt jnd}||
f}| j
v r j
| }nd }|d ur||u }||ju s||u rd}n6||ju r|	|= | }||u sJ ||u sJ  j	|= n|jd us|jd ur||rd}ntd|
 |s|sd S |jd ur|j}|jj}t|tjr d|}n|}d }|r| j
|< ||f  j	|< | jvr j|fi | |s||	vr|jr-d _ j||
}|jrU|j}dd	 |dD }|d
d fdd|D  7 }|dddu s`J  jsp|jj fd|i|S  j||jfi |}|jj fddi|}|d||f 7 }|jr|d j||jfi | 7 }||	|< |r|r|
|j |! < |s|r j||
S |r j||} j"|
r׈ j#|
}
| $|
7 }|S  j||
S d S )Nr  rV  TFz6CTE is stated as 'nest_here' in more than one locationr   z5Multiple, unrelated CTEs found with the same name: %rc                 S  s"   g | ]\}}}}}|s|p|qS r   r   )r   r  
proxy_namefallback_label_namerT  repeatedr   r   r   rU    s    z)SQLCompiler.visit_cte.<locals>.<listcomp>r  r   c                 3  s"    | ]} j j| jd V  qdS )r  N)r  format_label_namerC  )r   identr(  r   r   r)  -      
z(SQLCompiler.visit_cte.<locals>.<genexpr>subqueryr&  z AS %s
(%s)r  )%rc  r"  r   r  r   rD   rA  _get_reference_ctenestingr$  r#  r7   r  rz  rv  	_restates_is_clone_ofcompare
_cte_alias	visit_cte	recursiver%  r  format_aliasr  _generate_columns_plus_namesrw   rs  r  _generate_prefixes	_prefixes	_suffixesri  r)   _requires_quotesrZ  get_render_as_alias_suffix)r  r  r&  r  r  r  rV  r  r  	self_ctescte_name
is_new_cteembedded_in_current_named_cte_reference_cter  	cte_levelr  existing_cte_optscte_level_nameexisting_cteold_level_namenew_level_nameexisting_cte_reference_ctepre_alias_ctecte_pre_alias_namery  
col_source
recur_colsprefixesr   r   r(  r   r    s  










	
zSQLCompiler.visit_ctec                 K  s8   |j rd |d< |jr| j|fi |S | j|fi |S )NrV  )joins_implicitly_is_lateralvisit_lateralvisit_aliasrf  r   r   r   visit_table_valued_aliasa  s
   z$SQLCompiler.visit_table_valued_aliasc                 K     | j |fi |S r  )rc  rf  r   r   r   visit_table_valued_columni  r  z%SQLCompiler.visit_table_valued_columnc
              	     s  |rdvr j sJ  d< |	rdvrdv r|	d< |d urD|j u rD jjf||||| d}|rB|s=|rBd|f }|S   }d< |sN|r`t jtjr]d j}n j}|rij	 |S |r|	rt||	j
  <  jjfd|d}|rd|f }|j	 | } jr jr|dd	 fd
d jD  7 }|r |v r| |  |}|S  jjfd|iS )NrU  rW  )r&  r  iscrudr  lateralenclosing_aliasr  r  rV  T)r&  r  r   c                 3  sH    | ]}d j |j jrdjjj|jfi  ndf V  qdS )z%s%s %sr  N)r  rZ  r   _render_derived_w_typesr  ro  r  r   r  rV  r  r  r   r   r)    s"    
z*SQLCompiler.visit_alias.<locals>.<genexpr>r  )r	  r  r  r  r   r   rD   rA  r  r  ri  r)   r  _supports_derived_columns_render_derivedrw   rT  format_from_hint_text)r  rV  r&  r  r  r  r  r  r  rV  r  r   
alias_namer  r   r  r   r  l  s   



zSQLCompiler.visit_aliasc                 K  s   d|d< | j |fi |S )NTr  r  )r  r  r  r   r   r   visit_subquery  s   zSQLCompiler.visit_subqueryc                 K  s   d|d< d| j |fi | S )NTr  z
LATERAL %sr  )r  lateral_r  r   r   r   r
    s   zSQLCompiler.visit_lateralc                 K  sZ   d| j |fddi|| j| fi |f }|jd ur+|d|jj| fi | 7 }|S )Nz%s TABLESAMPLE %sr&  Tz REPEATABLE (%s))r  _get_methodr  seed)r  tablesampler&  r  ry  r   r   r   visit_tablesample  s   
zSQLCompiler.visit_tablesamplec                   s6    d j d fdd jD }d| S )Nr  r   c                 3  s>    | ]}|D ]}j tj|d  ji fi V  qqdS )r  N)r  r   r   _column_typesr  )r   chunkelemr  r  r  r   r   r)    s"    z-SQLCompiler._render_values.<locals>.<genexpr>r=  )
setdefaultr  rw   _data)r  r  r  tuplesr   r#  r   _render_values  s
   

zSQLCompiler._render_valuesc              	     s   j |fi  }|jrd }nt|jtjrd|j}n|j}|jr'd}nd}|rc|r:|d ur3|nd|j|	 < |r]d d< d||
j|d fd	d
|jD f }|S d||f }|S )Nr  zLATERAL r  z(unnamed VALUES element)FrS  z%s(%s)%s (%s)r   c                 3  r  r  r  rS  r  r   r   r)    s
    
z+SQLCompiler.visit_values.<locals>.<genexpr>z%s(%s))r'  _unnamedr  r   r   rD   rA  r	  ri  r)   r  r  rZ  rw   columns)r  r  r&  rV  r  r8  r   r  r   r  r   visit_values  s4   zSQLCompiler.visit_valuesc                 K  s   d| j |fi | dS r#  )r'  rf  r   r   r   visit_scalar_values     zSQLCompiler.visit_scalar_valuesc                 C  s   d| S )Nr   r   )r  alias_name_textr   r   r   r       z&SQLCompiler.get_render_as_alias_suffixr   r   r   r   r  c                 C  sN   |sJ |d u s|dkrd| _ d| _|jrtd| jt|||| d S )N*FTzxMost backends don't support SELECTing from a tuple() object.  If this is an ORM query, consider using the Bundle object.)r  r  r  r7   r  r  rk  r   r  r   r   r   r    s   
zSQLCompiler._add_to_result_mapc                 K  s&   | j d||d|du ri n|fi |S )aL  Render a column with necessary labels inside of a RETURNING clause.

        This method is provided for individual dialects in place of calling
        the _label_select_column method directly, so that the two use cases
        of RETURNING vs. SELECT can be disambiguated going forward.

        .. versionadded:: 1.4.21

        NF_label_select_column)r  r  r   r  column_clause_argsr  r   r   r   _label_returning_column8  s   z#SQLCompiler._label_returning_columnc                   s  j | j}|jr|s|r|}n}|r4| j}|
r&|  fdd}n|ur3|  fdd}nd}|	s<J dttjrT|urQt	|j
jfd}ny|}nv|ri|dus^J dt	|||jfd}na|urpd}n7ttjr|oj ojdu}n%ttjrd	}nttjrjp|}nttjsjdu rd}nd	}|r|s|
rJ j}t|tjst|n|}t	|||fd}n|}|j|	||d
 |j| fi |S )z.produce labeled columns present in a select().c                   s    | || f| d S r  r   r   r   r   r   )r  r   r   rE  s  r  z;SQLCompiler._label_select_column.<locals>.add_to_result_mapc                   s    | |f| | d S r  r   r4  r  r   r   r   rE  {  s   NzX_label_select_column is only relevant within the columns clause of a SELECT or RETURNING)r  z*proxy_name is required if 'name' is passedTF)r-  rE  rS  )r   r  r  _has_column_expressioncolumn_expressionr  r  r   rH   r  r   r  rX  rF   rW  r   
TextClauseUnaryExpressionwraps_column_expressionNamedColumn_non_anon_label_anon_name_labelrD   rJ  r  )r  r   r   r  r&  r2  r   r  r  r-  column_is_repeatedneed_column_expressionsrS  r  col_exprrE  result_exprrender_with_labelr   r5  r   r1  M  s   


z SQLCompiler._label_select_columnc                 C  s    |  ||}|r|d| 7 }|S r  )get_from_hint_text)r  r  r   hintr  hinttextr   r   r   r    s   z!SQLCompiler.format_from_hint_textc                 C  r  r  r   )r  byfromsr   r   r   get_select_hint_text  r  z SQLCompiler.get_select_hint_textc                 C  r  r  r   r  r   ry  r   r   r   rC    r  zSQLCompiler.get_from_hint_textc                 C  r  r  r   rH  r   r   r   get_crud_hint_text  r  zSQLCompiler.get_crud_hint_textc                 C  s
   d |S r  r  )r  
hint_textsr   r   r   get_statement_hint_text   r  z#SQLCompiler.get_statement_hint_textr  r  r  r  c                 K  sj   | j  }|r	| jn| j d }||| }|d }|d }	|r,|s,|j||	dd}
|
S |j||	d}
|
S )NrY  r  r  r   explicit_correlate_fromsimplicit_correlate_froms)rv  r  r  _get_display_froms
difference)r  select_stmtr&  r  r  r  rf  r  r  r  ri  r   r   r   _display_froms_for_select
  s$   z%SQLCompiler._display_froms_for_selecttranslate_select_structurec	              
     s  |d u sJ dd|	d< j fi |	}
|
j|	d< |
jj }|r*js*|
_|d up/|}jrUjfd i|	}|urU|
}}|j fi |	}
|
j|rZjnjd }|pl|ddpl|dd |rsds|d	|	v r||	d	= |
| ||}|		 
ddd
 d}jr\}}|r||d 7 }nd }jr|	 jr|jjfi |	7 }|jfi |	7 }dd  fdd|
jD D }r|d urttdd |
jD dd |jD fddjD _||
|||||	}jr-fddjD }|r-|d| 7 }jrI|r7|rI|s?tjnd }j|d| }jr\|djjfi |	 7 }jd |S )NzpSQLAlchemy 1.4 requires use of the translate_select_structure hook for structural translations of SELECT objectsFr-  rT  r&  rY  r  r  rE  )r@  r-  zSELECT r  c                 S  s   g | ]}|d ur|qS r  r   rS  r   r   r   rU    s
    z,SQLCompiler.visit_select.<locals>.<listcomp>c                   s4   g | ]\}}}}}j | ||||d 
qS ))r   r  r  r>  r?  r0  r   r   r  r  r   r  )r&  r2  r?  r  rQ  r  r   r   rU    s(    c                 S     g | ]	\}}}}}|qS r   r   r   r'  r  r  r   r  r   r   r   rU        c                 S  rU  r   r   rV  r   r   r   rU    rW  c              	     s4   g | ]\}}}}t ||t fd d|D |qS )c                 3  s    | ]	}  ||V  qd S r  rr  )r   or  r   r   r)    r   z6SQLCompiler.visit_select.<locals>.<listcomp>.<genexpr>)r   r+  )r   r'  r   r  r   rY  r   r   rU    s    
c                   s$   g | ]\}}|d  j jfv r|qS r/  r  r   )r   dialect_namehtr(  r   r   rU    s
    r  )r  _ambiguous_table_name_mapr  rv  r  rS  r  rs  _setup_select_stackr  rJ  _hints_setup_select_hintsr  r  r  r  get_select_precolumnscolumns_plus_namesr  r  r  _compose_select_body_statement_hintsrK  r"  rz  r  r  rl  )r  rQ  r&  insert_intor  r  select_wraps_forr  rV  r  r  r  is_embedded_selectnew_select_stmtcompile_state_wraps_forrf  ri  ry  	hint_textbyfrominner_columnsper_dialectr  r   )r&  r2  r?  r  rQ  r  r  r   visit_select.  s   







zSQLCompiler.visit_selectr   r  Tuple[str, _FromHintsType]c                   s*    fdd|j  D } |}||fS )Nc              	     s<   i | ]\\}}}|d  j jfv r||d|j ddi qS )r/  r   T)r  r  r   r  )r   r{  r  rE  r(  r   r   r4    s    
z3SQLCompiler._setup_select_hints.<locals>.<dictcomp>)r`  rK  rG  )r  r   rl  rk  r   r(  r   ra    s
   

zSQLCompiler._setup_select_hintsc                 C  s   |d }|d }|dkr||d< n"|r3|d }	t |	j}
t |j|
kr3tdd|
|d t |jf |rB|sB|j||dd}n|j||d}tt| }|	|}||||d	}| j
| |S )
Nr  r  r   r  z}All selectables passed to CompoundSelect must have identical numbers of columns; select #%d has %d columns, select #%d has %dr   r   rL  )r  r  r#   r  )rz  rV  rc  r7   r  rO  rP  ro   r*   r   rv  rk  )r  r   r  rf  r&  r  r  r  r  r  numcolsri  new_correlate_fromsall_correlate_fromsr  r   r   r   r_    sJ   


zSQLCompiler._setup_select_stackc	                   s  |d |7 }jt@ rti t jt@ }	|r_nd d}	|s'| }|rS|d7 }|jrB|d  fdd|D 7 }n|d fdd|D 7 }n|	 7 }|j
rpj|j
fdi}
|
rp|d|
 7 }|	r|d usxJ   |jr|j|fi 7 }|jrj|jfi }
|
r|d	|
 7 }|jr|j|fi 7 }|jr|j|fi 7 }|jd ur|j|fi 7 }|S )
Nr   Fz 
FROM c                   s&   g | ]}|j fd  dqS )T)r&  r  rV  r  r   frl  rV  r  r  r   r   rU  `  s    z4SQLCompiler._compose_select_body.<locals>.<listcomp>c                   s$   g | ]}|j fd  dqS )Tr&  rV  r  ru  )rV  r  r  r   r   rU  m  s    rV  z 
WHERE z	 
HAVING )rw   r.  r^  ra  ro   r_  rV  rstripr`  r  _where_criteriar  r}  _group_by_clausesr
  _having_criteria_order_by_clausesr  r  r  _for_update_argfor_update_clause)r  ry  r   r  rm  ri  rl  r  r  warn_lintingtr   rw  r   rd  ;  sn   

	
z SQLCompiler._compose_select_bodyc                   s*   d  fdd|D }|r|d7 }|S )Nr  c                 3  s:    | ]\}}|d v s|j jkr|jfi  V  qdS ))Nr/  Nrq  )r   prefixr\  r  r   r   r)    s    z1SQLCompiler._generate_prefixes.<locals>.<genexpr>r  )r  r  r  r  r  r   r  r   r    s   zSQLCompiler._generate_prefixesc                 C  s,  | j sdS |rB|dkrBt }t| j  D ]*}| j|  \}}}|jp'|j}||kp3|o3||d k}	|r8|	s9q| j | ||< qn| j }|sIdS tdd |D }
| 	|
d }|d
dd | D 7 }|d7 }|r|dkrt| D ]}| j|  \}}}| j |= | j||f= | j| = qw|S )	z
        include_following_stack
            Also render the nesting CTEs on the next stack. Useful for
            SQL structures like UNION or INSERT that can wrap SELECT
            statements containing nesting CTEs.
        r  r   c                 S  r  r   )r  )r   r  r   r   r   rU    r  z2SQLCompiler._render_cte_clause.<locals>.<listcomp>r  , 
c                 S  s   g | ]}|qS r   r   )r   txtr   r   r   rU    r  z
 )r"  r%   rb  r]  keysr$  r  r  rg   get_cte_preamblerw   r  r#  )r  r  r  r"  r  r  r  r  r  is_rendered_levelr%  cte_textr   r   r   r    s@   

zSQLCompiler._render_cte_clausec                 C  s   |rdS dS )NzWITH RECURSIVEWITHr   )r  r  r   r   r   r    s   zSQLCompiler.get_cte_preamblec                 K  s"   |j r
tjddd |jrdS dS )zcCalled when building a ``SELECT`` statement, position is just
        before column list.

        a  DISTINCT ON is currently supported only by the PostgreSQL dialect.  Use of DISTINCT ON for other backends is currently silently ignored, however this usage is deprecated, and will raise CompileError in a future release for all backends that do not support this syntax.z1.4versionr   r  )_distinct_onr%   warn_deprecated	_distinctr  r   r  r   r   r   rb    s   z!SQLCompiler.get_select_precolumnsc                 K  ,   | j |jttj fi |}|rd| S dS )z5allow dialects to customize how GROUP BY is rendered.z
 GROUP BY r  )r  r{  rB  r    comma_op)r  r   r  group_byr   r   r   r
    s   zSQLCompiler.group_by_clausec                 K  r  )z5allow dialects to customize how ORDER BY is rendered.z
 ORDER BY r  )r  r}  rB  r    r  )r  r   r  r  r   r   r   r    s   zSQLCompiler.order_by_clausec                 K  r  )Nz FOR UPDATEr   r  r   r   r   r    r  zSQLCompiler.for_update_clauser  rB   returning_colsSequence[ColumnElement[Any]]r  c                  s6    fddj dt|dD }dd| S )Nc              	     s6   g | ]\}}}}}j |f||||d  qS ))r  r>  r   r  )r3  rT  r  r  r  r  r   r   rU    s(    z0SQLCompiler.returning_clause.<locals>.<listcomp>T)cols
RETURNING r   )r  r   _select_iterablesrw   r  r  r  r  r  r)  r   r  r   returning_clause	  s   
zSQLCompiler.returning_clausec                 K  sf   d}|j d ur|d| j|j fi | 7 }|jd ur1|j d u r#|d7 }|d| j|jfi | 7 }|S )Nr  z
 LIMIT z

 LIMIT -1z OFFSET )_limit_clauser  _offset_clause)r  r   r  ry  r   r   r   r  )  s   


zSQLCompiler.limit_clausec           
      K  s   |d u r|j }|j}nddd}d}|jd ur5|j}|r%||r%| }| j|fi |}	|d|	 7 }n|r;|d7 }|d urf|rJ||rJ| }|d| j|fi ||d rZdnd|d	 rad
ndf 7 }|S )NF)ro  	with_tiesr  z
 OFFSET %s ROWSz
 OFFSET 0 ROWSz
 FETCH FIRST %s%s ROWS %sro  z PERCENTr  z	WITH TIESONLY)r  _fetch_clause_optionsr  _simple_int_clauserender_literal_executer  )
r  r   r  require_offset"use_literal_execute_for_simple_intr  fetch_clause_optionsry  offset_clause
offset_strr   r   r   r  3  s8   

zSQLCompiler.fetch_clausec	                 K  s   |r|j |j|< |s|r]| j|}
|r&|
r&| j|
d | j|j }n%| j|j}|
sK|rK|j|v rK| d||j }|| | j	d | }|r[||v r[| 
|||| |}|S dS )Nr  rV  r  )fullnameri  r  r\  r]  rZ  r   rA  r  r  r  )r  r   r&  r  r  r  
use_schemarV  rT  r  r`  r  	anon_namer   r   r   visit_table\  s:   


zSQLCompiler.visit_tablec                 K  s   |r|j tt|jjt|jj |jrd}n|j	r d}nd}|jj
| fd|d|| |jj
| fd|d| d |jj
| fd|i| S )Nz FULL OUTER JOIN z LEFT OUTER JOIN z JOIN Trx  z ON rV  )rj  rJ  r  rY  r)   rx   r*   r   ru   isouterr  onclause)r  rw   r&  rV  r  	join_typer   r   r   
visit_join  sP   

	zSQLCompiler.visit_joinc                   sB    fdd|j  D }|j|v r ||j||j d}||fS )Nc                   s*   i | ]\\}}}|d  j jfv r||qS rZ  r[  )r   r   r  rk  r(  r   r   r4    s
    
z1SQLCompiler._setup_crud_hints.<locals>.<dictcomp>T)r`  rK  r   r  )r  r  
table_textdialect_hintsr   r(  r   _setup_crud_hints  s   

zSQLCompiler._setup_crud_hintsr   rJ   r9  c                 C  s~   | j j}|j}|j}|du rdS |jr| j|jd}n| j|jd}||@ r*|S |j	r=t
d|d  d| j j ddS )a)  given a :class:`.Table`, return a usable sentinel column or
        columns for this dialect if any.

        Return None if no sentinel columns could be identified, or raise an
        error if a column was marked as a sentinel explicitly but isn't
        compatible with this dialect.

        Nr   zColumn z@ can't be explicitly marked as a sentinel column when using the a   dialect, as the particular type of default generation on this column is not currently compatible with this dialect's specific INSERT..RETURNING syntax which can receive the server-generated value in a deterministic way.  To remove this error, remove insert_sentinel=True from primary key autoincrement columns; these columns are automatically used as sentinels for supported dialects in any case.)r  "insertmanyvalues_implicit_sentinel _sentinel_column_characteristicsr)  
is_autoinc_sentinel_col_autoinc_lookuprs  default_characterization _sentinel_col_non_autoinc_lookupis_explicitr7   r*  r   )r  r   sentinel_optssentinel_characteristics	sent_colsbitmaskr   r   r   _get_sentinel_column_for_table  s*   z*SQLCompiler._get_sentinel_column_for_tablerW   compiled_parameters%List[_MutableCoreSingleExecuteParams]generic_setinputsizesrB  
batch_sizer5  r7  r  r   Iterator[_InsertManyValuesBatch]c           1      #  s   j }|d us
J |jsd  ntj|j  t
}	|jr&jjs&d}
d}njjr7|r<j	r<|j
d u s7|jr<d}
d}nd}
d}|
rlttdt
|dD ]\}\	}t|		g r` |gng d||	||
V  qMd S |rxtjjj|dnd |j}r|}d| d}||d}jj}|rtj}t|j}|| }t||| | }td	t
}td	t|}d }d}|	| |	| rdnd
 }|j}|d usJ rއfdd|D }d
 js2jrjni t
d
 }fdd|jrd}nd}ddfdd|D  | d}| fdd|D }
fdd|!|D }dnjd}t }i }|jrDd| dnd| dt |jD ]
}"|d  qPrj#}|d ushJ fddt|D } t| t$| d t|  ksJ j%rt&'j(}!t&)|! dd|r|d
| }"|d
| }#g |d
|< g |d
|< |r|}$nt|"}$rdd fddt*|$D D }jrc|j+}%|%t|"d
 krd  }&}'|"}(n|"d
 d  }&|"d
 d  }'fd!d|"D }(|jrdfd"dt|"D d d# })n|$ d d# })j%rN|%d
krN|'r2J d }*|%|$ |* }+t,fd$dt*|*|+D },|)|, })|d|)}-t,t-j./|(}.|&|. |' }.n?g }/|0 }.t|"D ]+\	|d%t1}0|jr|0d&t1}0|/2|0 |."	fd'd|D  qm|dd|/}-t|-|.||" r fd(d|#D ng |$|||d
V  |d7 }|sd S d S ))NTFzKSequence[Tuple[_DBAPISingleExecuteParams, _MutableCoreSingleExecuteParams]]r   r  r  r  __EXECMANY_TOKEN__zList[Sequence[Any]]r   c                   s$   g | ]\}}}}|| ||fqS r   r   )r   r  r'  r  st)rstr   r   rU    s    
zASQLCompiler._deliver_insertmanyvalues_batches.<locals>.<listcomp>c                   s@   | D ]}  ||}|jd|i jd| di }q|S )Nr   __EXECMANY_INDEX__)rs  rw  r  )r  	formattedr'  )r  r  r   r   apply_placeholders  s   zISQLCompiler._deliver_insertmanyvalues_batches.<locals>.apply_placeholdersz, _IMV_VALUES_COUNTERr  r   c                 3  s"    | ]\}}}} ||V  qd S r  r   )r   r  r  	bind_keys)r  r   r   r)    s
    

z@SQLCompiler._deliver_insertmanyvalues_batches.<locals>.<genexpr>c                 3  s.    | ]\}}}}|D ]	}  ||V  q
qd S r  rr  )r   r  r  r'  )r  r   r   r)    s    

c                   s   i | ]	}| d  | qS )r   r   r&  r%  r   r   r4    s    zASQLCompiler._deliver_insertmanyvalues_batches.<locals>.<dictcomp>z, _IMV_VALUES_COUNTER), z), r  c                   s   h | ]
\}}| v r|qS r   r   )r   r  r   )all_names_we_will_expandr   r   r     s
    z@SQLCompiler._deliver_insertmanyvalues_batches.<locals>.<setcomp>z\d+rh  c                 S  s   g | ]
\}}}|||fqS r   r   )r   new_keylen_r  r   r   r   rU    s    c                 3  s4    | ]} D ]\}}}| d | ||fV  qqdS )r  Nr   )r   rh  r'  r  r  )r  r   r   r)    s    r   c                 3  s    | ]	}|  V  qd S r  r   r2  )expand_pos_lower_indexexpand_pos_upper_indexr   r   r)    r  c                 3  s$    | ]\}}  d t|V  qdS )_IMV_VALUES_COUNTERN)rw  r   r   r  r  )executemany_values_w_commar   r   r)    s    
c                 3  s    | ]
} j  | V  qd S r  r  r   r  r(  r   r   r)  4  r  EXECMANY_INDEX__r  c                   s    i | ]}| d   | qS )__r   r&  )r  paramr   r   r4  U  s     c                   r  r   r   )r   r!  )_sentinel_from_paramsr   r   rU  c  r   )3r  r<  r  r  rz  r1  r  supports_default_metavaluesupports_multivalues_insertr  r:  r8  r  r   r  r?  	functoolspartialr  r  r2  rw  insertmanyvalues_max_parametersr  r4  minr]  r5  r  ro   r>  rw   r  rP  rJ  r#  r  r  r(  r)  r  rv  ranger6  r+  r  r  from_iterabler  r   rk  )1r  r  r  r  r  r  r7  r  imv	lenparamsuse_row_at_a_time
downgradedrG  compiled_paramimv_single_values_exprexecutemany_values
max_paramstotal_num_of_paramsnum_params_per_batchnum_params_outside_of_batchbatchescompiled_batchesrC  rH  r4  all_keysimv_values_counterformatted_values_clausekeys_to_replacebase_parametersr"  r#  all_expand_positionsescapedrD  compiled_batchrF  num_ins_paramsextra_params_leftextra_params_rightbatch_iteratorexpanded_values_stringrs  ri   rn  r@  rA  replaced_values_clausesfmvr   )r  r  r  r  r  r  r  r  r  r  r  r  r  r   !_deliver_insertmanyvalues_batches  s  
















 z-SQLCompiler._deliver_insertmanyvalues_batchesc                    s  |j | fi |}|j}|d ur||d< d}n j }|r-d _ js'| _ js-| _ jt t |d d}d } jrF|d u rFg }t	j
 |||fd|i|} jrr|d urrt|} jrr jd uro  j|7  _n| _|j}	|	s jjs jjs jjstd jj |jr jjstd jj  js|jr|jrtd	|j}	n|j}	 j}
 jj}d
}|jr| j||jfi |7 }|d7 }|
|j}|j r߈ !||\}}|j"r #|| ||7 }|	s|s|dd$dd |	D  7 }|j%}d }d }d} jp|j}|r|j&}|d urm|sJ dd |	D }g }|D ]!}||vr4d } n '|}||| vrDd } n|| q)|d u re jj(t)j*@ r[d}n
|d j+reJ dt,|t,| } j-|||d} j.r|d| 7 }nd }|j/d ur j0 jd d fddi|} j1r jj2r|st jnd }|d j3|dd|f 7 }n|d| 7 }n|	s|r|d7 }|rt4d jj5t6d|	||j|j7d u||rt|nd|d	 _8n|jr|dd$dd  |j9D f 7 }nd$d!d t6d|	D }|r|rX jj(t)j:@ rX|j;sXd} jj(t)j<@ }d$d"d  t=|	D }|rGd$ fd#d t=|	D }n|}|d$| d%| d&| d'7 }n
d}|d(| d)7 }t4d|t6d|	||j|j7d u||rxt|nd|||d* _8n|d(| d)7 }|j7d ur j0|j7fi |}|r|d| 7 }|r j.s|d| 7 } j1rˈ jj2s|st jnd } j3|dd| } j>d |S )+Nr  FTr  r   r  zWThe '%s' dialect with current database version settings does not support empty inserts.zcThe '%s' dialect with current database version settings does not support in-place multirow inserts.zcRETURNING cannot be determinstically sorted when using an INSERT which includes multi-row values().zINSERT zINTO  (%s)r   c                 S  s   g | ]\}}}}|qS r   r   )r   r  r  r   r   r   rU    rg  z,SQLCompiler.visit_insert.<locals>.<listcomp>c                 S  s   i | ]	\}}}}||qS r   r   )r   r  r  param_namesr   r   r   r4    s    
z,SQLCompiler.visit_insert.<locals>.<dictcomp>zTsentinel selection rules should have prevented us from getting here for this dialectr  r  rY  r  rf  z %s%sr  r  z DEFAULT VALUESr3  )r7  r8  r:  r;  r=  z
 VALUES %sc                 s  s(    | ]}d d dd |D  V  qdS )r  r   c                 s  s    | ]	\}}}}|V  qd S r  r   r   r  r  r   r   r   r)  g  r   z5SQLCompiler.visit_insert.<locals>.<genexpr>.<genexpr>Nr  )r   crud_param_setr   r   r   r)  e  s    
z+SQLCompiler.visit_insert.<locals>.<genexpr>c                 S  s   g | ]\}}}}|qS r   r   r  r   r   r   rU  m  s    
c                 s  s    | ]
\}}d | V  qdS pNr   r  r   r   r   r)        
c                 3  s8    | ]\}^}}  |j|j jd | V  qdS r  )rP  r   r  r  )r   r  r  r  r(  r   r   r)    s    

z SELECT z FROM (VALUES (z)) AS imp_sen(z#, sen_counter) ORDER BY sen_counterz	 VALUES (r  )r1  r2  r4  r6  r7  r8  r:  r;  r<  r=  r>  )?r  r  rv  r  r  r  rk  ro   r5  r   _get_crud_paramsrz  r  r  single_paramsr  supports_default_valuesr  supports_empty_insertr7   r  r   _has_multi_parametersr  r  
_returning_sort_by_parameter_orderr  r  r  format_tabler   r`  r  r  r  rw   use_insertmanyvaluesuse_sentinel_columnsr  r  rK  rT  _insert_sentinelr]  r  r  r   r  r"  cte_follows_insertr  r0  default_metavalue_tokenr   _post_values_clauser  all_multi_paramsrV  is_default_metavalue_onlyrW  r  rl  ) r  insert_stmtr  r  r  r  r  counted_bindparamcrud_params_structcrud_params_singler  r  ry  r  r  r  named_sentinel_paramsadd_sentinel_colsr=  r  _params_by_col_add_sentinel_col
param_namer  select_textr  rR  embed_sentinel_valuerender_bind_castscolnamescolnames_w_castpost_values_clauser   r(  r   visit_inserto  s  	


	







		
zSQLCompiler.visit_insertc                 C  r  )z3Provide a hook for MySQL to add LIMIT to the UPDATENr   )r  update_stmtr   r   r   update_limit_clause  r  zSQLCompiler.update_limit_clausec                 C  r  )z3Provide a hook for MySQL to add LIMIT to the DELETENr   )r  delete_stmtr   r   r   delete_limit_clause  r  zSQLCompiler.delete_limit_clausec                 K  s   d|d< |j | fddi|S )z|Provide a hook to override the initial table clause
        in an UPDATE statement.

        MySQL overrides this.

        Tr&  r  r  )r  r&  
from_tableextra_fromsr  r   r   r   update_tables_clause  s   z SQLCompiler.update_tables_clausec                 K     t d)z~Provide a hook to override the generation of an
        UPDATE..FROM clause.

        MySQL and MSSQL override this.

        zCThis backend does not support multiple-table criteria within UPDATEr  r  r&  r*  r+  
from_hintsr  r   r   r   update_from_clause  s   	zSQLCompiler.update_from_clausec                   s  |j || fi |}|j}|d ur||d< d}n| j }|r-d| _| js'|| _| js-|| _| jt@ rCti t	 }| jt
@ }|rB|| _nd }d}|j}t|}	|	rft	t|j  fdd|D }
 |}ng }
|jh}| j|||d d}|jr|| j||jfi |7 }| j||j|
fd|i|}tj| |||fi |}|j}|jr| ||\}}nd }|jr| || ||7 }|d	7 }|d
dd td|D 7 }| js|jr| j r|d| j!|| jp|j|d 7 }|r| j"||j|
|fd|i|}|r|d| 7 }|j#r | j$|j#fd|i|}|r |d| 7 }| %|}|r.|d| 7 }| js6|jrK| j sK|d| j!|| jpE|j|d 7 }| j&ra|sWt'| jnd }| j(|d| }|rq|d uskJ |j)dd | j*d |S )Nr  FTc                   s   g | ]}| vr|qS r   r   ru  
main_fromsr   r   rU  #  s    z,SQLCompiler.visit_update.<locals>.<listcomp>r  zUPDATE rV  z SET r   c                 s  s$    | ]\}}}}|d  | V  qdS )=Nr   )r   r  r  r  r   r   r   r)  S  s
    


z+SQLCompiler.visit_update.<locals>.<genexpr>zList[Tuple[Any, str, str, Any]]r  r    WHERE r  UPDATEr|  rY  )+r  r  rv  r  r  r  r.  r^  ra  ro   r_  rV  _extra_fromsr  r*   r   r   rk  r  r  r,  r   r  r  r`  r  r  r  rw   r   r  r  r  r  r0  rz  r  r'  r"  rz  r  r}  rl  )r  r&  r  r  r  r  rV  r  r+  is_multitablerender_extra_fromsr  ry  r  r  crud_paramsr  extra_from_textr  r  r  r   r1  r   visit_update  s   








zSQLCompiler.visit_updatec                 K  r-  )zProvide a hook to override the generation of an
        DELETE..FROM clause.

        This can be used to implement DELETE..USING for example.

        MySQL and MSSQL override this.

        zCThis backend does not support multiple-table criteria within DELETEr  r.  r   r   r   delete_extra_from_clause  s   z$SQLCompiler.delete_extra_from_clausec                 K  s   |j | fddd|S )NT)r&  r  r  )r  r(  r*  r+  r  r   r   r   delete_table_clause  s   zSQLCompiler.delete_table_clausec                 K  s  |j || fi |}|j}|d ur||d< d}n| j }|r-d| _| js'|| _| js-|| _| jt@ rCti t	 }| jt
@ }|rB|| _nd }d}|j}|jh|}	| j|	|	|d d}
|jrm|
| j||jfi |7 }
|
d7 }
z| j||j||d}W n ty   | ||j|}|r| j|j|d}Y nw tj| |||fi | |jr| ||\}}nd }|jr| || |
|7 }
| js|jr| jr|
d| j|| jp|j|d	 7 }
|r| j||j||fd
|i|}|r|
d| 7 }
|j r| j!|j fd
|i|}|r|
d| 7 }
| "|}|r|
d| 7 }
| js!|jr6| js6|
d| j|| jp0|j|d	 7 }
| j#rL|sBt$| jnd }| j%|d|
 }
|r\|d usVJ |j&dd | j'd |
S )Nr  FTr  zDELETE FROM )rV  r  r   rV  r4  r  DELETEr6  rY  )(r  r  rv  r  r  r  r.  r^  ra  ro   r_  rV  r7  r   r   rk  r  r  r>  	TypeErrorr  r   r  r`  r  r  r  r  r  r  r  r=  rz  r  r)  r"  rz  r  r}  rl  )r  r(  r  r  r  r  rV  r  r+  r  ry  r  r  r  r;  r  r  r  r   r   r   visit_delete  s   


	




zSQLCompiler.visit_deletec                 K     d| j | S )NzSAVEPOINT %sr  format_savepointr  savepoint_stmtr  r   r   r   visit_savepoint&     zSQLCompiler.visit_savepointc                 K  rC  )NzROLLBACK TO SAVEPOINT %srD  rF  r   r   r   visit_rollback_to_savepoint)     z'SQLCompiler.visit_rollback_to_savepointc                 K  rC  )NzRELEASE SAVEPOINT %srD  rF  r   r   r   visit_release_savepoint.  rK  z#SQLCompiler.visit_release_savepoint)r  r\   r  r  r  r  r,  r"  r-  r  r.  r\  r/  r0  r  r   )r  r  )r  r  )r  r_  )r  ra  )r  r  )NT)r  r  r  r  r  r  )NNTNTF)r  r  r  r  r  r  r  r  r  r  r:  r  r  r[   F)r  r[   r;  r  r  r  )r  r  )NFFNr   )NTr   N)r   rQ  rE  rR  rS  r  rF  r   rT  rU  r  r   r  r   r  )NF)r  r  rE  rR  r  r   r  r   )FNNr   )NFNN)FFFFF)r  r   r   rD   r  r   )r   r   r  r   )FFNNNN)r   r   r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r   )r  rM   r&  r  r  r  r  r  r  r  rV  r  r  r  r  r   r  r  )FFFNFFNN)
r   r   r   r   r   r   r   r   r  r  )NNNTFFT)FFNNNFN)r   r  r  rp  
r  rB   r  r  r  r  r  r   r  r   )NFF)FFFNTNN)r   rJ   r  r9  )r  r   r  rW   r  r  r  rB  r  r5  r7  r  r  r  r  r  rb  )r   r   r   r   EXTRACT_MAPr  r%   r  r  r   r  COMPOUND_KEYWORDSr  r  r  r  r  r  r  r  r  r	  r
  r  r  r  r  r  r  r  r  r  r  r  	frozensetr  r  r  r  r  postfetch_lastrowidr  r#  r  r   r!  r(  r*  r  r  patternrw  r  r  r'  r]  r  r/  rR  ro_memoized_propertyrW  rX  r\  r^  memoized_propertyr`  memoized_instancemethodrc  
contextlibcontextmanagerrg  rM  rL  r  r  r  r  r  r  r  rN  preload_moduler  r  r  r  r  r"  r'  r)  r9  r>  rJ  rM  rP  rc  rg  rk  rm  rq  rz  rY  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r#  r%  r)  r+  r/  r1  r:  r8  rN  r  r-  r\  r]  r^  r_  ra  r0  r  r  rg  rh  rp  rt  rx  r|  r  r  r  r  r  r  r  r  rm  rs  rw  r{  r  r  r  r  r  r  rP  r.   r  r  r  rA  r  r}  r  r#   _CTEOptsr  r  r  r  r  r
  r  r'  r*  r+  r  r  r3  r1  r  rG  rC  rI  rK  r  r   r  rR  rS  ro  ra  r_  rd  r  r  r  rb  r
  r  r  r  r  r  r  r  r  r,   
CLIENTSIDErK  rU  SENTINEL_DEFAULTNONErR  rS  r  r   rQ  r  r  r  r%  r'  r)  r,  r0  r<  r=  r>  rB  rH  rJ  rL  r   r   r   r   r  	  sH  
 

	





	 

"

;I
 
&
 F
R.
@001;4		*.
A	&		JR8					 &!%C Ld& %	 N4_5 ,0>  ^  t
  r  c                      s   e Zd ZdZdd Zed fddZdd Zd	d
 Z	dd Z
dd Zd+ddZdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Z  ZS ),StrSQLCompilera  A :class:`.SQLCompiler` subclass which allows a small selection
    of non-standard SQL features to render into a string value.

    The :class:`.StrSQLCompiler` is invoked whenever a Core expression
    element is directly stringified without calling upon the
    :meth:`_expression.ClauseElement.compile` method.
    It can render a limited set
    of non-standard SQL constructs to assist in basic stringification,
    however for more substantial custom or dialect-specific SQL constructs,
    it will be necessary to make use of
    :meth:`_expression.ClauseElement.compile`
    directly.

    .. seealso::

        :ref:`faq_sql_expression_string`

    c                 C  r  )Nz<name unknown>r   rL  r   r   r   rM  H  r  z$StrSQLCompiler._fallback_column_namezsqlalchemy.engine.urlc                   s`   |j dkr)tjj}|j|j   }|j|d | d}t|t	s)|j
|fi |S t ||S )Nr   )r/  )stringify_dialectr%   r  
engine_urlURLr   get_dialectstatement_compilerr  r_  r  r  r  )r  r  r  r  urlr  compilerr  r   r   r  K  s   

z,StrSQLCompiler.visit_unsupported_compilationc                 K  s,   d| j |jfi || j |jfi |f S )Nz%s[%s])r  rx   r   r"  r   r   r   visit_getitem_binaryY  r  z#StrSQLCompiler.visit_getitem_binaryc                 K     | j ||fi |S r  rg  r"  r   r   r   visit_json_getitem_op_binary_  r  z+StrSQLCompiler.visit_json_getitem_op_binaryc                 K  rh  r  ri  r"  r   r   r   !visit_json_path_getitem_op_binaryb  r  z0StrSQLCompiler.visit_json_path_getitem_op_binaryc                 K  s   d| j | dS )Nz<next sequence value: >)r  format_sequencer   r   r   r   r  e  s   zStrSQLCompiler.visit_sequencer  rB   r  r  r  r  r  r   r  r   c                  s&    fddt |D }dd| S )Nc              	     s   g | ]}  d |ddi qS )NTFr0  rS  r(  r   r   rU  r  s    z3StrSQLCompiler.returning_clause.<locals>.<listcomp>r  r   )r   r  rw   r  r   r(  r   r  j  s   
zStrSQLCompiler.returning_clausec                   s(   dd< dd  fdd|D  S )NTr&  r?  r   c                 3  &    | ]}|j fd  iV  qdS r  Nr  r   r  r/  r  r  r   r   r)  |  
    
z4StrSQLCompiler.update_from_clause.<locals>.<genexpr>r  r.  r   rq  r   r0  x     
z!StrSQLCompiler.update_from_clausec                   s(   dd< dd  fdd|D  S )NTr&  r   c                 3  rn  ro  r  rp  rq  r   r   r)    rr  z:StrSQLCompiler.delete_extra_from_clause.<locals>.<genexpr>r  r.  r   rq  r   r=    rs  z'StrSQLCompiler.delete_extra_from_clausec                 K  r  )NzSELECT 1 WHERE 1!=1r   r;  r   r   r   r8    r  z#StrSQLCompiler.visit_empty_set_exprc                 C  s   d| S )Nz[%s]r   rH  r   r   r   rC    r.  z!StrSQLCompiler.get_from_hint_textc                 K     | j |dfi |S )Nz
 <regexp> r0  r"  r   r   r   r    r  z+StrSQLCompiler.visit_regexp_match_op_binaryc                 K  rt  )Nz <not regexp> ru  r"  r   r   r   r    r  z/StrSQLCompiler.visit_not_regexp_match_op_binaryc                 K  s,   d|j j| fi ||jj| fi |f S )Nz<regexp replace>(%s, %s))rx   r  r   r"  r   r   r   r    r  z-StrSQLCompiler.visit_regexp_replace_op_binaryc                 K  r  )NzTRY_CAST(%s AS %s))r  r  rp  )r  r   r  r   r   r   visit_try_cast  r  zStrSQLCompiler.visit_try_castrO  )r   r   r   r   rM  r%   rZ  r  rg  rj  rk  r  r  r0  r=  r8  rC  r  r  r  rv  r  r   r   r  r   r_  4  s$    
		r_  c                   @  s  e Zd ZdZer			dpdqddZejdd Zejdd Z				drdsddZ
dd Zdd  Zd!d" Zd#d$ Zdtd&d'Z	dud(d)Zd*d+ Zd,d- Zd.d/ Z	dvd0d1Zd2d3 Zdtd4d5Zd6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zd@dA ZdBdC ZdDdE ZdudFdGZdHdI Z dJdK Z!dLdM Z"dNdO Z#dPdQ Z$dRdS Z%dTdU Z&dVdW Z'dXdY Z(dZd[ Z)d\d] Z*d^d_ Z+d`da Z,dbdc Z-ddde Z.dfdg Z/dhdi Z0djdk Z1dldm Z2dndo Z3dS )wDDLCompilerT.r  r\   r  r@   r  r  r  r  r  r  c                 C  r  r  r   r  r   r   r   r    s   zDDLCompiler.__init__c                 C  s   | j j| j d | jdS )Nr  )r  rd  r  r(  r   r   r   r    s   
zDDLCompiler.sql_compilerc                 C  r  r  )r  ro  r(  r   r   r   type_compiler  r  zDDLCompiler.type_compilerNr  r  r  r  r  r  r  c                 C  r  r  r   r  r   r   r   r    r  zDDLCompiler.construct_paramsc                 K  s   |j }t|jtjrD| }| j}||j}t|dkr%|d d}}n	|d |d }}|	d| |	d| |	d|
|j | j|j| S )Nr   r   r  rY  r   r"   r  )contextr  targetr"   rJ   r  r  format_table_seqrz  r$  r  r  rz  r  )r  ddlr  ry  r  pathr   schr   r   r   	visit_ddl  s   zDDLCompiler.visit_ddlc                 K  $   d}|j r	|d7 }|| j|j S )NzCREATE SCHEMA IF NOT EXISTS )if_not_existsr  format_schemar  )r  r   r  ry  r   r   r   visit_create_schema     zDDLCompiler.visit_create_schemac                 K  s6   d}|j r	|d7 }|| j|j7 }|jr|d7 }|S )NzDROP SCHEMA 
IF EXISTS  CASCADE)	if_existsr  r  r  cascader  dropr  ry  r   r   r   visit_drop_schema  s   zDDLCompiler.visit_drop_schemac                 K  sF  |j }| j}d}|jr|d|jd 7 }|d7 }|jr |d7 }|||d 7 }| |}|r6||d 7 }|d7 }d}d}|jD ]D}	|	j }
z"| j|	|
j	oO| d}|d urb||7 }d	}|d
| 7 }|
j	rgd}W qA t
jy } zt
d|j|
j|jd f |d }~ww | j||jd}|r||d
 | 7 }|d| | 7 }|S )Nz
CREATE r  zTABLE r  r  
Ffirst_pkr  	Tz (in table '%s', column '%s'): %sr   ) _include_foreign_key_constraintsz
)%s

)r  r  r  rw   r  r  create_table_suffixr)  r  r  r7   r  descriptionr   argscreate_table_constraintsinclude_foreign_key_constraintspost_create_table)r  r   r  r   r  ry  r  r  r  create_columnr   	processedceconstr   r   r   visit_create_table  s\   

zDDLCompiler.visit_create_tableFc                   sL   |j }|jrd S  j||d}d fdd|jD }|r$|d| 7 }|S )Nr  r  c                 3  s    | ]}  |V  qd S r  )r  r   r   r(  r   r   r)     s    

z2DDLCompiler.visit_create_column.<locals>.<genexpr>)r  systemget_column_specificationrw   constraints)r  r   r  r  r   ry  r  r   r(  r   visit_create_column  s   zDDLCompiler.visit_create_columnc                   st   g }j r|j  j}|d ur|| nt  | fddjD  ddd fdd|D D S )Nc                   s"   g | ]}|j ur| vr|qS r   )r  rS  )	omit_fkcsr   r   r   rU  8  s
    z8DDLCompiler.create_table_constraints.<locals>.<listcomp>z, 
	c                 s  s    | ]	}|d ur|V  qd S r  r   )r   r  r   r   r   r)  ?  s    	
z7DDLCompiler.create_table_constraints.<locals>.<genexpr>c                 3  s8    | ]}|  r jjrt|d ds |V  qdS )	use_alterFN)_should_create_for_compilerr  supports_alterr  r  r  r(  r   r   r)  A  s    

)r  rk  foreign_key_constraintsrP  ro   r  _sorted_constraintsrw   )r  r   r  r  r  all_fkcsr   )r  r  r   r   r  (  s"   

z$DDLCompiler.create_table_constraintsc                 K  r  )Nz
DROP TABLE r  )r  r  r  r  r  r   r   r   visit_drop_tableM  r  zDDLCompiler.visit_drop_tablec                 K  s   d| j |j S )Nz
DROP VIEW r  r  r  r  r  r  r   r   r   visit_drop_viewS  r  zDDLCompiler.visit_drop_viewc                 C  s   |j d u rtd|j d S )Nz,Index '%s' is not associated with any table.)r   r7   r  r   r  rh  r   r   r   _verify_index_tableV  s
   
zDDLCompiler._verify_index_tablec              	     s   |j } |  j}d}|jr|d7 }|jd u rtd|d7 }|jr)|d7 }|d j||d|j	|j
|dd	 fd
d|jD f 7 }|S )NzCREATE zUNIQUE z0CREATE INDEX requires that the index have a namezINDEX r  z%s ON %s (%s)include_schemar  r   c                 3  s"    | ]} j j|d ddV  qdS )FTrS  r  N)r  r  )r   r  r(  r   r   r)  s  r  z1DDLCompiler.visit_create_index.<locals>.<genexpr>)r  r  r  r   r   r7   r  r  _prepared_index_namer  r   rw   expressions)r  r   r  include_table_schemar  rh  r  ry  r   r(  r   visit_create_index\  s.   

zDDLCompiler.visit_create_indexc                 K  s>   |j }|jd u rtdd}|jr|d7 }|| j|dd S )Nz.DROP INDEX requires that the index have a namez
DROP INDEX r  Tr  )r  r   r7   r  r  r  )r  r  r  rh  ry  r   r   r   visit_drop_index|  s   
zDDLCompiler.visit_drop_indexc                 C  sX   |j d ur| j|j }nd }|r|r| j|}nd }| j|}|r*|d | }|S Nr  )r   r  r\  r]  format_index)r  rh  r  r`  schema_name
index_namer   r   r   r    s   
z DDLCompiler._prepared_index_namec                 K  s    d| j |jj| |jf S )NzALTER TABLE %s ADD %s)r  r  r  r   r  r  r   r  r   r   r   visit_add_constraint  s   
z DDLCompiler.visit_add_constraintc                 K  s(   d| j |j| j|jjt f S )NzCOMMENT ON TABLE %s IS %s)r  r  r  r  r  commentr$   Stringr  r   r   r   visit_set_table_comment  s   z#DDLCompiler.visit_set_table_commentc                 K  s   d| j |j S )NzCOMMENT ON TABLE %s IS NULLr  r  r   r   r   visit_drop_table_comment  s   z$DDLCompiler.visit_drop_table_commentc                 K  s.   d| j j|jddd| j|jjt f S )NzCOMMENT ON COLUMN %s IS %sT)	use_tabler  )r  format_columnr  r  r  r  r$   r  r  r   r   r   visit_set_column_comment  s   z$DDLCompiler.visit_set_column_commentc                 K  s   d| j j|jdd S )NzCOMMENT ON COLUMN %s IS NULLT)r  )r  r  r  r  r   r   r   visit_drop_column_comment  s   z%DDLCompiler.visit_drop_column_commentc                 K     t | t|r  r  r  r   r   r   visit_set_constraint_comment  rI  z(DDLCompiler.visit_set_constraint_commentc                 K  r  r  r  r  r   r   r   visit_drop_constraint_comment  rI  z)DDLCompiler.visit_drop_constraint_commentc                 C  s   g }|j d ur|d|j   |jd ur|d|j  |jd ur)|d|j  |jd ur6|d|j  |jd ur@|d |jd urJ|d |jd urW|d|j  |jd urf||jrcdnd	 d
	|S )NzINCREMENT BY %dzSTART WITH %dzMINVALUE %dzMAXVALUE %dzNO MINVALUEzNO MAXVALUEzCACHE %dCYCLEzNO CYCLEr  )
	incrementrk  rs  minvaluemaxvalue
nominvalue
nomaxvaluecachecyclerw   )r  identity_optionsry  r   r   r   get_identity_options  s$   










z DDLCompiler.get_identity_optionsc                 K  sP   d}|j r	|d7 }|| j|j7 }|r||7 }| |j}|r&|d| 7 }|S )NzCREATE SEQUENCE r  r  )r  r  rm  r  r  )r  r   r  r  ry  optionsr   r   r   visit_create_sequence  s   z!DDLCompiler.visit_create_sequencec                 K  r  )NzDROP SEQUENCE r  )r  r  rm  r  r  r   r   r   visit_drop_sequence  r  zDDLCompiler.visit_drop_sequencec                 K  sn   |j }|jd ur| j|}nd }|d u rtd|j  d| j|j j|jr*dnd||j	r3df S df S )Nz<Can't emit DROP CONSTRAINT for constraint %r; it has no namez%ALTER TABLE %s DROP CONSTRAINT %s%s%sr  r  r  )
r  r   r  format_constraintr7   r  r  r   r  r  )r  r  r  r   formatted_namer   r   r   visit_drop_constraint  s$   
z!DDLCompiler.visit_drop_constraintc                 K  s   | j |d | jjj|j|d }| |}|d ur!|d| 7 }|jd ur0|d| |j 7 }|jd urC| jj	rC|d| |j 7 }|j
sQ|jrM| jj	sQ|d7 }|S )Nr  )rn  z	 DEFAULT z	 NOT NULL)r  r  r  ro  r  r   get_column_default_stringcomputedidentitysupports_identity_columnsnullable)r  r   r  colspecr   r   r   r   r    s0   



z$DDLCompiler.get_column_specificationc                 C  r  Nr  r   r  r   r   r   r   r    r  zDDLCompiler.create_table_suffixc                 C  r  r  r   r  r   r   r   r    r  zDDLCompiler.post_create_tablec                 C  s    t |jtjr| |jjS d S r  )r  server_defaultr"   DefaultClauserender_default_stringargrL  r   r   r   r    s   z%DDLCompiler.get_column_default_stringc                 C  s*   t |tr| j|tjS | jj|ddS )NT)r  )r  r   r  r  r$   rf  r  )r  r   r   r   r   r  "  s
   
z!DDLCompiler.render_default_stringc                 K  s   |j r| |S | |S r  )is_column_levelvisit_column_check_constraintvisit_check_constraintr  r   r  r   r   r   &visit_table_or_column_check_constraint*  s   

z2DDLCompiler.visit_table_or_column_check_constraintc                 K  \   d}|j d ur| j|}|d ur|d| 7 }|d| jj|jddd 7 }|| |7 }|S Nr  CONSTRAINT %s z
CHECK (%s)FTr  r   r  r  r  r  r  define_constraint_deferrabilityr  r   r  ry  r  r   r   r   r  0     


z"DDLCompiler.visit_check_constraintc                 K  r  r  r  r  r   r   r   r  <  r  z)DDLCompiler.visit_column_check_constraintc                   s   t |dkrdS d}|jd ur j|}|d ur|d| 7 }|d7 }|dd fdd|jr2|jn|jD  7 }| |7 }|S )	Nr   r  r  zPRIMARY KEY r  r   c                 3      | ]
} j |jV  qd S r  r  rZ  r   rS  r(  r   r   r)  Q  r  z;DDLCompiler.visit_primary_key_constraint.<locals>.<genexpr>)	rz  r   r  r  rw   _implicit_generatedcolumns_autoinc_firstr)  r  r  r   r(  r   visit_primary_key_constraintH  s   
z(DDLCompiler.visit_primary_key_constraintc              	     s   | j  d}|jd ur| j |}|d ur|d| 7 }t|jd jj}|dd fdd|jD | || d fdd|jD f 7 }|| 	|7 }|| 
|7 }|| |7 }|S )	Nr  r  r   z"FOREIGN KEY(%s) REFERENCES %s (%s)r   c                 3      | ]
}  |jjV  qd S r  )rZ  parentr   ru  r  r   r   r)  e  r  z;DDLCompiler.visit_foreign_key_constraint.<locals>.<genexpr>c                 3  r  r  )rZ  r   r   ru  r  r   r   r)  k  r  )r  r   r  r]  r   r   r   rw   define_constraint_remote_tabledefine_constraint_matchdefine_constraint_cascadesr  )r  r   r  ry  r  remote_tabler   r  r   visit_foreign_key_constraint\  s,   
z(DDLCompiler.visit_foreign_key_constraintc                 C  s
   | |S )z=Format the remote table clause of a CREATE CONSTRAINT clause.)r  )r  r   r   r  r   r   r   r  t     
z*DDLCompiler.define_constraint_remote_tablec                   s   t |dkrdS d}|jd ur j|}|d ur|d| 7 }|d j|fi |d fdd|D f 7 }| |7 }|S )Nr   r  r  zUNIQUE %s(%s)r   c                 3  r  r  r  rS  r(  r   r   r)    s    z6DDLCompiler.visit_unique_constraint.<locals>.<genexpr>)rz  r   r  r  !define_unique_constraint_distinctrw   r  r  r   r(  r   visit_unique_constrainty  s   
z#DDLCompiler.visit_unique_constraintc                 K  r  r  r   r  r   r   r   r    r  z-DDLCompiler.define_unique_constraint_distinctc                 C  sL   d}|j d ur|d| j|j t 7 }|jd ur$|d| j|jt 7 }|S )Nr  z ON DELETE %sz ON UPDATE %s)ondeleter  validate_sql_phraseFK_ON_DELETEonupdateFK_ON_UPDATEr  r   ry  r   r   r   r    s   



z&DDLCompiler.define_constraint_cascadesc                 C  sL   d}|j d ur|j r|d7 }n|d7 }|jd ur$|d| j|jt 7 }|S )Nr  z DEFERRABLEz NOT DEFERRABLEz INITIALLY %s)r   r   r  r  FK_INITIALLYr  r   r   r   r    s   



z+DDLCompiler.define_constraint_deferrabilityc                 C  s    d}|j d ur|d|j  7 }|S )Nr  z	 MATCH %s)r  r  r   r   r   r    s   
z#DDLCompiler.define_constraint_matchc                 K  sD   d| j j|jddd }|jdu r|d7 }|S |jdu r |d7 }|S )NzGENERATED ALWAYS AS (%s)FTr  z STOREDz VIRTUAL)r  r  r  	persisted)r  	generatedr  ry  r   r   r   visit_computed_column  s   

z!DDLCompiler.visit_computed_columnc                 K  s2   d|j rdndf }| |}|r|d| 7 }|S )NzGENERATED %s AS IDENTITYALWAYSz
BY DEFAULTr  )alwaysr  )r  r  r  ry  r  r   r   r   visit_identity_column  s   
z!DDLCompiler.visit_identity_column)...)
r  r\   r  r@   r  r  r  r  r  r  r  r  rM  r  )FT)4r   r   r   r  r   r  r%   rV  r  rx  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r  r  r
  r  r   r   r   r   rw    st    	

	
4
%
 



rw  c                   @  s  e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zd`d d!Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Zd<d= Z d>d? Z!d@dA Z"dBdC Z#dDdE Z$dFdG Z%dHdI Z&dJdK Z'dLdM Z(dNdO Z)dPdQ Z*dRdS Z+dTdU Z,dVdW Z-dXdY Z.dZd[ Z/d\d] Z0d^d_ Z1dS )aGenericTypeCompilerc                 K  r  )NFLOATr   r  r   r   r   visit_FLOAT  r  zGenericTypeCompiler.visit_FLOATc                 K  r  )NDOUBLEr   r  r   r   r   visit_DOUBLE  r  z GenericTypeCompiler.visit_DOUBLEc                 K  r  )NzDOUBLE PRECISIONr   r  r   r   r   visit_DOUBLE_PRECISION  r  z*GenericTypeCompiler.visit_DOUBLE_PRECISIONc                 K  r  )NREALr   r  r   r   r   
visit_REAL  r  zGenericTypeCompiler.visit_REALc                 K  8   |j d u rdS |jd u rdd|j i S d|j |jd S )NNUMERICzNUMERIC(%(precision)s)	precisionz!NUMERIC(%(precision)s, %(scale)s)r  scaler  r  r   r   r   visit_NUMERIC     

z!GenericTypeCompiler.visit_NUMERICc                 K  r  )NDECIMALzDECIMAL(%(precision)s)r  z!DECIMAL(%(precision)s, %(scale)s)r  r  r  r   r   r   visit_DECIMAL  r  z!GenericTypeCompiler.visit_DECIMALc                 K  r  )NINTEGERr   r  r   r   r   visit_INTEGER  r  z!GenericTypeCompiler.visit_INTEGERc                 K  r  )NSMALLINTr   r  r   r   r   visit_SMALLINT  r  z"GenericTypeCompiler.visit_SMALLINTc                 K  r  )NBIGINTr   r  r   r   r   visit_BIGINT  r  z GenericTypeCompiler.visit_BIGINTc                 K  r  )N	TIMESTAMPr   r  r   r   r   visit_TIMESTAMP  r  z#GenericTypeCompiler.visit_TIMESTAMPc                 K  r  )NDATETIMEr   r  r   r   r   visit_DATETIME  r  z"GenericTypeCompiler.visit_DATETIMEc                 K  r  )NDATEr   r  r   r   r   
visit_DATE  r  zGenericTypeCompiler.visit_DATEc                 K  r  )NTIMEr   r  r   r   r   
visit_TIME  r  zGenericTypeCompiler.visit_TIMEc                 K  r  )NCLOBr   r  r   r   r   
visit_CLOB  r  zGenericTypeCompiler.visit_CLOBc                 K  r  )NNCLOBr   r  r   r   r   visit_NCLOB  r  zGenericTypeCompiler.visit_NCLOBNc                 C  sB   |}|r|d| 7 }n
|j r|d|j  7 }|jr|d|j 7 }|S )N(%d)z COLLATE "%s")lengthre  )r  r   r   length_overridery  r   r   r   _render_string_type  s   z'GenericTypeCompiler._render_string_typec                 K     |  |dS )NCHARr4  r  r   r   r   
visit_CHAR  r  zGenericTypeCompiler.visit_CHARc                 K  r5  )NNCHARr7  r  r   r   r   visit_NCHAR	  r  zGenericTypeCompiler.visit_NCHARc                 K  r5  )NVARCHARr7  r  r   r   r   visit_VARCHAR  r  z!GenericTypeCompiler.visit_VARCHARc                 K  r5  )NNVARCHARr7  r  r   r   r   visit_NVARCHAR  r  z"GenericTypeCompiler.visit_NVARCHARc                 K  r5  )NTEXTr7  r  r   r   r   
visit_TEXT  r  zGenericTypeCompiler.visit_TEXTc                 K  r  )NUUIDr   r  r   r   r   
visit_UUID  r  zGenericTypeCompiler.visit_UUIDc                 K  r  )NBLOBr   r  r   r   r   
visit_BLOB  r  zGenericTypeCompiler.visit_BLOBc                 K     d|j r	d|j  p
d S )NBINARYr1  r  r2  r  r   r   r   visit_BINARY  rd  z GenericTypeCompiler.visit_BINARYc                 K  rE  )N	VARBINARYr1  r  rG  r  r   r   r   visit_VARBINARY  rd  z#GenericTypeCompiler.visit_VARBINARYc                 K  r  )NBOOLEANr   r  r   r   r   visit_BOOLEAN!  r  z!GenericTypeCompiler.visit_BOOLEANc                 K  s0   |j r| jjs| j|dddS | j|fi |S )Nr6      )r3  )native_uuidr  supports_native_uuidr4  rB  r  r   r   r   
visit_uuid$  s   zGenericTypeCompiler.visit_uuidc                 K  r  r  )rD  r  r   r   r   visit_large_binary*  r  z&GenericTypeCompiler.visit_large_binaryc                 K  r  r  )rL  r  r   r   r   visit_boolean-  r  z!GenericTypeCompiler.visit_booleanc                 K  r  r  )r,  r  r   r   r   
visit_time0  r  zGenericTypeCompiler.visit_timec                 K  r  r  )r(  r  r   r   r   visit_datetime3  r  z"GenericTypeCompiler.visit_datetimec                 K  r  r  )r*  r  r   r   r   
visit_date6  r  zGenericTypeCompiler.visit_datec                 K  r  r  )r$  r  r   r   r   visit_big_integer9  r  z%GenericTypeCompiler.visit_big_integerc                 K  r  r  )r"  r  r   r   r   visit_small_integer<  r  z'GenericTypeCompiler.visit_small_integerc                 K  r  r  )r   r  r   r   r   visit_integer?  r  z!GenericTypeCompiler.visit_integerc                 K  r  r  )r  r  r   r   r   
visit_realB  r  zGenericTypeCompiler.visit_realc                 K  r  r  )r  r  r   r   r   visit_floatE  r  zGenericTypeCompiler.visit_floatc                 K  r  r  )r  r  r   r   r   visit_doubleH  r  z GenericTypeCompiler.visit_doublec                 K  r  r  )r  r  r   r   r   visit_numericK  r  z!GenericTypeCompiler.visit_numericc                 K  r  r  r<  r  r   r   r   visit_stringN  r  z GenericTypeCompiler.visit_stringc                 K  r  r  r]  r  r   r   r   visit_unicodeQ  r  z!GenericTypeCompiler.visit_unicodec                 K  r  r  r@  r  r   r   r   
visit_textT  r  zGenericTypeCompiler.visit_textc                 K  r  r  r`  r  r   r   r   visit_unicode_textW  r  z&GenericTypeCompiler.visit_unicode_textc                 K  r  r  r]  r  r   r   r   
visit_enumZ  r  zGenericTypeCompiler.visit_enumc                 K  s   t d| )NzKCan't generate DDL for %r; did you forget to specify a type on this Column?rK  r  r   r   r   r  ]  s
   zGenericTypeCompiler.visit_nullc                 K  s   | j || jfi |S r  )r  type_enginer  r  r   r   r   visit_type_decoratord  r,  z(GenericTypeCompiler.visit_type_decoratorc                 K  s   |j di |S rN  )get_col_specr  r   r   r   visit_user_definedg  rI  z&GenericTypeCompiler.visit_user_definedr  )2r   r   r   r  r  r  r  r  r  r   r"  r$  r&  r(  r*  r,  r.  r0  r4  r8  r:  r<  r>  r@  rB  rD  rH  rJ  rL  rP  rQ  rR  rS  rT  rU  rV  rW  rX  rY  rZ  r[  r\  r^  r_  ra  rb  rc  r  re  rg  r   r   r   r   r    s`    

r  c                   @  s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )StrSQLTypeCompilerc                 K  s@   z|j }W n ty   | j|fi | Y S w || fi |S r  )r  AttributeError_visit_unknown)r  r   r  r  r   r   r   r  l  s   
zStrSQLTypeCompiler.processc                 C  s   | dr| jS t|)Nvisit_)r>  rj  ri  )r  r'  r   r   r   __getattr__t  s   
zStrSQLTypeCompiler.__getattr__c                 K  s$   |j j|j j kr|j jS t|S r  )r  r   upperreprr  r   r   r   rj  z  s   z!StrSQLTypeCompiler._visit_unknownc                 K  r  r  r   r  r   r   r   r    r  zStrSQLTypeCompiler.visit_nullc                 K  s4   z|j }W n ty   t| Y S w |di |S rN  )rf  ri  rn  )r  r   r  rf  r   r   r   rg    s   
z%StrSQLTypeCompiler.visit_user_definedN)r   r   r   r  rl  rj  r  rg  r   r   r   r   rh  k  s    rh  c                   @     e Zd ZdddZdS )	_SchemaForObjectCallable_SchemaForObjectCallable__objr   r  r   c                 C  r  r  r   )r  rq  r   r   r   r        z!_SchemaForObjectCallable.__call__N)rq  r   r  r   r	  r   r   r   r   rp    r
  rp  c                   @  ro  )	r  r  rQ  r  r   c                 C  r  r  r   )r  r  r   r   r   r    rr  z _BindNameForColProtocol.__call__N)r  rQ  r  r   r	  r   r   r   r   r    r
  r  c                   @  s  e Zd ZU dZeZeZeZ	de
d< de
d< de
d< edZde
d	< 	 d
Zde
d< 					
dTddZdd Zdd ZdUddZdUddZdd ZdUddZdVd d!Zd"d# ZdWdXd&d'ZdWdYd)d*Zd+d, ZdZd-d.Z	dWd[d3d4Z	dWd\d7d8ZdWd9d:Ze d;dZd<d=Z!dZd>d?Z"dZd@dAZ#dBdC Z$dDdE Z%d]dFdGZ&dHdI Z'	dWdJdKZ(	
			
	d^dLdMZ)dZdNdOZ*ej+dPdQ Z,dRdS Z-dS )_r  z@Handle quoting and case-folding of identifiers based on options.r   initial_quotefinal_quotezMutableMapping[str, str]_stringsr"   rp  r\  Fr  _includes_none_schema_translaterz  NTc                 C  sN   || _ || _|p
| j| _|| _| jd | _|| _|| _i | _| j jdv | _	dS )as  Construct a new ``IdentifierPreparer`` object.

        initial_quote
          Character that begins a delimited identifier.

        final_quote
          Character that ends a delimited identifier. Defaults to
          `initial_quote`.

        omit_schema
          Prevent prepending schema name. Useful for databases that do
          not support schemae.
        r6   )r   r   N)
r  rs  rt  escape_quoteescape_to_quoteomit_schemaquote_case_sensitive_collationsru  r=  rv  )r  r  rs  rt  rw  rz  ry  r   r   r   r    s   zIdentifierPreparer.__init__c                   s@   | j | j }|j| j d |v   fdd}||_ |_|S )Nc                   sX   | j }| jr)|d us r)|d urd|v sd|v rtd| td|p$d ddS | j S )Nr  r  zJSquare bracket characters ([]) not supported in schema translate name '%s'z__[SCHEMA_%s]_noneFrZ  )r"   _use_schema_mapr7   r  r0   )r  r   includes_noner   r   symbol_getter  s   z@IdentifierPreparer._with_schema_translate.<locals>.symbol_getter)r  __new__rI  rJ  r\  rv  )r  r  prepr  r   r~  r   r    s   z)IdentifierPreparer._with_schema_translatec                   sD   | d  v rj std d   d<  fdd}td||S )Nzschema translate map which previously did not have `None` present as a key now has `None` present; compiled statement may lack adequate placeholders.  Please use consistent keys in successive schema_translate_map dictionaries.r{  c                   sV   |  d}| v r | }n|dv rtd|}|s&jj}|s&td|S )Nr6   )Nr{  zschema translate map which previously had `None` present as a key now no longer has it present; don't know how to apply schema for compiled statement. Please use consistent keys in successive schema_translate_map dictionaries.zLDialect has no default schema name; can't use None as dynamic schema target.)r   r7   r*  r  default_schema_namer  r]  )ri  r   r`  dr  r   r   rw    s   


z=IdentifierPreparer._render_schema_translates.<locals>.replacez(__\[SCHEMA_([^\]]+)\]))rv  r7   r*  r(  rv  )r  r  r  rw  r   r  r   r    s   z,IdentifierPreparer._render_schema_translatesr  r  c                 C  s&   | | j| j}| jr| dd}|S )zEscape an identifier.

        Subclasses should override this to provide database-dependent
        escaping behavior.
        r  rt  )rw  rw  rx  rv  r  r  r   r   r   _escape_identifier  s   z%IdentifierPreparer._escape_identifierc                 C  s   | | j| jS )zCanonicalize an escaped identifier.

        Subclasses should override this to provide database-dependent
        unescaping behavior that reverses _escape_identifier.
        )rw  rx  rw  r  r   r   r   _unescape_identifier!  s   z'IdentifierPreparer._unescape_identifierc                 C  s*   |dur| |std||jf |S )zkeyword sequence filter.

        a filter for elements that are intended to represent keyword sequences,
        such as "INITIALLY", "INITIALLY DEFERRED", etc.   no special characters
        should be present.

        .. versionadded:: 1.3

        Nz/Unexpected SQL phrase: %r (matching against %r))r  r7   r  rT  )r  r  r+  r   r   r   r  *  s   z&IdentifierPreparer.validate_sql_phrasec                 C  s   | j | | | j S )z~Quote an identifier.

        Subclasses should override this to provide database-dependent
        quoting behavior.
        )rs  r  rt  r  r   r   r   quote_identifier<  s   z#IdentifierPreparer.quote_identifierc                 C  s:   |  }|| jv p|d | jv p| jt| p||kS )z5Return True if the given identifier requires quoting.r   )r  reserved_wordsillegal_initial_characterslegal_charactersr  r   )r  r  lc_valuer   r   r   r  I  s   
z#IdentifierPreparer._requires_quotesc                 C  s   | j t| S )zjReturn True if the given identifier requires quoting, but
        not taking case convention into account.)r  r  r   r  r   r   r   r  S  s   z1IdentifierPreparer._requires_quotes_illegal_charsforcer   c                 C  s    |durt jddd | |S )a  Conditionally quote a schema name.


        The name is quoted if it is a reserved word, contains quote-necessary
        characters, or is an instance of :class:`.quoted_name` which includes
        ``quote`` set to ``True``.

        Subclasses can override this to provide database-dependent
        quoting behavior for schema names.

        :param schema: string schema name
        :param force: unused

            .. deprecated:: 0.9

                The :paramref:`.IdentifierPreparer.quote_schema.force`
                parameter is deprecated and will be removed in a future
                release.  This flag has no effect on the behavior of the
                :meth:`.IdentifierPreparer.quote` method; please refer to
                :class:`.quoted_name`.

        NzThe IdentifierPreparer.quote_schema.force parameter is deprecated and will be removed in a future release.  This flag has no effect on the behavior of the IdentifierPreparer.quote method; please refer to quoted_name().0.9r  )r%   r  rZ  )r  r"   r  r   r   r   r]  X  s   

zIdentifierPreparer.quote_schemar  c                 C  s   |durt jddd t|dd}|du r7|| jv r| j| S | |r-| || j|< n|| j|< | j| S |r>| |S |S )a  Conditionally quote an identifier.

        The identifier is quoted if it is a reserved word, contains
        quote-necessary characters, or is an instance of
        :class:`.quoted_name` which includes ``quote`` set to ``True``.

        Subclasses can override this to provide database-dependent
        quoting behavior for identifier names.

        :param ident: string identifier
        :param force: unused

            .. deprecated:: 0.9

                The :paramref:`.IdentifierPreparer.quote.force`
                parameter is deprecated and will be removed in a future
                release.  This flag has no effect on the behavior of the
                :meth:`.IdentifierPreparer.quote` method; please refer to
                :class:`.quoted_name`.

        NzThe IdentifierPreparer.quote.force parameter is deprecated and will be removed in a future release.  This flag has no effect on the behavior of the IdentifierPreparer.quote method; please refer to quoted_name().r  r  rZ  )r%   r  r  ru  r  r  )r  r  r  r   r   r   rZ    s    






zIdentifierPreparer.quotec                 C  s   | j r| |S |S r  )rz  rZ  )r  collation_namer   r   r   rd    s   
z#IdentifierPreparer.format_collationc                 C  s>   |  |j}| |}| js|r|d ur| |d | }|S r  )rZ  r   r\  ry  r]  )r  r  r  r   r`  r   r   r   rm    s   
z"IdentifierPreparer.format_sequencer  
Label[Any]r   r  c                 C  s   |  |p|jS r  rZ  r   )r  r  r   r   r   r   rD    s   zIdentifierPreparer.format_labelrV  Optional[AliasedReturnsRows]c                 C  s*   |d u r|d us
J |  |jS |  |S r  r  )r  rV  r   r   r   r   r    s   
zIdentifierPreparer.format_aliasc                 C  s"   |p|j }| |r| |}|S r  )r  r  r  )r  	savepointr   r  r   r   r   rE    s   


z#IdentifierPreparer.format_savepointzsqlalchemy.sql.namingc                 C  sZ   t jj}|jtu r|||j}|d u rd S n|j}|jdkr&| j||dS | j	||dS )Nrh  )_alembic_quote)
r%   r  
sql_namingr   r+   _constraint_name_for_tabler   r  truncate_and_render_index_name#truncate_and_render_constraint_name)r  r   r  namingr   r   r   r   r    s    

z$IdentifierPreparer.format_constraintc                 C     | j jp| j j}| |||S r  )r  max_index_name_lengthrB   _truncate_and_render_maxlen_namer  r   r  max_r   r   r   r       z1IdentifierPreparer.truncate_and_render_index_namec                 C  r  r  )r  max_constraint_name_lengthrB  r  r  r   r   r   r     r  z6IdentifierPreparer.truncate_and_render_constraint_namec                 C  s^   t |tjr t||kr|d|d  d t|dd   }n| j| |s*|S | |S )Nr   rM  r  )	r  r   rD   rz  r%   md5_hexr  validate_identifierrZ  )r  r   r  r  r   r   r   r    s   &
z3IdentifierPreparer._truncate_and_render_maxlen_namec                 C  
   |  |S r  )r  r  r   r   r   r    r  zIdentifierPreparer.format_indexc                 C  sF   |du r|j }| |}| |}| js!|r!|r!| |d | }|S )z'Prepare a quoted table and schema name.Nr  )r   rZ  r\  ry  r]  )r  r   r  r   r  r`  r   r   r   r    s   

zIdentifierPreparer.format_tablec                 C  r  )zPrepare a quoted schema name.r|  r  r   r   r   r  )  r  z IdentifierPreparer.format_schemac                 C  s(   |durt |tjr||}| |S )Prepare a quoted column name.N)r  r   rD   r  rZ  )r  r   rC  r   r   r   r  .  s
   


z$IdentifierPreparer.format_label_namec                 C  s   |du r|j }|durt|tjr||}t|dds3|r.| j|j||dd | | S | |S |rB| j|j||dd | S |S )r  NrW  F)r  r   r  )	r   r  r   rD   r  r  r  r   rZ  )r  r   r  r   
table_namer  rC  r   r   r   r  <  s6   


z IdentifierPreparer.format_columnc                 C  s@   |  |}| js|r|r| || j|ddfS | j|ddfS )z(Format table name and schema as a tuple.Fr  )r\  ry  r]  r  )r  r   r  r`  r   r   r   r{  i  s   
z#IdentifierPreparer.format_table_seqc                 C  s@   dd | j | j| | jfD \}}}td|||d }|S )Nc                 s  s    | ]}t |V  qd S r  )r(  r)  r  r   r   r   r)  |  s
    
z4IdentifierPreparer._r_identifiers.<locals>.<genexpr>zM(?:(?:%(initial)s((?:%(escaped)s|[^%(final)s])+)%(final)s|([^\.]+))(?=\.|$))+)initialfinalr  )rs  rt  r  r(  r*  )r  r  r  escaped_finalrr   r   r   _r_identifiersz  s   

z!IdentifierPreparer._r_identifiersc                   s(    j } fdddd ||D D S )z:Unpack 'schema.table.column'-like strings into components.c                   s   g | ]}  |qS r   )r  r  r(  r   r   rU    s    z;IdentifierPreparer.unformat_identifiers.<locals>.<listcomp>c                 S  s   g | ]\}}|p	|qS r   r   )r   ar3  r   r   r   rU    rg  )r  findall)r  identifiersr  r   r(  r   unformat_identifiers  s   
z'IdentifierPreparer.unformat_identifiers)rz  Nrz  TF)r  r   r  r   )r  r   r  r  r  )r"   r   r  r   r  r   )r  r   r  r   r  r   )T)r  r  r   r  r  r   )rV  r  r   r  r  r   )TN)FNNFN).r   r   r   r   RESERVED_WORDSr  LEGAL_CHARACTERSr  ILLEGAL_INITIAL_CHARACTERSr  r   r  
attrgetterr\  rv  r  r  r  r  r  r  r  r  r  r]  rZ  rd  rm  rD  r  rE  r%   rZ  r  r  r  r  r  r  r  r  r  r{  rV  r  r  r   r   r   r   r    sl   
 

$
)
	


'4

		





-
r  )r   
__future__r   rm  collections.abcabcrF  rX  enumr   r  r  r  r(  timer   r  r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r   r   r   r   r   r    r!   r"   r#   r$   r%   r  _typingr'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   type_apir3   visitorsr4   r5   r7   r8   util.typingr9   r:   r;   
annotationr<   r=   r>   r  r?   r|  r@   dmlrA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   engine.cursorrT   engine.interfacesrU   rV   rW   rX   rY   rZ   r[   r\   r]   r   _FromHintsTyper  r*  Ir  LEGAL_CHARACTERS_PLUS_SPACEr  r   r  r  r  r  UNICODEr  r  r?  r@  r  or_r  mulrv  modnegltlenegtgeeqis_distinct_fromis_not_distinct_from	concat_opr.  not_match_opr7  r6  r  r{  rC  r  r  r   existsdistinct_opinvany_opall_opdesc_opasc_opnulls_first_opnulls_last_opbitwise_xor_opbitwise_or_opbitwise_and_opbitwise_not_opbitwise_lshift_opbitwise_rshift_oprB  r   r   r   r   r   r   r   r   r   r   r   cuberollupgrouping_setsr   r   rP  _CompoundSelectKeywordr   	UNION_ALLr   
EXCEPT_ALLr   INTERSECT_ALLrQ  r   r   r  r  r  r  r  r  r  r0  r?  rK  rX  r\  r+  r]  r^  r_  r`  
namedtuplera  r  EnsureKWArgr  BinaryElementRoleCompilerColumnElementr  r  r  r_  rw  r  rh  rp  r  r  r   r   r   r   <module>   s~  a
	
 !#$%&1
=wB L

&                                          Uo     .!