U
    d                 ^   @  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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$m/Z0 dd&l1m2Z2 dd'l1m3Z3 dd(l%m4Z4 dd)l%m5Z5 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d0l<m=Z= dd1l>m?Z? dd2l>m@Z@ d3d4l$mAZA d3d%l$m/Z/ d3d5lBmCZC d3d6lBmDZD d3d7lBmEZE ej"rdd8lFmGZG dd9l%mHZH dd:l%mIZI dd;lJmKZK dd<lLmMZM dd=lNmOZO dd>lNmPZP dd?lNmQZQ dd@l(mRZR ddAl(mSZS ddBl(mTZT ddCl(mUZU ddDl(mVZV ddEl)mWZW ddFl-mXZX ddGl-mYZY ddHl-mZZZ ddIl-m[Z[ ddJl-m\Z\ ddKl-m]Z] ddLl-m^Z^ ddMl-m_Z_ ddNl<m`Z` d3dOlambZb d3dPlcmdZd d3dQlcmeZe d3dRlcmfZf d3dSlcmgZg d3dTlcmhZh edUeif ZjdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dddddddddddddddddddddddddddddddddddddddddddddddddddddh^ZkeldejmZneldejmZodd epddD qdgZreldejmZseldejmZteldejmZueldejvZweldejvZxdZyeyddddddĜZze*j{de*j|de*j}de*j~de*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jde*jdi)Ze)jde)jde)jde)jde)jde)jde)jde)jde)jde)jde)jde)jde)jde)jdiZded< ddd ddddddddd	d
dddZe-jjde-jjde-jjde-jjde-jjde-jjdiZG dd deZG dd deDZdZded< dZded< d3Zded< dZÐded < G d!d" d"eEZG d#d$ d$eĐd%d&ZG d'd( d(eZG d)d* d*eZG d+d, d,e	ZG d-d. d.e	ZeeɃ\ZZZZG d/d0 d0eϐd0d1d2gZG d3d4 d4ZG d5d6 d6e/j҃ZG d7d8 d8e+je e(jՃZG d9d: d:e+je e(jՃZG d;d< d<eуZG d=d> d>e؃ZG d?d@ d@eуZG dAdB dBeӃZG dCdD dDeۃZG dEdF dFeDZG dGdH dHeDZG dIdJ dJZdS (K  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)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)_from_objects)
_NONE_NAME)
Executable)NO_ARG)ClauseElement)quoted_name)Column)	TupleType)
TypeEngine)prefix_anon_map)	Visitable   )exc)Literal)Protocol)	TypedDict)_AnnotationDict)_AmbiguousTableNameMap)CompileState)CacheKey)ExecutableDDLElement)Insert)
UpdateBase)
ValuesBase)_truncated_label)BindParameter)ColumnClause)ColumnElement)Label)Function)AliasedReturnsRows)CompoundSelectState)CTE)
FromClause)NamedFromClause)ReturnsRows)Select)SelectState)_BindProcessorType)CursorResultMetaData)_CoreSingleExecuteParams)_ExecuteOptions)_MutableCoreSingleExecuteParams)Dialect)SchemaTranslateMapTyperI   allZanalyseZanalyzeandanyarrayasZascZ
asymmetricauthorizationZbetweenbinaryZbothZcaser   checkcollatecolumn
constraintcreateZcrosscurrent_dateZcurrent_rolecurrent_timecurrent_timestampcurrent_userdefault
deferrabledescZdistinctZdoelseendexceptfalseforZforeignfreezefromfullZgrantgroupZhavingZilikein	initiallyinnerZ	intersectZintoisZisnulljoinleadingleftlikelimit	localtimelocaltimestampZnaturalnewnotZnotnullnulloffoffsetoldononlyororderouteroverlapsZplacingZprimaryZ
referencesrightselectsession_usersetZsimilarZsome	symmetrictableZthentoZtrailingtrueunionuniqueuserZusingverbosewhenwherez^[A-Z0-9_$]+$z^[A-Z0-9_ $]+$c                 C  s   h | ]}t |qS  )str).0xr   r   ;/tmp/pip-unpacked-wheel-wbr0uges/sqlalchemy/sql/compiler.py	<setcomp>   s     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 >> coalesceZCURRENT_DATEZCURRENT_TIMEZCURRENT_TIMESTAMPZCURRENT_USERZ	LOCALTIMEZLOCALTIMESTAMPrandomsysdateZSESSION_USERUSERZCUBEZ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                   @  s2   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   Q  s   
	r   c                   @  s"   e Zd ZddddddddZdS )	_ResultMapAppenderr   zSequence[Any]r   Noner   r   r   type_returnc                 C  s   d S Nr   selfr   r   r   r   r   r   r   __call__q  s    z_ResultMapAppender.__call__Nr   r   r   r   r   r   r   r   r   p  s   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_fromsrK   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_compoundrK   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dZed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   	statementrP   
parameters%Mapping[str, _BindProcessorType[Any]]
processorsOptional[Sequence[str]]positiontupzMapping[str, List[str]]Zparameter_expansionr   r   c                   s,    j dk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   r4   InvalidRequestErrortupler   r   r   r   positional_parameters  s
    
z#ExpandedState.positional_parametersc                 C  s   | j S )z.synonym for :attr:`.ExpandedState.parameters`.r   r   r   r   r   additional_parameters  s    z#ExpandedState.additional_parametersN)r   r   r   r   r   propertyr   r   r   r   r   r   r     s   

r   c                   @  s2   e Zd ZU dZded< ded< ded< ded	< d
S )_InsertManyValueszErepresents state to use for executing an "insertmanyvalues" statementr   is_default_exprr   single_values_exprList[crud._CrudParamElementStr]insert_crud_paramsintnum_positional_params_countedNr   r   r   r   r   r     s
   
r   c                   @  s   e Zd ZdZdZdZdS )CompilerStater   r   r3   N)r   r   r   	COMPILINGSTRING_APPLIEDNO_STATEMENTr   r   r   r   r    s   r  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   r3   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 ZdS )
FromLinterzKrepresents current state for the "cartesian product" detection
    feature.Nc                   s   | j }|sdS t| j}t|}|d k	r8|}|| n| }t|g}|r|r|  |   fdd|D }|	 fdd|D  |
| qL|r||fS dS d S )N)NNc                   s   h | ]} |kr|qS r   r   r   Zedgenoder   r   r     s      z"FromLinter.lint.<locals>.<setcomp>c                 3  s   | ]}||    V  qd S r   )indexr  r  r   r   r   #  s     z"FromLinter.lint.<locals>.<genexpr>)fromsr   edgesremovepopcollectionsdequepopleftdiscard
extendleftdifference_update)r   startr  r  the_rest
start_withstackZ	to_remover   r  r   lint  s&    

zFromLinter.lintc                   sV      \}}|rR|}|rRd}d fdd|D }|j| j| d}t| d S )NzSELECT 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)r  )r   from_r   r   r   r   :  s    z"FromLinter.warn.<locals>.<genexpr>)r  r  )r   ru   r   r  r$   warn)r   r  r  r  templateZ	froms_strmessager   r   r   r#  ,  s     zFromLinter.warn)N)r   r   r   r   r   r#  r   r   r   r   r    s   
%r  r  r  c                      sP  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ddddd d!d"d#Zd$d% fd&d'Zed(d) Zd*d+ Zd,d- Zed.d/ Zd0d1dd2d3d4Zdd%d5d6Zd@d8d9dd:d;d<d=Zed>d? Z  Z S )A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   stringr  stateFzOptional[CursorResultMetaData]_cached_metadataz"Optional[List[ResultColumnsEntry]]_result_columns Optional[SchemaTranslateMapType]schema_translate_maprQ   execution_optionsIdentifierPreparerpreparerr8   _annotationszOptional[CompileState]r   dml_compile_stateOptional[CacheKey]	cache_keyfloat	_gen_timerS   r   Mapping[str, Any]dialectr   r.  render_schema_translatecompile_kwargsc                 C  s   || _ | j j| _|r(|| _| j|| _|dk	rtj| _|| _|j	| _
|j| _| j
rntrft|tsft|j| _| j| jf|| _|r| 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

         .. versionadded:: 1.1

         .. seealso::

            :ref:`schema_translating`

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


        N)r:  identifier_preparerr1  r.  _with_schema_translater  r  r*  r   Zsupports_executioncan_executer2  r   
isinstancer*   AssertionErrorZ_execution_optionsr/  processr)  _render_schema_translatesr  r  r   r7  r   r:  r   r.  r;  r<  r   r   r   __init__  s2    
 
zCompiled.__init__r   r   c                   s   |    t  S r   )_init_compiler_clssuper__init_subclass__cls	__class__r   r   rH    s    zCompiled.__init_subclass__c                 C  s   d S r   r   rI  r   r   r   rF    s    zCompiled._init_compiler_clsc                 C  s$   | j r|| ||S t| jd S r   )r?  Z_execute_compiledr4   ZObjectNotExecutableErrorr   )r   
connectionZdistilled_paramsr/  r   r   r   _execute_on_connection  s      zCompiled._execute_on_connectionc                 K  s   t | t||d S r   r4   UnsupportedCompilationErrorr   r   elementerrkwr   r   r   visit_unsupported_compilation  s    z&Compiled.visit_unsupported_compilationc                 C  s
   t  dS )zReturn a Compiled that is capable of processing SQL expressions.

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

        NNotImplementedErrorr   r   r   r   sql_compiler  s    zCompiled.sql_compilerr2   r   )objkwargsr   c                 K  s   |j | f|S r   _compiler_dispatch)r   rY  rZ  r   r   r   rB    s    zCompiled.processc                 C  s   | j tjkr| jS dS dS )z3Return the string text of the generated SQL or DDL.r(  N)r*  r  r  r)  r   r   r   r   __str__  s    zCompiled.__str__T"Optional[_CoreSingleExecuteParams]&Optional[Sequence[BindParameter[Any]]])Optional[_MutableCoreSingleExecuteParams]paramsextracted_parametersescape_namesr   c                 C  s
   t  dS )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.
        NrV  r   rb  rc  rd  r   r   r   construct_params  s    zCompiled.construct_paramsc                 C  s   |   S )z0Return the bind params for this compiled object.rf  r   r   r   r   rb    s    zCompiled.params)NNT)!r   r   r   r   r   r   r)  is_sqlis_ddlr+  r,  r.  r$   
EMPTY_DICTr/  r2  r   r3  r5  immutabledictrE  rH  classmethodrF  rN  rU  r   rX  rB  r]  rf  rb  __classcell__r   r   rK  r   r&  D  sF   
:


	
   r&  c                   @  sH   e Zd ZdZdZddddZddd	d
ddZdddddddZdS )TypeCompilerz2Produces DDL specification for TypeEngine objects.z	visit_\w+rS   r:  c                 C  s
   || _ d S r   ro  )r   r:  r   r   r   rE    s    zTypeCompiler.__init__r   r   r   )r   rT  r   c                 K  s0   |j r"| jj|j kr"|j | jj }|j| f|S r   )Z_variant_mappingr:  r   r\  r   r   rT  r   r   r   rB    s    zTypeCompiler.process	Exceptionr   )rR  rS  rT  r   c                 K  s   t | ||d S r   )r4   rP  rQ  r   r   r   rU  "  s    z*TypeCompiler.visit_unsupported_compilationN)r   r   r   r   Zensure_kwargrE  rB  rU  r   r   r   r   rn    s
   rn  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rR  r   
_alt_namesr   c                 C  s   || _ || _|f| | _d S r   rt  )r   colr   	alt_namesr   r   r   rE  3  s    z_CompileLabel.__init__c                 C  s   | j jS r   rR  	proxy_setr   r   r   r   ry  8  s    z_CompileLabel.proxy_setc                 C  s   | j jS r   rR  r   r   r   r   r   r   <  s    z_CompileLabel.typec                 K  s   | S r   r   r   rT  r   r   r   
self_group@  s    z_CompileLabel.self_groupN)r   )r   r   r   r   __visit_name__	__slots__rE  r   ry  r   r|  r   r   r   r   rr  *  s   


rr  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

    Zilike_case_insensitive_operandrR  Z
comparatorc                 C  s   || _ |j| _d S r   r  )r   rR  r   r   r   rE  U  s    zilike_case_insensitive.__init__c                 C  s   | j jS r   rx  r   r   r   r   ry  Y  s    z ilike_case_insensitive.proxy_setc                 C  s   | j jS r   rz  r   r   r   r   r   ]  s    zilike_case_insensitive.typec                 K  s   | S r   r   r{  r   r   r   r|  a  s    z!ilike_case_insensitive.self_groupc                 C  s   t | j|S r   )r  rR  _with_binary_element_type)r   r   r   r   r   r  d  s    
z0ilike_case_insensitive._with_binary_element_typeN)r   r   r   r   r}  r~  rE  r   ry  r   r|  r  r   r   r   r   r  D  s   

