File: //opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/sql/__pycache__/crud.cpython-37.pyc
B
��4]�d � @ s" d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ e�d d
�Ze�d �Z e�d �Z
e�d
�Z dd� Z dd� Z
d:dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd;d d!�Zd<d"d#�ZG d$d%� d%ej�Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!dS )=ziFunctions used by compiler.py to determine the parameters rendered
within INSERT and UPDATE statements.
� N� )�dml)�elements� )�exc)�util�REQUIREDa1
Placeholder for the value within a :class:`.BindParameter`
which is required to be present when the statement is passed
to :meth:`.Connection.execute`.
This symbol is typically used when a :func:`.expression.insert`
or :func:`.expression.update` statement is compiled without parameter
values present.
�ISINSERT�ISUPDATE�ISDELETEc K s� | j }| j}| j}|p*|p*|p*t| j�dk}|tkrBd| _d| _ n6|tkrXd| _d| _ n |tkrl|sxd| _n dsxtd��z |ttfkr�t | |f|�S W d |r�|| _ || _|| _X d S )Nr FTz(ISINSERT, ISUPDATE, or ISDELETE expected)
�isinsert�isupdateZisdelete�len�stackr r
r �AssertionError�_get_crud_params)�compiler�stmtZlocal_stmt_type�kwZrestore_isinsertZrestore_isupdateZrestore_isdeleteZshould_restore� r �F/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/sql/crud.py�_setup_crud_params' s, r c
s� g �_ g �_g �_g �_�jdkrB|jdkrB�fdd�|jjD �S |jrT|jd �n|j�t �|�\� }}�jdkrzi }nt
� �fdd��jD ��}g }�dk r�t �|�� ||� i }�j r�|j
r܈r�t�|�|||||� �j�r|j�rt�|||� ||||� nt�|||� ||||� |�rp��rpt|��� fdd��D ���|�}|�rpt�dd �d
d� |D �� ��|j�r�t�|||�}|S ) aP create a set of tuples representing column/string pairs for use
in an INSERT or UPDATE statement.
Also generates the Compiled object's postfetch, prefetch, and
returning column collections, used for default handling and ultimately
populating the ResultProxy's prefetch_cols() and postfetch_cols()
collections.
Nc s g | ]}|t � |d dd�f�qS )NT)�required)�_create_bind_param)�.0�c)r r r �
<listcomp>Z s z$_get_crud_params.<locals>.<listcomp>r c 3 s& | ] }�r|�kr� |�t fV qd S )N)r )r �key)�_column_as_key�stmt_parametersr r � <genexpr>r s z#_get_crud_params.<locals>.<genexpr>c 3 s | ]}� |�V qd S )Nr )r �k)r r r r � s zUnconsumed column names: %sz, c s s | ]}d | V qdS )z%sNr )r r r r r r � s )� postfetch�insert_prefetch�update_prefetch� returningZ column_keys�
parameters�table�columns�_has_multi_parameters� _key_getters_for_crud_column�dict�_get_stmt_parameters_paramsr
� _extra_froms�_get_multitable_paramsr � select_names� _scan_insert_from_select_cols�
_scan_cols�set� intersection�
differencer � CompileError�join� _extend_values_for_multiparams) r r r �_getattr_col_key�_col_bind_namer& �values�
check_columnsZcheckr )r r r r r F s~
r TFc K s> |d kr|j }tj|||j|d�}d|_|r:|j| f|�}|S )N)�type_r T)r r �
BindParameter�typeZ_is_crudZ_compiler_dispatch)r �col�value�processr �namer Z bindparamr r r r � s r c sZ | j r<|jr<t|j�� � fdd�}� fdd�}� fdd�}ntj}t�d� }}|||fS )Nc s2 t �| �}t| d�r*| j� kr*| jj|fS |S d S )Nr' )r r �hasattrr' rB )r Zstr_key)�_etr r r � s
z4_key_getters_for_crud_column.<locals>._column_as_keyc s" | j � kr| j j| jfS | jS d S )N)r' rB r )r? )rD r r r8 � s
z6_key_getters_for_crud_column.<locals>._getattr_col_keyc s& | j � kr d| j j| jf S | jS d S )Nz%s_%s)r' rB r )r? )rD r r r9 � s
z4_key_getters_for_crud_column.<locals>._col_bind_namer )r
r- r2 r r �operator�
attrgetter)r r r r8 r9 r )rD r r* � s
r* c s� t | ��\} }
} } � �fdd��jD �}
�j| _g }�jrlt|
�}x(�jjD ] }||krL|jrL|
� |� qLW xN|
D ]F}||�}||kr�||kr�|�
|� |� |d f� qrt | �|||� qrW |r�|� |� | j�
� | _t| jj�tdd� |D �� | j_d S )Nc s g | ]}�j j� |� �qS r )r' r )r rB )r r r r r � s z1_scan_insert_from_select_cols.<locals>.<listcomp>c s s | ]\}}|V qd S )Nr )r r? �exprr r r r s z0_scan_insert_from_select_cols.<locals>.<genexpr>)�_get_returning_modifiersr/ ZselectZ_insert_from_selectZ#include_insert_from_select_defaultsr2 r' r( �default�append�pop�&_append_param_insert_select_hasdefault�extendZ _generate�tupleZ _raw_columns)r r r& r8 r r9 r; r: r �need_pks�implicit_returning�implicit_return_defaults�postfetch_lastrowid�colsZadd_select_colsZcol_setr? r �col_keyr )r r r r0 � s,
r0 c s� t | ��\} }
} } �jr^� fdd��jD �}
t|
���fdd�|
D ��fdd��jjD � }n�jj}�xN|D �]D}||�}||kr�||kr�t| �|||||
| ||�
qn| j�r�|jr�| r�|
s�| r�|�jj k r�|
r�t
| �|||� nt | �|||� n�|j d k �rt
| �|| ||� n�|jd k �rV| �r@|| k�r@| j�|� n|j�s�| j�|� nD| �rt|| k�rt| j�|� n&|j�r�|�jj k �r�|j�s�t|� qn| jrnt| �|| ||� qnW d S )Nc s g | ] }� |��qS r r )r r )r r r r 1 s z _scan_cols.<locals>.<listcomp>c s g | ]}� j j| �qS r )r' r )r r )r r r r 4 s c s g | ]}|j � kr|�qS r )r )r r )� ordered_keysr r r 5 s )rH Z_parameter_orderingr2 r' r r( �_append_param_parameterr � primary_key�_autoincrement_column�!_append_param_insert_pk_returning�_append_param_insert_pkrI � _append_param_insert_hasdefault�server_defaultr% rJ r" �nullable�"_warn_pk_with_no_anticipated_valuer
�_append_param_update)r r r& r8 r r9 r; r: r rO rP rQ rR Zparameter_orderingrS r rT r )r rU r r r1 s`
r1 c
C s� |� |�}
t�|
�rJt| ||
f|
tk|js2||�n
d||� d�| ��}
n�t|
tj�rn|
jj rn|
�
� }
|j|
_|j r�|r�| j �
|� | j|
�� f| �}
nP|r�||kr�| j �
|� | j|
�� f| �}
n$|j s�| j�
|� | j|
�� f| �}
|�
||
f� d S )Nz%s_m0)r rB )rK r � _is_literalr r r) �
isinstancer= r> Z_isnullZ_clonerW r% rJ rA �
self_groupr" ) r r r rT r& r9 rP rQ r: r r@ r r r rV x s.
rV c C s� |j dk r�|j jrV| jjrH|j jr*| jjsH| j|j f|�}|�||f� | j�|� q�|j j r�|�|| j|j j
� � f|�f� | j�|� q�|�|t | |�f� n2||j
jks�|jdk r�| j�|� n|js�t|� dS )a� Create a primary key expression in the INSERT statement and
possibly a RETURNING clause for it.
If the column has a Python-side default, we will create a bound
parameter for it and "pre-execute" the Python function. If
the column has a SQL expression default, or is a sequence,
we will add it directly into the INSERT statement and add a
RETURNING element to get the new value. If the column has a
server side default or is marked as the "autoincrement" column,
we will add a RETRUNING element to get at the value.
If all the above tests fail, that indicates a primary key column with no
noted default generation capabilities that has no parameter passed;
raise an exception.
N)rI � is_sequence�dialect�supports_sequences�optional�sequences_optionalrA rJ r% �is_clause_element�argrb �"_create_insert_prefetch_bind_paramr' rX r\ r] r^ )r r r r: r �procr r r rY � s"
rY c C s" t | |d ||d�}| j�|� |S )N)rA rB )r r# rJ )r r rA rB �paramr r r rj � s rj c C s" t | |d ||d�}| j�|� |S )N)rA rB )r r$ rJ )r r rA rB rl r r r �"_create_update_prefetch_bind_param� s rm c @ s e Zd ZdZdd� Zdd� ZdS )�_multiparam_columnTc C s4 || _ d|j|d f | _|| _|j| _|j| _d S )Nz%s_m%dr )�indexr �originalrI r> )�selfrp ro r r r �__init__� s
z_multiparam_column.__init__c C s" t |t�o |j| jko |j| jkS )N)ra rn r rp )rq �otherr r r �__eq__� s
z_multiparam_column.__eq__N)�__name__�
__module__� __qualname__Z_is_multiparam_columnrr rt r r r r rn � s rn c C sb |j st�d| ��nH|j jr4| j|j j�� f|�S t||�}t|t j
�rTt | |�S t | |�S d S )Nz�INSERT value for column %s is explicitly rendered as a boundparameter in the VALUES clause; a Python-side value or SQL expression is required)
rI r r5 rh rA ri rb rn ra r ZInsertrj rm )r r r ro r r? r r r � _process_multiparam_default_bind� s
rx c C sr |j dk r|j jr6| jjs6||jjkrL| jjs6| jjrL|�|t| |�f� n"|j dkrn|j dkrn|j
snt |� dS )a� Create a bound parameter in the INSERT statement to receive a
'prefetched' default value.
The 'prefetched' value indicates that we are to invoke a Python-side
default function or expliclt SQL expression before the INSERT statement
proceeds, so that we have a primary key value available.
if the column has no noted default generation capabilities, it has
no value passed in either; raise an exception.
N) rI rc rd re r' rX Z"preexecute_autoincrement_sequencesrJ rj r\ r] r^ )r r r r: r r r r rZ � s
rZ c C s� |j jrl| jjr�|j jr | jjs�| j|j f|�}|�||f� |rX||krX| j�|� q�|j s�| j
�|� nn|j j r�| j|j j �
� f|�}|�||f� |r�||kr�| j�|� q�|j s�| j
�|� n|�|t| |�f� d S )N)rI rc rd re rf rg rA rJ r% rW r" rh ri rb rj )r r r rQ r: r rk r r r r[ $ s" r[ c C sz |j jr:| jjrv|j jr | jjsv|j }|�||�� f� n<|j jr^|j j �
� }|�||f� n|�|t | |dd�f� d S )NF)rA ) rI rc rd re rf rg rJ Z
next_valuerh ri rb rj )r r r r: r rk r r r rL ? s rL c C s� |j d k rx|j jsx|j jrb|�|| j|j j�� f|�f� |rT||krT| j�|� qv| j�|� q�|�|t | |�f� nT|j
d k r�|r�||kr�| j�|� q�| j�|� n"|r�|j dk r�||kr�| j�|� d S )NT) �onupdaterc rh rJ rA ri rb r% r" rm �server_onupdate�_return_defaults)r r r rQ r: r r r r r_ P s
r_ c
C sd t dd� |�� D ��}t� } x�|jD ]�}
x�|
jD ]|} | |kr0| �|
� | ||| �< || } t�| �r�t| | | | t k|| �d�} n | j
� | � | j | �
� f|�} |� | | f� q0W q$W x�| D ]�}
x�|
jD ]�} | |kr�q�q�| jd k �rB| jj�sB| jj�r$|� | | j | jj�
� f|�f� | j
� | � n |� | t| | || �d�f� q�| jd k r�| j
� | � q�W q�W d S )Nc s s | ]\}}t �|�|fV qd S )N)r Z_clause_element_as_expr)r r rl r r r r x s z)_get_multitable_params.<locals>.<genexpr>)r rB )rB )r+ �itemsr2 r- r �addr r` r r r" rJ rA rb ry rc rh ri rm rz )
r r r r; r9 r8 r: r Znormalized_paramsZaffected_tables�tr r@ r r r r. l sJ
r. c C s� |}|g}x�t |jdd � �D ]�\}}g }x�|D ]�\}} ||ksJ|j|kr�||krV|n|j}
t�||
�r�t| |||
fdd|j|d f i|��} q�| j||
�� f|�} nt| ||||�} |� || f� q0W |� |� q W |S )Nr rB z%s_m%d)
� enumerater& r r r` r rA rb rx rJ ) r r r: r Zvalues_0�i�row� extensionr? rl r Z new_paramr r r r7 � s(
r7 c C s� xz|� � D ]n\}}||�}|d k r0|�||� q
t�|�rX| jtjd ||jd�f|�}n| j|�� f|�}|�||f� q
W d S )N)r< ) r| �
setdefaultr r` rA r= r> rb rJ ) r r& r r r: r r! �vZcolkeyr r r r, � s
r, c C s� | j o | j o |j o |j }|o0| jjo0|jj}| j rD|o@|j}n"| jrb| jjo^|jjo^|j}nd}|r�|jdkr�t |jj
�}n
t |j�}|o�| jj }||||fS )NFT) r ZinlineZ
_returningr) rd rP r' r{ r
r2 r rR )r r rO rP rQ rR r r r rH � s.
rH c C s> d| j j| j| j jf }t| j j�dkr0|d7 }t�|� d S )Na Column '%s.%s' is marked as a member of the primary key for table '%s', but has no Python-side or server-side default generator indicated, nor does it indicate 'autoincrement=True' or 'nullable=True', and no explicit value is passed. Primary key columns typically may not store NULL.r a0 Note that as of SQLAlchemy 1.1, 'autoincrement=True' must be indicated explicitly for composite (e.g. multicolumn) primary keys if AUTO_INCREMENT/SERIAL/IDENTITY behavior is expected for one of the columns in the primary key. CREATE TABLE statements are impacted by this change as well on most backends.)r' �fullnamerB r rW r �warn)r �msgr r r r^ s r^ )TFN)TN)TN)"�__doc__rE � r r r r Zsymbolr r r
r r r r r* r0 r1 rV rY rj rm Z
ColumnElementrn rx rZ r[ rL r_ r. r7 r, rH r^ r r r r �<module> s@
u
$0\.)
% @ +