Edit File: default.cpython-37.pyc
B ��4]�� � @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ e�dejejB �Ze�dejejB �ZG dd� dej�ZG dd� de�ZG dd� dej �Z e e_ dS )a Default implementations of per-dialect sqlalchemy.engine classes. These are semi-private implementation classes which are only of importance to database dialect authors; dialects will usually use the classes here as the base class for their own corresponding classes. � N� )� interfaces)� reflection)�result� )�event)�exc)�pool)� processors)�types)�util)�compiler)� expression)�schemaz-\s*(?:UPDATE|INSERT|CREATE|DELETE|DROP|ALTER)z \s*SELECTc @ sn e Zd ZdZejZejZej Z ej Z dZ dZdZdZeZdZdZdZdZdZdZdZdZdZdZ dZ dZ dZ e �!de �"d�fde j#fde �"d �fd e �"d �fd e j#fd e j#fd e j#fde j$fg�Z%dZ&e j'r�dZ(dZ)dZ*dZ+ndZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0i Z1dZ2dZ3dZ4dZ5dZ6dZ7dZ8dZ9dZ:e �!� Z;e j<dd�d[dd��Z=e j>dd � �Z?e@d d � �ZAe@d d � �ZBeCd!d"� �ZDd#d$� ZEd%d&� ZFd\d'd(�ZGd)d*� ZHd+d,� ZId-d.� ZJd]d/d0�ZKd1d2� ZLd3d4� ZMd5d6� ZNd7d8� ZOd9d:� ZPd;d<� ZQd=d>� ZRd?d@� ZSdAdB� ZTdCdD� ZUe j>dEdF� �ZVdGdH� ZWdIdJ� ZXdKdL� ZYdMdN� ZZdOdP� Z[d^dQdR�Z\d_dSdT�Z]d`dUdV�Z^dWdX� Z_dYdZ� Z`dS )a�DefaultDialectz!Default implementation of DialectTFr �convert_unicode�forceZ pool_timeoutZecho�debugZ echo_poolZ pool_recycleZ pool_sizeZ max_overflowZpool_threadlocalN� use_encoding�defaulti' �named� )z1.3z�The :paramref:`.create_engine.convert_unicode` parameter and corresponding dialect-level parameters are deprecated, and will be removed in a future release. Modern DBAPIs support Python Unicode natively and this parameter is unnecessary.)r �utf-8�staticc K sv t | dd�s t�d| j � || _|| _d| _d | _|| _|d k rJ|| _ n | jd k r`| jj | _ n| j | _ |d k rv|| _ | j dk| _| � | �| _ | �| �| _|d k r�|| _|d k r�|| _|| _| | _| dkr�d| _n$| dkr�d| _| dk| _n t�d ��| �r | | jk�r t�d | | jf ��| | _| jd k�r<t�|�| _ n| jd k �rVt�| j�| _ t � | j�| _ t�| j�| _ d S ) NZported_sqla_06Tz2The %s dialect is not yet ported to the 0.6 formatF)Zqmark�formatZnumericr )Zdynamic� dynamic_warnr z?empty_in_strategy may be 'static', 'dynamic', or 'dynamic_warn'zQLabel length of %d is greater than this dialect's maximum identifier length of %dr )!�getattrr �warn�namer �encoding� positionalZ_ischema�dbapi� paramstyle�default_paramstyle�implicit_returning�preparerZidentifier_preparer� type_compiler�supports_right_nested_joins�supports_native_boolean�case_sensitive�empty_in_strategyZ_use_static_inZ_warn_on_empty_inr � ArgumentError�max_identifier_length� label_length�description_encodingr Z to_unicode_processor_factory�_description_decoder�codecs� getencoder�_encoderZ_decoder) �selfr r r" r! r$ r' r) r( r* r- �kwargsr r �L/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/engine/default.py�__init__� sZ zDefaultDialect.__init__c C s t �� S )N)�weakref�WeakKeyDictionary)r3 r r r5 � _type_memos s zDefaultDialect._type_memosc C s | j d | j S )N�+)r Zdriver)r3 r r r5 �dialect_description s z"DefaultDialect.dialect_descriptionc C s | j S )N)�supports_sane_rowcount)r3 r r r5 � supports_sane_rowcount_returning s z/DefaultDialect.supports_sane_rowcount_returningc C s t | dtj�S )NZ poolclass)r r Z QueuePool)�cls�urlr r r5 �get_pool_class s z DefaultDialect.get_pool_classc C s� y| � |�| _W n tk r* d | _Y nX y| �|�| _W n tk rV d | _Y nX y| �|j�| _W n tk r� d | _Y nX | �|�| _ | j d k r�| � |�r�d | _ | _ d S )N) Z_get_server_version_info�server_version_info�NotImplementedErrorZ_get_default_schema_nameZdefault_schema_nameZget_isolation_level� connection�default_isolation_level�_check_unicode_returns�returns_unicode_stringsr. �_check_unicode_descriptionr/ )r3 rC r r r5 � initialize s& zDefaultDialect.initializec C s dS )a� return a callable which sets up a newly created DBAPI connection. This is used to set dialect-wide per-connection options such as isolation modes, unicode modes, etc. If a callable is returned, it will be assembled into a pool listener that receives the direct DBAPI connection, with all wrappers removed. If None is returned, no listener will be generated. Nr )r3 r r r5 � on_connect= s zDefaultDialect.on_connectc s� t jr�jst j� nt j� �jr*��� �ni �� ���fdd��t�t� d�t � d��t�t� d�t � d��g}|r|||7 }�fdd�|D �}|� dd g�r�d S |dhkS d S ) Nc s� � t �| g�j�d��}y,�j�� }��||�� |�� }|�� W n2 tj k rv } zt � d| � dS d }~X Y nX t |d t j �S d S )N)�dialectz2Exception attempting to detect unicode returns: %rFr )r �select�compilerC �cursor�_cursor_execute�fetchone�closer Z DBAPIErrorr r � isinstance� text_type)�test� statementrM �row�de)�cast_torC � parametersr3 r r5 � check_unicodeV s z<DefaultDialect._check_unicode_returns.<locals>.check_unicodez'test plain returns'�< z'test unicode returns'c s h | ] }� |��qS r r )�.0rS )rY r r5 � <setcomp>{ s z8DefaultDialect._check_unicode_returns.<locals>.<setcomp>TFZ conditional)r �py2k�supports_unicode_statements� binary_typerR r �execute_sequence_formatr �cast�literal_column�sqltypesZVARCHARZUnicode� issuperset)r3 rC �additional_tests�tests�resultsr )rW rY rC rX r3 r5 rE K s& z%DefaultDialect._check_unicode_returnsc C st t jr| jst j}nt j}|j�� }z@|�|t� t� d�� d�g�j | d��� t |jd d t j�S |�� X d S )Nz'x'Z some_label)rJ r )r r] r^ r_ rR rC rM �executer rK rb �labelrL rQ � descriptionrP )r3 rC rW rM r r r5 rG � s z)DefaultDialect._check_unicode_descriptionc C s t �|| j�S )a) Provide a database-specific :class:`.TypeEngine` object, given the generic object which comes from the types module. This method looks for a dictionary called ``colspecs`` as a class or instance-level variable, and passes on to :func:`.types.adapt_type`. )rc Z adapt_type�colspecs)r3 �typeobjr r r5 �type_descriptor� s z DefaultDialect.type_descriptorc K s t j�|�}|j||||f|�S )N)r Z InspectorZ from_engine� reflecttable)r3 rC �tableZinclude_columnsZexclude_columnsZ resolve_fks�optsZinspr r r5 rn � s zDefaultDialect.reflecttablec K s d| j ||fd|i|��iS )z�Compatibility method, adapts the result of get_primary_keys() for those dialects which don't implement get_pk_constraint(). Zconstrained_columnsr )Zget_primary_keys)r3 �connZ table_namer �kwr r r5 �get_pk_constraint� s z DefaultDialect.get_pk_constraintc C s& t |�| jkr"t�d|| jf ��d S )Nz7Identifier '%s' exceeds maximum length of %d characters)�lenr, r ZIdentifierError)r3 �identr r r5 �validate_identifier� s z"DefaultDialect.validate_identifierc O s | j j||�S )N)r! �connect)r3 ZcargsZcparamsr r r5 rw � s zDefaultDialect.connectc C s |� � }|�|j� g |gS )N)Ztranslate_connect_args�update�query)r3 r? rp r r r5 �create_connect_args� s z"DefaultDialect.create_connect_argsc sb d|kr*|d �t �|d���fdd��}d|kr^t�|d �� � |_t �|d�� fdd��}d S )N�isolation_levelZengine_connectc s |s�� | � � d S )N)�_set_connection_isolation)rC �branch)r{ r3 r r5 � set_isolation� s zBDefaultDialect.set_engine_execution_options.<locals>.set_isolation�schema_translate_mapc s � | _ d S )N)�schema_for_object)rC r} )�getterr r5 �set_schema_translate_map� s zMDefaultDialect.set_engine_execution_options.<locals>.set_schema_translate_map)r Z listens_forr �_schema_getterr� )r3 �enginerp r~ r� r )r� r{ r3 r5 � set_engine_execution_options� s z+DefaultDialect.set_engine_execution_optionsc C s8 d|kr| � ||d � d|kr4t�|d �}||_d S )Nr{ r )r| r r� r� )r3 rC rp r� r r r5 � set_connection_execution_options� s z/DefaultDialect.set_connection_execution_optionsc C s6 |� � rt�d� | �|j|� |jjj�| j� d S )Nz�Connection is already established with a Transaction; setting isolation_level may implicitly rollback or commit the existing transaction, or have no effect until next transaction) Zin_transactionr r �set_isolation_levelrC Z_connection_recordZfinalize_callback�append�reset_isolation_level)r3 rC �levelr r r5 r| � s z(DefaultDialect._set_connection_isolationc C s d S )Nr )r3 �dbapi_connectionr r r5 �do_begin� s zDefaultDialect.do_beginc C s |� � d S )N)Zrollback)r3 r� r r r5 � do_rollback� s zDefaultDialect.do_rollbackc C s |� � d S )N)Zcommit)r3 r� r r r5 � do_commit� s zDefaultDialect.do_commitc C s |� � d S )N)rP )r3 r� r r r5 �do_close� s zDefaultDialect.do_closec C s t t�dg�j| d��S )Nr )rJ )�strr rK rL )r3 r r r5 � _dialect_specific_select_one� s z+DefaultDialect._dialect_specific_select_onec C sp d }y(|� � }z|�| j� W d |�� X W n: | jjk rf } z| �|||�rTdS � W d d }~X Y nX dS d S )NFT)rM rh r� rP r! �Error� is_disconnect)r3 r� rM �errr r r5 �do_ping s zDefaultDialect.do_pingc C s dt �ddd � S )z�Create a random two-phase transaction ID. This id will be passed to do_begin_twophase(), do_rollback_twophase(), do_commit_twophase(). Its format is unspecified. z _sa_%032xr r � )�random�randint)r3 r r r5 � create_xid s zDefaultDialect.create_xidc C s |� t�|�� d S )N)rh r ZSavepointClause)r3 rC r r r r5 � do_savepoint s zDefaultDialect.do_savepointc C s |� t�|�� d S )N)rh r ZRollbackToSavepointClause)r3 rC r r r r5 �do_rollback_to_savepoint s z'DefaultDialect.do_rollback_to_savepointc C s |� t�|�� d S )N)rh r ZReleaseSavepointClause)r3 rC r r r r5 �do_release_savepoint! s z#DefaultDialect.do_release_savepointc C s |� ||� d S )N)� executemany)r3 rM rT rX �contextr r r5 �do_executemany$ s z DefaultDialect.do_executemanyc C s |� ||� d S )N)rh )r3 rM rT rX r� r r r5 � do_execute' s zDefaultDialect.do_executec C s |� |� d S )N)rh )r3 rM rT r� r r r5 �do_execute_no_params* s z#DefaultDialect.do_execute_no_paramsc C s dS )NFr )r3 �erC rM r r r5 r� - s z DefaultDialect.is_disconnectc C s | � || j� d S )N)r� rD )r3 Z dbapi_connr r r5 r� 0 s z$DefaultDialect.reset_isolation_level) Fr NNNNTNr N)N)N)N)N)N)a�__name__� __module__� __qualname__�__doc__r Z SQLCompiler�statement_compiler� DDLCompiler� ddl_compilerZGenericTypeCompilerr& �IdentifierPreparerr% Zsupports_alterZsupports_commentsZinline_commentsZdefault_sequence_base�tupler` Zsupports_views�supports_sequences�sequences_optional�"preexecute_autoincrement_sequences�postfetch_lastrowidr$ r' Zcte_follows_insertZsupports_native_enumr( Z#non_native_boolean_check_constraint� supports_simple_order_by_label�tuple_in_valuesr � immutabledictZ bool_or_strZasintZasboolZengine_config_typesZsupports_native_decimalZpy3kr^ �supports_unicode_bindsrF r. r r, Zmax_index_name_lengthr< � supports_sane_multi_rowcountrk r# Zsupports_default_valuesZsupports_empty_insertZsupports_multivalues_insert� supports_server_side_cursorsrA Zconstruct_argumentsZrequires_name_normalizeZreflection_optionsZ dbapi_exception_translation_mapZdeprecated_paramsr6 �memoized_propertyr9 �propertyr; r= � classmethodr@ rH rI rE rG rm rn rs rv rw rz r� r� r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r5 r + s� * ? 7 r c @ s< e Zd ZejZejZejZ ej Z dZ dZ dZdZdZdZdS )�StrCompileDialectTFN)r� r� r� r ZStrSQLCompilerr� r� r� ZStrSQLTypeCompilerr& r� r% r� r� r� r$ r( r� r r r r5 r� 7 s r� c @ s� e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZe�� Zedd� �Zedd� �Zdd� Zed d � �Zed d � �Zejd d� �Zejdd� �Z ejdd� �Z ejdd� �Z ejdd� �Z ejdd� �Z dd� Z!e"dd � �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/d0� Z-e"d1d2� �Z.d3d4� Z/d5d6� Z0d7d8� Z1d9d:� Z2d;d<� Z3d=d>� Z4d?d@� Z5dPdAdB�Z6dCdD� Z7dZ8dQdFdG�Z9dHdI� Z:dJdK� Z;dLdM� Z<dNdO� Z=dS )R�DefaultExecutionContextFNc C s� | � | �}||_||_|j|_| |_}d|_|j|_|jrVt|j�|_|j� |j� |j s|t � |�|_ |�|j �d |_nt � |� |_|_ |�� |_g |_|jr�|�� g|_ni g|_|S )z8Initialize execution context for a DDLElement construct.Tr )�__new__�root_connection�_dbapi_connectionrJ �compiled�isddl�execution_options�_execution_options�dictrx r^ r rR �unicode_statementr2 rT � create_cursorrM �compiled_parametersr r` rX )r> rJ rC r� Z compiled_ddlr3 r� r r r5 � _init_ddl] s( z!DefaultExecutionContext._init_ddlc s\ | � | �}||_||_|j|_� |_� js.t�� j�|j �|_� j � j � j f|_ t�� �|_�jsv|j�|jj�|_n|j|_� j|_� j|_� j|_� j|_|s�� �� g|_n&� fdd�t |�D �|_t |�dk|_ |� � |_ |js�|js�|j�r d|_!t"� jj#�|_$t"� j%�o� jj# �|_&|jj'�s4|jj(�rN|j �rF|�)� n|�*� � j+�� j,�rj|�-� ��}n� j.�rz|jj/}g }� j.�r�x�|jD ]Z�g }x>|D ]6} | �k�r�|�0�| �| �� n|�0�| � �q�W |�0��1|�� �q�W n^�j } xT|jD ]J�| �r$t2���fdd��D ��}nt2��fdd��D ��}|�0|� �q�W ��1|�|_3|S )z6Initialize execution context for a Compiled construct.c s g | ]\}}� j ||d ��qS ))Z _group_number)�construct_params)r[ �grp�m)r� r r5 � <listcomp>� s z:DefaultExecutionContext._init_compiled.<locals>.<listcomp>r Tc 3 s< | ]4}�� |�d |�kr*�| � | �n� | fV qdS )r N)r2 )r[ �key)�compiled_paramsrJ r r r5 � <genexpr>� s z9DefaultExecutionContext._init_compiled.<locals>.<genexpr>c 3 s2 | ]*}||�kr �| � | �n� | fV qd S )Nr )r[ r� )r� r r r5 r� � s )4r� r� r� rJ r� Z can_execute�AssertionErrorr� �unionr� Z_result_columnsZ_ordered_columnsZ_textual_ordered_columns�result_column_structr rR r� r^ �encoder rT �isinsert�isupdate�isdeleteZ isplaintext�is_textr� r� � enumeratert r� r� rM �is_crud�boolZ _returning�_is_explicit_returning� returning�_is_implicit_returning�insert_prefetch�update_prefetch� _process_executemany_defaults� _process_executesingle_defaultsZ_bind_processorsZ contains_expanding_parameters�_expand_in_parametersr � positiontupr� r` r� rX ) r> rJ rC r� r� rX r3 r� �paramr� r� r )r� r� rJ r r5 �_init_compiled~ sz z&DefaultExecutionContext._init_compiledc s �j rt�d���jjr(�jjr(td��i �_�jd }|jrDg }nd}i �i }�x�|jrb�jj n�jj D �]r� �jj � }|j �rΈ �kr�|� �n�|� � �}|s�g �|� < �j� |jr�|jn|jg��� < n�t|d ttf��r<� fdd�t|d�D � �|� < �jj�rdnd d ���fd d �t|�D �� �� < n<� fd d�t|d�D � �|� < d ��fdd ��D ���� < |��� ��� �fdd ��D �� |j�r�|�dd � �D �� dd� �D ��j� < qj|jrj|�� � qjW �fdd�}t�d|�j ��_ |S )z�handle special 'expanding' parameters, IN tuples that are rendered on a per-parameter basis for an otherwise fixed SQL statement string. z7'expanding' parameters can't be used with executemany()zQ'expanding' bind parameters not supported with 'numeric' paramstyle at this time.r Nc s6 g | ].\}}t |d �D ]\}}d� ||f |f�qqS )r z%s_%s_%s)r� )r[ �i� tuple_element�j�value)r r r5 r� - s zADefaultExecutionContext._expand_in_parameters.<locals>.<listcomp>r zVALUES � z, c 3 s8 | ]0\� �d d� � ���fdd�t��D �� V qdS )z(%s)z, c 3 s6 | ].\}}�j jd �� t�� | d i V qdS )r r N)r� � bindtemplatert )r[ r� r� )r� r3 � to_updater� r r5 r� 6 s zJDefaultExecutionContext._expand_in_parameters.<locals>.<genexpr>.<genexpr>N)�joinr� )r[ )r3 r� )r� r� r5 r� 4 s z@DefaultExecutionContext._expand_in_parameters.<locals>.<genexpr>c s g | ]\}}d � |f |f�qS )z%s_%sr )r[ r� r� )r r r5 r� B s c 3 s" | ]\}}� j jd |i V qdS )r N)r� r� )r[ r� r� )r3 r r5 r� F s c 3 s& | ] \}}� �kr|�� fV qd S )Nr )r[ r� r� )r r r r5 r� L s c s s | ]\}}|V qd S )Nr )r[ r r� r r r5 r� Q s c S s g | ] \}}|�qS r r )r[ � expand_keyr� r r r5 r� S s c s � | � d� S )Nr )�group)r� )�replacement_expressionsr r5 �process_expandingX s zHDefaultExecutionContext._expand_in_parameters.<locals>.process_expandingz\[EXPANDING_(\S+)\]) r� r �InvalidRequestErrorr� r Z_numeric_bindsrB �_expanded_parametersr� r� �bindsZ expanding�popZvisit_empty_set_exprZ_expanding_in_types�typerQ r� �listr� rJ r� r� rx �extendr� �re�subrT ) r3 r� r r� r� Zto_update_setsZ parameter�valuesr� r )r r r� r3 r� r5 r� � sf z-DefaultExecutionContext._expand_in_parametersc s | � | �}||_||_|j|_d|_|j|_|sP|jjrF� �� g|_ q�i g|_ ndt |d � j�rh||_ nLt |d t �r�� j r�||_ q�� fdd�|D �p�i g|_ n� fdd�|D �|_ t |�dk|_� j s�t |tj�r�||_� �|�d |_n | |_|_|�� |_|S )z8Initialize execution context for a string SQL statement.Tr c s g | ]� � �fd d�� D ��qS )c s i | ]}� | �� |�d �qS )r )r2 )r[ �k)�drJ r r5 � <dictcomp>{ s zFDefaultExecutionContext._init_statement.<locals>.<listcomp>.<dictcomp>r )r[ )rJ )r� r5 r� { s z;DefaultExecutionContext._init_statement.<locals>.<listcomp>c s g | ]}� � |��qS r )r` )r[ �p)rJ r r5 r� � s r )r� r� r� rJ r� r� r� r r` rX rQ r� r^ rt r� r rR r� r2 rT r� rM )r> rJ rC r� rT rX r3 r )rJ r5 �_init_statement` s6 z'DefaultExecutionContext._init_statementc C s4 | � | �}||_||_|j|_|j|_|�� |_|S )z;Initialize execution context for a ColumnDefault construct.)r� r� r� rJ r� r� r� rM )r> rJ rC r� r3 r r r5 � _init_default� s z%DefaultExecutionContext._init_defaultc C s | j jS )N)r� r� )r3 r r r5 r� � s z DefaultExecutionContext.enginec C s | j jS )N)r� � postfetch)r3 r r r5 �postfetch_cols� s z&DefaultExecutionContext.postfetch_colsc C s$ | j r| jjS | jr | jjS dS d S )Nr )r� r� r� r� r� )r3 r r r5 � prefetch_cols� s z%DefaultExecutionContext.prefetch_colsc C s | j j d S )N)r� r� )r3 r r r5 �returning_cols� s z&DefaultExecutionContext.returning_colsc C s | j �dd�S )N� no_parametersF)r� �get)r3 r r r5 r � s z%DefaultExecutionContext.no_parametersc C s@ | j �d| j r | jr tjp d�}|tjkr8| �| j�S |S d S )N� autocommitF)r� r r� rT r ZPARSE_AUTOCOMMIT�should_autocommit_textr� )r3 r r r r5 �should_autocommit� s z)DefaultExecutionContext.should_autocommitc C s� | j }t|tj�r*| jjs*| j�|�d }| jjr>| j�� }ni }|j | j ||| d� | j � � d }|dk r�|� | j| j j d d �}|r�||�S |S )a Execute a string statement on the current cursor, returning a scalar result. Used to fire off sequences, default phrases, and "select lastrowid" types of statements individually or in the context of a parent INSERT or UPDATE statement. r )r� Nr )r� rQ r rR rJ r^ r2 r r` rN rM rO �_cached_result_processorrj )r3 �stmt�type_rq Zdefault_params�r�procr r r5 �_execute_scalar� s z'DefaultExecutionContext._execute_scalarc C s | j �� S )N)r� Z_branch)r3 r r r5 rC � s z"DefaultExecutionContext.connectionc C s t �|�S )N)�AUTOCOMMIT_REGEXP�match)r3 rT r r r5 r � s z.DefaultExecutionContext.should_autocommit_textc C sv | j js dS | j jrd| j�dd�o`| jr8t| jjtj �p`| j sPt| jjtj �o`| jo`t � | j�}n| j�dd�}|S )NFZstream_resultsT) rJ r� Zserver_side_cursorsr� r r� rQ rT r Z SelectableZ TextClause�SERVER_SIDE_CURSOR_REr )r3 Zuse_server_sider r r5 �_use_server_side_cursor� s z/DefaultExecutionContext._use_server_side_cursorc C s* | � � rd| _| �� S d| _| j�� S d S )NTF)r �_is_server_side�create_server_side_cursorr� rM )r3 r r r5 r� s z%DefaultExecutionContext.create_cursorc C s t � �d S )N)rB )r3 r r r5 r s z1DefaultExecutionContext.create_server_side_cursorc C s d S )Nr )r3 r r r5 �pre_exec s z DefaultExecutionContext.pre_execc C s d S )Nr )r3 r r r5 � post_exec s z!DefaultExecutionContext.post_execc C s |� | j|�S )z�Return a 'result processor' for a given type as present in cursor.description. This has a default implementation that dialects can override for context-sensitive result type handling. )r rJ )r3 r ZcolnameZcoltyper r r5 �get_result_processor s z,DefaultExecutionContext.get_result_processorc C s | j jS )a� return self.cursor.lastrowid, or equivalent, after an INSERT. This may involve calling special cursor functions, issuing a new SELECT on the cursor (or a new one), or returning a stored value that was calculated within post_exec(). This function will only be called for dialects which support "implicit" primary key generation, keep preexecute_autoincrement_sequences set to False, and when no explicit id value was bound to the statement. The function is called once, directly after post_exec() and before the transaction is committed or ResultProxy is generated. If the post_exec() method assigns a value to `self._lastrowid`, the value is used in place of calling get_lastrowid(). Note that this method is *not* equivalent to the ``lastrowid`` method on ``ResultProxy``, which is a direct proxy to the DBAPI ``lastrowid`` accessor in all cases. )rM � lastrowid)r3 r r r5 � get_lastrowid' s z%DefaultExecutionContext.get_lastrowidc C s d S )Nr )r3 r� r r r5 �handle_dbapi_exceptionC s z.DefaultExecutionContext.handle_dbapi_exceptionc C s | j rt�| �S t�| �S d S )N)r r ZBufferedRowResultProxyZ ResultProxy)r3 r r r5 �get_result_proxyF s z(DefaultExecutionContext.get_result_proxyc C s | j jS )N)rM �rowcount)r3 r r r5 r L s z DefaultExecutionContext.rowcountc C s | j jS )N)rJ r< )r3 r r r5 r< P s z.DefaultExecutionContext.supports_sane_rowcountc C s | j jS )N)rJ r� )r3 r r r5 r� S s z4DefaultExecutionContext.supports_sane_multi_rowcountc C s� | j r:| js:| js,| jjs,| jjr,| �� n| js:| �� | � � }| j r�| jrv|� � }|| _ | � |� |� � d |_q�| js�|� � d |_nB| jr�| jr�|� � }|| _ |� � d |_n|jd kr�|j |� � |S )N)r� r� r� r� �inlinerJ r� � _setup_ins_pk_from_lastrowid�_setup_ins_pk_from_emptyr rO �returned_defaults�%_setup_ins_pk_from_implicit_returningZ _soft_closeZ _metadatar� r� r )r3 r rU r r r5 �_setup_crud_result_proxyV s6 z0DefaultExecutionContext._setup_crud_result_proxyc s� | j jd �| j jj}| jd �| �� ��d k r||j� � d k r^� j�| j d �}|d k r^|���� ���fdd�|j D �| _ n��fdd�|j D �| _ d S )Nr r c s( g | ] }|� kr�n�� �|�d ��qS )N)r )r[ �c)� autoinc_colr� � key_getterr r r5 r� � s zHDefaultExecutionContext._setup_ins_pk_from_lastrowid.<locals>.<listcomp>c s g | ]}� � �|�d ��qS )N)r )r[ r# )r� r% r r5 r� � s ) r� � _key_getters_for_crud_columnrT ro r� r Z_autoincrement_columnr� r rJ � primary_key�inserted_primary_key)r3 ro r r )r$ r� r% r r5 r } s z4DefaultExecutionContext._setup_ins_pk_from_lastrowidc s< | j jd �| j jj}| jd � � �fdd�|jD �| _d S )Nr r c s g | ]}� � �|�d ��qS )N)r )r[ r# )r� r% r r5 r� � s zDDefaultExecutionContext._setup_ins_pk_from_empty.<locals>.<listcomp>)r� r& rT ro r� r' r( )r3 ro r )r� r% r5 r � s z0DefaultExecutionContext._setup_ins_pk_from_emptyc s\ �d krd | _ d S | jjd �| jjj}| jd � �fdd�� �fdd�|jD �D �| _ d S )Nr r c s$ g | ] \}}|d kr � | n|�qS )Nr )r[ �colr� )rU r r5 r� � s zQDefaultExecutionContext._setup_ins_pk_from_implicit_returning.<locals>.<listcomp>c s g | ]}|� � �|�d �f�qS )N)r )r[ r) )r� r% r r5 r� � s )r( r� r& rT ro r� r' )r3 rU ro r )r� r% rU r5 r! � s z=DefaultExecutionContext._setup_ins_pk_from_implicit_returningc C s | j s | jot| jj�S )N)r� r� r� r� r )r3 r r r5 �lastrow_has_defaults� s z,DefaultExecutionContext.lastrow_has_defaultsc sT t | jd�sdS i }xv| jjD ]j}|j�| j�}t|�}|�| jj�� � dk r�|rb� |kr�||kr�|rv� |ksv||kr�� ||< q d||< q W | jjr�| jj � || j | j | j | � | jj�rng }xb| jjD ]V} | jj| }|�|d�� � dk r�| | jk�r|�� gt| j| � � q�|�� � q�W y| j j|� W n8 tk �rj } z| j�| ddd| � W dd} ~ X Y nX n�i } x�| jj�� D ]�\}} |�|d�� � dk �r�|�r�|�| | �} | jj�s�| j� | �d } | | jk�r�| � � fdd�| j| D �� n� | | < �q�W y| j jf | � W n8 tk �rN } z| j�| ddd| � W dd} ~ X Y nX dS )a& Given a cursor and ClauseParameters, call the appropriate style of ``setinputsizes()`` on the cursor, using DB-API types from the bind parameter's ``TypeEngine`` objects. This method only called by those dialects which require it, currently cx_oracle. � bind_namesNr c 3 s | ]}|� fV qd S )Nr )r[ r� )�dbtyper r5 r� s z:DefaultExecutionContext.set_input_sizes.<locals>.<genexpr>) �hasattrr� r+ r� Z_unwrapped_dialect_implrJ Zget_dbapi_typer! Z _has_events�dispatchZdo_setinputsizesrM rT rX r r� r� r r� r� rt r� Z setinputsizes� BaseExceptionr� Z_handle_dbapi_exception�itemsr� r2 rx ) r3 � translateZ include_typesZ exclude_typesZ inputsizesZ bindparamZ dialect_implZdialect_impl_clsZpositional_inputsizesr� r� Zkeyword_inputsizesr )r, r5 �set_input_sizes� sh " z'DefaultExecutionContext.set_input_sizesc C s� |j r| �||�S |jr(|| _|�| �S |jrv| j}|jsJt� |j|�}n|j}t� |g�j |d�}|� |di �� � S |jS d S )N)�bindr )� is_sequenceZ fire_sequenceZ is_callable�current_column�argZis_clause_elementrC Z _arg_is_typedr Z type_coercerK rL Z_execute_compiledZscalar)r3 �columnr r rq Z default_argr# r r r5 � _exec_default s z%DefaultExecutionContext._exec_defaultTc s� y| j �| j}W n tk r. t�d��Y nX |r�| jr�| jjjr�|j rh|j d � |j j �|j i}n|j �|j i}d� | jjj d �� }|�� �fdd�|D �� |S �S dS )aF Return a dictionary of parameters applied to the current row. This method can only be used in the context of a user-defined default generation function, e.g. as described at :ref:`context_default_functions`. When invoked, a dictionary is returned which includes entries for each column/value pair that is part of the INSERT or UPDATE statement. The keys of the dictionary will be the key value of each :class:`.Column`, which is usually synonymous with the name. :param isolate_multiinsert_groups=True: indicates that multi-valued INSERT constructs created using :meth:`.Insert.values` should be handled by returning only the subset of parameters that are local to the current column default invocation. When ``False``, the raw parameters of the statement are returned including the naming convention used in the case of multi-valued INSERT. .. versionadded:: 1.2 added :meth:`.DefaultExecutionContext.get_current_parameters` which provides more functionality over the existing :attr:`.DefaultExecutionContext.current_parameters` attribute. .. seealso:: :attr:`.DefaultExecutionContext.current_parameters` :ref:`context_default_functions` zdget_current_parameters() can only be invoked in the context of a Python side column default functionr r c 3 s" | ]}|�d |� f fV qdS )z%s_m%dNr )r[ r� )�indexrX r r5 r� n s zADefaultExecutionContext.get_current_parameters.<locals>.<genexpr>N)�current_parametersr5 �AttributeErrorr r� r� r� rT Z_has_multi_parametersZ_is_multiparam_columnr9 �originalr� rX �keysrx )r3 Zisolate_multiinsert_groupsr7 r� r= r )r9 rX r5 �get_current_parameters: s& z.DefaultExecutionContext.get_current_parametersc C s$ |j d krd S | �||j |j�S d S )N)r r8 r� )r3 r7 r r r5 �get_insert_defaultt s z*DefaultExecutionContext.get_insert_defaultc C s$ |j d krd S | �||j |j�S d S )N)�onupdater8 r� )r3 r7 r r r5 �get_update_defaultz s z*DefaultExecutionContext.get_update_defaultc C s | j jd }i }| j j}| j j}x&|D ] }|jr&|jjr&|jj||< q&W x&|D ] }|jrN|jjrN|jj||< qNW x�| jD ]�}|| _ x<|D ]4}||kr�|| }n | � |�}|d k r�||||�< q�W x<|D ]4}||kr�|| }n | � |�}|d k r�||||�< q�W qxW | ` d S )Nr ) r� r& r� r� r � is_scalarr6 r@ r� r: r? rA )r3 r% Zscalar_defaultsr� r� r# r� �valr r r5 r� � s2 z5DefaultExecutionContext._process_executemany_defaultsc C s� | j jd }| jd | _}xN| j jD ]B}|jrJ|jjsJ|jjrJ|jj}n | � |�}|d k r&||||�< q&W x.| j j D ]"}| � |�}|d k rv||||�< qvW | `d S )Nr r ) r� r& r� r: r� r r4 rB r6 r? r� rA )r3 r% r� r# rC r r r5 r� � s z7DefaultExecutionContext._process_executesingle_defaults)NNN)T)>r� r� r� r� r� r� r� r� r� r� r� rT r� r r� r� Z_translate_colnamer r� r� r� r� r� r� r� r� r� r� r r r r r r r� rC r r r� r r r r r r r r r< r� r" r r r! r* r2 r8 r: r> r? rA r� r� r r r r5 r� H sl ! tn 0 ! ' W :%r� ) r� r0 r� r� r7 r� r r r r r r r r rc r Zsqlr r r rL �I�UNICODEr r ZDialectr r� ZExecutionContextr� Zexecution_ctx_clsr r r r5 �<module> sB z