r  c                   @  sl  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e7fdUdVdWdXddYdZd[d\d]Z8e9d4d^d_d`Z:ej;dd^dadbZ<e9dcdd Z=e9dedf Z>e9dgdh Z?ej@did^djdkZAejBdld^dmdnZCeDjEdodp ZFdqdr ZGdsdt ZHej@dud^dvdwZIdxdy ZJe9dzd{ ZKdd|dd}d~ddZLdd|dddddddddZMejBdd ZNe9dd ZOdddd}dddZPeQddd ZRde	d< ej@dd^ddZSej@eQddd ZTej@eQddd ZUdd ZVdddZWdd ZXdddZYdddZZdddZ[dd Z\dd Z]dddddddZd&dddZ^dd Z_dd Z`dd Zadd Zbdd Zcdd ZddddZedddZfddÄ Zgddń ZhddǄ ZiddɄ Zjdd˄ Zkdd̈́ Zlddτ Zmddф Znddӄ ZoddՄ Zpddׄ Zqddل Zrddۄ Zsdd݄ Ztdd߄ Zudd Zvdd ZwddddZd&dddZxdd Zydd Zzdd Z{dddZ|dd Z}dd Z~dddZdd Zdd Zdd Zdd Zdd 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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d0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zd@dA ZdBdC ZdDdE ZdFdG ZdHdI ZddJdKZdLdM ZedfdNdOZdPdQ ZdRdS Zd&dTd&dUdVdWZd&d&dXdYdZZdd&ddd4d[d\d6dZd&d]	d^d_Zd`da ZdddddedfdbdddcdddedfdZd4dg	dhdiZdjdk Zdldm ZddndoZdpdq Zdrds ZddtduZdvdw ZddxdyZdzd{ Zd|d} Zd&d&dd~d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ZdZe	d< dddZΐdd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dddZd&d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Đ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ل ZdS (  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]]]Z	postfetchr   zSequence[Column[Any]]insert_prefetchupdate_prefetchNz&Optional[Sequence[ColumnElement[Any]]]implicit_returningisplaintextzDict[str, BindParameter[Any]]bindszDict[BindParameter[Any], str]
bind_nameszList[_CompilerStackEntry]r  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_stringz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_translaterI  r   r   r   rF    s    zSQLCompiler._init_compiler_clsc                 C  s2   t d| j}t d| d| _| j| _d S )Nr(  r  r  )reescaperu   r  compiler  r  )rJ  regr   r   r   r    s    z SQLCompiler._init_bind_translaterS   r'  r4  r   r  r   )r:  r   r5  column_keysfor_executemanylintingrZ  c                 K  s  || _ || _|r>dd |d D }dd |d D }	|	|f| _|| _|| _i | _t | _g | _	g | _
|j| _| jr|jd | _}
|
r|jdkrdnd| _t| _nt|j | _d	| _|jp|j| _t | _i | _i | _tj| ||f| | js| js| jrxtrt |t!st"| js*| jrxtr@t |t#s@t"|j$rPd
| _%n(| jrx| jrr| j&j'rx|j(rxd
| _%t|j | _)| j*t+j,kr| jr| jr| -  n| .  | j/r| j0dd
d}| j1|d
d 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>  s      z(SQLCompiler.__init__.<locals>.<dictcomp>r   c                 S  s   i | ]}||gqS r   r   r  r   r   r   r    s      r   r   r   r  NTFrd  _no_postcompile)_populate_self)2r  r5  r  r  r  r  r$   Zcolumn_dictr  r  r,  
positional
paramstyle
startswithr  r  _pyformat_templater  BIND_TEMPLATESr  label_lengthmax_identifier_lengthr1   anon_maptruncated_names_truncated_countersr&  rE  r  r  r  r   r@  r>   rA  r?   Z_inliner  r:  Zinsert_executemany_returningZ_return_defaultsr  r*  r  r  _process_numeric_process_positionalr  rf  #_process_parameters_for_postcompile)r   r:  r   r5  r  r  r  rZ  cksmckbmnbr   r   r   r   rE    st    



 zSQLCompiler.__init__r   c                 C  s   | j dkrdS | j jS d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  r   r   r   r   r   insert_single_values_expr  s    
z%SQLCompiler.insert_single_values_exprc                 C  s:   | j r| j S | jdk	r2t| jr2dd | jjD S 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  s   | j S )zVbackwards compatibility; returns the
        effective_returning collection.

        )r  r   r   r   r   	returningH  s    zSQLCompiler.returningc              
   C  sB   z| j d d W S  tk
r< } ztd|W 5 d}~X Y nX dS )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)r  
IndexError)r   Zier   r   r   current_executableP  s    zSQLCompiler.current_executablec                 C  s   t | jt | j S r   )listr  r  r   r   r   r   prefetchs  s    zSQLCompiler.prefetchzDict[Any, Any]c                 C  s   i S r   r   r   r   r   r   _global_attributesw  s    zSQLCompiler._global_attributesz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  sp   | j | j }}g d | _ | _z0| jr8| jd }d|d< nd}| j | jfV  W 5 |r\|d || | _ | _X dS )z;special API to support the use case of 'nested result sets'Fr   r  TN)r,  r  r  r  )r   Zresult_columnsZordered_columnsentryr   r   r   _nested_result  s    


zSQLCompiler._nested_resultc                   s  j r
tjtjkstjr$tjjdkr6dnjjdksFtdg dddfdd	 t	j
 j_jrd
d j D tjtkstfddD _ n_ jrg t	j
 jj} fddjjD }tjj||jjd_d S )Nr   %sr   r   zre.Match[str]r   )mr   c                   s:   |  d}|r|  S |  d |  dS d S )Nr   r3   r   )rp   append)r  Znormal_bind)placeholder	positionsr   r   find_position  s    

z6SQLCompiler._process_positional.<locals>.find_positionc                 S  s   i | ]\}}||qS r   r   )r   kvr   r   r   r    s      z3SQLCompiler._process_positional.<locals>.<dictcomp>c                   s   g | ]}  ||qS r   getr   r   )reverse_escaper   r   r    s    z3SQLCompiler._process_positional.<locals>.<listcomp>c              
     s4   g | ],}|d  |d t j |d |d fqS )r   r   r3   r   )r  sub_positional_patternr   r  )r  r   r   r   r    s   r   r   r  r  )r   rA  r*  r  r  r  r:  r  r  r  r  r)  r  itemslenr  r   r  r   r   r  )r   r   r  r   )r  r  r  r  r   r   r    sJ    


  


zSQLCompiler._process_positionalc           	        sd  j s
tjtjkstd}i  jrZjd k	rZtfddj	
 D j	
 }n
j	
 }|D ]T}| krvqhj| }|jks|jkrd  |< qhj | }|d7 }| |< qh|_t _jrt }fdd  D  t |kstj fddj_jr`jj  }dd	 jjD }tjj||jjd
_d S )Nr   c                 3  s   | ]}| j kr|V  qd S r   )r  r  r   r   r   r     s   
z/SQLCompiler._process_numeric.<locals>.<genexpr>c                   s    i | ]\}} j |||qS r   )r  r  )r   r   posr   r   r   r    s    z0SQLCompiler._process_numeric.<locals>.<dictcomp>c                   s    |  d S Nr   rp   r  	param_posr   r   <lambda>      z.SQLCompiler._process_numeric.<locals>.<lambda>c                 S  s$   g | ]}|d  |d d|d fqS )r   r   r  r   r   r  r   r   r   r  &  s   z0SQLCompiler._process_numeric.<locals>.<listcomp>r  )r  rA  r*  r  r  r  r  	itertoolschainr  valuesr  r  r  r  next_numeric_posr  r   r  r  r  _pyformat_patternr  r)  r   r  r   r   r  )	r   numr   	bind_namebindphZ
len_beforer   r  r   )r  r   r   r    sd    

	






 zSQLCompiler._process_numericzVMutableMapping[str, Union[_BindProcessorType[Any], Sequence[_BindProcessorType[Any]]]]c                   s   dd  fdd j D D S )Nc                 S  s   i | ]\}}|d k	r||qS r   r   r   r   valuer   r   r   r  ?  s    z0SQLCompiler._bind_processors.<locals>.<dictcomp>c                 3  sN   | ]F} 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   Z	elem_typer   r   r   r   F  s   z9SQLCompiler._bind_processors.<locals>.<genexpr>.<genexpr>N)	r  r   _is_tuple_typer  r:  r   r   r/   types)r   	bindparamr   r   r   r   A  s   
z/SQLCompiler._bind_processors.<locals>.<genexpr>)r  r   r   r   r   _bind_processors6  s
    	

zSQLCompiler._bind_processorsc                 C  s   t | jdkS r  )r  r  r   r   r   r   is_subqueryP  s    zSQLCompiler.is_subqueryc                 C  s   | S r   r   r   r   r   r   rX  S  s    zSQLCompiler.sql_compilerr^  r   )rb  rd  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

        Tr  )rf  r  )r   rb  rd  r   r   r   r   construct_expanded_stateW  s    z$SQLCompiler.construct_expanded_stater_  zOptional[int]rR   )rb  rc  rd  _group_number_checkr  r   c                   s2  | j r2|s2| jdk	st|s(t| jjS td|o>t| j}|r| j	dkrZt
dn
| j	d }| j}	|	dk	svt|	\ }
 fddt||D }nd}|ri }| j D ]\}}|r| j||n|}|j|kr||j ||< q||kr || ||< q|rD|jrD|r.tjd|j|f dd	ntjd
|j dd	q|rX|||}n|}|jrp|j||< q|j||< q|S i }| j D ]\}}|r| j||n|}|r|jr|rtjd|j|f dd	ntjd
|j dd	|r|||}n|}|jr|j||< n
|j||< q|S d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   r  Z	extractedr  r  r   r   r    s
   
  z0SQLCompiler.construct_params.<locals>.<dictcomp>z@A value is required for bind parameter %r, in parameter group %dZcd3x)codez)A value is required for bind parameter %r)r  r  rA  dictr   r4   r   r   r  r5  CompileErrorr  zipr  r  r  r   requiredcallableeffective_valuer  )r   rb  rc  rd  r  r  r  Zhas_escaped_namesZorig_extracted
ckbm_tupler  Zresolved_extractedpdr  r   escaped_nameZvalue_paramr   r  r   rf  n  s    





 zSQLCompiler.construct_paramsc                   s   | j jjj  fddi }| j}| jD ]F}||krHq:|jjrrfddtt	|jj
