Current File : //opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/sql/__pycache__/compiler.cpython-37.pyc
B
��4]�� � _ @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
Z
dd
lm Z dd lm Z d d
lm
Z
d dlmZ edddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdlg^�Ze�dmej�Ze�dnej�Zdodp� eddq�D ��drg�Ze�dsej�Ze�dsej�Ze�dtej�Ze�duej�Z e�dvej�Z dwdxdydzd{d|�Z ej d}ej d~ej!dej"d�ej#d�ej$d�ej%d�ej&d�ej'd�ej(d�ej)d�ej*d�ej+d�ej,d�ej-d�ej.d�ej/d�ej0d�ej1d�ej2d�ej3d�ej4d�ej5d�ej6d�ej7d�ej8d�ej9d�ej:d�ej;d�ej<d�ej=d�ej>d�ej?d�ej@d�ejAd�ejBd�ejCd�i%ZDejEd�ejFd�ejGd�ejHd�ejId�ejJd�ejKd�ejLd�ejMd�ejNd�ejOd�ejPd�ejQd�ejRd�iZSd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d��ZTe
jUjVd�e
jUjWd�e
jUjXd�e
jUjYd�e
jUjZd�e
jUj[d�iZ\G d�dƄ d�e]�Z^G d�dȄ d�e�_ej`e]��ZaG d�dʄ d�e jb�ZcG d�d̄ d�e^�ZdG d�d΄ d�ed�ZeG d�dЄ d�e^�ZfG d�d҄ d�ea�ZgG d�dԄ d�eg�ZhG d�dք d�e]�ZidS )�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`.
� N� )�crud)�elements)� functions)� operators)�schema)�
selectable)�sqltypes)�visitors� )�exc)�util�allZanalyseZanalyze�and�anyZarray�asZascZ
asymmetricZ
authorizationZbetween�binaryZbothZcase�castZcheck�collate�column�
constraint�createZcross� current_dateZ current_role� current_time�current_timestamp� current_user�default�
deferrableZdescZdistinctZdo�else�end�except�false�forZforeignZfreeze�from�fullZgrant�groupZhavingZilike�in� initially�innerZ intersectZinto�isZisnull�joinZleading�leftZlike�limit� localtime�localtimestampZnatural�new�notZnotnullZnullZoff�offset�oldZonZonly�or�orderZouterZoverlapsZplacingZprimaryZ
references�right�select� session_user�setZsimilarZsome� symmetric�tableZthen�toZtrailing�true�union�unique�userZusing�verboseZwhen�wherez
^[A-Z0-9_$]+$z^[A-Z0-9_ $]+$c C s h | ] }t |��qS � )�str)�.0�xrB rB �J/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/sql/compiler.py� <setcomp>� s rG �
�$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 :%(name)s)�pyformatZqmark�format�numericZnamedz AND z OR z + z * 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, z FROM z AS z IS z IS NOT z COLLATE zEXISTS z DISTINCT zNOT zANY zALL z DESCz ASCz NULLS FIRSTz NULLS LAST�coalesceZ CURRENT_DATEZ CURRENT_TIMEZCURRENT_TIMESTAMPZ CURRENT_USERZ LOCALTIMEZLOCALTIMESTAMP�random�sysdateZ SESSION_USERZUSERZCUBEZROLLUPz
GROUPING SETS�month�day�year�second�hour�doy�minute�quarter�dow�week�epoch� milliseconds� microseconds�
timezone_hour�timezone_minute)rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` �UNIONz UNION ALL�EXCEPTz
EXCEPT ALL� INTERSECTz
INTERSECT ALLc @ s� e Zd ZdZdZe�� Zdde�� fdd�Ze� dd�dd� �Z
d d
� Z e d d � �Z
d
d� Zdd� Zddd�Ze dd� �Zdd� Zdd� ZdS )�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.
Nc C sb || _ || _| j j| _|r(| j�|�| _|dk r^|| _|j| _| jrL|j| _ | j
| jf|�| _ dS )aq Construct a new :class:`.Compiled` object.
:param dialect: :class:`.Dialect` to compile against.
:param statement: :class:`.ClauseElement` to be compiled.
:param bind: Optional Engine or Connection to compile this
statement against.
: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) �dialect�bindZidentifier_preparer�preparer�_with_schema_translate� statementZsupports_execution� can_execute�_execution_options�execution_options�process�string)�selfre ri rf �schema_translate_mapZcompile_kwargsrB rB rF �__init__ s
zCompiled.__init__z0.7z�The :meth:`.Compiled.compile` method is deprecated and will be removed in a future release. The :class:`.Compiled` object now runs its compilation within the constructor, and this method does nothing.c C s dS )zDProduce the internal string representation of this element.
NrB )ro rB rB rF �compileA s
zCompiled.compilec C s$ | j r|�| ||�S t�| j��d S )N)rj �_execute_compiledr ZObjectNotExecutableErrorri )ro Z
connection� multiparams�paramsrB rB rF �_execute_on_connectionM s z Compiled._execute_on_connectionc C s
t � �dS )z�Return a Compiled that is capable of processing SQL expressions.
If this compiler is one, it would likely just return 'self'.
N)�NotImplementedError)ro rB rB rF � sql_compilerS s zCompiled.sql_compilerc K s |j | f|�S )N)�_compiler_dispatch)ro �obj�kwargsrB rB rF rm ] s zCompiled.processc C s
| j pdS )z3Return the string text of the generated SQL or DDL.� )rn )ro rB rB rF �__str__` s zCompiled.__str__c C s
t � �dS )z�Return 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.
N)rw )ro ru rB rB rF �construct_paramse s zCompiled.construct_paramsc C s | � � S )z0Return the bind params for this compiled object.)r~ )ro rB rB rF ru o s zCompiled.paramsc O s* | j }|dkr tjddd��|�| ||�S )z Execute this compiled object.Nz>This Compiled object is not bound to any Engine or Connection.Z2afi)�code)rf r ZUnboundExecutionErrorrs )ro rt ru �erB rB rF �executet s zCompiled.executec O s | j ||��� S )zJExecute this compiled object and return the result's
scalar value.)r� �scalar)ro rt ru rB rB rF r� � s zCompiled.scalar)N)�__name__�
__module__� __qualname__�__doc__Z_cached_metadatar
�
immutabledictrl rq Z
deprecatedrr rv �propertyrx rm r} r~ ru r� r� rB rB rB rF rd � s"
)
rd c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� TypeCompilerz2Produces DDL specification for TypeEngine objects.z visit_\w+c C s
|| _ d S )N)re )ro re rB rB rF rq � s zTypeCompiler.__init__c K s |j | f|�S )N)ry )ro �type_�kwrB rB rF rm � s zTypeCompiler.processN)r� r� r� r� Z ensure_kwargrq rm rB rB rB rF r� � s r� c @ sB e Zd ZdZdZdZddd�Zedd� �Zed d
� �Z d d � Z
d
S )�
_CompileLabelz;lightweight label object which acts as an expression.Label.�label)�element�namerB c C s || _ || _|f| | _d S )N)r� r� �
_alt_names)ro �colr� � alt_namesrB rB rF rq � s z_CompileLabel.__init__c C s | j jS )N)r� � proxy_set)ro rB rB rF r� � s z_CompileLabel.proxy_setc C s | j jS )N)r� �type)ro rB rB rF r� � s z_CompileLabel.typec K s | S )NrB )ro r� rB rB rF �
self_group� s z_CompileLabel.self_groupN)rB ) r� r� r� r� �__visit_name__� __slots__rq r� r� r� r� rB rB rB rF r� � s
r� c @ s� e Zd ZdZeZeZd Z Z Z
dZ dZ dZ
dZdZdZdZdZdZd ZZ�ddd�Zedd � �Zejd
d � �Ze j d d
� �Z dd� Z ej dd� �Z!dd� Z"edd� �Z#�ddd�Z$edd� �Z%e�&d�dd � �Z'd d � Z(�dd d �Z)�dd!d"�Z*�dd#d$�Z+�d d%d&�Z,d'd(� Z-�d
d)d*�Z.d+d,� Z/d-d.� Z0d/d0� Z1d1d2� Z2d3d4� Z3d5d6� Z4d7d8� Z5�d d9d:�Z6d;d<� Z7d=d>� Z8d?d@� Z9dAdB� Z:dCdD� Z;dEdF� Z<dGdH� Z=dIdJ� Z>dKdL� Z?dMdN� Z@dOdP� ZAdQdR� ZB�d dSdT�ZCdUdV� ZDdWdX� ZEdYdZ� ZF�d
d\d]�ZGd^d_� ZHd`da� ZIdbdc� ZJddde� ZKdfdg� ZLdhdi� ZMdjdk� ZNdldm� ZOdndo� ZP�ddpdq�ZQdrds� ZRdtdu� ZSdvdw� ZTdxdy� ZUdzd{� ZV�dd|d}�ZWd~d� ZXd�d�� ZYej d�d�� �ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� Zbd�d�� Zcd�d�� Zdd�d�� Zed�d�� Zf�dd�d��Zgd�d�� Zhd�d�� Zid�d�� Zjd�d�� Zkd�d�� Zld�d�� Zm�dd�d��Zn�dd�d��Zo�dd�d��Zpd�d�� Zq�dd�d��Zrd�d�� Zsd�d�� Zt�dd�d��Zud�d�� Zvd�d�� Zwd�d�� Zxd�d�� Zyd�dÄ Zzd�dń Z{d�dDŽ Z|e�}d�e~� fd�e~� fg�Z�dd�d˄Z��dd�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��dd�d�Z��dd�d�Z�d�d� Z��dd�d�Z�d�d� Z�d�d� Z�d�d� Z��dd�d��Z�ej d�d�� �Z�d�d�� Z�d�d�� Z��d d�d��Z�d�d�� Z��d �d� Z��d�d� Z�dS ( � SQLCompilerzsDefault implementation of :class:`.Compiled`.
Compiles :class:`.ClauseElement` objects into SQL strings.
FNTrB c K s� || _ |pt|dd�| _i | _t�� | _g | _g | _|j | _ | j rTg | _
|j dk| _ t
|j | _d| _|jpp|j| _t�| j�| _i | _tj| ||f|� | js�| js�| jr�|jr�|j| _ | j r�| j r�| � � dS )a Construct a new :class:`.SQLCompiler` object.
:param dialect: :class:`.Dialect` to be used
:param statement: :class:`.ClauseElement` to be compiled
:param column_keys: a list of column names to be compiled into an
INSERT or UPDATE statement.
:param inline: whether to generate INSERT statements as "inline", e.g.
not formatted to return any generated defaults
:param kwargs: additional keyword arguments to be consumed by the
superclass.
�inlineFrM N) � column_keys�getattrr� �bindsr
Z column_dict�
bind_names�stack�_result_columns�
positional� positiontup�
paramstyle�_numeric_binds�BIND_TEMPLATES� bindtemplate�ctes� label_length�max_identifier_lengthZ PopulateDict�
_process_anon�anon_map�truncated_namesrd rq �isinsert�isupdate�isdelete�
_returning� returning�_apply_numbered_params)ro re ri r� r� r{ rB rB rF rq � s*
zSQLCompiler.__init__c C s t | j| j �S )N)�list�insert_prefetch�update_prefetch)ro rB rB rF �prefetch< s zSQLCompiler.prefetchc C s& t �� | _i | _d| _| jr"i | _dS )z�Initialize collections related to CTEs only if
a CTE is located, to save on the overhead of
these collections otherwise.
FN)r
� OrderedDictr� � ctes_by_name�ctes_recursiver� �cte_positional)ro rB rB rF �_init_cte_state@ s
zSQLCompiler._init_cte_statec c sp | j | j }}g d | _ | _z0| jr8| jd }d|d< nd}| j | jfV W d|r\|�d� || | _ | _X dS )z;special API to support the use case of 'nested result sets'F���T�need_result_map_for_nestedN)r� �_ordered_columnsr� �pop)ro Zresult_columnsZordered_columns�entryrB rB rF �_nested_resultN s
zSQLCompiler._nested_resultc s( t �d�� t�d� fdd�| j�| _d S )Nr z
\[_POSITION\]c s t t�� ��S )N)rC r
�next)�m)�poscountrB rF �<lambda>i � z4SQLCompiler._apply_numbered_params.<locals>.<lambda>)� itertools�count�re�subrn )ro rB )r� rF r� f s
z"SQLCompiler._apply_numbered_paramsc s" t dd� � fdd�� jD �D ��S )Nc s s" | ]\}}|d k r||fV qd S )NrB )rD �key�valuerB rB rF � <genexpr>o s z/SQLCompiler._bind_processors.<locals>.<genexpr>c 3 s& | ] }� j | |j�� j�fV qd S )N)r� r� Z_cached_bind_processorre )rD � bindparam)ro rB rF r� q s )�dictr� )ro rB )ro rF �_bind_processorsl s
z SQLCompiler._bind_processorsc C s t | j�dkS )Nr )�lenr� )ro rB rB rF � is_subqueryz s zSQLCompiler.is_subqueryc C s | S )NrB )ro rB rB rF rx } s zSQLCompiler.sql_compilerc C s2 |r�i }x�| j D ]�}| j | }|j|kr8||j ||< q||krN|| ||< q|r�|jr�|rvtjd|j|f dd��q�tjd|j dd��q|jr�|j||< q|j||< qW |S i }xt| j D ]j}|r�|jr�|r�tjd|j|f dd��ntjd|j dd��|j�r|j|| j | < q�|j|| j | < q�W |S dS )z5return a dictionary of bind parameter keys and valuesz@A value is required for bind parameter %r, in parameter group %dZcd3x)r z)A value is required for bind parameter %rN)r� r� Zrequiredr ZInvalidRequestError�callable�effective_valuer� )ro ru Z
_group_number�_checkZpdr� r� rB rB rF r~ � sL
z SQLCompiler.construct_paramsc C s | j dd�S )zoReturn the bind param dictionary embedded into this
compiled object, for those values that are present.F)r� )r~ )ro rB rB rF ru � s zSQLCompiler.paramszsqlalchemy.engine.resultc C s |j �| j�S )z(utility method used for unit tests only.)ZResultMetaData�_create_result_mapr� )ro �resultrB rB rF r� � s z SQLCompiler._create_result_mapc C s dS )z�Called 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| rB )ro rB rB rF � default_from� s zSQLCompiler.default_fromc K s d|j j| f|� d S )N�(�))r� ry )ro Zgrouping�asfromr{ rB rB rF �visit_grouping� s zSQLCompiler.visit_groupingc
K s� | j rl| jjrl| j d d }|j\}}}|r2|}n|}|jj} | d k rl| j|krl| �|| j �rl|jj|d< | j|jfd|i|��S )Nr� r �render_label_as_label�within_columns_clause) r� re Z supports_simple_order_by_label�_label_resolve_dictr� Z_order_by_label_elementr� Zshares_lineagerm )
ro r� r� r{ r � with_cols�
only_froms� only_colsZ resolve_dictZ
order_by_elemrB rB rF �visit_label_reference� s
z!SQLCompiler.visit_label_referencec K s� | j s| �|j�S | j d d }|j\}}}y |r>||j }n
||j }W n& tk
rp t�|jtj d� Y n X ||d< | j|fd|i|��S d S )Nr� r z6Can't resolve label reference for ORDER BY / GROUP BY.r� r� )
r� rm Z _text_clauser� r� �KeyErrorr Z_no_text_coercionr � CompileError) ro r� r� r{ r r� r� r� r� rB rB rF � visit_textual_label_reference� s z)SQLCompiler.visit_textual_label_referencec
K s� |o| }||k}|s|r>t |jtj�r8| �d|j�} n|j} |r�|d k rf|| |j|| f|j |j� |jj| fddd�|��t t
j | j �
|| � S |r�| j �
|| �S |jj| fddi|��S d S )N�colidentT)r� �within_label_clauser� F)�
isinstancer� r �_truncated_label�_truncated_identifierr� r� r� ry � OPERATORSr �as_rg � format_label)
ro r� �add_to_result_mapr� r� r� r� Zrender_label_with_asZrender_label_onlyZ labelnamerB rB rF � visit_label s$
2zSQLCompiler.visit_labelc C s t �d��d S )Nz:Cannot compile Column object until its 'name' is assigned.)r r� )ro r rB rB rF �_fallback_column_name= s z!SQLCompiler._fallback_column_namec K s� |j }}|d kr | �|�}|j}|s>t|tj�r>| �d|�}|d k r^||||||jf|j� |rn| � |�}n | j
� |�}|j }|d ks�|r�|j
s�|S | j
�|�} | r�| j
�| �d }
nd}
|j } t| tj�r�| �d| �} |
| j
� | � d | S d S )Nr� �.r| �alias)r� r� �
is_literalr� r r� r� r� r� �escape_literal_columnrg �quoter: Znamed_with_column�schema_for_object� quote_schema) ro r r� �
include_tabler{ r� Z orig_namer� r: �effective_schemaZ
schema_prefixZ tablenamerB rB rF � visit_columnB s.
zSQLCompiler.visit_columnc K s | j �|j�S )N)rg �format_collation� collation)ro r� r� rB rB rF �visit_collationj s zSQLCompiler.visit_collationc K s |j S )N)r� )ro Z
fromclauser{ rB rB rF �visit_fromclausem s z SQLCompiler.visit_fromclausec K s |j S )N)r� )ro �indexr{ rB rB rF � visit_indexp s zSQLCompiler.visit_indexc K s ||d< | j jj|jf|�S )N�type_expression)re �
type_compilerrm r� )ro �
typeclauser� rB rB rF �visit_typeclauses s z SQLCompiler.visit_typeclausec C s | j jr|�dd�}|S )N�%z%%)rg �_double_percents�replace)ro �textrB rB rF �post_process_textw s z SQLCompiler.post_process_textc C s | j jr|�dd�}|S )Nr z%%)rg r r )ro r rB rB rF r� | s z!SQLCompiler.escape_literal_columnc s< � ��fdd�}�j s d�_t�dd� t�|���j���S )Nc s: | � d�}|�jkr(�j�j| f� �S �j|f� �S d S )Nr )r% Z _bindparamsrm �bindparam_string)r� r� )r� ro �
textclauserB rF � do_bindparam� s
z2SQLCompiler.visit_textclause.<locals>.do_bindparamTc S s
| � d�S )Nr )r% )r� rB rB rF r� � r� z.SQLCompiler.visit_textclause.<locals>.<lambda>)r� � isplaintext�BIND_PARAMS_ESCr� � BIND_PARAMSr r )ro r r� r rB )r� ro r rF �visit_textclause� s z SQLCompiler.visit_textclausec K s� | j }|r| jn| j d }|p>|dkr4|�dd�p>|�dd�}|rt|j | _| _x |jD ]} | j| d| jd� qZW | j|j f|�}
|r�|r�d|
}
|
S ) Nr� r � need_result_map_for_compoundFr� T)r� r� z(%s))
r� �_default_stack_entry�getr� r� �_textual_ordered_columnsZ column_argsrm �_add_to_result_mapr� ) ro Ztaf�compound_indexr� �parensr� �toplevelr� �populate_result_map�cr rB rB rF �visit_text_as_from� s" z SQLCompiler.visit_text_as_fromc K s dS )NZNULLrB )ro �exprr� rB rB rF �
visit_null� s zSQLCompiler.visit_nullc K s | j jr dS dS d S )Nr< �1)re �supports_native_boolean)ro r r� rB rB rF �
visit_true� s zSQLCompiler.visit_truec K s | j jr dS dS d S )Nr! �0)re r )ro r r� rB rB rF � visit_false� s zSQLCompiler.visit_falsec s^ |j }|d krd}n
t|j }|�dd� � �fdd�|jD �D ��}|jrZ�jjrZd| }|S )N� c s s | ]}|r|V qd S )NrB )rD �srB rB rF r� � s z/SQLCompiler.visit_clauselist.<locals>.<genexpr>c 3 s | ]}|j �f� �V qd S )N)ry )rD r )r� ro rB rF r� � s zVALUES )�operatorr� r* �clausesZ
_tuple_valuesre Ztuple_in_values)ro Z
clauselistr� �sepr rB )r� ro rF �visit_clauselist� s
z SQLCompiler.visit_clauselistc K s� d}|j d k r&||j j| f|�d 7 }x>|jD ]4\}}|d|j| f|� d |j| f|� d 7 }q.W |jd k r�|d|jj| f|� d 7 }|d7 }|S )NzCASE r# zWHEN z THEN zELSE ZEND)r� ry ZwhensZelse_)ro �clauser{ rE Zcondr� rB rB rF �
visit_case� s
.
zSQLCompiler.visit_casec K s |j j| f|�S )N)Ztyped_expressionry )ro Z type_coercer� rB rB rF �visit_type_coerce� s z SQLCompiler.visit_type_coercec K s$ d|j j| f|�|jj| f|�f S )NzCAST(%s AS %s))r) ry r )ro r r{ rB rB rF �
visit_cast� s zSQLCompiler.visit_castc K s� d|d t jkrdn\|d t jkr&dnJ|d dk rTd| jt �t|d ��f|�f n d| jt �|d �f|�f |d t jkr�dn\|d t jkr�dnJ|d dk r�d| jt �t|d ��f|�f n d| jt �|d �f|�f f S ) Nz %s AND %sr zUNBOUNDED PRECEDINGz CURRENT ROWz %s PRECEDINGz %s FOLLOWINGr zUNBOUNDED FOLLOWING)r ZRANGE_UNBOUNDEDZ
RANGE_CURRENTrm �literal�abs)ro �range_r� rB rB rF �_format_frame_clause� s ..z SQLCompiler._format_frame_clausec s� |j r d�j|j f� � }n |jr8d�j|jf� � }nd }d|jj�f� �d�� �fdd�d|jfd|jffD �|r||gng �f S ) NzRANGE BETWEEN %szROWS BETWEEN %sz %s OVER (%s)r# c s6 g | ].\}}|d k rt |�rd||j�f� �f �qS )Nz%s BY %s)r� ry )rD Zwordr) )r{ ro rB rF �
<listcomp> s z*SQLCompiler.visit_over.<locals>.<listcomp>Z PARTITIONZORDER)r/ r0 Zrowsr� ry r* Z partition_by�order_by)ro Zoverr{ r/ rB )r{ ro rF �
visit_over s zSQLCompiler.visit_overc K s$ d|j j| f|�|jj| f|�f S )Nz %s WITHIN GROUP (ORDER BY %s))r� ry r2 )ro Z withingroupr{ rB rB rF �visit_withingroup$ s z SQLCompiler.visit_withingroupc K s$ d|j j| f|�|jj| f|�f S )Nz%s FILTER (WHERE %s))�funcry Z criterion)ro Z
funcfilterr{ rB rB rF �visit_funcfilter* s z SQLCompiler.visit_funcfilterc K s* | j �|j|j�}d||jj| f|�f S )NzEXTRACT(%s FROM %s))� extract_mapr �fieldr ry )ro �extractr{ r8 rB rB rF �
visit_extract0 s zSQLCompiler.visit_extractc s� |d k r ||j |j d|j� t� d|j �� d �}|rB||f|�S t�|jd �}|rd|jr�|d7 }n*|j }� j� |�r�� j�
|�n|}|d }d� � fdd�|j D �|g �d� j
|f|�i S d S )NrB z
visit_%s_funcz%(expr)sr� c s( g | ] }� j �|�r � j �|�n|�qS rB )rg � _requires_quotes_illegal_charsr� )rD �tok)ro rB rF r1 M s z.SQLCompiler.visit_function.<locals>.<listcomp>r )r� r� r� �lower� FUNCTIONSr � __class__Z _has_argsrg r; r� r* Z packagenames�function_argspec)ro r5 r� r{ �dispr� rB )ro rF �visit_function7 s$
zSQLCompiler.visit_functionc K s | � |j�S )N)�visit_sequence�sequence)ro Z
next_valuer� rB rB rF �visit_next_value_funcW s z!SQLCompiler.visit_next_value_funcc K s t d| jj ��d S )Nz2Dialect '%s' does not support sequence increments.)rw re r� )ro rD r� rB rB rF rC Z s zSQLCompiler.visit_sequencec K s |j j| f|�S )N)Z clause_exprry )ro r5 r{ rB rB rF r@ ` s z SQLCompiler.function_argspecr c s( �j }|r�jn�j d }|p2|dko2|�dd�}�j �|d |d ||d�� �j�|j�} d| d �� ��fd d
�t|j�D ��}
|
�j |ft
f d � i����7 }
|
�j |f��7 }
|
|j d k s�|j
d k r�j|f��p�d 7 }
�jr�|r���� |
}
�j �d� � �r |�r d
|
d S |
S d S )Nr� r r F�correlate_froms� asfrom_froms)rF rG r r r# c 3 s, | ]$\}}|j �f� d |d����V qdS )F)r� r r N)ry )rD �ir )r� r{ ro rB rF r� z s z4SQLCompiler.visit_compound_select.<locals>.<genexpr>r� r| r� r� )r� r r �append�compound_keywords�keywordr* � enumerate�selects�group_by_clauser� �order_by_clause�
_limit_clause�_offset_clause� limit_clauser� �_render_cte_clauser� ) ro Zcsr� r r r{ r r� Zneed_result_maprK r rB )r� r{ ro rF �visit_compound_selectc s4
z!SQLCompiler.visit_compound_selectc C s( d|j ||rd| ndf }t| |d �S )Nz
visit_%s_%s%s�_r| )r� r� )ro � operator_Z
qualifier1Z
qualifier2ZattrnamerB rB rF �_get_operator_dispatch� s
z"SQLCompiler._get_operator_dispatchc K s� |j rR|jrt�d��| �|j dd�}|r:|||j f|�S | j|t|j f|�S nL|jr�| �|jdd�}|r||||jf|�S | j|t|j f|�S n
t�d��d S )NzFUnary expression does not support operator and modifier simultaneously�unaryr% �modifierz,Unary expression has no operator or modifier)r% rY r r� rW � _generate_generic_unary_operatorr� � _generate_generic_unary_modifier)ro rX r� rA rB rB rF � visit_unary� s& zSQLCompiler.visit_unaryc K s6 |j s| jjr | j|jf|�S d| j|jf|� S d S )Nz%s = 1)�_is_implicitly_booleanre r rm r� )ro r� r% r� rB rB rF �visit_istrue_unary_operator� s z'SQLCompiler.visit_istrue_unary_operatorc K s: |j s| jjr"d| j|jf|� S d| j|jf|� S d S )NzNOT %sz%s = 0)r] re r rm r� )ro r� r% r� rB rB rF � visit_isfalse_unary_operator� s z(SQLCompiler.visit_isfalse_unary_operatorc K s d| j |tjd� S )NzNOT %s)�override_operator)� visit_binaryr �match_op)ro r r% r� rB rB rF �visit_notmatch_op_binary� s z$SQLCompiler.visit_notmatch_op_binaryc C s t �d� d S )Nz�The IN-predicate was invoked with an empty sequence. This results in a contradiction, which nonetheless can be expensive to evaluate. Consider alternative strategies for improved performance.)r
�warn)ro rB rB rF �_emit_empty_in_warning� s z"SQLCompiler._emit_empty_in_warningc K s2 | j jr dS | j jr | �� | �|j|jk�S d S )Nz1 != 1)re �_use_static_in�_warn_on_empty_inre rm r+ )ro r r% r� rB rB rF �visit_empty_in_op_binary� s
z$SQLCompiler.visit_empty_in_op_binaryc K s2 | j jr dS | j jr | �� | �|j|jk�S d S )Nz1 = 1)re rf rg re rm r+ )ro r r% r� rB rB rF �visit_empty_notin_op_binary� s
z'SQLCompiler.visit_empty_notin_op_binaryc C s t d| jj ��d S )Nz3Dialect '%s' does not support empty set expression.)rw re r� )ro Z
element_typesrB rB rF �visit_empty_set_expr� s z SQLCompiler.visit_empty_set_exprc K s� | j r*t|jtj�r*t|jtj�r*d|d< |p2|j}| �|dd �}|rT|||f|�S y t| }W n t k
r� t
� | |��Y nX | j ||f|�S d S )NT�
literal_bindsr )
�ansi_bind_rulesr� r+ r Z
BindParameterr5 r% rW r� r� r ZUnsupportedCompilationError�_generate_generic_binary)ro r r` �eager_groupingr� rV rA �opstringrB rB rF ra � s
zSQLCompiler.visit_binaryc K s | j |jf|�S )N)rm Z sql_function)ro r� r% r� rB rB rF �&visit_function_as_comparison_op_binary s z2SQLCompiler.visit_function_as_comparison_op_binaryc K sT | j jr,| j|jf|�d | j|jf|� S | j|jf|�d | j|jf|� S d S )Nz %% z % )rg r rm r+ r5 )ro r r% r� rB rB rF �visit_mod_binary s $z SQLCompiler.visit_mod_binaryc K s$ |j |d< | j|d|j d f|�S )Nrn r# )rn rm ro )ro r� r% r� rB rB rF �visit_custom_op_binary s
z"SQLCompiler.visit_custom_op_binaryc K s | j ||jd f|�S )Nr# )rZ ro )ro r� r% r� rB rB rF � visit_custom_op_unary_operator" s z*SQLCompiler.visit_custom_op_unary_operatorc K s | j |d|j f|�S )Nr# )r[ ro )ro r� r% r� rB rB rF � visit_custom_op_unary_modifier' s z*SQLCompiler.visit_custom_op_unary_modifierc K s\ |� dd�}d|d< |jj| fd|i|��| |jj| fd|i|�� }|rX|rXd| }|S )N�
_in_binaryFTrn z(%s))r r+ ry r5 )ro r ro rn r� ru r rB rB rF rm , s z$SQLCompiler._generate_generic_binaryc K s ||j j| f|� S )N)r� ry )ro rX ro r� rB rB rF rZ A s z,SQLCompiler._generate_generic_unary_operatorc K s |j j| f|�| S )N)r� ry )ro rX ro r� rB rB rF r[ D s z,SQLCompiler._generate_generic_unary_modifierc C s t jdtjd�S )Nz'%')r� )r Zliteral_columnr �
STRINGTYPE)ro rB rB rF �_like_percent_literalG s z!SQLCompiler._like_percent_literalc K s2 |� � }| j}|�|j��|�|_| j||f|�S )N)�_clonerw �__add__r5 �visit_like_op_binary)ro r r% r� �percentrB rB rF �visit_contains_op_binaryK s z$SQLCompiler.visit_contains_op_binaryc K s2 |� � }| j}|�|j��|�|_| j||f|�S )N)rx rw ry r5 �visit_notlike_op_binary)ro r r% r� r{ rB rB rF �visit_notcontains_op_binaryQ s z'SQLCompiler.visit_notcontains_op_binaryc K s, |� � }| j}|�|j�|_| j||f|�S )N)rx rw �__radd__r5 rz )ro r r% r� r{ rB rB rF �visit_startswith_op_binaryW s z&SQLCompiler.visit_startswith_op_binaryc K s, |� � }| j}|�|j�|_| j||f|�S )N)rx rw r r5 r} )ro r r% r� r{ rB rB rF � visit_notstartswith_op_binary] s z)SQLCompiler.visit_notstartswith_op_binaryc K s, |� � }| j}|�|j�|_| j||f|�S )N)rx rw ry r5 rz )ro r r% r� r{ rB rB rF �visit_endswith_op_binaryc s z$SQLCompiler.visit_endswith_op_binaryc K s, |� � }| j}|�|j�|_| j||f|�S )N)rx rw ry r5 r} )ro r r% r� r{ rB rB rF �visit_notendswith_op_binaryi s z'SQLCompiler.visit_notendswith_op_binaryc K sL |j �dd �}d|jj| f|�|jj| f|�f |rFd| �|tj� nd S )N�escapez
%s LIKE %sz ESCAPE r| )� modifiersr r+ ry r5 �render_literal_valuer rv )ro r r% r� r� rB rB rF rz o s z SQLCompiler.visit_like_op_binaryc K sL |j �dd �}d|jj| f|�|jj| f|�f |rFd| �|tj� nd S )Nr� z%s NOT LIKE %sz ESCAPE r| )r� r r+ ry r5 r� r rv )ro r r% r� r� rB rB rF r} | s z#SQLCompiler.visit_notlike_op_binaryc K sL |j �dd �}d|jj| f|�|jj| f|�f |rFd| �|tj� nd S )Nr� zlower(%s) LIKE lower(%s)z ESCAPE r| )r� r r+ ry r5 r� r rv )ro r r% r� r� rB rB rF �visit_ilike_op_binary� s z!SQLCompiler.visit_ilike_op_binaryc K sL |j �dd �}d|jj| f|�|jj| f|�f |rFd| �|tj� nd S )Nr� z lower(%s) NOT LIKE lower(%s)z ESCAPE r| )r� r r+ ry r5 r� r rv )ro r r% r� r� rB rB rF �visit_notilike_op_binary� s z$SQLCompiler.visit_notilike_op_binaryc K s& |j �dd�}| j||r dndf|�S )Nr9 Fz BETWEEN SYMMETRIC z BETWEEN )r� r rm )ro r r% r� r9 rB rB rF �visit_between_op_binary� s z#SQLCompiler.visit_between_op_binaryc K s& |j �dd�}| j||r dndf|�S )Nr9 Fz NOT BETWEEN SYMMETRIC z
NOT BETWEEN )r� r rm )ro r r% r� r9 rB rB rF �visit_notbetween_op_binary� s
z&SQLCompiler.visit_notbetween_op_binaryc
K s& |s<|j �| j�}|jr<|�|�}| j|fd||d�|��S |sJ|r�| jr�|jd krn|jd krnt �
d|j ��| j |fddi|��S | �
|�}|| jkr�| j| } | |k r�| js�|jr�| j�|j�s�t �
d|j ��n"| js�|jr�t �
d|j |j f ��| | j|j < | j|< | j|fd|ji|��S )NT)�skip_bind_expressionr� rk z@Bind parameter '%s' without a renderable value not allowed here.r� zIBind parameter '%s' conflicts with unique bind parameter of the same namez�bindparam() name '%s' 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_%s').� expanding)r� � dialect_implre Z_has_bind_expression�bind_expressionrm rl r� r� r r� r� �render_literal_bindparam�_truncate_bindparamr� r> r� � intersectionZ_is_crudr
r� )
ro r� r� rk r� r{ �implr� r� ZexistingrB rB rF �visit_bindparam� sB
zSQLCompiler.visit_bindparamc K s |j }| �||j�S )N)r� r� r� )ro r� r� r� rB rB rF r� � s z$SQLCompiler.render_literal_bindparamc C s( |� | j�}|r||�S td| ��dS )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(Don't know how to literal-quote value %rN)Z_cached_literal_processorre rw )ro r� r� Z processorrB rB rF r� � s
z SQLCompiler.render_literal_valuec C s@ || j kr| j | S |j}t|tj�r2| �d|�}|| j |< |S )Nr� )r� r� r� r r� r� )ro r� Z bind_namerB rB rF r� � s
z SQLCompiler._truncate_bindparamc C s� ||f| j kr | j ||f S |�| j�}t|�| jd kr�| j �|d�}|dt| jd d�� d t|�dd � }|d | j |< n|}|| j ||f< |S )N� r r rU r )r� Z apply_mapr� r� r� r �max�hex)ro Z ident_classr� ZanonnameZcounterZ truncnamerB rB rF r� s ,z!SQLCompiler._truncated_identifierc C s
|| j S )N)r� )ro r� rB rB rF �
_anonymize s zSQLCompiler._anonymizec C s< |� dd�\}}| j�|d�}|d | j|< |d t|� S )Nr# r rU )�splitr� r rC )ro r� �identZderivedZanonymous_counterrB rB rF r� " s zSQLCompiler._process_anonc K sJ | j r&|d k r|�|� n | j�|� |r8d| _d| S | jd|i S d S )NTz([EXPANDING_%s])r� )r� rI r� � contains_expanding_parametersr� )ro r� �positional_namesr� r� rB rB rF r
( s z SQLCompiler.bindparam_stringc s� � � � ||d< t|jtj�r.� �d|j�}n|j}d}d} |� jkr�� j| }
||
k} ||
jksj||
krpd}n"|
|jkr�� j|
= nt �
d| ��|s�|r�|j d k r�|j } |j j} t| tj�rԈ �d| �} n|} d } |�rD|� j|< d|j j
k�rd� jk�r� j�d|j j
d i�� _| � jk�r2� j| f|� | �sD|� jk�rD|j�rRd� _� j�||�}
|j�r�t|jtj��r�|j}n(t|jtj��r�|jjd }n
d�s�t�dd � t� |j �D �}|
d
d � � fd d
�|D �� 7 }
� j �r�g |d< � j |< |
d|jj!� fddi|�� 7 }
|j"�r:|
d� j#||j"f|� 7 }
|
� j|< |�r�|�sd| �rd� j�||�S | �r�� j�|| �}
� j�$|��r�� j�%|�}|
� �&|�7 }
|
S � j�||�S d S )N� visiting_cter� TFz5Multiple, unrelated CTEs found with the same name: %rZ
autocommitr c S s g | ]}|d k r|�qS )NrB )rD r rB rB rF r1 � s z)SQLCompiler.visit_cte.<locals>.<listcomp>z(%s)z, c 3 s | ]}� j �|�V qd S )N)rg �
format_column)rD r� )ro rB rF r� � s z(SQLCompiler.visit_cte.<locals>.<genexpr>r� z AS
r� r# )'r� r� r� r r� r� r� Z _restatesr� r r� Z
_cte_aliasr� rk rl r= � visit_cte� recursiver� rg � format_alias�originalr �Select�CompoundSelectrM �AssertionErrorr
Z unique_list�
inner_columnsr* r� r� ry � _suffixes�_generate_prefixes�_requires_quotesr� �get_render_as_alias_suffix)ro �cter� �ashint� fromhintsr� r{ Zcte_nameZ
is_new_cteZ embedded_in_current_named_cteZ existing_cteZ
pre_alias_cteZcte_pre_alias_namer Z
col_sourceZ
recur_colsrB )ro rF r� 6 s�
zSQLCompiler.visit_ctec K s� |s|r,t |jtj�r&| �d|j�}n|j}|r>| j�||�S |r�|jj| fddi|��| � | j�||�� }|r�||kr�| �
|||| |�}|S |jj| f|�S d S )Nr� r� T) r� r� r r� r� rg r� r� ry r� �format_from_hint_text) ro r� r� r� �iscrudr� r{ Z
alias_name�retrB rB rF � visit_alias� s zSQLCompiler.visit_aliasc K s d|d< d| j |f|� S )NT�lateralz
LATERAL %s)r� )ro r� r� rB rB rF �
visit_lateral� s zSQLCompiler.visit_lateralc K sR d| j |fddi|��|�� j| f|�f }|jd k rN|d|jj| f|� 7 }|S )Nz%s TABLESAMPLE %sr� Tz REPEATABLE (%s))r� Z _get_methodry Zseed)ro Z tablesampler� r� r rB rB rF �visit_tablesample� s
z SQLCompiler.visit_tablesamplec C s d| S )Nz AS rB )ro Zalias_name_textrB rB rF r� � s z&SQLCompiler.get_render_as_alias_suffixc C s | j �||||f� d S )N)r� rI )ro �keynamer� �objectsr� rB rB rF r � s z SQLCompiler._add_to_result_mapc s~ � j ��j�}|jr2|r2|�� �} � �fdd�}
n� } |rB�j}
nd}
|sR| } �nt� tj�r�| � k r|t | � j
� j fd�} n| } n�|dk r�|r�t | |� j fd�} n�|r�t� tj
�r� js� jdk r�t� jtj�s�t | t�� j
�� jfd�} nvt� tj��s8t� tj��r� j�r8t� d��r*t� tj��r8t | � j�} n*| � k �r^t | t�� j
�� jfd�} n| } |j||
d� | j �f|�S )z.produce labeled columns present in a select().c s �� | |� f| |� d S )N)r )r� r� r� r� )r ro rB rF r� � s z;SQLCompiler._label_select_column.<locals>.add_to_result_mapN)r� r� )r� r� ) r� r� re Z_has_column_expressionZcolumn_expressionr r� r ZLabelr� r� r� �
_key_labelZ ColumnClauser� r: r r� Z
_as_truncatedr� Z
TextClauseZUnaryExpressionZwraps_column_expression�hasattrr ZFunctionZ
anon_label�updatery ) ro r6 r r r� �column_clause_argsr� r� r� Zcol_exprr� Z result_exprrB )r ro rF �_label_select_column� sV
z SQLCompiler._label_select_columnc C s | � ||�}|r |d| 7 }|S )Nr# )�get_from_hint_text)ro �sqltextr: Zhintr� �hinttextrB rB rF r� ; s z!SQLCompiler.format_from_hint_textc C s d S )NrB )ro ZbyfromsrB rB rF �get_select_hint_textA s z SQLCompiler.get_select_hint_textc C s d S )NrB )ro r: r rB rB rF r� D s z SQLCompiler.get_from_hint_textc C s d S )NrB )ro r: r rB rB rF �get_crud_hint_textG s z SQLCompiler.get_crud_hint_textc C s
d� |�S )Nr# )r* )ro Z
hint_textsrB rB rF �get_statement_hint_textJ s z#SQLCompiler.get_statement_hint_textc s" i � i g�� ��fdd���|�S )z�Rewrite any "a JOIN (b JOIN c)" expression as
"a JOIN (select * from b JOIN c) AS anon", to support
databases that can't parse a parenthesized join correctly
(i.e. sqlite < 3.7.16).
c
s� | �d kr�d | S | � kr(� | S | � � }� | < |j�r^|j�r^t|jtj��r^|�� �|jf|�|_�|jf|�}tj |j
gdd�� � }x |j D ]}|j
|_|j|_q�W tt|jj
j |j ��}|||j
j< |||j
j< |||jj
j< |||jj
j< x8t�d �� �D ]$\}}||k�r|| �d |< �qW �d �|� ||_�|jf|�|_n�|j�r�d|d< |jf d�i|�� nn|j�r�|j�r�|�dd �dk} | �r���i � d|d< |jf d�i|�� | �r�d= n|jf d�i|�� |S )Nr� T)�
use_labelsZselect_containerZtransform_clueZcloneZ
inside_select) rx Z
is_selectableZ_is_joinr� r5 r Z FromGroupingZ_reset_exportedr+ r� r� r� r r� r� r� Z_labelr� �zipr� �itemsr� �onclauseZ_is_from_containerZ_copy_internalsZ
_is_selectr rI )
r� r� Znewelemr5 Z selectable_r Ztranslate_dict�k�vZbarrier_select)�cloned�column_translate�visitrB rF r� W sR
z=SQLCompiler._transform_select_for_nested_joins.<locals>.visitrB )ro r6 rB )r� r� r� rF �"_transform_select_for_nested_joinsM s Qz.SQLCompiler._transform_select_for_nested_joinsc sF t dd� |jD ���t �fdd�|jD ��� � fdd�| jD �| _d S )Nc s s | ]}|j |fV qd S )N)r� )rD r rB rB rF r� � s zESQLCompiler._transform_result_map_for_nested_joins.<locals>.<genexpr>c 3 s | ]}� |j |fV qd S )N)r� )rD r )� inner_colrB rF r� � s c s2 g | ]*\}}}}||t � fd d�|D ��|f�qS )c s g | ]}� � ||��qS rB )r )rD r� )�drB rF r1 � s zQSQLCompiler._transform_result_map_for_nested_joins.<locals>.<listcomp>.<listcomp>)�tuple)rD r� r� Zobjs�typ)r� rB rF r1 � s zFSQLCompiler._transform_result_map_for_nested_joins.<locals>.<listcomp>)r� r� r� )ro r6 �transformed_selectrB )r� r� rF �&_transform_result_map_for_nested_joins� s
z2SQLCompiler._transform_result_map_for_nested_joinsrF rG c C s\ | j }|r| jn| j d }|d }|d }|rJ|sJ|j|�|�dd�}n|j||d�}|S )Nr� rF rG rB )�explicit_correlate_froms�implicit_correlate_froms)r� r �_get_display_froms�
difference) ro r6 r� r� r r� rF rG �fromsrB rB rF �_display_froms_for_select� s
z%SQLCompiler._display_froms_for_selectc sf �j o | o �j o �jj }
|
rJ����} �j| f� |||dd�| ��} �j }
|
r\�jn�jd }|
p�|dkr~|�dd�p�|�dd���s�d| kr�| d= |
r��r����| � | S �� �|� |�}| �
� ��� ddd �� d
} �j �r
��
��\}}|�r| |d 7 } nd }�j�r,| �j��jf| �7 } | �j�f| �7 } d d
� � ����fdd
��jD �D �}��r�|d k �r�ttdd
� �jD �dd
� |jD �����fdd
��jD ��_��| �|||| �} �j�r��fdd
��jD �}|�r�| d ��|� 7 } �j�r|
�r��� | } �j�r4| d �j��jf| � 7 } �j�d� � �sL|�r^|�r^d| d S | S d S )NT)r� r r� r �nested_join_translationr� r r Fr� r� )r� r� zSELECT r# c S s g | ]}|d k r|�qS )NrB )rD r rB rB rF r1 & s z,SQLCompiler.visit_select.<locals>.<listcomp>c
s& g | ] \}}�j �|�� �|d ��qS ))r� )r� )rD r� r )r� r� r r6 ro rB rF r1 ( s c S s g | ] \}}|�qS rB rB )rD r� r� rB rB rF r1 ; s c S s g | ] \}}|�qS rB rB )rD r� r� rB rB rF r1 = s c s2 g | ]*\}}}}||t � fd d�|D ��|f�qS )c 3 s | ]}� � ||�V qd S )N)r )rD �o)� translaterB rF r� D s z6SQLCompiler.visit_select.<locals>.<listcomp>.<genexpr>)r� )rD r� r� rz r� )r� rB rF r1 D s c s$ g | ] \}}|d � j jfkr|�qS )�*)re r� )rD � dialect_nameZht)ro rB rF r1 N s r� r� ) r� r� re Zsupports_right_nested_joinsr� � visit_selectr r r� �_setup_select_stack�copyr� �_hints�_setup_select_hints� _prefixesr� �get_select_precolumnsZ_columns_plus_namesr� r� r� �_compose_select_bodyZ_statement_hintsr� r� rS r� r� )ro r6 r� r r� r r� Zselect_wraps_forr� r{ Zneeds_nested_translationr� r r r� r� � hint_text�byfromr� Z per_dialectrB )r� r� r r6 ro r� rF r� � s�
zSQLCompiler.visit_selectc s. t � fdd�|j�� D ��}� �|�}||fS )Nc s> g | ]6\\}}}|d � j jfkr||d|j� dd�i f�qS )r� r� T)r� )re r� ry )rD �from_re r� )ro rB rF r1 g s
z3SQLCompiler._setup_select_hints.<locals>.<listcomp>)r� r� r� r� )ro r6 r� r� rB )ro rF r� d s
z SQLCompiler._setup_select_hintsc C sp |d }|d }|r.|s.|j |�|�dd�}n|j ||d�}ttj|� �}|�|�} || |d�}
| j�|
� |S )NrF rG rB )r� r� )rG rF r )r� r� r8 r �
_from_objectsr= r� rI ) ro r6 r� r� r� rF rG r� Znew_correlate_fromsZall_correlate_fromsZ new_entryrB rB rF r� s s"
z SQLCompiler._setup_select_stackc sB |d� |�7 }|rb|d7 }|jrB|d� � ��fdd�|D ��7 }qn|d� ��fdd�|D ��7 }n |��� 7 }|jd k r�|jj�f��}|r�|d| 7 }|jjr�|�j|f��7 }|jd k r�|jj�f��}|r�|d| 7 }|j jr�|�j
|f��7 }|j d k �s|j d k �r |�j
|f��7 }|jd k �r>|�j|f��7 }|S )Nz, z
FROM c s$ g | ] }|j �fd � d�����qS )T)r� r� )ry )rD �f)r� r{ ro rB rF r1 � s z4SQLCompiler._compose_select_body.<locals>.<listcomp>c s" g | ]}|j �fd di� ���qS )r� T)ry )rD r� )r{ ro rB rF r1 � s z
WHERE z
HAVING )r* r� r� � _whereclausery �_group_by_clauser& rN Z_having�_order_by_clauserO rP rQ rR Z_for_update_arg�for_update_clause)ro r r6 r� r� r� r{ �trB )r� r{ ro rF r� � s:
z SQLCompiler._compose_select_bodyc s* d� � �fdd�|D ��}|r&|d7 }|S )Nr# c 3 s4 | ],\}}|d ks |�j jkr|j�f� �V qd S )N)re r� ry )rD �prefixr� )r� ro rB rF r� � s z1SQLCompiler._generate_prefixes.<locals>.<genexpr>)r* )ro �stmt�prefixesr� r) rB )r� ro rF r� � s
z SQLCompiler._generate_prefixesc sb � j r(t� fdd�� jD �g �� j � _� �� j�d }|d�dd� � j�� D ��7 }|d7 }|S )Nc s g | ]}� j | �qS rB )r� )rD r� )ro rB rF r1 � s z2SQLCompiler._render_cte_clause.<locals>.<listcomp>r# z,
c S s g | ]}|�qS rB rB )rD ZtxtrB rB rF r1 � s z
)r� �sumr� r� �get_cte_preambler� r* �values)ro Zcte_textrB )ro rF rS � s
z SQLCompiler._render_cte_clausec C s |rdS dS d S )NzWITH RECURSIVEZWITHrB )ro r� rB rB rF r� � s z SQLCompiler.get_cte_preamblec K s |j r
dp dS )zcCalled when building a ``SELECT`` statement, position is just
before column list.
z DISTINCT r| )Z _distinct)ro r6 r� rB rB rF r� � s z!SQLCompiler.get_select_precolumnsc K s$ |j j| f|�}|r d| S dS dS )z5allow dialects to customize how GROUP BY is rendered.z
GROUP BY r| N)r� ry )ro r6 r� Zgroup_byrB rB rF rN � s zSQLCompiler.group_by_clausec K s$ |j j| f|�}|r d| S dS dS )z5allow dialects to customize how ORDER BY is rendered.z
ORDER BY r| N)r� ry )ro r6 r� r2 rB rB rF rO � s zSQLCompiler.order_by_clausec K s dS )Nz FOR UPDATErB )ro r6 r� rB rB rF r� � s z SQLCompiler.for_update_clausec C s t �d��d S )Nz@RETURNING is not supported by this dialect's statement compiler.)r r� )ro r� �returning_colsrB rB rF �returning_clause� s z SQLCompiler.returning_clausec K s^ d}|j d k r&|d| j|j f|� 7 }|jd k rZ|j d krB|d7 }|d| j|jf|� 7 }|S )Nr| z
LIMIT z
LIMIT -1z OFFSET )rP rm rQ )ro r6 r� r rB rB rF rR s
zSQLCompiler.limit_clausec
K sv |s|rn| j �|�}|r<|r<| j �|�d | j �|j� } n| j �|j�} |rj||krj| �| ||| |�} | S dS d S )Nr� r| )rg r� r� r� r� r� )
ro r: r� r� r� r� �
use_schemar{ r� r� rB rB rF � visit_table
s
zSQLCompiler.visit_tablec K sd |j r d}n|jrd}nd}|jj| fddi|��| |jj| fddi|�� d |jj| f|� S )Nz FULL OUTER JOIN z LEFT OUTER JOIN z JOIN r� Tz ON )r$ Zisouterr+ ry r5 r� )ro r* r� r{ Z join_typerB rB rF �
visit_join* s zSQLCompiler.visit_joinc sF t � fdd�|j�� D ��}|j|kr>� �||j||j d�}||fS )Nc s, g | ]$\\}}}|d � j jfkr||f�qS )r� )re r� )rD r: re r� )ro rB rF r1 < s
z1SQLCompiler._setup_crud_hints.<locals>.<listcomp>T)r� r� r� r: r� )ro r� �
table_text�
dialect_hintsrB )ro rF �_setup_crud_hints9 s
z SQLCompiler._setup_crud_hintsc s� | j }| j �t� t� |d�� tj| |tjf|�}|sZ| jjsZ| jjsZt �
d| jj ��|j r�| jj
szt �
d| jj ��|d }n|}| j� | jj}d}|jr�|| j||jf|�7 }|d7 }� �|j�} |jr�| �|| �\}
} || 7 }|s�|�s|dd�� fd d
�|D �� 7 }| j�s"|j�rN| �|| j�p4|j�} | j�rR|d | 7 }nd } |jd k �r�| j| j f|�} | j �r�|�r�| jj �r�|d | � � | f 7 }n |d
| 7 }nX|�s�|�r�|d7 }nB|j �r�|dd�dd� |D �� 7 }n |dd�dd
� |D �� 7 }|j d k �r0| j|j f|�}
|
�r0|d |
7 }| �rJ| j�sJ|d | 7 }| j �rn|�rn| jj �sn| � � | }| j �!d� |�r�d| d S |S d S )N)rF rG 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.r zINSERT zINTO z (%s)z, c s g | ]}� � |d ��qS )r )r� )rD r )rg rB rF r1 � s z,SQLCompiler.visit_insert.<locals>.<listcomp>r# z %s%sz %sz DEFAULT VALUESz
VALUES %sc s s& | ] }d d� dd� |D �� V qdS )z(%s)z, c s s | ]}|d V qdS )r NrB )rD r rB rB rF r� � s z5SQLCompiler.visit_insert.<locals>.<genexpr>.<genexpr>N)r* )rD Zcrud_param_setrB rB rF r� � s z+SQLCompiler.visit_insert.<locals>.<genexpr>z VALUES (%s)c S s g | ] }|d �qS )r rB )rD r rB rB rF r1 � s r� r� r� )"r� rI r8 r �_setup_crud_paramsZISINSERTre �supports_default_valuesZsupports_empty_insertr r� r� Z_has_multi_parametersZsupports_multivalues_insertrg r� r� � format_tabler: r� r� r* r� r� r� �returning_precedes_valuesr6 rm Z_insert_from_selectr� Zcte_follows_insertrS Z_post_values_clauser� )ro Z insert_stmtr� r� r � crud_paramsZcrud_params_singler� r r� rU r� Z select_textZpost_values_clauserB )rg rF � visit_insertG s�
zSQLCompiler.visit_insertc C s dS )z3Provide a hook for MySQL to add LIMIT to the UPDATENrB )ro � update_stmtrB rB rF �update_limit_clause� s z SQLCompiler.update_limit_clausec K s d|d< |j | fddi|��S )z|Provide a hook to override the initial table clause
in an UPDATE statement.
MySQL overrides this.
Tr� r� )ry )ro r �
from_table� extra_fromsr� rB rB rF �update_tables_clause� s z SQLCompiler.update_tables_clausec K s t d��dS )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 UPDATEN)rw )ro r r r �
from_hintsr� rB rB rF �update_from_clause� s z SQLCompiler.update_from_clausec s( �j }|j}t|�}|rHtt�|j����fdd�|D �}��|�}n g }|jh}�j �|||d�� d} |j r�| �j
||j f|�7 } �j ||j|f|�}
t j
�|t jf|�} |jrȈ�||
�\} }
nd } | |
7 } | d7 } |o�j� | d�� �fdd�| D ��7 } �j�s|j�r:�j�r:| d ��|�j�p2|j� 7 } |�rh�j||j|| f|�}
|
�rh| d |
7 } |jd k �r��j|jf|�}|�r�| d
| 7 } ��|�}|�r�| d | 7 } �j�s�|j�r�j�s�| d ��|�j�p�|j� 7 } �j�r|�r�� � | } �j � d � |�r d | d
S | S d S )Nc s g | ]}|� kr|�qS rB rB )rD r� )�
main_fromsrB rF r1 � s z,SQLCompiler.visit_update.<locals>.<listcomp>)rF rG r zUPDATE z SET z, c 3 s, | ]$}|d j �� d�d |d V qdS )r )r� �=r N)ry )rD r )r� ro rB rF r�
s z+SQLCompiler.visit_update.<locals>.<genexpr>r# z WHERE r� r� r� ) r� � _extra_froms�boolr8 r r� r: r= rI r� r� r r r� ZISUPDATEr� r� �'render_table_with_column_in_update_fromr* r� r� r� r� r r� rm r r� rS r� )ro r r� r� r r Z
is_multitableZrender_extra_fromsrF r r� r r� �extra_from_textr� rR rB )r� r ro rF � visit_update� sz
zSQLCompiler.visit_updatec C s t �| | j�S )N)r � _key_getters_for_crud_columnri )ro rB rB rF r 6
s z(SQLCompiler._key_getters_for_crud_columnc K s t d��dS )z�Provide 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 DELETEN)rw )ro r r r r r� rB rB rF �delete_extra_from_clause:
s z$SQLCompiler.delete_extra_from_clausec C s |j | ddd�S )NT)r� r� )ry )ro � delete_stmtr r rB rB rF �delete_table_clauseJ
s z SQLCompiler.delete_table_clausec K s� | j }tj| |tjf|� |j}|jh�|�}| j �|||d�� d}|jrd|| j ||jf|�7 }|d7 }| �
||j|�}|j r�| � ||�\} }nd } ||7 }|j
r�| jr�|d| �||j
� 7 }|r�| j||j|| f|�}
|
r�|d|
7 }|jd k �r|jj| f|�} | �r|d| 7 }|j
�r@| j�s@|d| �||j
� 7 }| j�rZ|�rZ| �� | }| j �d� |�rxd| d S |S d S ) N)rF rG r zDELETE zFROM r# z WHERE r� r� r� )r� r r� ZISDELETEr r: r= rI r� r� r r� r� r� r� r� r r� ry r� rS r� ) ro r r� r� r r rF r r� r� r r� rB rB rF � visit_deleteM
s\
zSQLCompiler.visit_deletec C s d| j �|� S )Nz SAVEPOINT %s)rg �format_savepoint)ro �savepoint_stmtrB rB rF �visit_savepoint�
s zSQLCompiler.visit_savepointc C s d| j �|� S )NzROLLBACK TO SAVEPOINT %s)rg r )ro r rB rB rF �visit_rollback_to_savepoint�
s z'SQLCompiler.visit_rollback_to_savepointc C s d| j �|� S )NzRELEASE SAVEPOINT %s)rg r )ro r rB rB rF �visit_release_savepoint�
s z#SQLCompiler.visit_release_savepoint)NF)NNT)F)F)F)NFFN)NT)NFT)N)FTr )NF)F)FFF)NF)FFNN)FFFN)F)NT)F)FTNr FNF)FFFNT)F)F)F)F)�r� r� r� r� � EXTRACT_MAPr7 �COMPOUND_KEYWORDSrJ r� r� r� r
r� r� r
r� rl r r� r� r� r� rq r� r� r
Zmemoized_instancemethodr� �
contextlib�contextmanagerr� r� �memoized_propertyr� r� rx r~ ru � dependenciesr� r� r� r� r� r� r� r� r� r� r r r r� r r r r r" r( r* r+ r, r0 r3 r4 r6 r: rB rE rC r@ rT rW r\ r^ r_ rc re rh ri rj ra rp rq rr rs rt rm rZ r[ rw r| r~ r� r� r� r� rz 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� � frozensetr r� r� r� r� r� r� rS r� r� rN rO r� r� rR r� r� r� r r r r r r r r r r r r rB rB rB rF r� � sH G 8 " + ' 2
8
v
L] 6
m
f J r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�StrSQLCompileraw 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:`.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:`.ClauseElement.compile`
directly.
.. seealso::
:ref:`faq_sql_expression_string`
c C s dS )Nz<name unknown>rB )ro r rB rB rF r� �
s z$StrSQLCompiler._fallback_column_namec K s$ d| j |jf|�| j |jf|�f S )Nz%s[%s])rm r+ r5 )ro r r% r� rB rB rF �visit_getitem_binary�
s z#StrSQLCompiler.visit_getitem_binaryc K s | j ||f|�S )N)r" )ro r r% r� rB rB rF � visit_json_getitem_op_binary�
s z+StrSQLCompiler.visit_json_getitem_op_binaryc K s | j ||f|�S )N)r" )ro r r% r� rB rB rF �!visit_json_path_getitem_op_binary�
s z0StrSQLCompiler.visit_json_path_getitem_op_binaryc K s d| j �|� S )Nz<next sequence value: %s>)rg �format_sequence)ro �seqr� rB rB rF rC �
s z StrSQLCompiler.visit_sequencec s& � fdd�t �|�D �}dd�|� S )Nc s g | ]}� � d |ddi ��qS )NTF)r� )rD r )ro rB rF r1 �
s z3StrSQLCompiler.returning_clause.<locals>.<listcomp>z
RETURNING z, )r Z_select_iterablesr* )ro r� r� �columnsrB )ro rF r� �
s
z StrSQLCompiler.returning_clausec s dd� � ��fdd�|D �� S )NzFROM z, c 3 s& | ] }|j �fd � d����V qdS )T)r� r� N)ry )rD r� )r r� ro rB rF r� �
s z4StrSQLCompiler.update_from_clause.<locals>.<genexpr>)r* )ro r r r r r� rB )r r� ro rF r �
s z!StrSQLCompiler.update_from_clausec s dd� � ��fdd�|D �� S )Nz, c 3 s& | ] }|j �fd � d����V qdS )T)r� r� N)ry )rD r� )r r� ro rB rF r� �
s z:StrSQLCompiler.delete_extra_from_clause.<locals>.<genexpr>)r* )ro r r r r r� rB )r r� ro rF r �
s z'StrSQLCompiler.delete_extra_from_clauseN) r� r� r� r� r� r" r# r$ rC r� r r rB rB rB rF r! �
s r! c @ sB e Zd Zejdd� �Zejdd� �ZdLdd�Zdd � Zd
d � Z d d
� Z
dd� Z dMdd�Z dNdd�Z
dd� Zdd� Zdd� ZdOd d �Zd d � ZdPd d!�Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� 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(dS )Q� DDLCompilerc C s | j �| j d �S )N)re Zstatement_compiler)ro rB rB rF rx �
s zDDLCompiler.sql_compilerc C s | j jS )N)re r )ro rB rB rF r �
s zDDLCompiler.type_compilerNc C s d S )NrB )ro ru rB rB rF r~ �
s z DDLCompiler.construct_paramsc K s� |j }t|jtj�r�|�� }| j}|�|j�}t|�dkrJ|d d }}n|d |d }}|� d|� |� d|� |� d|�
|j�� | j � |j
| �S )Nr r r| r� r: r �fullname)�contextr� �targetr ZTabler� rg �format_table_seqr� �
setdefaultr� rx r ri )ro Zddlr{ r* rg �pathr: ZschrB rB rF � visit_ddl�
s zDDLCompiler.visit_ddlc C s | j �|j�}d| S )NzCREATE SCHEMA )rg �
format_schemar� )ro r r rB rB rF �visit_create_schema s z DDLCompiler.visit_create_schemac C s( | j �|j�}d| }|jr$|d7 }|S )Nz DROP SCHEMA z CASCADE)rg r0 r� �cascade)ro �dropr r rB rB rF �visit_drop_schema s
z DDLCompiler.visit_drop_schemac
C sR |j }| j}d}|jr*|d�|j�d 7 }|d|�|� d 7 }| �|�}|rZ||d 7 }|d7 }d}d}x�|jD ]�}|j } yD| j|| jo�| d�}
|
d k r�||7 }d}|d |
7 }| jr�d
}W qr t j
k
�r } z.t � t �
t �
d �|j| j| jd f �� W d d } ~ X Y qrX qrW | j||jd
�} | �r<||d | 7 }|d| �|� 7 }|S )Nz
CREATE r# zTABLE r� �
F)�first_pkz,
� Tz (in table '%s', column '%s'): %sr )� _include_foreign_key_constraintsz
)%s
)r� rg r� r* r� �create_table_suffixr' rm � primary_keyr r� r
Zraise_from_cause�u� descriptionr� �args�create_table_constraintsZ include_foreign_key_constraints�post_create_table)
ro r r: rg r r9 Z separatorr6 Z
create_columnr Z processedZce�constrB rB rF �visit_create_table sF
0
z DDLCompiler.visit_create_tableFc sL |j }|jrd S � j||d�}d�� fdd�|jD ��}|rH|d| 7 }|S )N)r6 r# c 3 s | ]}� � |�V qd S )N)rm )rD r )ro rB rF r� E s z2DDLCompiler.visit_create_column.<locals>.<genexpr>)r� �system�get_column_specificationr* � constraints)ro r r6 r r r@ rB )ro rF �visit_create_column= s z DDLCompiler.visit_create_columnc st g }�j r|��j � �j}|d k r0|�|�� nt� � |�� �fdd��jD �� d�dd� �fdd�|D �D ��S )Nc s" g | ]}|�j k r|� kr|�qS rB )r: )rD r )� omit_fkcsr: rB rF r1 ^ s z8DDLCompiler.create_table_constraints.<locals>.<listcomp>z,
c s s | ]}|d k r|V qd S )NrB )rD �prB rB rF r� e s z7DDLCompiler.create_table_constraints.<locals>.<genexpr>c 3 s@ | ]8}|j d ks|� � �r� jjr.t|dd�s� �|�V qd S )NZ use_alterF)Z _create_rulere Zsupports_alterr� rm )rD r )ro rB rF r� g s
)r: rI Zforeign_key_constraintsr� r8 �extendZ_sorted_constraintsr* )ro r: r8 rD Zall_fkcsrB )rF ro r: rF r> L s
z$DDLCompiler.create_table_constraintsc C s d| j �|j� S )Nz
DROP TABLE )rg r� r� )ro r3 rB rB rF �visit_drop_tableu s z DDLCompiler.visit_drop_tablec C s d| j �|j� S )Nz
DROP VIEW )rg r� r� )ro r3 rB rB rF �visit_drop_viewx s zDDLCompiler.visit_drop_viewc C s |j d krt�d|j ��d S )Nz,Index '%s' is not associated with any table.)r: r r� r� )ro r� rB rB rF �_verify_index_table{ s
z DDLCompiler._verify_index_tableTc sj |j }� �|� � j}d}|jr(|d7 }|d� j||d�|j|j|d�d�� fdd�|jD ��f 7 }|S ) NzCREATE zUNIQUE zINDEX %s ON %s (%s))�include_schema)r� z, c 3 s | ]}� j j|d dd�V qdS )FT)r� rk N)rx rm )rD r )ro rB rF r� � s z1DDLCompiler.visit_create_index.<locals>.<genexpr>) r� rK rg r> �_prepared_index_namer� r: r* Z expressions)ro r rL Zinclude_table_schemar� rg r rB )ro rF �visit_create_index� s
z DDLCompiler.visit_create_indexc C s |j }d| j|dd� S )Nz
DROP INDEX T)rL )r� rM )ro r3 r� rB rB rF �visit_drop_index� s z DDLCompiler.visit_drop_indexc C sX |j d k r| j�|j �}nd }|r4|r4| j�|�}nd }| j�|�}|rT|d | }|S )Nr� )r: rg r� r� � format_index)ro r� rL r� Z schema_nameZ
index_namerB rB rF rM � s
z DDLCompiler._prepared_index_namec C s d| j �|jj�| �|j�f S )NzALTER TABLE %s ADD %s)rg r� r� r: rm )ro r rB rB rF �visit_add_constraint� s z DDLCompiler.visit_add_constraintc C s( d| j �|j�| j�|jjt�� �f S )NzCOMMENT ON TABLE %s IS %s)rg r� r� rx r� �commentr �String)ro r rB rB rF �visit_set_table_comment� s z#DDLCompiler.visit_set_table_commentc C s d| j �|j� S )NzCOMMENT ON TABLE %s IS NULL)rg r� r� )ro r3 rB rB rF �visit_drop_table_comment� s z$DDLCompiler.visit_drop_table_commentc C s. d| j j|jddd�| j�|jjt�� �f S )NzCOMMENT ON COLUMN %s IS %sT)� use_tabler� )rg r� r� rx r� rR r rS )ro r rB rB rF �visit_set_column_comment� s
z$DDLCompiler.visit_set_column_commentc C s d| j j|jdd� S )Nz COMMENT ON COLUMN %s IS NULLT)rV )rg r� r� )ro r3 rB rB rF �visit_drop_column_comment� s z%DDLCompiler.visit_drop_column_commentc C s� d| j �|j� }|jjd k r.|d|jj 7 }|jjd k rJ|d|jj 7 }|jjd k rf|d|jj 7 }|jjd k r�|d|jj 7 }|jjd k r�|d7 }|jjd k r�|d7 }|jj d k r�|d|jj 7 }|jj
d kr�|d
7 }|jj d k r�|d 7 }|S ) NzCREATE SEQUENCE %sz INCREMENT BY %dz START WITH %dz MINVALUE %dz MAXVALUE %dz NO MINVALUEz NO MAXVALUEz CACHE %dTz ORDERz CYCLE) rg r% r� Z increment�startZminvalueZmaxvalueZ
nominvalueZ
nomaxvalue�cacher4 �cycle)ro r r rB rB rF �visit_create_sequence� s*
z!DDLCompiler.visit_create_sequencec C s d| j �|j� S )NzDROP SEQUENCE %s)rg r% r� )ro r3 rB rB rF �visit_drop_sequence� s z DDLCompiler.visit_drop_sequencec C s^ |j }|jd k r | j�|�}nd }|d kr:t�d|j ��d| j�|j j�||jrVdpXdf S )Nz<Can't emit DROP CONSTRAINT for constraint %r; it has no namez#ALTER TABLE %s DROP CONSTRAINT %s%sz CASCADEr| ) r� r� rg �format_constraintr r� r� r: r2 )ro r3 r �formatted_namerB rB rF �visit_drop_constraint� s
z!DDLCompiler.visit_drop_constraintc K sT | j �|�d | jjj|j|d� }| �|�}|d k rB|d| 7 }|jsP|d7 }|S )Nr# )r z DEFAULT z NOT NULL)rg r� re r rm r� �get_column_default_stringZnullable)ro r r{ Zcolspecr rB rB rF rC � s
z$DDLCompiler.get_column_specificationc C s dS )Nr| rB )ro r: rB rB rF r9
s z DDLCompiler.create_table_suffixc C s dS )Nr| rB )ro r: rB rB rF r? s z DDLCompiler.post_create_tablec C sP t |jtj�rHt |jjtj�r2| j�|jjt j
�S | jj |jjdd�S nd S d S )NT)rk ) r� Zserver_defaultr Z
DefaultClause�argr
Z string_typesrx r� r rv rm )ro r rB rB rF ra s z%DDLCompiler.get_column_default_stringc C s\ d}|j d k r.| j�|�}|d k r.|d| 7 }|d| jj|jddd� 7 }|| �|�7 }|S )Nr| zCONSTRAINT %s z
CHECK (%s)FT)r� rk )r� rg r^ rx rm r� � define_constraint_deferrability)ro r r r_ rB rB rF �visit_check_constraint s
z"DDLCompiler.visit_check_constraintc C s\ d}|j d k r.| j�|�}|d k r.|d| 7 }|d| jj|jddd� 7 }|| �|�7 }|S )Nr| zCONSTRAINT %s z
CHECK (%s)FT)r� rk )r� rg r^ rx rm r� rc )ro r r r_ rB rB rF � visit_column_check_constraint, s
z)DDLCompiler.visit_column_check_constraintc s� t |�dkrdS d}|jd k r>� j�|�}|d k r>|d| 7 }|d7 }|dd�� fdd�|jrd|jn|jD �� 7 }|� �|�7 }|S ) Nr r| zCONSTRAINT %s z PRIMARY KEY z(%s)z, c 3 s | ]}� j �|j�V qd S )N)rg r� r� )rD r )ro rB rF r� B s z;DDLCompiler.visit_primary_key_constraint.<locals>.<genexpr>) r� r� rg r^ r* Z_implicit_generatedZcolumns_autoinc_firstr' rc )ro r r r_ rB )ro rF � visit_primary_key_constraint8 s
z(DDLCompiler.visit_primary_key_constraintc s� | j � d}|jd k r4| j �|�}|d k r4|d| 7 }t|j�d jj}|dd�� fdd�|jD ��| �||� �d�� fdd�|jD ��f 7 }|| � |�7 }|| �
|�7 }|| � |�7 }|S ) Nr| zCONSTRAINT %s r z"FOREIGN KEY(%s) REFERENCES %s (%s)z, c 3 s | ]}� � |jj�V qd S )N)r� �parentr� )rD r� )rg rB rF r� V s z;DDLCompiler.visit_foreign_key_constraint.<locals>.<genexpr>c 3 s | ]}� � |jj�V qd S )N)r� r r� )rD r� )rg rB rF r� \ s ) rg r� r^ r� r r r: r* � define_constraint_remote_table�define_constraint_match�define_constraint_cascadesrc )ro r r r_ Z remote_tablerB )rg rF � visit_foreign_key_constraintL s$
z(DDLCompiler.visit_foreign_key_constraintc C s
|� |�S )z=Format the remote table clause of a CREATE CONSTRAINT clause.)r� )ro r r: rg rB rB rF rh d s z*DDLCompiler.define_constraint_remote_tablec sh t |�dkrdS d}|jd k r6� j�|�}|d| 7 }|dd�� fdd�|D �� 7 }|� �|�7 }|S )Nr r| zCONSTRAINT %s z UNIQUE (%s)z, c 3 s | ]}� j �|j�V qd S )N)rg r� r� )rD r )ro rB rF r� q s z6DDLCompiler.visit_unique_constraint.<locals>.<genexpr>)r� r� rg r^ r* rc )ro r r r_ rB )ro rF �visit_unique_constrainti s
z#DDLCompiler.visit_unique_constraintc C sL d}|j d k r&|d| j�|j t� 7 }|jd k rH|d| j�|jt� 7 }|S )Nr| z
ON DELETE %sz
ON UPDATE %s)Zondeleterg �validate_sql_phrase� FK_ON_DELETEZonupdate� FK_ON_UPDATE)ro r r rB rB rF rj v s
z&DDLCompiler.define_constraint_cascadesc C sL d}|j d k r&|j r |d7 }n|d7 }|jd k rH|d| j�|jt� 7 }|S )Nr| z DEFERRABLEz NOT DEFERRABLEz
INITIALLY %s)r r' rg rm � FK_INITIALLY)ro r r rB rB rF rc � s
z+DDLCompiler.define_constraint_deferrabilityc C s d}|j d k r |d|j 7 }|S )Nr| z MATCH %s)�match)ro r r rB rB rF ri � s
z#DDLCompiler.define_constraint_match)N)F)N)FT)F))r� r� r� r
r rx r r~ r/ r1 r4 rA rE r> rI rJ rK rN rO rM rQ rT rU rW rX r\ r] r` rC r9 r? ra rd re rf rk rh rl rj rc ri rB rB rB rF r( �
sH
1
(
r( c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUS )V�GenericTypeCompilerc K s dS )NZFLOATrB )ro r� r� rB rB rF � visit_FLOAT� s z GenericTypeCompiler.visit_FLOATc K s dS )NZREALrB )ro r� r� rB rB rF �
visit_REAL� s z GenericTypeCompiler.visit_REALc K s< |j d krdS |jd kr&dd|j i S d|j |jd� S d S )NZNUMERICzNUMERIC(%(precision)s)� precisionz!NUMERIC(%(precision)s, %(scale)s))ru �scale)ru rv )ro r� r� rB rB rF �
visit_NUMERIC� s
z!GenericTypeCompiler.visit_NUMERICc K s< |j d krdS |jd kr&dd|j i S d|j |jd� S d S )NZDECIMALzDECIMAL(%(precision)s)ru z!DECIMAL(%(precision)s, %(scale)s))ru rv )ru rv )ro r� r� rB rB rF �
visit_DECIMAL� s
z!GenericTypeCompiler.visit_DECIMALc K s dS )NZINTEGERrB )ro r� r� rB rB rF �
visit_INTEGER� s z!GenericTypeCompiler.visit_INTEGERc K s dS )NZSMALLINTrB )ro r� r� rB rB rF �visit_SMALLINT� s z"GenericTypeCompiler.visit_SMALLINTc K s dS )NZBIGINTrB )ro r� r� rB rB rF � visit_BIGINT� s z GenericTypeCompiler.visit_BIGINTc K s dS )N� TIMESTAMPrB )ro r� r� rB rB rF �visit_TIMESTAMP� s z#GenericTypeCompiler.visit_TIMESTAMPc K s dS )NZDATETIMErB )ro r� r� rB rB rF �visit_DATETIME� s z"GenericTypeCompiler.visit_DATETIMEc K s dS )NZDATErB )ro r� r� rB rB rF �
visit_DATE� s z GenericTypeCompiler.visit_DATEc K s dS )NZTIMErB )ro r� r� rB rB rF �
visit_TIME� s z GenericTypeCompiler.visit_TIMEc K s dS )NZCLOBrB )ro r� r� rB rB rF �
visit_CLOB� s z GenericTypeCompiler.visit_CLOBc K s dS )NZNCLOBrB )ro r� r� rB rB rF � visit_NCLOB� s z GenericTypeCompiler.visit_NCLOBc C s0 |}|j r|d|j 7 }|jr,|d|j 7 }|S )Nz(%d)z
COLLATE "%s")�lengthr� )ro r� r� r rB rB rF �_render_string_type� s z'GenericTypeCompiler._render_string_typec K s | � |d�S )NZCHAR)r� )ro r� r� rB rB rF �
visit_CHAR� s z GenericTypeCompiler.visit_CHARc K s | � |d�S )NZNCHAR)r� )ro r� r� rB rB rF � visit_NCHAR� s z GenericTypeCompiler.visit_NCHARc K s | � |d�S )NZVARCHAR)r� )ro r� r� rB rB rF �
visit_VARCHAR� s z!GenericTypeCompiler.visit_VARCHARc K s | � |d�S )NZNVARCHAR)r� )ro r� r� rB rB rF �visit_NVARCHAR� s z"GenericTypeCompiler.visit_NVARCHARc K s | � |d�S )NZTEXT)r� )ro r� r� rB rB rF �
visit_TEXT� s z GenericTypeCompiler.visit_TEXTc K s dS )NZBLOBrB )ro r� r� rB rB rF �
visit_BLOB� s z GenericTypeCompiler.visit_BLOBc K s d|j rd|j pd S )NZBINARYz(%d)r| )r� )ro r� r� rB rB rF � visit_BINARY� s z GenericTypeCompiler.visit_BINARYc K s d|j rd|j pd S )NZ VARBINARYz(%d)r| )r� )ro r� r� rB rB rF �visit_VARBINARY� s z#GenericTypeCompiler.visit_VARBINARYc K s dS )NZBOOLEANrB )ro r� r� rB rB rF �
visit_BOOLEAN� s z!GenericTypeCompiler.visit_BOOLEANc K s | j |f|�S )N)r� )ro r� r� rB rB rF �visit_large_binary� s z&GenericTypeCompiler.visit_large_binaryc K s | j |f|�S )N)r� )ro r� r� rB rB rF �
visit_boolean� s z!GenericTypeCompiler.visit_booleanc K s | j |f|�S )N)r� )ro r� r� rB rB rF �
visit_time� s z GenericTypeCompiler.visit_timec K s | j |f|�S )N)r~ )ro r� r� rB rB rF �visit_datetime� s z"GenericTypeCompiler.visit_datetimec K s | j |f|�S )N)r )ro r� r� rB rB rF �
visit_date� s z GenericTypeCompiler.visit_datec K s | j |f|�S )N)r{ )ro r� r� rB rB rF �visit_big_integer
s z%GenericTypeCompiler.visit_big_integerc K s | j |f|�S )N)rz )ro r� r� rB rB rF �visit_small_integer
s z'GenericTypeCompiler.visit_small_integerc K s | j |f|�S )N)ry )ro r� r� rB rB rF �
visit_integer
s z!GenericTypeCompiler.visit_integerc K s | j |f|�S )N)rt )ro r� r� rB rB rF �
visit_real
s z GenericTypeCompiler.visit_realc K s | j |f|�S )N)rs )ro r� r� rB rB rF � visit_float
s z GenericTypeCompiler.visit_floatc K s | j |f|�S )N)rw )ro r� r� rB rB rF �
visit_numeric
s z!GenericTypeCompiler.visit_numericc K s | j |f|�S )N)r� )ro r� r� rB rB rF � visit_string
s z GenericTypeCompiler.visit_stringc K s | j |f|�S )N)r� )ro r� r� rB rB rF �
visit_unicode
s z!GenericTypeCompiler.visit_unicodec K s | j |f|�S )N)r� )ro r� r� rB rB rF �
visit_text
s z GenericTypeCompiler.visit_textc K s | j |f|�S )N)r� )ro r� r� rB rB rF �visit_unicode_text
s z&GenericTypeCompiler.visit_unicode_textc K s | j |f|�S )N)r� )ro r� r� rB rB rF �
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 r� )ro r� r� rB rB rF r "
s z GenericTypeCompiler.visit_nullc K s | j |�| j�f|�S )N)rm Z type_enginere )ro r� r� rB rB rF �visit_type_decorator)
s z(GenericTypeCompiler.visit_type_decoratorc K s |j f |�S )N)Z get_col_spec)ro r� r� rB rB rF �visit_user_defined,
s z&GenericTypeCompiler.visit_user_definedN)-r� r� r� rs rt rw rx ry rz 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 rB rB rF rr � sT rr c @ s e Zd Zdd� Zdd� ZdS )�StrSQLTypeCompilerc C s |� d�r| jS t|��d S )NZvisit_)�
startswith�_visit_unknown�AttributeError)ro r� rB rB rF � __getattr__1
s
z StrSQLTypeCompiler.__getattr__c K s d|j j S )Nz%s)r? r� )ro r� r� rB rB rF r� 7
s z!StrSQLTypeCompiler._visit_unknownN)r� r� r� r� r� rB rB rB rF r� 0
s r� c @ s e Zd ZdZeZeZeZ e
� d�Z d5dd�Z
dd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zd6dd�Zd7dd�Zdd� Zd8d d �Zd9d d �Zd:d d!�Zd;d"d#�Ze � d$�d%d&� �Z d'd(� Z d<d)d*�Z d+d,� Z!d=d-d.�Z"d>d/d0�Z#e j$d1d2� �Z%d3d4� Z&dS )?�IdentifierPreparerz@Handle quoting and case-folding of identifiers based on options.N�"TFc C sN || _ || _|p| j| _|| _| jd | _|| _|| _i | _| j jdk| _ dS )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.
r )rL rK N)
re �
initial_quote� final_quote� escape_quote�escape_to_quote� omit_schema� quote_case_sensitive_collations�_stringsr� r )ro re r� r� r� r� r� rB rB rF rq G
s zIdentifierPreparer.__init__c C s, | j �| j �}|j�| j� t�|�|_|S )N)r? �__new__�__dict__r� r �_schema_getterr� )ro rp ZpreprB rB rF rh k
s z)IdentifierPreparer._with_schema_translatec C s& |� | j| j�}| jr"|� dd�}|S )z�Escape an identifier.
Subclasses should override this to provide database-dependent
escaping behavior.
r z%%)r r� r� r )ro r� rB rB rF �_escape_identifierq
s z%IdentifierPreparer._escape_identifierc C s |� | j| j�S )z�Canonicalize an escaped identifier.
Subclasses should override this to provide database-dependent
unescaping behavior that reverses _escape_identifier.
)r r� r� )ro r� rB rB rF �_unescape_identifier}
s z'IdentifierPreparer._unescape_identifierc C s* |dk r&|� |�s&t�d||jf ��|S )z�keyword 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))rq r r� �pattern)ro r� ZregrB rB rF rm �
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� r� r� )ro r� rB rB rF �quote_identifier�
s
z#IdentifierPreparer.quote_identifierc C s< |� � }|| jkp:|d | jkp:| j�t�|�� p:||kS )z5Return True if the given identifier requires quoting.r )r= �reserved_words�illegal_initial_characters�legal_charactersrq r
� text_type)ro r� Zlc_valuerB rB rF 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� rq r
r� )ro r� rB rB rF r; �
s z1IdentifierPreparer._requires_quotes_illegal_charsc C s |dk rt �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`.
Nz�The 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().)r
�warn_deprecatedr� )ro r �forcerB rB rF r� �
s z IdentifierPreparer.quote_schemac C s� |dk rt �d� t|dd�}|dkrl|| jkr:| j| S | �|�rV| �|�| j|< n
|| j|< | j| S n|rz| �|�S |S dS )a Conditionally quote an identfier.
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`.
Nz�The 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� )ro r� r� rB rB rF r� �
s
zIdentifierPreparer.quotec C s | j r| �|�S |S d S )N)r� r� )ro Zcollation_namerB rB rF r� s
z#IdentifierPreparer.format_collationc C s> | � |j�}| �|�}| js:|r:|d k r:| �|�d | }|S )Nr� )r� r� r� r� r� )ro rD r� r� r� rB rB rF r% s
z"IdentifierPreparer.format_sequencec C s | � |p |j�S )N)r� r� )ro r� r� rB rB rF r� s z IdentifierPreparer.format_labelc C s | � |p |j�S )N)r� r� )ro r� r� rB rB rF r� ! s z IdentifierPreparer.format_aliasc C s" |p|j }| �|�r | �|�}|S )N)r� r� r� )ro Z savepointr� r� rB rB rF r $ s
z#IdentifierPreparer.format_savepointzsqlalchemy.sql.namingc C s� t |jtj�r>|�||j�}|d krDt |jtj�r6d S |j}n|j}t |tj�r�|jdkrl| j j
ph| j j }n| j j }t |�|kr�|d|d � d t
�|�dd � }n | j �|� | �|�S )Nr� r � rU ���)r� r� r Z _defer_nameZ_constraint_name_for_tabler: Z_defer_none_namer� r� re Zmax_index_name_lengthr� r� r
Zmd5_hexZvalidate_identifierr� )ro Znamingr r� Zmax_rB rB rF r^ - s"
( z$IdentifierPreparer.format_constraintc C s
| � |�S )N)r^ )ro r� rB rB rF rP K s z IdentifierPreparer.format_indexc C sF |dkr|j }| �|�}| �|�}| jsB|rB|rB| �|�d | }|S )z'Prepare a quoted table and schema name.Nr� )r� r� r� r� r� )ro r: r� r� r� r� rB rB rF r� N s
z IdentifierPreparer.format_tablec C s
| � |�S )z Prepare a quoted schema name.)r� )ro r� rB rB rF r0 [ s z IdentifierPreparer.format_schemac C sp |dkr|j }t|dd�sJ|r>| j|j||d�d | �|� S | �|�S n"|rh| j|j||d�d | S |S dS )z Prepare a quoted column name.Nr� F)r� r� r� )r� r� r� r: r� )ro r rV r� Z
table_namer� rB rB rF r� ` s
z IdentifierPreparer.format_columnc C sD | � |�}| js0|r0|r0| �|�| j|dd�fS | j|dd�fS dS )z(Format table name and schema as a tuple.F)r� N)r� r� r� r� )ro r: r� r� rB rB rF r, � s
z#IdentifierPreparer.format_table_seqc C s@ dd� | j | j| �| j�fD �\}}}t�d|||d� �}|S )Nc S s g | ]}t �|��qS rB )r� r� )rD r$ rB rB rF r1 � s z5IdentifierPreparer._r_identifiers.<locals>.<listcomp>zM(?:(?:%(initial)s((?:%(escaped)s|[^%(final)s])+)%(final)s|([^\.]+))(?=\.|$))+)�initial�finalZescaped)r� r� r� r� rr )ro r� r� Z
escaped_final�rrB rB rF �_r_identifiers� s z!IdentifierPreparer._r_identifiersc s( � j }� fdd�dd� |�|�D �D �S )z:Unpack 'schema.table.column'-like strings into components.c s g | ]}� � |��qS rB )r� )rD rH )ro rB rF r1 � s z;IdentifierPreparer.unformat_identifiers.<locals>.<listcomp>c S s g | ]\}}|p|�qS rB rB )rD �a�brB rB rF r1 � s )r� �findall)ro Z identifiersr� rB )ro rF �unformat_identifiers� s
z'IdentifierPreparer.unformat_identifiers)r� Nr� TF)N)N)T)N)N)N)TN)FNNF)T)'r� r� r� r� �RESERVED_WORDSr� �LEGAL_CHARACTERSr� �ILLEGAL_INITIAL_CHARACTERSr� r r� r� rq rh r� r� rm r� r� r; r� r� r� r% r� r� r r
r r^ rP r� r0 r� r, r r� r� rB rB rB rF r� ;
sF
%
2
r� )jr� r r� r� r| r r r r r r r r
r r
r8 r� rr �Ir� ZLEGAL_CHARACTERS_PLUS_SPACE�ranger= r� rn ro rp �UNICODEr r r� �and_�or_�add�mulr� Zdiv�mod�truediv�neg�lt�le�ne�gt�ge�eqZis_distinct_fromZisnot_distinct_fromZ concat_oprb Z notmatch_opZin_opZnotin_opZcomma_opr� r� �is_Zisnotr �existsZ distinct_op�invZany_opZall_opZdesc_opZasc_opZ
nullsfirst_opZ nullslast_opr� rO r r r r r- r. rP rQ r7 r? ZcubeZrollupZ
grouping_setsr> r r� ra Z UNION_ALLrb Z
EXCEPT_ALLrc Z
INTERSECT_ALLr �objectrd Zwith_metaclassZEnsureKWArgTyper� Z Visitabler� r� r! r( rr r� r� rB rB rB rF �<module> s�
= 7
Mini Shell