Edit File: 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_columns�checkr )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' rC )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' rC r )r@ )rE 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' rC r )r@ )rE 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 )rE 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 rC )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/ �selectZ_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 )rI Z_parameter_orderingr2 r' r r( �_append_param_parameterr � primary_key�_autoincrement_column�!_append_param_insert_pk_returning�_append_param_insert_pkrK � _append_param_insert_hasdefault�server_defaultr% rL r" �nullable�"_warn_pk_with_no_anticipated_valuer �_append_param_update)r r r& r8 r r9 r; r: r rQ rR rS rT Zparameter_orderingrU r rV r )r rW 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 rC )rM r � _is_literalr r r) � isinstancer> r? Z_isnullZ_clonerY r% rL rB � self_groupr" ) r r r rV r& r9 rR rS r: r rA r r r rX x s. rX 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)rK � is_sequence�dialect�supports_sequences�optional�sequences_optionalrB rL r% �is_clause_element�argrd �"_create_insert_prefetch_bind_paramr' rZ r^ r_ r` )r r r r: r �procr r r r[ � s" r[ c C s" t | |d ||d�}| j�|� |S )N)rB rC )r r# rL )r r rB rC �paramr r r rl � s rl c C s" t | |d ||d�}| j�|� |S )N)rB rC )r r$ rL )r r rB rC rn r r r �"_create_update_prefetch_bind_param� s ro c @ s e Zd ZdZdd� Zdd� ZdS )�_multiparam_columnTc C s4 || _ d|j|d f | _|| _|j| _|j| _d S )Nz%s_m%dr )�indexr �originalrK r? )�selfrr rq r r r �__init__� s z_multiparam_column.__init__c C s" t |t�o |j| jko |j| jkS )N)rc rp r rr )rs �otherr r r �__eq__� s z_multiparam_column.__eq__N)�__name__� __module__� __qualname__Z_is_multiparam_columnrt rv r r r r rp � s rp 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) rK r r5 rj rB rk rd rp rc r ZInsertrl ro )r r r rq r r@ r r r � _process_multiparam_default_bind� s rz 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) rK re rf rg r' rZ Z"preexecute_autoincrement_sequencesrL rl r^ r_ r` )r r r r: r r r r r\ � s r\ 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)rK re rf rg rh ri rB rL r% rY r" rj rk rd rl )r r r rS r: r rm 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)rB ) rK re rf rg rh ri rL Z next_valuerj rk rd rl )r r r r: r rm r r r rN ? s rN 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) �onupdatere rj rL rB rk rd r% r" ro �server_onupdate�_return_defaults)r r r rS r: r r r r ra P s ra 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 rn r r r r x s z)_get_multitable_params.<locals>.<genexpr>)r rC )rC )r+ �itemsr2 r- r �addr rb r r r" rL rB rd r{ re rj rk ro r| ) r r r r; r9 r8 r: r Znormalized_paramsZaffected_tables�tr rA 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 rC z%s_m%d) � enumerater& r r rb r rB rd rz rL ) r r r: r Zvalues_0�i�row� extensionr@ rn r � 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 rb rB r> r? rd rL ) 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 �inlineZ _returningr) rf rR r' r} r r2 r rT )r r rQ rR rS rT r r r rI � s. rI 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' �fullnamerC r rY r �warn)r �msgr r r r` s r` )TFN)TN)TN)"�__doc__rF � r r r r �symbolr r r r r r r r* r0 r1 rX r[ rl ro Z ColumnElementrp rz r\ r] rN ra r. r7 r, rI r` r r r r �<module> s@ u $0\.) % @ +