D ||< q:|j||< q:|S )Nc                   sD   |   }|d k	r<d ks(|kr<d ks8|kr<|S d S d S r   )_unwrapped_dialect_implZget_dbapi_type)typZdbtype)dbapir:  exclude_typesinclude_typesr   r   lookup_type  s    z<SQLCompiler._get_set_input_sizes_lookup.<locals>.lookup_typec                   s   g | ]} |qS r   r   )r   r$  )r(  r   r   r    s   z;SQLCompiler._get_set_input_sizes_lookup.<locals>.<listcomp>)r:  Zinclude_set_input_sizesZexclude_set_input_sizesr%  r  r  r   r  r   r/   r  )r   Z
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  rg  r   r   r   r   rb    s    zSQLCompiler.params)r   r  r   c                   s  i }j }|dkrj}jr8g }j}|dkr@j}nd }}j}td|td|i }i i }	d}
jr|dk	r|}jrg }
n
j	 }j
}|D ]p |r|  n }j  }|jkr|kr||}j||d|< q|jkr|kr|	| }d}n0| }|||}|\}}||	|< ||< |js|| |jjr|dk	snt| fddt|dD  n| fdd|D  |
dk	r|
d	d |D  n|dk	r|d
d |D  dd |D | < q|dk	r|  qfdd}tj||}|
dk	r~|dk	sDtfddt|
jD jfdd|}||
 t|||||}|r|_ |_|j _jrt!|jpd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  s`   | ]X\}}t |d D ]D\}} kr  |d   dk	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&   | ]\}} kr|  fV  qd S r   r   )r   r   r  )r   single_processorsr   r   r     s   c                 s  s   | ]\}}|V  qd S r   r   r   r   r  r   r   r   r     s    c                 s  s   | ]\}}|V  qd S r   r   r4  r   r   r   r     s     c                 S  s   g | ]\}}|qS r   r   )r   Z
expand_keyr  r   r   r   r    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   r3   ~~r   r   c                 3  s   | ]}d  |f V  qdS z%s%s%sNr   )r   expbe_leftbe_rightr   r   r     s   z]SQLCompiler._process_parameters_for_postcompile.<locals>.process_expanding.<locals>.<genexpr>)rp   splitru   )r  r   exprtok)replacement_expressionsr8  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   r    s    zCSQLCompiler._process_parameters_for_postcompile.<locals>.<dictcomp>c                   s    |  d S r  r  r  r   r   r   r    r  zASQLCompiler._process_parameters_for_postcompile.<locals>.<lambda>r   )#r  r)  r  r  r   r  r   r  r  r  r  r  r  r  r  render_literal_bindparamr  $_literal_execute_expanding_parameterliteral_executeupdater   r  rA  r.  extendr  r  r  _post_compile_patternr  r  r   r   r  r  )r   r   r  Zexpanded_parametersZpre_expanded_stringZnew_positiontupZpre_expanded_positiontupr   Znew_processorsZto_update_setsZnumeric_positiontupnamesZebnr"  	parameterr  	to_updater  Zleep_resreplacement_exprr?  r   Zexpanded_stater   )r   r  r>  r   r3  r2  r   r  $  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$   	preloadedZengine_cursorrO   Z_create_description_match_mapr,  )r   cursorr   r   r   _create_result_map  s    zSQLCompiler._create_result_map_BindNameForColProtocol_get_bind_name_for_colzCallable[[Any], str]c                 C  s
   | j }|S r   )rO  )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 k	st| jj}tr6t|t	s6t|j
}fdd|jD d |j  d k	r j| jd  | jkrfdd}|nd |dd |jD  fdd}|S )Nc                   s"   g | ]}t d  |d|fqS )r  N)operatormethodcallerr   rv  )param_key_getterr   r   r  
  s   zKSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>.<listcomp>c                   s    |  | }|d k	r|S | S d S r   r  )	lastrowidr   Zparam_value)autoinc_keyr   r   _autoinc_getter  s    zPSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>._autoinc_getterc                 S  s   g | ]
}|j qS r   r  rT  r   r   r   r  1  s     c                   sN   dk	r   dkr.fddD S  fddD S 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   rP  rv  r   r   r   r   >  s     zWSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>.get.<locals>.<genexpr>c                 3  s8   | ]0\}}| kr(d k	r$q.n|V  qd S r   r   rY  )autoinc_colautoinc_getterrV  r   r   r   r   @  s   r   rV  r   )rZ  r[  getterslastrowid_processorrow_fnr\  r   r  3  s    zDSQLCompiler._inserted_primary_key_from_lastrowid_getter.<locals>.get)r$   rK  engine_resultrQ  r   rA  r   r   r@  r=   r   primary_keyZ_autoincrement_columnr   Z_cached_result_processorr:  r  result_tuple)r   resultr   r   rX  r  r   )rZ  r[  rW  r]  r^  rU  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 k	s*t| jj	}trDt
|tsDt| j|j}| j}|d k	sbtdd t|D tdfdd|jD  |dd |jD  fd	d
}|S )Nr3   )rc  c                 S  s   i | ]\}}||qS r   r   )r   idxrv  r   r   r   r  `  s      zKSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.<dictcomp>z'List[Tuple[Callable[[Any], Any], bool]]c                   s<   g | ]4}|kr"t | d fnt d |ddfqS )Tr  NF)rR  
itemgetterrS  rT  )rU  retr   r   r  d  s   zKSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.<listcomp>c                 S  s   g | ]
}|j qS r   r  rT  r   r   r   r  o  s     c                   s    fddD S )Nc                 3  s&   | ]\}}|r|n| V  qd S r   r   )r   rP  Zuse_rowr   rowr   r   r   r  s   zWSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.get.<locals>.<genexpr>r   )ri  r   )r]  r_  rh  r   r  q  s    zDSQLCompiler._inserted_primary_key_from_returning_getter.<locals>.get)typingr   Zenginerc  r$   rK  r`  r   rA  r   r@  r=   rQ  r   r  r.  r   ra  rb  )r   rc  r   r   r  r  r   )r]  rU  rg  r_  r   +_inserted_primary_key_from_returning_getterM  s*    z7SQLCompiler._inserted_primary_key_from_returning_getterc                 C  s   dS )zCalled when a SELECT statement has no froms, and no FROM clause is
        to be appended.

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

        r(  r   r   r   r   r   default_fromy  s    zSQLCompiler.default_fromc                 K  s   d|j j| f| d S Nr  r  rR  r\  )r   groupingasfromrZ  r   r   r   visit_grouping  s    zSQLCompiler.visit_groupingc                 K  s   d|j j| f| d S rm  rn  )r   ro  rZ  r   r   r   visit_select_statement_grouping  s    z+SQLCompiler.visit_select_statement_groupingc              
   K  s   | j r| jjrztd| j d d }W n. tk
rT } ztd|W 5 d }~X Y nX |j\}}}|rl|}	n|}	|jj	}
|
d k	r|
j
|	kr|
|	|
j
 r|jj	|d< | j|jfd|i|S )N'Union[SelectState, CompoundSelectState]r  r   ECan't resolve label reference for ORDER BY / GROUP BY / DISTINCT etc.render_label_as_labelwithin_columns_clause)r  r:  Zsupports_simple_order_by_labelr   KeyErrorr4   r  _label_resolve_dictrR  Z_order_by_label_elementr   Zshares_lineagerB  )r   rR  rv  rZ  r   ke	with_cols
only_froms	only_colsZresolve_dictZorder_by_elemr   r   r   visit_label_reference  sP    		z!SQLCompiler.visit_label_referencec              
   K  s   | j s| |jS ztd| j d d }W n8 tk
rb } ztj|jdtj	|d W 5 d }~X Y nX |j
\}}}z|r||j }	n
||j }	W n8 tk
r }
 ztj|jdtj	|
d W 5 d }
~
X Y n X |	|d< | j|	fd|i|S d S )Nrs  r  r   rt  )extraZexc_clsrS  ru  rv  )r  rB  Z_text_clauser   rw  r   Z_no_text_coercionrR  r4   r  rx  )r   rR  rv  rZ  r   ry  rz  r{  r|  rv  rS  r   r   r   visit_textual_label_reference  sD    

z)SQLCompiler.visit_textual_label_referencec                 K  s   |o| }||k}	|	s|r>t |jtjr8| d|j}
n|j}
|r|d k	rj||
|j||
f|j | |j |jj| fddd|t	t
j  | j||
 S |	r| j||
S |jj| fddi|S d S )NcolidentT)rv  within_label_clauserv  F)r@  r   r   r@   _truncated_identifierru  r   rR  r\  	OPERATORSr   as_r1  format_label)r   rs  add_to_result_mapr  rv  ru  result_map_targetsrT  Zrender_label_with_asZrender_label_onlyZ	labelnamer   r   r   visit_label  sJ    
zSQLCompiler.visit_labelc                 C  s   t dd S )Nz:Cannot compile Column object until its 'name' is assigned.r4   r  r   r^   r   r   r   _fallback_column_name	  s    z!SQLCompiler._fallback_column_namec                 K  s   |j }| j|f|S r   )Z	_resolvedrB  )r   rR  rT  Zsql_elementr   r   r   visit_lambda_element	  s    z SQLCompiler.visit_lambda_elementColumnClause[Any]zOptional[_ResultMapAppender]r   z Optional[_AmbiguousTableNameMap])r^   r  include_tabler  ambiguous_table_name_maprZ  r   c                 K  sB  |j  }}|d kr| |}|j}	|	s>t|tjr>| d|}|d k	rx|||jf| }
|jrh|
|jf7 }
||||
|j	 |	r| 
|}n| j|}|j}|d ks|r|js|S | j|}|r| j|d }nd}trt|tst|j }|s|r||kr|| }t|tjr&| d|}|| j| d | S d S )Nr  r  r(  alias)r   r  
is_literalr@  r   r@   r  r   	_tq_labelr   escape_literal_columnr1  quoter   Znamed_with_columnschema_for_objectquote_schemar   rJ   rA  )r   r^   r  r  r  r  rZ  r   	orig_namer  targetsr   effective_schemaZschema_prefixZ	tablenamer   r   r   visit_column	  sF    	

zSQLCompiler.visit_columnc                 K  s   | j |jS r   )r1  format_collation	collationr   rR  rT  r   r   r   visit_collationY	  s    zSQLCompiler.visit_collationc                 K  s   |j S r   r   )r   Z
fromclauserZ  r   r   r   visit_fromclause\	  s    zSQLCompiler.visit_fromclausec                 K  s   |j S r   r  )r   r  rZ  r   r   r   visit_index_	  s    zSQLCompiler.visit_indexc                 K  s&   ||d< | j |d< | jjj|jf|S )Ntype_expressionr=  )r1  r:  type_compiler_instancerB  r   )r   
typeclauserT  r   r   r   visit_typeclauseb	  s    
zSQLCompiler.visit_typeclausec                 C  s   | j jr|dd}|S Nr  %%r1  _double_percentsreplacer   textr   r   r   post_process_texti	  s    zSQLCompiler.post_process_textc                 C  s   | j jr|dd}|S r  r  r  r   r   r   r  n	  s    z!SQLCompiler.escape_literal_columnc              	     sR    fdd}j sd_|r2|d d ftj tdd t|jS )Nc                   s:   |  d}|jkr(jj| f S j|f S d S r  )rp   Z_bindparamsrB  bindparam_string)r  r   rT  r   
textclauser   r   do_bindparamt	  s    

z2SQLCompiler.visit_textclause.<locals>.do_bindparamTc                 S  s
   |  dS r  r  r  r   r   r   r  	  r  z.SQLCompiler.visit_textclause.<locals>.<lambda>)	r  r  r#   NULLTYPEBIND_PARAMS_ESCr  BIND_PARAMSr  r  )r   r  r  rT  r  r   r  r   visit_textclauses	  s     
zSQLCompiler.visit_textclausec                 K  s   | j  }|r| jn| j d }t t |d}| j | |jrJ| || |pl|dkrb|ddpl|dd}|r|j | _| _	|j ot
|j| _|jD ]}	| j|	d| jd q| j|jf|}
| jr|st| j nd }| j|d	|
 }
| j d |
S )
Nr  r   r   r"   r   r   Fr   T)rv  r  nesting_level)r  _default_stack_entryr   r  _independent_ctes_dispatch_independent_ctesr  r  r  r  r   Zcolumn_argsr  rB  _add_to_result_maprR  r  r  _render_cte_clauser  )r   Ztafcompound_indexrp  rT  toplevelr  	new_entrypopulate_result_mapr  r  r  r   r   r   visit_textual_select	  sD    

	

z SQLCompiler.visit_textual_selectc                 K  s   dS NNULLr   r   r<  rT  r   r   r   
visit_null	  s    zSQLCompiler.visit_nullc                 K  s   | j jrdS dS d S )Nr   1r:  supports_native_booleanr  r   r   r   
visit_true	  s    zSQLCompiler.visit_truec                 K  s   | j jrdS dS d S )Nrk   0r  r  r   r   r   visit_false	  s    zSQLCompiler.visit_falsec                   s$   | dd  fdd|D D S )Nc                 s  s   | ]}|r|V  qd S r   r   r   sr   r   r   r   	  s   z7SQLCompiler._generate_delimited_list.<locals>.<genexpr>c                 3  s   | ]}|j f V  qd S r   r[  r  rT  r   r   r   r   	  s     ru   )r   r   	separatorrT  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 S t	tj }|
dd  fdd|D D S d S )Nr   r   c                 s  s   | ]}|r|V  qd S r   r   r  r   r   r   r   	  s   z;SQLCompiler._generate_delimited_and_list.<locals>.<genexpr>c                 3  s   | ]}|j f V  qd S r   r[  r  r  r   r   r   	  s     )r   ZBooleanClauseListZ_process_clauses_for_booleanr   and_ZTrue_Z
_singletonZFalse_r\  r  ru   )r   clausesrT  Zlccr  r   r  r   _generate_delimited_and_list	  s    

z(SQLCompiler._generate_delimited_and_listc                 K  s   d| j |f| S N(%s))visit_clauselist)r   
clauselistrT  r   r   r   visit_tuple	  s    zSQLCompiler.visit_tuplec                 K  s0   |j }|d krd}n
t|j  }| j|j|f|S Nr  )rR  r  r  r  )r   r  rT  sepr   r   r   r  	  s
    
zSQLCompiler.visit_clauselistc              
   K  sz   |j }| |dd }|r&|||f|S zt| }W n0 tk
rb } zt| ||W 5 d }~X Y nX | j|j|f|S d S )NZexpression_clauselist)rR  _get_operator_dispatchr  rw  r4   rP  r  r  )r   r  rT  	operator_dispopstringrS  r   r   r   visit_expression_clauselist	  s$        z'SQLCompiler.visit_expression_clauselistc                 K  s   d}|j d k	r&||j j| f|d 7 }|jD ]4\}}|d|j| f| d |j| f| d 7 }q,|jd k	r|d|jj| f| d 7 }|d7 }|S )NzCASE r  zWHEN z THEN zELSE ZEND)r  r\  ZwhensZelse_)r   clauserZ  r   Zcondrc  r   r   r   
visit_case
  s*    

zSQLCompiler.visit_casec                 K  s   |j j| f|S r   )Ztyped_expressionr\  )r   Ztype_coercerT  r   r   r   visit_type_coerce
  s    zSQLCompiler.visit_type_coercec                 K  s$   d|j j| f||jj| f|f S )NzCAST(%s AS %s))r  r\  r  )r   r   rZ  r   r   r   
visit_cast
  s    zSQLCompiler.visit_castc              	   K  s   d|d t jkrdn\|d t jkr&dnJ|d dk rTd| jt t|d f|f nd| jt |d f|f |d t jkrdn\|d t jkrdnJ|d dk rd| jt t|d f|f nd| jt |d f|f f S )	Nz	%s AND %sr   zUNBOUNDED PRECEDINGzCURRENT ROWz%s PRECEDINGz%s FOLLOWINGr   zUNBOUNDED FOLLOWING)r   ZRANGE_UNBOUNDEDZRANGE_CURRENTrB  literalabs)r   range_rT  r   r   r   _format_frame_clause#
  s0    

z SQLCompiler._format_frame_clausec                   s   |j jf }|jr,dj|jf  }n |jrHdj|jf  }nd }d|d fddd|jfd|jffD |r|gng  f S )	NzRANGE BETWEEN %szROWS BETWEEN %sz%s OVER (%s)r  c                   s6   g | ].\}}|d k	rt |rd||jf f qS )Nz%s BY %s)r  r\  )r   wordr  rZ  r   r   r   r  J
  s    z*SQLCompiler.visit_over.<locals>.<listcomp>Z	PARTITIONORDER)rR  r\  r  r  Zrowsru   Zpartition_byorder_by)r   ZoverrZ  r  r  r   r  r   
visit_over:
  s4    	zSQLCompiler.visit_overc                 K  s$   d|j j| f||jj| f|f S )Nz%s WITHIN GROUP (ORDER BY %s))rR  r\  r  )r   ZwithingrouprZ  r   r   r   visit_withingroupW
  s    zSQLCompiler.visit_withingroupc                 K  s$   d|j j| f||jj| f|f S )Nz%s FILTER (WHERE %s))funcr\  Z	criterion)r   Z
funcfilterrZ  r   r   r   visit_funcfilter]
  s    zSQLCompiler.visit_funcfilterc                 K  s*   | j |j|j}d||jj| f|f S )NzEXTRACT(%s FROM %s))extract_mapr  fieldr<  r\  )r   extractrZ  r  r   r   r   visit_extractc
  s
    zSQLCompiler.visit_extractc                 K  s*   | j |jf|}| j|f|}d||f S )Nz(%s).%s)visit_functionfnr  )r   rR  rT  Zcompiled_fnZcompiled_colr   r   r   visit_scalar_function_columnj
  s    z(SQLCompiler.visit_scalar_function_columnzFunction[Any])r  r  rZ  r   c                   s   |d k	r||j |j d|j td|j   d }|rD||f|}nt| jd   rj|jr d7  n6|j  j	
 st tjrj	 n   d  d fdd|jD  g dj|f|i }|jr|d7 }|S )	Nr   zvisit_%s_funcz%(expr)sr  c                   s4   g | ],}j |s t tjr,j |n|qS r   )r1  _requires_quotes_illegal_charsr@  r   r-   r  )r   r=  r   r   r   r   r  
  s   

z.SQLCompiler.visit_function.<locals>.<listcomp>r<  z WITH ORDINALITY)r   r   getattrlowerr   r  Z_deannotaterL  Z	_has_argsr1  r  r@  r   r-   r  ru   Zpackagenamesfunction_argspecZ_with_ordinality)r   r  r  rZ  r  r  r   r  r   r  o
  s:    


	zSQLCompiler.visit_functionc                 K  s   |  |jS r   )visit_sequencesequence)r   Z
next_valuerT  r   r   r   visit_next_value_func
  s    z!SQLCompiler.visit_next_value_funcc                 K  s   t d| jj d S )Nz2Dialect '%s' does not support sequence increments.rW  r:  r   )r   r  rT  r   r   r   r  
  s
    zSQLCompiler.visit_sequencec                 K  s   |j j| f|S r   )Zclause_exprr\  )r   r  rZ  r   r   r   r  
  s    zSQLCompiler.function_argspecc                   sb  j  }|j|f}|r(js(|_|j}|r8jnj d }|pV| oV|dd}	|dkrh||d< j |d |d |||	d |jr| j	|j
 }
d	|
 d	  fd
dt|jD }dd< |j|ftf d i7 }|j|f7 }|jr$|j|f7 }jrR|s<tj nd }j|dd| }j d |S )Nr  r   Fr   r   r   r   )r   r   r"   r   r   r  c                 3  s*   | ]"\}}|j f |d V  qdS ))rp  r  Nr[  )r   r/  r  rp  rZ  r   r   r   r   
  s    z4SQLCompiler.visit_compound_select.<locals>.<genexpr>r  rp  Tr  include_following_stack)r  _compile_state_factoryr   r   r  r  r  r  r  compound_keywordskeywordru   r.  Zselectsgroup_by_clauser  order_by_clause_has_row_limiting_clause_row_limit_clauser  r  r  r  )r   csrp  r  rZ  r  r   Zcompound_stmtr  Zneed_result_mapr  r  r  r   r  r   visit_compound_select
  sX    


	 z!SQLCompiler.visit_compound_selectc                 K  s*   |j d k	r| j|f|S | j|f|S d S r   )_fetch_clausefetch_clauselimit_clause)r   r  rZ  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  Z
qualifier1Z
qualifier2attrnamer   r   r   r  
  s    z"SQLCompiler._get_operator_dispatchc                 K  s   |d k	r"||f7 }||d< ||d< |j rt|jr8td| |j dd}|r\|||j f|S | j|t|j  f|S nL|jr| |jdd}|r|||jf|S | j|t|j f|S n
tdd S )Nr  r  zFUnary expression does not support operator and modifier simultaneouslyunaryrR  modifierz,Unary expression has no operator or modifier)rR  r  r4   r  r   _generate_generic_unary_operatorr   _generate_generic_unary_modifier)r   r  r  r  rT  r  r   r   r   visit_unary
  sN    
      zSQLCompiler.visit_unaryc                 K  sx   | j jrP| j|jf|d | jt|j|jjjt	j
kr>|jjnt	
 f| S | j|jf|d | j|jf| S d S )N / )r:  div_is_floordivrB  rw   r   ZCastr   r   _type_affinityr#   ZNumericr   r[   rR  rT  r   r   r   visit_truediv_binary  s,    
z SQLCompiler.visit_truediv_binaryc                 K  sh   | j jr<|jjjtjkr<| j|jf|d | j|jf| S d| j|jf|d | j|jf|  S d S )Nr  z	FLOOR(%s))	r:  r  r   r   r  r#   ZIntegerrB  rw   r  r   r   r   visit_floordiv_binary7  s"    z!SQLCompiler.visit_floordiv_binaryc                 K  s6   |j s| jjr| j|jf|S d| j|jf| S d S )Nz%s = 1Z_is_implicitly_booleanr:  r  rB  rR  r   rR  rR  rT  r   r   r   visit_is_true_unary_operatorH  s    z(SQLCompiler.visit_is_true_unary_operatorc                 K  s:   |j s| jjr"d| j|jf| S d| j|jf| S d S )NNOT %sz%s = 0r  r  r   r   r   visit_is_false_unary_operatorQ  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_binaryZ  s     z%SQLCompiler.visit_not_match_op_binaryc                 K  s   d| j |t| f| S r  )_generate_generic_binaryr  r  r   r   r   visit_not_in_op_binary_  s     z"SQLCompiler.visit_not_in_op_binaryc                 K  sv   |t jkr4t|dkr.dddd |D  S dS n>|t jkrht|dkrbdddd |D  S d	S n
| |S d S )
Nr   z(%s)) OR (1 = 1r   c                 s  s   | ]
}d V  qdS r  Nr   r   rR  r   r   r   r   k  s     z6SQLCompiler.visit_empty_set_op_expr.<locals>.<genexpr>zNULL) OR (1 = 1z(%s)) AND (1 != 1c                 s  s   | ]
}d V  qdS r(  r   r)  r   r   r   r   r  s     zNULL) AND (1 != 1)r   	not_in_opr  ru   in_opvisit_empty_set_expr)r   r   	expand_oprT  r   r   r   visit_empty_set_op_exprg  s    

z#SQLCompiler.visit_empty_set_op_exprc                 K  s   t d| jj d S )Nz3Dialect '%s' does not support empty set expression.r  )r   Zelement_typesrT  r   r   r   r,  y  s
    z SQLCompiler.visit_empty_set_exprc           	        sL  j j}|sP|jr<jjr$dndj jj }nj gj}n|js~|jrt	|d t
jrt	|d ttfs|jrtdjjrdnddfddt|D  }n|r*j}||}|r|dstd	|dd
}|d |d   d fdd|D }ndfdd|D }d|fS )NVALUES r(  r   z?bind_expression() on TupleType not supported with literal_bindsr   c                 3  s8   | ]0\}}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  Z
param_typer   r   r   r     s   z[SQLCompiler._literal_execute_expanding_parameter_literal_binds.<locals>.<genexpr>.<genexpr>N)ru   r  r   r  )r   r/  r0  rH  r   r   r   r     s   
 zQSQLCompiler._literal_execute_expanding_parameter_literal_binds.<locals>.<genexpr>r3   )unexpected format for expanding parameterr5  r   r   c                 3  s&   | ]}d   |jf V  qdS r6  r+  r   r   r  r9  r:  rH  r   r   r   r     s   c                 3  s   | ]} | jV  qd S r   r2  r3  r0  r   r   r     s   r   )r   r#  r:  r  tuple_in_valuesr.  r  r-  _isnullr@  collections_abcr   r   bytes_has_bind_expressionrW  ru   r.  rF  searchrp   rA  r;  )	r   rH  r  bind_expression_templatetyp_dialect_implreplacement_expressionZpost_compile_patternr  r=  r   r4  r   2_literal_execute_expanding_parameter_literal_binds  s\      



z>SQLCompiler._literal_execute_expanding_parameter_literal_bindsc                   sX  j r|S j}j|jr2jnjjjrZj	rZfdd nfdd |sg j
rjjj}njgj}nj
sЈjr t|d tjr t|d ttfs jrtfddt|dD |jrdnd	d
 fddt|D  }n0fddt|dD d
 fddD }|fS )Nc                   s    j d| i S Nr   )render_bind_castr   r  )bind_templaterH  r   r<  r   r   _render_bindtemplate  s
    
zNSQLCompiler._literal_execute_expanding_parameter.<locals>._render_bindtemplatec                   s    d| i S r?  r   r  )rA  r   r   rB    s    r   c                   s6   g | ].\}}t |d D ]\}}d ||f |fqqS )r   r,  r-  )r   r/  r0  r1  r  r  r   r   r    s    zDSQLCompiler._literal_execute_expanding_parameter.<locals>.<listcomp>r   r/  r(  r   c                 3  s8   | ]0\ d d  fddtD  V  qdS )r  r   c                 3  s.   | ]&\}} t  |  d  V  qdS )r   N)r  )r   r1  r  )rB  r/  rI  r0  r   r   r     s   zMSQLCompiler._literal_execute_expanding_parameter.<locals>.<genexpr>.<genexpr>N)ru   r.  )r   )rB  rI  )r/  r0  r   r     s   
zCSQLCompiler._literal_execute_expanding_parameter.<locals>.<genexpr>c                   s    g | ]\}}d  |f |fqS )z%s_%sr   )r   r/  r  r  r   r   r    s   c                 3  s   | ]\}} |V  qd S r   r   r  )rB  r   r   r     s    )rC  r>  r:  r   r#  r  r  r  _bind_typing_render_castsr@  r  r.  r  r-  r6  r@  r7  r   r   r8  Z	_is_arrayrA  r.  r5  ru   )r   r   rH  r  r:  r=  r   )rB  rA  r   rH  r   rI  r<  r   rB    sd     	  



z0SQLCompiler._literal_execute_expanding_parameterc              
   K  s  |rft |jrf|d k	rJ|d }|jt|jj|g |j	j|g  n|jt|jj|j	j | j
rt|jtjrt|j	tjrd|d< |p|j}| |dd }	|	r|	||f|S zt| }
W n0 tk
r } zt| ||W 5 d }~X Y nX | j||
f||d|S d S )Nenclosing_lateralTrC  r[   )from_linterlateral_from_linter)r   Zis_comparisonrR  r  rD  r  productrw   r(   r   r  r@  r   rA   r  r  rw  r4   rP  r&  )r   r[   r"  eager_groupingrE  rF  rT  rD  r  r  r  rS  r   r   r   r#    sN    	 
 zSQLCompiler.visit_binaryc                 K  s   | j |jf|S r   )rB  Zsql_functionr  r   r   r   &visit_function_as_comparison_op_binaryO  s    z2SQLCompiler.visit_function_as_comparison_op_binaryc                 K  sT   | j jr,| j|jf|d | j|jf| S | j|jf|d | j|jf| S d S )Nz %% r   )r1  r  rB  rw   r   r  r   r   r   visit_mod_binaryR  s    zSQLCompiler.visit_mod_binaryc                 K  s*   |j |d< | j|d| |j d f|S )NrH  r  )rH  r&  r  r  r  r   r   r   visit_custom_op_binary`  s    
z"SQLCompiler.visit_custom_op_binaryc                 K  s   | j || |jd f|S r  )r  r  r  r  r   r   r   visit_custom_op_unary_operatorh  s     z*SQLCompiler.visit_custom_op_unary_operatorc                 K  s   | j |d| |j f|S r  )r  r  r  r  r   r   r   visit_custom_op_unary_modifierm  s     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| }|rb|rbd| }|S )N
_in_binaryFT
_binary_oprH  r  )r  rR  rw   r\  r   )r   r[   r  rH  rT  rN  r  r   r   r   r&  r  s0    

z$SQLCompiler._generate_generic_binaryc                 K  s   ||j j| f| S r   rn  r   r  r  rT  r   r   r   r    s    z,SQLCompiler._generate_generic_unary_operatorc                 K  s   |j j| f|| S r   rn  rP  r   r   r   r    s    z,SQLCompiler._generate_generic_unary_modifierc                 C  s   t jdtjdS )Nz'%')r   )r   Zliteral_columnr#   
STRINGTYPEr   r   r   r   _like_percent_literal  s    z!SQLCompiler._like_percent_literalc                 K  s   d|j j| f| dS )Nzlower(r  rn  r  r   r   r   $visit_ilike_case_insensitive_operand  s    z0SQLCompiler.visit_ilike_case_insensitive_operandc                 K  s2   |  }| j}||j||_| j||f|S r   _clonerR  concatr   visit_like_op_binaryr   r[   rR  rT  percentr   r   r   visit_contains_op_binary  s    z$SQLCompiler.visit_contains_op_binaryc                 K  s2   |  }| j}||j||_| j||f|S r   rU  rR  rV  r   visit_not_like_op_binaryrX  r   r   r   visit_not_contains_op_binary  s    z(SQLCompiler.visit_not_contains_op_binaryc                 K  sB   |  }| j}t|j|_|t|j||_| j||f|S r   rU  rR  r  rw   rV  r   visit_ilike_op_binaryrX  r   r   r   visit_icontains_op_binary  s    z%SQLCompiler.visit_icontains_op_binaryc                 K  sB   |  }| j}t|j|_|t|j||_| j||f|S r   rU  rR  r  rw   rV  r   visit_not_ilike_op_binaryrX  r   r   r   visit_not_icontains_op_binary  s    z)SQLCompiler.visit_not_icontains_op_binaryc                 K  s,   |  }| j}||j|_| j||f|S r   )rU  rR  _rconcatr   rW  rX  r   r   r   visit_startswith_op_binary  s    z&SQLCompiler.visit_startswith_op_binaryc                 K  s,   |  }| j}||j|_| j||f|S r   )rU  rR  rd  r   r\  rX  r   r   r   visit_not_startswith_op_binary  s    z*SQLCompiler.visit_not_startswith_op_binaryc                 K  s<   |  }| j}t|j|_|t|j|_| j||f|S r   )rU  rR  r  rw   rd  r   r_  rX  r   r   r   visit_istartswith_op_binary  s
    z'SQLCompiler.visit_istartswith_op_binaryc                 K  s<   |  }| j}t|j|_|t|j|_| j||f|S r   )rU  rR  r  rw   rd  r   rb  rX  r   r   r   visit_not_istartswith_op_binary  s
    z+SQLCompiler.visit_not_istartswith_op_binaryc                 K  s,   |  }| j}||j|_| j||f|S r   rT  rX  r   r   r   visit_endswith_op_binary  s    z$SQLCompiler.visit_endswith_op_binaryc                 K  s,   |  }| j}||j|_| j||f|S r   r[  rX  r   r   r   visit_not_endswith_op_binary  s    z(SQLCompiler.visit_not_endswith_op_binaryc                 K  s<   |  }| j}t|j|_|t|j|_| j||f|S r   r^  rX  r   r   r   visit_iendswith_op_binary  s
    z%SQLCompiler.visit_iendswith_op_binaryc                 K  s<   |  }| j}t|j|_|t|j|_| j||f|S r   ra  rX  r   r   r   visit_not_iendswith_op_binary  s
    z)SQLCompiler.visit_not_iendswith_op_binaryc                 K  sL   |j dd }d|jj| f||jj| f|f |rFd| |tj nd S )Nr  z
%s LIKE %s ESCAPE r(  	modifiersr  rw   r\  r   r+  r#   rQ  r   r[   rR  rT  r  r   r   r   rW    s    z SQLCompiler.visit_like_op_binaryc                 K  sL   |j dd }d|jj| f||jj| f|f |rFd| |tj nd S )Nr  z%s NOT LIKE %srm  r(  rn  rp  r   r   r   r\    s    z$SQLCompiler.visit_not_like_op_binaryc                 K  s:   |t jkr*| }t|j|_t|j|_| j||f|S r   )r   Zilike_oprU  r  rw   r   rW  r  r   r   r   r_    s
    
z!SQLCompiler.visit_ilike_op_binaryc                 K  s:   |t jkr*| }t|j|_t|j|_| j||f|S r   )r   Znot_ilike_oprU  r  rw   r   r\  r  r   r   r   rb    s
    
z%SQLCompiler.visit_not_ilike_op_binaryc                 K  s&   |j dd}| j||rdndf|S )Nr   Fz BETWEEN SYMMETRIC z	 BETWEEN ro  r  r&  r   r[   rR  rT  r   r   r   r   visit_between_op_binary  s     
z#SQLCompiler.visit_between_op_binaryc                 K  s&   |j dd}| j||rdndf|S )Nr   Fz NOT BETWEEN SYMMETRIC z NOT BETWEEN rq  rr  r   r   r   visit_not_between_op_binary  s    
z'SQLCompiler.visit_not_between_op_binaryc                 K  s   t d| jj d S Nz/%s dialect does not support regular expressionsr4   r  r:  r   r  r   r   r   visit_regexp_match_op_binary  s
    z(SQLCompiler.visit_regexp_match_op_binaryc                 K  s   t d| jj d S ru  rv  r  r   r   r    visit_not_regexp_match_op_binary$  s
    z,SQLCompiler.visit_not_regexp_match_op_binaryc                 K  s   t d| jj d S )Nz;%s dialect does not support regular expression replacementsrv  r  r   r   r   visit_regexp_replace_op_binary*  s
    z*SQLCompiler.visit_regexp_replace_op_binaryc                 K  s  |s|j | j}|jr||}	| j|	fd||o8|j ||d|}
|jrtd|
}|sft	dd|
d|
d|
df }
|r| j|fd|
d	|}d
| S |
S |s|p|jp|o| j}|p|j}nd}|r| j|fddi|}|jrd
| }|S | |}|| jkr| j| }||k	r|js@|jrp|j|jsp|j|jsptd| nh|j|jkrtd|f nH|js|jr|jr|jrtdntd|j d|j d| | j|j< | j|< | j}|r4|\}}|jD ]*}|j|kr||j }|| | q|jrBd| _|r||rTd| _|rl|  j|hO  _n|  j|hO  _| j|f||j|j d|}|jrd
| }|S )NT)skip_bind_expressionrv  literal_bindsrC  render_postcompilez&^(.*)\(__\[POSTCOMPILE_(\S+?)\]\)(.*)$r1  z$(__[POSTCOMPILE_%s~~%s~~REPL~~%s~~])r3   r   r   )rv  r;  r  Frv  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:  r9  bind_expressionrB  r~  r  matchrA  rp   rA  rC  r  _truncate_bindparamr  r   ry  intersectionZ_cloned_setr4   r  Z_is_crudr   r  r  Z
isoutparamhas_out_parametersr  r  r  r  )r   r  rv  r{  rz  rC  r|  rZ  implr  wrappedr  rg  r}  r   existingr   r  r  Zbpcbr   r   r   visit_bindparam0  s    


	 


	


zSQLCompiler.visit_bindparamc                 C  s
   t  d S r   rV  )r   r   Z
dbapi_typesqltextr   r   r   r@    s    zSQLCompiler.render_bind_castc           
      K  s   |t k	r|}nZ|jd krb|jd krb|dd }|rR|tjtjfkrRtd|j	f | j
tjf|S |j}|jr| j}||||d\}}	|	S | ||jS d S )NrO  zBound parameter '%s' rendering literal NULL in a SQL expression; comparisons to NULL should not use operators outside of 'is' or 'is not')r;  )r+   r  r  r  r   is_is_notr$   Zwarn_limitedr   rB  r#   r  r  r~  r>  r+  r   )
r   r  r+  r;  rT  r  opZleeprI  rJ  r   r   r   rA    s(    
z$SQLCompiler.render_literal_bindparamc              
   C  s   | | j}|r`z
||W S  tk
r\ } z$tdt| d| d|W 5 d}~X Y q|X ntdt| d| dS )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.

        z Could not render literal value "z" with datatype z); see parent stack trace for more detail.Nz:No literal value renderer is available for literal value ")Z_cached_literal_processorr:  rq  r4   r  sql_utilZ_repr_single_value)r   r  r   	processorer   r   r   r+    s    
	z SQLCompiler.render_literal_valuec                 C  s@   || j kr| j | S |j}t|tjr2| d|}|| j |< |S )Nr  )r  r   r@  r   r@   r  )r   r  r
  r   r   r   r    s    


zSQLCompiler._truncate_bindparamr@   )ident_classr   r   c                 C  s   ||f| j kr| j ||f S || j}t|| jd kr| j|d}|dt| jd d d t|dd   }|d | j|< n|}|| j ||f< |S )N   r   r   r  r3   )	r  	apply_mapr  r  r  r  r  maxhex)r   r  r   ZanonnamecounterZ	truncnamer   r   r   r  $  s    z!SQLCompiler._truncated_identifier)r   r   c                 C  s
   || j  S r   r  r   r   r   r   r   
_anonymize9  s    zSQLCompiler._anonymizezOptional[TypeEngine[Any]]zOptional[Set[str]])	r   r}  r~  escaped_fromr  accumulate_bind_namesvisited_bindparamrT  r   c                   s  |d k	r| | |d k	r$|| |sR j|rR j fdd|}	|}|	}|rh j||i _|rd| }
|r||
S |d k	r| j}|j	r 
|||
}
|
S  jtjkrĈ jd|i }
n jd|i }
|d k	r jjr| j}|j
r 
|||
}
|
S )Nc                   s    j | d S )Nr   )r  rp   r  r   r   r   r  X  r  z.SQLCompiler.bindparam_string.<locals>.<lambda>z__[POSTCOMPILE_%s]r   )addr  r  r:  r  r  r   r#  r:  Zrender_literal_castr@  r*  r  r  r  r  rC  )r   r   r}  r~  r  r  r  r  rT  new_namerg  Z	type_implr   r   r   r  <  sL    


zSQLCompiler.bindparam_stringc                 C  sF   |  }|dd  t|j|jD ]\}}|j| fd|i| q"d S )Ncte_opts)copyr  r  r  Z_independent_ctes_optsr\  )r   stmtrT  Zlocal_kwcteoptr   r   r   r    s     z&SQLCompiler._dispatch_independent_ctesrH   zOptional[_FromHintsType]zOptional[CTE]zOptional[FromLinter]zselectable._CTEOpts)	r  rp  ashint	fromhintsvisiting_cterE  r  rZ  r   c                   s     }	|	 jkst||d< |j}
t|
tjr< d|
}
d}d}| }|j	pV|j	}| j
kr j
| \}}}||
ks~t||
f} j| }|j	r|j	rtd||
f}|rt jnd}||
f }} j|= | j|< ||f  j
|< n6|rt jnd}||
f}| jkr" j| }nd }|d k	r||k}||jksN||krTd}nx||jkr|	|= | }||ks|t||kst j
|= n8|jd k	s|jd k	r||rd}ntd|
 |s|sd S |jd k	r|j}|jj}t|tjr d|}n|}d }|r\| j|< ||f  j
|< | jkrT j|f| |s\||	kr\|jrrd _ j||
}|jr|j}dd	 |dD }|d
d fdd|D  7 }|dddkst js|jj fd|i|S  j||jf|}|jj fddi|}|d||f 7 }|j rT|d j||j f| 7 }||	|< |r|rr|
|j!|< |s|r j||
S |r̈ j||} j"|
r j#|
}
| $|
7 }|S  j||
S d S )Nr  r  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_namer  repeatedr   r   r   r    s   z)SQLCompiler.visit_cte.<locals>.<listcomp>r  r   c                 3  s    | ]} j j| jd V  qdS )r  N)r1  format_label_namer  )r   identr   r   r   r      s
    z(SQLCompiler.visit_cte.<locals>.<genexpr>subqueryrp  z AS %s
(%s)r  )%r  r  rA  r   r@  r   r@   r  _get_reference_ctenestingr  r  r4   r  r  r  Z	_restatesZ_is_clone_ofcompareZ
_cte_alias	visit_cte	recursiver  r1  format_aliasrR  _generate_columns_plus_namesru   r  r\  _generate_prefixes	_prefixes	_suffixesr  _requires_quotesr  get_render_as_alias_suffix)r   r  rp  r  r  r  rE  r  rZ  Z	self_ctescte_nameZ
is_new_cteZembedded_in_current_named_cteZ_reference_cter  	cte_levelr  Zexisting_cte_optsZcte_level_nameZexisting_cteZold_level_nameZnew_level_nameZexisting_cte_reference_cteZpre_alias_cteZcte_pre_alias_namer  Z
col_sourceZ
recur_colsprefixesrs   r   r   r   r    s   





 
	  
zSQLCompiler.visit_ctec                 K  s4   |j rd |d< |jr"| j|f|S | j|f|S d S )NrE  )Zjoins_implicitly_is_lateralvisit_lateralvisit_aliasr  r   r   r   visit_table_valued_aliasT  s
    z$SQLCompiler.visit_table_valued_aliasc                 K  s   | j |f|S r   )r  r  r   r   r   visit_table_valued_column\  s    z%SQLCompiler.visit_table_valued_columnc
              	     s  |r:dkr j st d< |	r:dkr:dkr:|	d< |d k	r|j kr jjf||||| d}|r|sz|rd|f }|S   }d< |s|rt jtjrd j}n j}|r҈j	
 |S |r|	r||	j <  jjfd|d}|rd|f }|j	
 | } jr^ jr^|dd	 fd
d jD  7 }|r |kr| |  |}|S  jjfd|iS d S )NrD  rF  )rp  r  iscrudr  lateralenclosing_aliasr  r  r  T)rp  r  r   c                 3  sB   | ]:}d j |j jr2djjj|jf ndf V  qdS )z%s%s %sr(  N)r1  r  r   Z_render_derived_w_typesr:  r  rB  r   rT  r  rZ  r   r   r   r     s   z*SQLCompiler.visit_alias.<locals>.<genexpr>r  )r  rA  rR  r\  r@  r   r   r@   r  r1  r  r  r  Z_supports_derived_columnsZ_render_derivedru   r  format_from_hint_text)r   r  rp  r  r  r  r  r  r  rE  rZ  rs   Z
alias_namerg  r   r  r   r  _  s    



 
   zSQLCompiler.visit_aliasc                 K  s   d|d< | j |f|S )NTr  r  )r   r  rT  r   r   r   visit_subquery  s    zSQLCompiler.visit_subqueryc                 K  s   d|d< d| j |f| S )NTr  z
LATERAL %sr  )r   Zlateral_rT  r   r   r   r    s    zSQLCompiler.visit_lateralc                 K  sR   d| j |fddi|| j| f|f }|jd k	rN|d|jj| f| 7 }|S )Nz%s TABLESAMPLE %srp  Tz REPEATABLE (%s))r  Z_get_methodr\  seed)r   Ztablesamplerp  rT  r  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  s8   | ]0}|D ]&}j tj|d  ji fV  q
qdS )r  N)rB  r   r   Z_column_typesr|  )r   chunkelemrR  rT  r   r   r   r     s    z-SQLCompiler._render_values.<locals>.<genexpr>r/  )
setdefaultr{  ru   _data)r   rR  rT  Ztuplesr   r  r   _render_values  s
    
zSQLCompiler._render_valuesc              	     s   j |f }|jrd }n$t|jtjr8d|j}n|j}|jrJd}nd}|r|rl|d k	rb|nd|j|< |rd||	j
|d fdd|jD f }nd	||f }|S )
Nr  zLATERAL r(  z(unnamed VALUES element)z%s(%s)%s (%s)r   c                 3  s$   | ]}|j fd di V  qdS )r  FNr[  r  r  r   r   r      s   z+SQLCompiler.visit_values.<locals>.<genexpr>z%s(%s))r  Z_unnamedr@  r   r   r@   r  r  r  r  r1  r  ru   columns)r   rR  rp  rE  rT  r  r   r  r   r  r   visit_values  s0    zSQLCompiler.visit_valuesc                 K  s   d| j |f| dS rm  )r  r  r   r   r   visit_scalar_values  s    zSQLCompiler.visit_scalar_valuesc                 C  s   d| S )Nr   r   )r   Zalias_name_textr   r   r   r    s    z&SQLCompiler.get_render_as_alias_suffixr   r   r   c                 C  sF   |d k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  r4   r  r,  r  r   r   r   r   r   r    s    zSQLCompiler._add_to_result_mapc                 K  s"   | j d||d|dkri n|f|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_argsrT  r   r   r   _label_returning_column&  s    z#SQLCompiler._label_returning_columnc                   s  j | j}|jr(|s|r(|}n}|rh| j}|
rL|  fdd}ql|k	rl|  fdd}nd}|	sxtdttj	r|k	rt
|jjfd}n|}n|r|dk	stdt
|||jfd}n|k	rd}n~ttjr|oj ojdk	}nTttjr d	}n@ttjr<jp8|}n$ttjs\jdkr\d}nd	}|r|s||
rvtj}t|tjst|n|}t
|||fd}n|}|j|	||d
 |j| f|S )z.produce labeled columns present in a select().c                   s    | |d| d S )Nr   r   r   r   r   r   )r  r   r   r  a  s    z;SQLCompiler._label_select_column.<locals>.add_to_result_mapc                   s    | |f| | d S r   r   r  r  r^   r   r   r  i  s       NzX_label_select_column is only relevant within the columns clause of a SELECT or RETURNING)rw  z*proxy_name is required if 'name' is passedTF)rv  r  r  )r   r  r:  Z_has_column_expressionZcolumn_expressionr  rA  r@  r   rD   rr  r   rR  r  rB   r  r   Z
TextClauseZUnaryExpressionZwraps_column_expressionZNamedColumnZ_non_anon_labelZ_anon_name_labelr@   rD  r\  )r   r   r^   r  rp  r  r   r  r  rv  column_is_repeatedneed_column_expressionsr  r  Zcol_exprr  Zresult_exprZrender_with_labelr   r  r   r  ;  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  s   d S r   r   )r   Zbyfromsr   r   r   get_select_hint_text  s    z SQLCompiler.get_select_hint_textc                 C  s   d S r   r   r   r   r  r   r   r   r    s    zSQLCompiler.get_from_hint_textc                 C  s   d S r   r   r  r   r   r   get_crud_hint_text  s    zSQLCompiler.get_crud_hint_textc                 C  s
   d |S r  r  )r   Z
hint_textsr   r   r   get_statement_hint_text  s    z#SQLCompiler.get_statement_hint_textr   r  r   r   c                 K  sh   | j  }|r| jn| j d }||| }|d }|d }	|rV|sV|j||	dd}
n|j||	d}
|
S )Nr  r   r   r   Zexplicit_correlate_fromsZimplicit_correlate_froms)r  r  r  _get_display_froms
difference)r   select_stmtrp  r  rT  r  r  r   r   r   r  r   r   r   _display_froms_for_select  s"    z%SQLCompiler._display_froms_for_selecttranslate_select_structurec	              
     s  |d kst dd|	d< jf|	}
|
j|	d< |
jj }|rPjsP|
_|d k	pZ|}jrjfd i|	}|k	r|
}}|jf|	}
|
j|rjnjd }|p|ddp|dd |rdsd	|	kr|	d	= 	|
| ||}|	
 ddd
 d}jrJ\}}|rN||d 7 }nd }jrb|	 jr|jjf|	7 }|jf|	7 }dd  fdd|
jD D }r|d k	rttdd |
jD dd |jD fddjD _||
|||||	}jrRfddjD }|rR|d| 7 }jr|rf|r|svtjnd }j|d| }jr|djjf|	 7 }jd |S )NzpSQLAlchemy 1.4 requires use of the translate_select_structure hook for structural translations of SELECT objectsFrv  r  rp  r  r   r   r  )r  rv  zSELECT r  c                 S  s   g | ]}|d k	r|qS r   r   r  r   r   r   r    s   z,SQLCompiler.visit_select.<locals>.<listcomp>c                   s4   g | ],\}}}}}j | ||||d 
qS ))r   r  r  r  r  r  r   r   r  r  r^   r  )rp  r  r  r  r  r   r   r   r    s$   c                 S  s   g | ]\}}}}}|qS r   r   r   r   r  r  r   r  r   r   r   r    s   c                 S  s   g | ]\}}}}}|qS r   r   r  r   r   r   r    s   c              	     s4   g | ],\}}}}t ||t fd d|D |qS )c                 3  s   | ]}  ||V  qd S r   r  )r   o	translater   r   r     s     z6SQLCompiler.visit_select.<locals>.<listcomp>.<genexpr>)r   r   )r   r   r   rY  r   r  r   r   r    s   
   c                   s$   g | ]\}}|d  j jfkr|qS r  r:  r   )r   dialect_nameZhtr   r   r   r    s   r  )rA  r  Z_ambiguous_table_name_mapr   r  r   r  r  r  _setup_select_stackr  rD  _hints_setup_select_hintsr  r  r  r  get_select_precolumnscolumns_plus_namesr  r  r,  _compose_select_bodyZ_statement_hintsr  r  r  r  r  r  )r   r  rp  insert_intor  r  Zselect_wraps_forr  rE  rZ  r   r  Zis_embedded_selectZnew_select_stmtZcompile_state_wraps_forr  r  r  	hint_textbyfrominner_columnsZper_dialectr  r   )rp  r  r  r  r  r   r  r   visit_select  s    
 
 

      


 zSQLCompiler.visit_selectr   zTuple[str, _FromHintsType])r   r   c                   s*    fdd|j  D } |}||fS )Nc              	     s<   i | ]4\\}}}|d  j jfkr||d|j ddi qS )r  r   T)r  r:  r   r\  )r   r"  r:  r  r   r   r   r    s   
 z3SQLCompiler._setup_select_hints.<locals>.<dictcomp>)r  r  r  )r   r   r  r  r   r   r   r    s
    

zSQLCompiler._setup_select_hintsc                 C  s   |d }|d }|dkr"||d< nD|rf|d }	t |	j}
t |j|
krftdd|
|d t |jf |r|s|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   r  )r   r   r"   r   )r  r  r  r4   r  r  r  r   r(   r   r  r  )r   r   r   r  rp  r  r  r   r   r   Znumcolsr  Znew_correlate_fromsZall_correlate_fromsr  r   r   r   r    sJ    


zSQLCompiler._setup_select_stackc	                   s  |d |7 }jt@ r:ti t jt@ }	|rB_nd d}	|sN| }|r|d7 }|jr|d  fdd|D 7 }q|d fdd|D 7 }n|	 7 }|j
rj|j
fdi}
|
r|d|
 7 }|	rd k	st  |jr|j|f7 }|jr<j|jf}
|
r<|d	|
 7 }|jrV|j|f7 }|jrp|j|f7 }|jd k	r|j|f7 }|S )
Nr   Fz 
FROM c                   s&   g | ]}|j fd  dqS )T)rp  r  rE  r[  r   fr  rE  rZ  r   r   r   r  N  s   z4SQLCompiler._compose_select_body.<locals>.<listcomp>c                   s$   g | ]}|j fd  dqS )Trp  rE  r[  r  )rE  rZ  r   r   r   r  [  s   rE  z 
WHERE z	 
HAVING )ru   r  r
  r  r   r  rE  rstripr  rl  _where_criteriar  rA  r#  _group_by_clausesr  Z_having_criteria_order_by_clausesr	  r
  r  Z_for_update_argfor_update_clause)r   r  r   r   r  r  r  r  rZ  Zwarn_lintingtr   r  r   r  )  sl    

	z SQLCompiler._compose_select_bodyc                   s*   d  fdd|D }|r&|d7 }|S )Nr  c                 3  s4   | ],\}}|d ks|j jkr|jf V  qdS ))Nr  Nr  )r   prefixr  r  r   r   r     s    z1SQLCompiler._generate_prefixes.<locals>.<genexpr>r  )r   r  r  rT  r  r   r  r   r    s    zSQLCompiler._generate_prefixesc                 C  s0  | j s
dS |r|dkrt }t| j  D ]T}| j|  \}}}|jpN|j}||kpf|of||d k}	|r,|	srq,| j | ||< q,n| j }|sdS tdd |D }
| 	|
d }|d
dd | D 7 }|d7 }|r,|dkr,t| D ]8}| j|  \}}}| j |= | j||f= | j| = q|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  s   g | ]
}|j qS r   )r  )r   r  r   r   r   r    s     z2SQLCompiler._render_cte_clause.<locals>.<listcomp>r  , 
c                 S  s   g | ]}|qS r   r   )r   txtr   r   r   r    s     z
 )r  r$   r  r  keysr  r  r  rW   get_cte_preambleru   r  r  )r   r  r  r  r  r  r  r  r  Zis_rendered_levelr  Zcte_textr   r   r   r    s>    

zSQLCompiler._render_cte_clausec                 C  s   |rdS dS d S )NzWITH RECURSIVEZ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(  )Z_distinct_onr$   warn_deprecatedZ	_distinctr   r   rT  r   r   r   r    s    z!SQLCompiler.get_select_precolumnsc                 K  s,   | j |jttj f|}|r$d| S dS dS )z5allow dialects to customize how GROUP BY is rendered.z
 GROUP BY r(  N)r  r  r  r   comma_op)r   r   rT  Zgroup_byr   r   r   r    s     zSQLCompiler.group_by_clausec                 K  s,   | j |jttj f|}|r$d| S dS dS )z5allow dialects to customize how ORDER BY is rendered.z
 ORDER BY r(  N)r  r  r  r   r  )r   r   rT  r  r   r   r   r	    s     zSQLCompiler.order_by_clausec                 K  s   dS )Nz FOR UPDATEr   r  r   r   r   r    s    zSQLCompiler.for_update_clauser>   Sequence[ColumnElement[Any]]r  returning_colsr  rT  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  )r  r  rT  r  r   r  r   r   r     s$   z0SQLCompiler.returning_clause.<locals>.<listcomp>T)cols
RETURNING r   )r  r   _select_iterablesru   r   r  r  r  rT  r  r   r  r   returning_clause  s    	 zSQLCompiler.returning_clausec                 K  s^   d}|j d k	r&|d| j|j f| 7 }|jd k	rZ|j d krB|d7 }|d| j|jf| 7 }|S )Nr(  z
 LIMIT z

 LIMIT -1z OFFSET )Z_limit_clauserB  _offset_clause)r   r   rT  r  r   r   r   r    s    


zSQLCompiler.limit_clausec           
      K  s   |d kr|j }|j}n
ddd}d}|jd k	rf|j}|rJ||rJ| }| j|f|}	|d|	 7 }n|rr|d7 }|d k	r|r||r| }|d| j|f||d rdnd|d	 rd
ndf 7 }|S )NF)rY  	with_tiesr(  z
 OFFSET %s ROWSz
 OFFSET 0 ROWSz
 FETCH FIRST %s%s ROWS %srY  z PERCENTr  z	WITH TIESZONLY)r  Z_fetch_clause_optionsr  Z_simple_int_clauseZrender_literal_executerB  )
r   r   r  Zrequire_offsetZ"use_literal_execute_for_simple_intrT  Zfetch_clause_optionsr  Zoffset_clauseZ
offset_strr   r   r   r  "  s8    

zSQLCompiler.fetch_clausec	                 K  s   |r|j |j|< |s|r| j|}
|rL|
rL| j|
d | j|j }nJ| j|j}|
s|r|j|kr| d||j }|| | j	d | }|r||kr| 
|||| |}|S dS d S )Nr  r  r(  )fullnamer  r1  r  r  r  r   r  r  r  r  )r   r   rp  r  r  r  
use_schemarE  r  rZ  r  rg  Z	anon_namer   r   r   visit_tableK  sD    
    zSQLCompiler.visit_tablec                 K  s   |r |j t|jj|jj |jr,d}n|jr8d}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 Tr  z ON rE  )r  rD  r  rG  rw   r(   r   ro   Zisouterr\  Zonclause)r   ru   rp  rE  rZ  Z	join_typer   r   r   
visit_join{  sT       	zSQLCompiler.visit_joinc                   sB    fdd|j  D }|j|kr: ||j||j d}||fS )Nc                   s*   i | ]"\\}}}|d  j jfkr||qS r  r  )r   r   r:  r  r   r   r   r    s   
 z1SQLCompiler._setup_crud_hints.<locals>.<dictcomp>T)r  r  r   r  )r   r  
table_textdialect_hintsr   r   r   _setup_crud_hints  s    

   zSQLCompiler._setup_crud_hintsc                 C  s$   | j jo"| j jo"| j jp"| jp"| jS r   )r:  supports_multivalues_insertZuse_insertmanyvaluesZ!use_insertmanyvalues_wo_returningr  r,  )r   r   r   r   r   (_insert_stmt_should_use_insertmanyvalues  s    z4SQLCompiler._insert_stmt_should_use_insertmanyvaluesc           &      #  s  	j }|d k	std|j d}t}|jr^	jjs^tdD ]\}|||fV  q@d S ||d}	jj	}	|	rt	j
}
t|j}|
| }t||	| | }t}d }d}|| || rdnd }|j}|d k	std 	jsp	jr	jni td }	fddddfd	d
|D  d}|fdd
|D }fdd||D }d}nd}t }i }d|j d}t  |jD ]} |d  q r	j}|d k	st fddt|D }t|t|d t| kst	jr0t	j}t| dd|}|r|d| }g |d|< rxdd fdd
tt|D D }	jrp|j}|t|d krd }}|}n4|d d  }|d d  }fdd
|D }|t| d d }	jrF|dkrF|r
td } |t| |  }!t	fdd
t| |!D }"||" }|d|}#ttj !|}$||$ | }$nbg }%|" }$t|D ]:\|%#|dt$ |$fdd|D  q|dd|%}#|#|$|||fV  |d7 }q0d S )Nr  r  r   Z__EXECMANY_TOKEN__r   c                   s@   | D ]6}  ||}|jd|i jd| di }q|S )Nr   Z__EXECMANY_INDEX__)r  r  r  )r  	formattedr   )r  r   r   r   apply_placeholders  s    zISQLCompiler._deliver_insertmanyvalues_batches.<locals>.apply_placeholdersr   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   r  )r   r  r)  r   )r  r   r   r     s   
 c                   s   i | ]}| d  | qS )r   r   r   r   r   r   r    s    zASQLCompiler._deliver_insertmanyvalues_batches.<locals>.<dictcomp>r(  z), r   c                   s   h | ]\}}| kr|qS r   r   )r   re  r   )all_names_we_will_expandr   r   r   *  s   z@SQLCompiler._deliver_insertmanyvalues_batches.<locals>.<setcomp>z\d+r  c                 S  s   g | ]\}}}|||fqS r   r   )r   Znew_keylen_r$  r   r   r   r  D  s   zASQLCompiler._deliver_insertmanyvalues_batches.<locals>.<listcomp>c                 3  s2   | ]*} D ] \}}}| d | ||fV  q
qdS )r  Nr   )r   r  r   r+  r$  )generic_setinputsizesr   r   r   F  s    r   c                 3  s   | ]}|  V  qd S r   r   r  )expand_pos_lower_indexexpand_pos_upper_indexr   r   r   X  s   c                 3  s   | ]} j  | V  qd S r   r@  r   r/  r   r   r   r   r  s   ZEXECMANY_INDEX__c                   s    i | ]}| d   | qS )__r   r   )r/  paramr   r   r    s      )%r  rA  r   r  r   r:  supports_default_metavaluer.  r  Zinsertmanyvalues_max_parametersr  r  minr  r  r  r   ru   r  r  rD  r   r  r  r  r  r  r  ranger  r   r  r  from_iterabler  r  r   )&r   r   r   r,  Z
batch_sizeZimvZexecutemany_valuesZ	lenparamsZbatchnumZ
max_paramsZtotal_num_of_paramsZnum_params_per_batchZnum_params_outside_of_batchZbatchesZprocessed_setinputsizesZtotal_batchesr  Zall_keysZformatted_values_clauseZkeys_to_replaceZbase_parametersZexecutemany_values_w_commar  r   Zall_expand_positionsescapedbatchZnum_ins_paramsZextra_params_leftZextra_params_rightZbatch_iteratorZexpanded_values_stringr  ri   r  Zreplaced_statementZreplaced_parametersZreplaced_values_clausesr   )
r*  r(  r  r-  r.  r,  r/  r2  r   r   r   !_deliver_insertmanyvalues_batches  s&    	





  





 
 z-SQLCompiler._deliver_insertmanyvalues_batchesc                 K  s  |j || f|}|j}| j }|r@d| _| js4|| _| js@|| _| jt t |d d}d }| jrvd|k}|svg }t	j
| |||fd|i|}| jr|d k	rt|}| jr| jd k	r|  j|7  _n|| _|j}	|	s
| jjs
| jjs
| jjs
td| jj |jr6| jjs.td| jj |j}	n|j}	| j}
| jj}d}|jrl|| j||jf|7 }|d	7 }|
|j}|jr| ||\}}|jr|  || ||7 }|	s|s|d
d!dd |	D  7 }| j"s|j#r| j$|| j"p|j#|d}| j%r |d| 7 }nd }|j&d k	r| j'| jd d fddi|}| j(r| jj)r|snt| jnd }|d| j*|dd|f 7 }n|d| 7 }n|	s|r|d7 }|rT| +|rTt,d| jj-t.d|	|| _/nv|jr|dd!dd |j0D  7 }nNd!dd t.d|	D }|d| 7 }|rT| +|rTt,d|t.d|	|| _/|j1d k	r| j'|j1f|}|r|d| 7 }|r| j%s|d| 7 }| j(r| jj)s|st| jnd }| j*|dd| }| j2d |S )NTr  r   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.zINSERT zINTO  (%s)r   c                 S  s   g | ]\}}}}|qS r   r   )r   r  r<  r   r   r   r    s    
 z,SQLCompiler.visit_insert.<locals>.<listcomp>r  r  r  r   r  z %s%sr  r  z DEFAULT VALUESr   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   C  s    
 z5SQLCompiler.visit_insert.<locals>.<genexpr>.<genexpr>Nr  )r   Zcrud_param_setr   r   r   r   A  s   z+SQLCompiler.visit_insert.<locals>.<genexpr>c                 S  s   g | ]\}}}}|qS r   r   r<  r   r   r   r  K  s   
z VALUES (%s)F)3r  r   r  r  r3  r   r  r   r  r   _get_crud_paramsr  r  r  single_paramsr:  supports_default_valuesr3  Zsupports_empty_insertr4   r  r   Z_has_multi_parametersr%  r1  r  r  format_tabler   r  r$  r  r  ru   r  
_returningr  r  r   rB  r  Zcte_follows_insertr  r&  r   Zdefault_metavalue_tokenr   r  Zall_multi_paramsZ_post_values_clauser  )r   Zinsert_stmtr  rT  r   r  Zcounted_bindparamZhas_visiting_ctecrud_params_structZcrud_params_singler1  r?  r  r"  r  r  Zselect_textr  r  Zpost_values_clauser   r   r   visit_insert  s^    	

 

 




zSQLCompiler.visit_insertc                 C  s   dS )z3Provide a hook for MySQL to add LIMIT to the UPDATENr   )r   update_stmtr   r   r   update_limit_clausez  s    zSQLCompiler.update_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.

        Trp  r  r[  )r   rD  
from_tableextra_fromsrT  r   r   r   update_tables_clause~  s    z SQLCompiler.update_tables_clausec                 K  s   t ddS )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 UPDATENrV  r   rD  rF  rG  
from_hintsrT  r   r   r   update_from_clause  s    	zSQLCompiler.update_from_clausec                   sp  |j || f|}|j}| j }|r@d| _| js4|| _| js@|| _|j}t|}|r~tt	|j
  fdd|D } |}ng }|j
h}| j|||d d}	|jr|	| j||jf|7 }	| j||j
|f|}
tj| |||f|}|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pz|j|d 7 }	|r| j||j
||f|}|r|	d| 7 }	|jr| j|jf|}|r|	d| 7 }	|  |}|r|	d| 7 }	| js
|jr4| js4|	d| j|| jp(|j|d 7 }	| j!r`|sLt"| jnd }| j#|d|	 }	| j$d |	S )NTc                   s   g | ]}| kr|qS r   r   r  Z
main_fromsr   r   r    s     z,SQLCompiler.visit_update.<locals>.<listcomp>r  zUPDATE z SET r   c                 s  s"   | ]\}}}}|d  | V  qdS )=Nr   )r   r  r<  r  r   r   r   r     s   
z+SQLCompiler.visit_update.<locals>.<genexpr>zList[Tuple[Any, str, str, Any]]r  r;   WHERE r  r  )%r  r   r  r  r3  r   _extra_fromsr   r   r(   r   r   r  r  r  rH  r   r=  r>  r  r$  r  r  ru   r   r  rA  r  r  rK  r  r  rE  r  r  r  r  )r   rD  rT  r   r  rG  Zis_multitableZrender_extra_fromsr   r  r"  rB  Zcrud_paramsr#  extra_from_textr  r  r  r   rL  r   visit_update  s     
       
 



zSQLCompiler.visit_updatec                 K  s   t ddS )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 DELETENrV  rI  r   r   r   delete_extra_from_clause  s    z$SQLCompiler.delete_extra_from_clausec                 C  s   |j | dddS )NT)rp  r  r[  )r   delete_stmtrF  rG  r   r   r   delete_table_clause  s    zSQLCompiler.delete_table_clausec                 K  s  |j || f|}|j}| j }|r@d| _| js4|| _| js@|| _|j}|jh|}| j	|||d d}|j
r|| j||j
f|7 }|d7 }| ||j|}tj| |||f| |jr| ||\}	}nd }	|jr| || ||7 }| js|jr"| jr"|d| j|| jp|j|d 7 }|rP| j||j||	f|}
|
rP|d|
 7 }|jrz| j|jf|}|rz|d| 7 }| js|jr| js|d| j|| jp|j|d 7 }| jr|st| jnd }| j|d| }| jd	 |S )
NTr  zDELETE FROM r  r;  rN  r  r  )r  r   r  r  r3  r   rO  r   r   r  r  r  rT  r   r=  r  r$  r  r  r  rA  r  r  rR  r  r  r  r  r  r  )r   rS  rT  r   r  rG  r   r  r"  r#  rP  r  r  r   r   r   visit_delete  s         


zSQLCompiler.visit_deletec                 K  s   d| j | S )NzSAVEPOINT %sr1  format_savepointr   Zsavepoint_stmtrT  r   r   r   visit_savepoint~  s    zSQLCompiler.visit_savepointc                 K  s   d| j | S )NzROLLBACK TO SAVEPOINT %srW  rY  r   r   r   visit_rollback_to_savepoint  s    z'SQLCompiler.visit_rollback_to_savepointc                 K  s   d| j | S )NzRELEASE SAVEPOINT %srW  rY  r   r   r   visit_release_savepoint  s    z#SQLCompiler.visit_release_savepoint)NT)NNTNTF)F)F)F)F)NFFNr   )NTr   N)N)NF)N)FN)Nr   )N)NFNN)F)FFFFF)FFNNNN)FFFNFFNN)F)FN)N)NNNTFFT)F)FFNNNFN)NF)NFF)FFFNTNN)FN)N)r   r   r   r   EXTRACT_MAPr  r$   rk  r  r   rh  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  rj  r  r  Zpostfetch_lastrowidr  r   r  r  r  r  r  rF  r  patternr  rl  rF  r  r	  rE  r   r  Zro_memoized_propertyr  r  r  r  memoized_propertyr  Zmemoized_instancemethodr  
contextlibcontextmanagerr  r  r  r  r  rX  r  rf  r)  rb  r  preload_modulerM  rQ  rd  rk  rl  rq  rr  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.  r,  r>  rB  r#  rI  rJ  rK  rL  rM  r&  r  r  rR  rS  rZ  r]  r`  rc  re  rf  rg  rh  ri  rj  rk  rl  rW  r\  r_  rb  rs  rt  rw  rx  ry  r  r@  r+   rA  r+  r  r  r  r  r  r"   Z_CTEOptsr  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rj  r   r  r  r  r  r  r  r  r  r  r  r  r  r	  r  r  r  r  r   r!  r$  r&  r9  rC  rE  rH  rK  rQ  rR  rT  rV  rZ  r[  r\  r   r   r   r   r  j  s  

+
		




	

"

>M
       
&
 H
P*	
0
0    
1   ;  
5	.  
A	  
'		LT   3					     %!     *E0 L       c'        %
       N4_ 5 !  ,      0 s X
v_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dd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  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  s   dS )Nz<name unknown>r   r  r   r   r   r    s    z$StrSQLCompiler._fallback_column_namezsqlalchemy.engine.urlc                   sT   |j dkrFtjj}|j|j   }||d }t|t	sF|
|S t ||S )Nre   )Zstringify_dialectr$   rK  Z
engine_urlURLr`   get_dialectstatement_compilerr@  re  rB  rG  rU  )r   rR  rS  rT  urlr:  compilerrK  r   r   rU    s    


z,StrSQLCompiler.visit_unsupported_compilationc                 K  s$   d| j |jf|| j |jf|f S )Nz%s[%s])rB  rw   r   r  r   r   r   visit_getitem_binary  s    z#StrSQLCompiler.visit_getitem_binaryc                 K  s   | j ||f|S r   rk  r  r   r   r   visit_json_getitem_op_binary  s    z+StrSQLCompiler.visit_json_getitem_op_binaryc                 K  s   | j ||f|S r   rl  r  r   r   r   !visit_json_path_getitem_op_binary  s    z0StrSQLCompiler.visit_json_path_getitem_op_binaryc                 K  s   d| j | S )Nz<next sequence value: %s>)r1  format_sequence)r   seqrT  r   r   r   r    s    zStrSQLCompiler.visit_sequencer>   r  r   r   r   r  c                  s&    fddt |D }dd| S )Nc              	     s   g | ]}  d |ddi qS )NTFr  r  r   r   r   r    s   z3StrSQLCompiler.returning_clause.<locals>.<listcomp>r  r   )r   r  ru   r  r   r   r   r    s    
zStrSQLCompiler.returning_clausec                   s(   dd< dd  fdd|D  S )NTrp  rU  r   c                 3  s$   | ]}|j fd  iV  qdS r  Nr[  r   r  rJ  rT  r   r   r   r     s   z4StrSQLCompiler.update_from_clause.<locals>.<genexpr>r  rI  r   rs  r   rK    s    z!StrSQLCompiler.update_from_clausec                   s(   dd< dd  fdd|D  S )NTrp  r   c                 3  s$   | ]}|j fd  iV  qdS rq  r[  rr  rs  r   r   r     s   z:StrSQLCompiler.delete_extra_from_clause.<locals>.<genexpr>r  rI  r   rs  r   rR    s    z'StrSQLCompiler.delete_extra_from_clausec                 K  s   dS )NzSELECT 1 WHERE 1!=1r   rp  r   r   r   r,    s    z#StrSQLCompiler.visit_empty_set_exprc                 C  s   d| S )Nz[%s]r   r  r   r   r   r    s    z!StrSQLCompiler.get_from_hint_textc                 K  s   | j |df|S )Nz
 <regexp> r&  r  r   r   r   rw    s    z+StrSQLCompiler.visit_regexp_match_op_binaryc                 K  s   | j |df|S )Nz <not regexp> rt  r  r   r   r   rx    s    z/StrSQLCompiler.visit_not_regexp_match_op_binaryc                 K  s:   |j d }d|jj| f||jj| f||j| f|f S )Nreplacementz<regexp replace>(%s, %s, %s))ro  rw   r\  r   )r   r[   rR  rT  ru  r   r   r   ry    s    
z-StrSQLCompiler.visit_regexp_replace_op_binary)r   r   r   r   r  r$   rd  rU  rk  rm  rn  r  r  rK  rR  r,  r  rw  rx  ry  rm  r   r   rK  r   re    s    		re  c                   @  s  e Zd ZdZer(dgddddddd	d
Zejdd Zejdd Z	dhdddddddZ
dd Zdd Zdd Zdd Zdidd Zdjd!d"Zd#d$ Zd%d& Zd'd( Zdkd)d*Zd+d, Zdld-d.Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Zdmd?d@ZdAdB Z dCdD Z!dEdF Z"dGdH Z#dIdJ Z$dKdL Z%dMdN Z&dOdP Z'dQdR Z(dSdT Z)dUdV Z*dWdX Z+dYdZ Z,d[d\ Z-d]d^ Z.d_d` Z/dadb Z0dcdd Z1dedf Z2dS )nDDLCompilerT.rS   r<   r-  r   r8  r9  c                 C  s   d S r   r   rD  r   r   r   rE    s    zDDLCompiler.__init__c                 C  s   | j j| j d | jdS )Nr.  )r:  rh  r.  r   r   r   r   rX    s
      zDDLCompiler.sql_compilerc                 C  s   | j jS r   )r:  r  r   r   r   r   type_compiler  s    zDDLCompiler.type_compilerNr^  r_  r`  ra  c                 C  s   d S r   r   re  r   r   r   rf    s    zDDLCompiler.construct_paramsc                 K  s   |j }t|jtjr| }| j}||j}t|dkrJ|d d }}n|d |d  }}|	d| |	d| |	d|
|j | j|j| S )Nr   r   r(  r  r   r!   r  )contextr@  targetr!   ZTabler  r1  format_table_seqr  r  r@  rX  r  r   )r   ddlrZ  ry  r1  pathr   Zschr   r   r   	visit_ddl  s    zDDLCompiler.visit_ddlc                 K  s$   d}|j r|d7 }|| j|j S )NzCREATE SCHEMA IF NOT EXISTS )if_not_existsr1  format_schemarR  )r   r`   rT  r  r   r   r   visit_create_schema'  s    zDDLCompiler.visit_create_schemac                 K  s6   d}|j r|d7 }|| j|j7 }|jr2|d7 }|S )NzDROP SCHEMA 
IF EXISTS  CASCADE)	if_existsr1  r  rR  cascader   droprT  r  r   r   r   visit_drop_schema-  s    zDDLCompiler.visit_drop_schemac                 K  sV  |j }| j}d}|jr*|d|jd 7 }|d7 }|jr@|d7 }|||d 7 }| |}|rl||d 7 }|d7 }d}d}|jD ]}	|	j }
zD| j|	|
j	o| d}|d k	r||7 }d	}|d
| 7 }|
j	rd}W q t
jk
r } z$t
d|j|
j|jd f |W 5 d }~X Y qX q| 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

)rR  r1  r  ru   r  r@  create_table_suffixr  rB  ra  r4   r  descriptionr   argscreate_table_constraintsZinclude_foreign_key_constraintspost_create_table)r   r`   rT  r   r1  r  r  r  r  Zcreate_columnr^   	processedZceconstr   r   r   visit_create_table6  sX    

 
zDDLCompiler.visit_create_tableFc                   sL   |j }|jrd S  j||d}d fdd|jD }|rH|d| 7 }|S )Nr  r  c                 3  s   | ]}  |V  qd S r   )rB  r   r_   r   r   r   r   q  s    z2DDLCompiler.visit_create_column.<locals>.<genexpr>)rR  systemget_column_specificationru   constraints)r   r`   r  rT  r^   r  r  r   r   r   visit_create_columnj  s    zDDLCompiler.visit_create_columnc                   st   g }j r|j  j}|d k	r0|| nt  | fddjD  ddd fdd|D D S )Nc                   s"   g | ]}|j k	r| kr|qS r   )ra  r  )	omit_fkcsr   r   r   r    s   
 z8DDLCompiler.create_table_constraints.<locals>.<listcomp>z, 
	c                 s  s   | ]}|d k	r|V  qd S r   r   )r   pr   r   r   r     s   	z7DDLCompiler.create_table_constraints.<locals>.<genexpr>c                 3  s6   | ].}|  r jjr$t|d ds |V  qdS )Z	use_alterFN)Z_should_create_for_compilerr:  Zsupports_alterr  rB  r  r   r   r   r     s
   
)ra  r  Zforeign_key_constraintsr  r   rE  Z_sorted_constraintsru   )r   r   r  rT  r  Zall_fkcsr   )r  r   r   r   r  y  s"    

z$DDLCompiler.create_table_constraintsc                 K  s$   d}|j r|d7 }|| j|j S )Nz
DROP TABLE r  )r  r1  r@  rR  r  r   r   r   visit_drop_table  s    zDDLCompiler.visit_drop_tablec                 K  s   d| j |j S )Nz
DROP VIEW r1  r@  rR  r   r  rT  r   r   r   visit_drop_view  s    zDDLCompiler.visit_drop_viewc                 C  s   |j d krtd|j d S )Nz,Index '%s' is not associated with any table.)r   r4   r  r   r   r  r   r   r   _verify_index_table  s    
zDDLCompiler._verify_index_tablec              	     s   |j } |  j}d}|jr(|d7 }|jd kr<td|d7 }|jrR|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r  r{  N)rX  rB  )r   r<  r   r   r   r     s     z1DDLCompiler.visit_create_index.<locals>.<genexpr>)rR  r  r1  r   r   r4   r  r  _prepared_index_namer@  r   ru   Zexpressions)r   r`   r  Zinclude_table_schemarT  r  r1  r  r   r   r   visit_create_index  s0    

 zDDLCompiler.visit_create_indexc                 K  s>   |j }|jd k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  )rR  r   r4   r  r  r  )r   r  rT  r  r  r   r   r   visit_drop_index  s    
zDDLCompiler.visit_drop_indexc                 C  sX   |j d k	r| j|j }nd }|r4|r4| j|}nd }| j|}|rT|d | }|S Nr  )r   r1  r  r  format_index)r   r  r  r  Zschema_nameZ
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)r1  r@  rR  r   rB  r   r`   rT  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)r1  r@  rR  rX  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  )r1  format_columnrR  rX  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  )r1  r  rR  r  r   r   r   visit_drop_column_comment  s     z%DDLCompiler.visit_drop_column_commentc                 K  s   t | t|d S r   rO  r  r   r   r   visit_set_constraint_comment  s    z(DDLCompiler.visit_set_constraint_commentc                 K  s   t | t|d S r   rO  r  r   r   r   visit_drop_constraint_comment  s    z)DDLCompiler.visit_drop_constraint_commentc                 C  s   g }|j d k	r|d|j   |jd k	r8|d|j  |jd k	rR|d|j  |jd k	rl|d|j  |jd k	r|d |jd k	r|d |jd k	r|d|j  |jd k	r||jrdnd	 |j	d k	r||j	rd
nd d
|S )NzINCREMENT BY %dzSTART WITH %dzMINVALUE %dzMAXVALUE %dzNO MINVALUEzNO MAXVALUEzCACHE %dr  zNO ORDERZCYCLEzNO CYCLEr  )	incrementr  r  ZminvalueZmaxvalueZ
nominvalueZ
nomaxvaluecacher   cycleru   )r   Zidentity_optionsr  r   r   r   get_identity_options  s(    










z DDLCompiler.get_identity_optionsc                 K  sP   d}|j r|d7 }|| j|j7 }|r0||7 }| |j}|rL|d| 7 }|S )NzCREATE SEQUENCE r  r  )r  r1  ro  rR  r  )r   r`   r	  rT  r  optionsr   r   r   visit_create_sequence)  s    z!DDLCompiler.visit_create_sequencec                 K  s$   d}|j r|d7 }|| j|j S )NzDROP SEQUENCE r  )r  r1  ro  rR  r  r   r   r   visit_drop_sequence6  s    zDDLCompiler.visit_drop_sequencec                 K  sj   |j }|jd k	r| j|}nd }|d kr:td|j  d| j|j j|jrTdnd||j	rbdn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  )
rR  r   r1  format_constraintr4   r  r@  r   r  r  )r   r  rT  r_   formatted_namer   r   r   visit_drop_constraint<  s     
z!DDLCompiler.visit_drop_constraintc                 K  s   | j |d | jjj|j|d }| |}|d k	rB|d| 7 }|jd k	r`|d| |j 7 }|jd k	r| jj	r|d| |j 7 }|j
s|jr| jj	s|d7 }|S )Nr  )r  z	 DEFAULT z	 NOT NULL)r1  r  r:  r  rB  r   get_column_default_stringZcomputedidentityZsupports_identity_columnsZnullable)r   r^   rZ  Zcolspecre   r   r   r   r  O  s4    
 

z$DDLCompiler.get_column_specificationc                 C  s   dS Nr(  r   r   r   r   r   r   r  j  s    zDDLCompiler.create_table_suffixc                 C  s   dS r  r   r  r   r   r   r  m  s    zDDLCompiler.post_create_tablec                 C  s$   t |jtjr| |jjS d S d S r   )r@  Zserver_defaultr!   ZDefaultClauserender_default_stringargr  r   r   r   r  p  s    z%DDLCompiler.get_column_default_stringc                 C  s.   t |tr| j|tjS | jj|ddS d S )NT)r{  )r@  r   rX  r+  r#   rQ  rB  )r   re   r   r   r   r  v  s    
 z!DDLCompiler.render_default_stringc                 K  s   |j r| |S | |S d S r   )Zis_column_levelvisit_column_check_constraintvisit_check_constraint)r   r_   rT  r   r   r   &visit_table_or_column_check_constraint~  s    
z2DDLCompiler.visit_table_or_column_check_constraintc                 K  s\   d}|j d k	r.| j|}|d k	r.|d| 7 }|d| jj|jddd 7 }|| |7 }|S Nr(  CONSTRAINT %s z
CHECK (%s)FTr  r   r1  r  rX  rB  r  define_constraint_deferrabilityr   r_   rT  r  r  r   r   r   r    s    

  
z"DDLCompiler.visit_check_constraintc                 K  s\   d}|j d k	r.| j|}|d k	r.|d| 7 }|d| jj|jddd 7 }|| |7 }|S r  r  r  r   r   r   r    s    

  
z)DDLCompiler.visit_column_check_constraintc                   s   t |dkrdS d}|jd k	r> j|}|d k	r>|d| 7 }|d7 }|dd fdd|jrd|jn|jD  7 }| |7 }|S )	Nr   r(  r  zPRIMARY KEY r  r   c                 3  s   | ]} j |jV  qd S r   r1  r  r   r  r   r   r   r     s   z;DDLCompiler.visit_primary_key_constraint.<locals>.<genexpr>)	r  r   r1  r  ru   Z_implicit_generatedZcolumns_autoinc_firstr  r  r  r   r   r   visit_primary_key_constraint  s    
z(DDLCompiler.visit_primary_key_constraintc              	     s   | j  d}|jd k	r4| j |}|d k	r4|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  s   | ]}  |jjV  qd S r   )r  parentr   r  r1  r   r   r     s    z;DDLCompiler.visit_foreign_key_constraint.<locals>.<genexpr>c                 3  s   | ]}  |jjV  qd S r   )r  r^   r   r  r  r   r   r     s    )r1  r   r  r  r   r^   r   ru   define_constraint_remote_tabledefine_constraint_matchdefine_constraint_cascadesr  )r   r_   rT  r  r  Zremote_tabler   r  r   visit_foreign_key_constraint  s0    
  z(DDLCompiler.visit_foreign_key_constraintc                 C  s
   | |S )z=Format the remote table clause of a CREATE CONSTRAINT clause.)r@  )r   r_   r   r1  r   r   r   r    s    z*DDLCompiler.define_constraint_remote_tablec                   sp   t |dkrdS d}|jd k	r> j|}|d k	r>|d| 7 }|dd fdd|D  7 }| |7 }|S )Nr   r(  r  zUNIQUE (%s)r   c                 3  s   | ]} j |jV  qd S r   r  r  r   r   r   r     s     z6DDLCompiler.visit_unique_constraint.<locals>.<genexpr>)r  r   r1  r  ru   r  r  r   r   r   visit_unique_constraint  s    
z#DDLCompiler.visit_unique_constraintc                 C  sL   d}|j d k	r&|d| j|j t 7 }|jd k	rH|d| j|jt 7 }|S )Nr(  z ON DELETE %sz ON UPDATE %s)Zondeleter1  validate_sql_phraseFK_ON_DELETEZonupdateFK_ON_UPDATEr   r_   r  r   r   r   r    s    

 

 z&DDLCompiler.define_constraint_cascadesc                 C  sL   d}|j d k	r&|j r|d7 }n|d7 }|jd k	rH|d| j|jt 7 }|S )Nr(  z DEFERRABLEz NOT DEFERRABLEz INITIALLY %s)rf   rr   r1  r  FK_INITIALLYr  r   r   r   r    s    



 z+DDLCompiler.define_constraint_deferrabilityc                 C  s    d}|j d k	r|d|j  7 }|S )Nr(  z	 MATCH %s)r  r  r   r   r   r    s    
z#DDLCompiler.define_constraint_matchc                 K  sB   d| j j|jddd }|jdkr,|d7 }n|jdkr>|d7 }|S )NzGENERATED ALWAYS AS (%s)FTr  z STOREDz VIRTUAL)rX  rB  r  Z	persisted)r   	generatedrT  r  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 IDENTITYZALWAYSz
BY DEFAULTr:  )alwaysr  )r   r  rT  r  r  r   r   r   visit_identity_column  s    
z!DDLCompiler.visit_identity_column)...)NNT)F)N)FT)F)N)3r   r   r   ri  r   rE  r$   ra  rX  rx  rf  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   rv    sr      


   	4
 
&   
 



rv  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S )_GenericTypeCompilerc                 K  s   dS )NFLOATr   rp  r   r   r   visit_FLOAT  s    zGenericTypeCompiler.visit_FLOATc                 K  s   dS )NZDOUBLEr   rp  r   r   r   visit_DOUBLE  s    z GenericTypeCompiler.visit_DOUBLEc                 K  s   dS )NzDOUBLE PRECISIONr   rp  r   r   r   visit_DOUBLE_PRECISION  s    z*GenericTypeCompiler.visit_DOUBLE_PRECISIONc                 K  s   dS )NZREALr   rp  r   r   r   
visit_REAL  s    zGenericTypeCompiler.visit_REALc                 K  s<   |j d krdS |jd kr&dd|j i S d|j |jd S d S )NZNUMERICzNUMERIC(%(precision)s)	precisionz!NUMERIC(%(precision)s, %(scale)s)r  Zscaler  rp  r   r   r   visit_NUMERIC  s    

z!GenericTypeCompiler.visit_NUMERICc                 K  s<   |j d krdS |jd kr&dd|j i S d|j |jd S d S )NZDECIMALzDECIMAL(%(precision)s)r  z!DECIMAL(%(precision)s, %(scale)s)r  r  rp  r   r   r   visit_DECIMAL&  s    

z!GenericTypeCompiler.visit_DECIMALc                 K  s   dS )NZINTEGERr   rp  r   r   r   visit_INTEGER1  s    z!GenericTypeCompiler.visit_INTEGERc                 K  s   dS )NZSMALLINTr   rp  r   r   r   visit_SMALLINT4  s    z"GenericTypeCompiler.visit_SMALLINTc                 K  s   dS )NZBIGINTr   rp  r   r   r   visit_BIGINT7  s    z GenericTypeCompiler.visit_BIGINTc                 K  s   dS )N	TIMESTAMPr   rp  r   r   r   visit_TIMESTAMP:  s    z#GenericTypeCompiler.visit_TIMESTAMPc                 K  s   dS )NZDATETIMEr   rp  r   r   r   visit_DATETIME=  s    z"GenericTypeCompiler.visit_DATETIMEc                 K  s   dS )NZDATEr   rp  r   r   r   
visit_DATE@  s    zGenericTypeCompiler.visit_DATEc                 K  s   dS )NZTIMEr   rp  r   r   r   
visit_TIMEC  s    zGenericTypeCompiler.visit_TIMEc                 K  s   dS )NZCLOBr   rp  r   r   r   
visit_CLOBF  s    zGenericTypeCompiler.visit_CLOBc                 K  s   dS )NZNCLOBr   rp  r   r   r   visit_NCLOBI  s    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")lengthr  )r   r   r   length_overrider  r   r   r   _render_string_typeL  s    z'GenericTypeCompiler._render_string_typec                 K  s   |  |dS )NCHARr  rp  r   r   r   
visit_CHARW  s    zGenericTypeCompiler.visit_CHARc                 K  s   |  |dS )NZNCHARr  rp  r   r   r   visit_NCHARZ  s    zGenericTypeCompiler.visit_NCHARc                 K  s   |  |dS )NZVARCHARr  rp  r   r   r   visit_VARCHAR]  s    z!GenericTypeCompiler.visit_VARCHARc                 K  s   |  |dS )NZNVARCHARr  rp  r   r   r   visit_NVARCHAR`  s    z"GenericTypeCompiler.visit_NVARCHARc                 K  s   |  |dS )NZTEXTr  rp  r   r   r   
visit_TEXTc  s    zGenericTypeCompiler.visit_TEXTc                 K  s   dS )NZBLOBr   rp  r   r   r   
visit_BLOBf  s    zGenericTypeCompiler.visit_BLOBc                 K  s   d|j rd|j  pd S )NBINARYr  r(  r  rp  r   r   r   visit_BINARYi  s    z GenericTypeCompiler.visit_BINARYc                 K  s   d|j rd|j  pd S )NZ	VARBINARYr  r(  r  rp  r   r   r   visit_VARBINARYl  s    z#GenericTypeCompiler.visit_VARBINARYc                 K  s   dS )NZBOOLEANr   rp  r   r   r   visit_BOOLEANo  s    z!GenericTypeCompiler.visit_BOOLEANc                 K  s   | j |dddS )Nr      )r  r  rp  r   r   r   
visit_uuidr  s    zGenericTypeCompiler.visit_uuidc                 K  s   | j |f|S r   )r  rp  r   r   r   visit_large_binaryu  s    z&GenericTypeCompiler.visit_large_binaryc                 K  s   | j |f|S r   )r  rp  r   r   r   visit_booleanx  s    z!GenericTypeCompiler.visit_booleanc                 K  s   | j |f|S r   )r  rp  r   r   r   
visit_time{  s    zGenericTypeCompiler.visit_timec                 K  s   | j |f|S r   )r  rp  r   r   r   visit_datetime~  s    z"GenericTypeCompiler.visit_datetimec                 K  s   | j |f|S r   )r  rp  r   r   r   
visit_date  s    zGenericTypeCompiler.visit_datec                 K  s   | j |f|S r   )r  rp  r   r   r   visit_big_integer  s    z%GenericTypeCompiler.visit_big_integerc                 K  s   | j |f|S r   )r  rp  r   r   r   visit_small_integer  s    z'GenericTypeCompiler.visit_small_integerc                 K  s   | j |f|S r   )r  rp  r   r   r   visit_integer  s    z!GenericTypeCompiler.visit_integerc                 K  s   | j |f|S r   )r  rp  r   r   r   
visit_real  s    zGenericTypeCompiler.visit_realc                 K  s   | j |f|S r   )r  rp  r   r   r   visit_float  s    zGenericTypeCompiler.visit_floatc                 K  s   | j |f|S r   )r  rp  r   r   r   visit_double  s    z GenericTypeCompiler.visit_doublec                 K  s   | j |f|S r   )r  rp  r   r   r   visit_numeric  s    z!GenericTypeCompiler.visit_numericc                 K  s   | j |f|S r   r  rp  r   r   r   visit_string  s    z GenericTypeCompiler.visit_stringc                 K  s   | j |f|S r   r  rp  r   r   r   visit_unicode  s    z!GenericTypeCompiler.visit_unicodec                 K  s   | j |f|S r   r  rp  r   r   r   
visit_text  s    zGenericTypeCompiler.visit_textc                 K  s   | j |f|S r   r  rp  r   r   r   visit_unicode_text  s    z&GenericTypeCompiler.visit_unicode_textc                 K  s   | j |f|S r   r  rp  r   r   r   
visit_enum  s    zGenericTypeCompiler.visit_enumc                 K  s   t d| d S )NzKCan't generate DDL for %r; did you forget to specify a type on this Column?r  rp  r   r   r   r    s
    zGenericTypeCompiler.visit_nullc                 K  s   | j || jf|S r   )rB  Ztype_enginer:  rp  r   r   r   visit_type_decorator  s    z(GenericTypeCompiler.visit_type_decoratorc                 K  s   |j f |S r   )get_col_specrp  r   r   r   visit_user_defined  s    z&GenericTypeCompiler.visit_user_defined)N)1r   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  r  r  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k
r,   | j|f| Y S X || f|S d S r   )r\  AttributeError_visit_unknown)r   r   rT  r\  r   r   r   rB    s
    
zStrSQLTypeCompiler.processc                 C  s   | dr| jS t|d S )NZvisit_)r  r"  r!  )r   r   r   r   r   __getattr__  s    
zStrSQLTypeCompiler.__getattr__c                 K  s(   |j j|j j kr|j jS t|S d S r   )rL  r   upperreprrp  r   r   r   r"    s    z!StrSQLTypeCompiler._visit_unknownc                 K  s   dS r  r   rp  r   r   r   r    s    zStrSQLTypeCompiler.visit_nullc                 K  s6   z
|j }W n tk
r&   t| Y S X |f |S d S r   )r  r!  r%  )r   r   rT  r  r   r   r   r    s
    
z%StrSQLTypeCompiler.visit_user_definedN)r   r   r   rB  r#  r"  r  r  r   r   r   r   r     s
   r   c                   @  s   e Zd ZdddddZdS )_SchemaForObjectCallabler   r   )rY  r   c                 C  s   d S r   r   )r   rY  r   r   r   r     s    z!_SchemaForObjectCallable.__call__Nr   r   r   r   r   r&    s   r&  c                   @  s   e Zd ZdddddZdS )rN  r  r   )rv  r   c                 C  s   d S r   r   )r   rv  r   r   r   r     s    z _BindNameForColProtocol.__call__Nr   r   r   r   r   rN    s   rN  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	< dQddZdd Zdd ZdddddZdddddZdd ZdddddZdddddZd d! ZdRdd"dd#d$d%ZdSdd"dd&d'd(Zd)d* ZdTd+d,ZdUd-d.dd/d0d1ZdVd2d.dd3d4d5ZdWd6d7Zed8dXd9d:Z dYd;d<Z!dZd=d>Z"d?d@ Z#dAdB Z$d[dCdDZ%dEdF Z&d\dGdHZ'd]dIdJZ(d^dKdLZ)ej*dMdN Z+dOdP Z,dS )_r0  z@Handle quoting and case-folding of identifiers based on options.r   initial_quotefinal_quotezMutableMapping[str, str]_stringsr!   r&  r  r!  NTFc                 C  sN   || _ || _|p| j| _|| _| jd | _|| _|| _i | _| j jdk| _	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.
        r3   )r   r   N)
r:  r'  r(  escape_quoteescape_to_quoteomit_schemaquote_case_sensitive_collationsr)  r  r  )r   r:  r'  r(  r*  r-  r,  r   r   r   rE    s    zIdentifierPreparer.__init__c                   s2   | j | j }|j| j  fdd}||_|S )Nc                   sX   | j }| krN| jrN|d k	r:d|ks,d|kr:td| td|pDd ddS | j S d S )Nr  r  zJSquare bracket characters ([]) not supported in schema translate name '%s'z__[SCHEMA_%s]_noneFr  )r!   Z_use_schema_mapr4   r  r-   )rY  r   rw  r   r   symbol_getter$  s    
 z@IdentifierPreparer._with_schema_translate.<locals>.symbol_getter)rL  __new____dict__rD  r  )r   r.  prepr0  r   rw  r   r>     s
    z)IdentifierPreparer._with_schema_translatec                   s4   | d  kr d   d<  fdd}t d||S )Nr.  c                   s6   |  d} | }|s,jj}|s,td|S )Nr3   zLDialect has no default schema name; can't use None as dynamic schema target.)rp   r:  Zdefault_schema_namer4   r  r  )r  r   r  dr   r   r   r  :  s    
z=IdentifierPreparer._render_schema_translates.<locals>.replacez(__\[SCHEMA_([^\]]+)\]))r  r  )r   r   r.  r  r   r4  r   rC  5  s
    z,IdentifierPreparer._render_schema_translates)r  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  r  )r  r*  r+  r  r   r  r   r   r   _escape_identifierI  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.
        )r  r+  r*  r6  r   r   r   _unescape_identifierU  s    z'IdentifierPreparer._unescape_identifierc                 C  s*   |dk	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  r4   r  r`  )r   rR  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.
        )r'  r7  r(  r6  r   r   r   quote_identifierp  s    z#IdentifierPreparer.quote_identifierr   c                 C  s:   |  }|| jkp8|d | jkp8| jt| p8||kS )z5Return True if the given identifier requires quoting.r   )r  reserved_wordsillegal_initial_characterslegal_charactersr  r   )r   r  Zlc_valuer   r   r   r  }  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   r6  r   r   r   r    s    z1IdentifierPreparer._requires_quotes_illegal_charsr   )r!   forcer   c                 C  s    |dk	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  r  )r   r!   r=  r   r   r   r    s    
zIdentifierPreparer.quote_schema)r  r=  r   c                 C  s   |dk	rt jddd t|dd}|dkrp|| jkr>| j| S | |rZ| || j|< n
|| j|< | j| S n|r~| |S |S d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  r  )r$   r  r  r)  r  r9  )r   r  r=  r   r   r   r    s     





zIdentifierPreparer.quotec                 C  s   | j r| |S |S d S r   )r-  r  )r   Zcollation_namer   r   r   r    s    
z#IdentifierPreparer.format_collationc                 C  s>   |  |j}| |}| js:|r:|d k	r:| |d | }|S r  )r  r   r  r,  r  )r   r  r  r   r  r   r   r   ro    s    
z"IdentifierPreparer.format_sequencez
Label[Any]r  )rs  r   r   c                 C  s   |  |p|jS r   )r  r   )r   rs  r   r   r   r   r    s    zIdentifierPreparer.format_labelzOptional[AliasedReturnsRows])r  r   r   c                 C  s.   |d kr |d k	st | |jS | |S d S r   )rA  r  r   )r   r  r   r   r   r   r    s    zIdentifierPreparer.format_aliasc                 C  s"   |p|j }| |r| |}|S r   )r  r  r9  )r   Z	savepointr   r  r   r   r   rX    s    


z#IdentifierPreparer.format_savepointzsqlalchemy.sql.namingc                 C  s^   t jj}|jtkr.|||j}|d kr4d S n|j}|jdkrL| j||dS | j	||dS d S )Nr  )_alembic_quote)
r$   rK  Z
sql_namingr   r)   Z_constraint_name_for_tabler   r}  truncate_and_render_index_name#truncate_and_render_constraint_name)r   r_   r?  Znamingr   r   r   r   r    s$    
 
  z$IdentifierPreparer.format_constraintc                 C  s   | j jp| j j}| |||S r   )r:  Zmax_index_name_lengthr   _truncate_and_render_maxlen_namer   r   r?  max_r   r   r   r@  (  s      z1IdentifierPreparer.truncate_and_render_index_namec                 C  s   | j jp| j j}| |||S r   )r:  Zmax_constraint_name_lengthr  rB  rC  r   r   r   rA  4  s      z6IdentifierPreparer.truncate_and_render_constraint_namec                 C  sb   t |tjr@t||krL|d|d  d t|dd   }n| j| |sT|S | |S d S )Nr      r  )	r@  r   r@   r  r$   Zmd5_hexr:  Zvalidate_identifierr  )r   r   rD  r?  r   r   r   rB  @  s    (z3IdentifierPreparer._truncate_and_render_maxlen_namec                 C  s
   |  |S r   )r  r  r   r   r   r  L  s    zIdentifierPreparer.format_indexc                 C  sF   |dkr|j }| |}| |}| jsB|rB|rB| |d | }|S )z'Prepare a quoted table and schema name.Nr  )r   r  r  r,  r  )r   r   r  r   rc  r  r   r   r   r@  O  s    

zIdentifierPreparer.format_tablec                 C  s
   |  |S )zPrepare a quoted schema name.r/  r  r   r   r   r  ]  s    z IdentifierPreparer.format_schemac                 C  s(   |dk	rt |tjr||}| |S )Prepare a quoted column name.N)r@  r   r@   r  r  )r   r   r  r   r   r   r  b  s    
 
z$IdentifierPreparer.format_label_namec                 C  s   |dkr|j }|dk	r,t|tjr,||}t|ddsh|r\| j|j||dd | | S | |S n"|r| j|j||dd | S |S dS )rG  Nr  F)r  r   r  )	r   r@  r   r@   r  r  r@  r   r  )r   r^   r  r   Z
table_namer  r  r   r   r   r  p  s@    
 
    z IdentifierPreparer.format_columnc                 C  sD   |  |}| js0|r0|r0| || j|ddfS | j|ddfS dS )z(Format table name and schema as a tuple.Fr  N)r  r,  r  r@  )r   r   r  r  r   r   r   r{    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finalr7  )r'  r(  r7  r  r  )r   rH  rI  Zescaped_finalrr   r   r   _r_identifiers  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   )r8  r0  r   r   r   r    s   z;IdentifierPreparer.unformat_identifiers.<locals>.<listcomp>c                 S  s   g | ]\}}|p|qS r   r   )r   ar  r   r   r   r    s     )rK  findall)r   ZidentifiersrJ  r   r   r   unformat_identifiers  s    
z'IdentifierPreparer.unformat_identifiers)r!  Nr!  TF)N)N)T)N)N)N)T)T)T)TN)N)FNNFN)T)-r   r   r   r   RESERVED_WORDSr:  LEGAL_CHARACTERSr<  ILLEGAL_INITIAL_CHARACTERSr;  r   rR  
attrgetterr  rE  r>  rC  r7  r8  r  r9  r  r  r  r  r  ro  r  r  rX  r$   rd  r  r@  rA  rB  r  r@  r  r  r  r{  ra  rK  rN  r   r   r   r   r0    sf   
     
$	
'4
  	
	


 
     
-

r0  )r   
__future__r   r  collections.abcabcr7  rb  enumr   r  rR  r  timer   rj  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.   r/   Ztype_apir0   Zvisitorsr1   r2   r4   Zutil.typingr5   r6   r7   
annotationr8   r9   r:   r5  r;   r|  r<   Zdmlr=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   Zengine.cursorrO   Zengine.interfacesrP   rQ   rR   rS   rT   r   Z_FromHintsTyperO  r  IrP  ZLEGAL_CHARACTERS_PLUS_SPACEr5  r   rQ  r  r  r  UNICODEr  r  r  r  r  or_r  mulr  modnegltlenegtgeeqZis_distinct_fromZis_not_distinct_fromZ	concat_opr$  Znot_match_opr+  r*  r  r"  r  r  r  r]   existsZdistinct_opinvZany_opZall_opZdesc_opZasc_opZnulls_first_opZnulls_last_opZbitwise_xor_opZbitwise_or_opZbitwise_and_opZbitwise_not_opZbitwise_lshift_opZbitwise_rshift_opr  r   ra   rb   rc   rd   rz   r{   r   r   r   r   ZcubeZrollupZgrouping_setsr   r   r]  Z_CompoundSelectKeywordr   Z	UNION_ALLr   Z
EXCEPT_ALLr   ZINTERSECT_ALLr^  r   r   r   r   r   r   r   r   r   r   r  r  r   r	  r
  r  r  
namedtupler  r&  ZEnsureKWArgrn  ZBinaryElementRoleZCompilerColumnElementrr  r  r  re  rv  r  r   r&  rN  r0  r   r   r   r   <module>	   sT  a                                           1                    
=	(A O  &                                      Hg     )!