Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
local
/
lib
/
python3.10
/
dist-packages
/
sqlalchemy
/
engine
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: default.cpython-310.pyc
o [�h�� � @ 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 ddlmZ e�dejejB �Ze�dejejB �ZG dd� dej�Z G dd� de �Z G 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)�schema)� quoted_namez-\s*(?:UPDATE|INSERT|CREATE|DELETE|DROP|ALTER)z \s*SELECTc @ s� 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'rodZ(dZ)dZ*dZ+ndZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1i Z2dZ3dZ4dZ5dZ6dZ7dZ8dZ9 dZ:dZ;e �!� Z< e j=dd� dadd��Z>e j?dd � �Z@eAd d � �ZBeAd d � �ZCeDd!d"� �ZEd#d$� ZFd%d&� ZGd'd(� ZHdbd)d*�ZId+d,� ZJd-d.� ZKd/d0� ZLdbd1d2�ZMd3d4� ZNd5d6� ZOd7d8� ZPd9d:� ZQd;d<� ZRd=d>� ZSd?d@� ZTdAdB� ZUdCdD� ZVdEdF� ZWe j?dGdH� �ZXdIdJ� ZYdKdL� ZZdMdN� Z[dOdP� Z\dQdR� Z]dbdSdT�Z^dbdUdV�Z_dbdWdX�Z`dYdZ� Zad[d\� Zbd]d^� Zcd_d`� ZddS )c�DefaultDialectz!Default implementation of DialectTFr �convert_unicode�force� pool_timeout�echo�debug� echo_pool� pool_recycle� pool_size� max_overflow�pool_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 s` t | dd�st�d| j � || _|| _d| _d | _|| _|d ur%|| _ n| jd ur0| jj | _ n| j | _ |d ur;|| _ | j dv | _| � | �| _ | �| �| _|d urT|| _|d ur[|| _|| _| | _| dkrid| _n| dv rvd| _| dk| _nt�d ��| | _| jr�| j| _| | _| jd kr�t� |�| _ n | jd ur�t� | j�| _ t � | j�| _ t� | j�| _!d S ) N�ported_sqla_06Tz2The %s dialect is not yet ported to the 0.6 formatF)�qmark�format�numericr! )�dynamic� dynamic_warnr' z?empty_in_strategy may be 'static', 'dynamic', or 'dynamic_warn'r )"�getattrr �warn�namer �encoding� positional�_ischema�dbapi� paramstyle�default_paramstyle�implicit_returning�preparer�identifier_preparer� type_compiler�supports_right_nested_joins�supports_native_boolean�case_sensitive�empty_in_strategy�_use_static_in�_warn_on_empty_inr � ArgumentError�#_user_defined_max_identifier_length�max_identifier_length� label_length�description_encodingr � to_unicode_processor_factory�_description_decoder�codecs� getencoder�_encoder�_decoder) �selfr r+ r/ r. r1 r5 r7 r6 r8 r= r> �kwargsr r �D/usr/local/lib/python3.10/dist-packages/sqlalchemy/engine/default.py�__init__� sh �� �� �� ��zDefaultDialect.__init__c C s t �� S �N)�weakref�WeakKeyDictionary�rF r r rH � _type_memos � zDefaultDialect._type_memosc C s | j d | j S )N�+)r* �driverrM r r rH �dialect_description s z"DefaultDialect.dialect_descriptionc C s | j S )z�True if this dialect supports sane rowcount even if RETURNING is in use. For dialects that don't support RETURNING, this is synomous with supports_sane_rowcount. )�supports_sane_rowcountrM r r rH � supports_sane_rowcount_returning s z/DefaultDialect.supports_sane_rowcount_returningc C s t | dtj�S )N� poolclass)r( r � QueuePool)�cls�urlr r rH �get_pool_class% � z DefaultDialect.get_pool_classc C s� z| � |�| _W n ty d | _Y nw z| �|�| _W n ty) d | _Y nw z | �|j�| _W n ty? d | _Y nw | �|�| _ | j d urV| � |�rVd | _ | _ | j sc| �|�}|rc|| _| jrw| j| jkryt�d| j| jf ��d S d S )NzQLabel length of %d is greater than this dialect's maximum identifier length of %d)�_get_server_version_info�server_version_info�NotImplementedError�_get_default_schema_name�default_schema_name�get_isolation_level� connection�default_isolation_level�_check_unicode_returns�returns_unicode_stringsr? �_check_unicode_descriptionrA r< � _check_max_identifier_lengthr= r> r r; )rF ra �max_ident_lengthr r rH � initialize) sP � � � � � � � � ���zDefaultDialect.initializec C � 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 rM r r rH � on_connectU s zDefaultDialect.on_connectc C ri )z�Perform a connection / server version specific check to determine the max_identifier_length. If the dialect's class level max_identifier_length should be used, can return None. .. versionadded:: 1.3.9 Nr )rF ra r r rH rf c s z+DefaultDialect._check_max_identifier_lengthc s� t jr �js t j� nt j� �jr��� �ni �� ���fdd��t�t� d�t � d��t�t� d�t � d��g}|r>||7 }�fdd�|D �}|� dd g�rPd S |dhkS ) Nc s� � t �| g�j�d��}z�j�� }��||�� |�� }|�� W n tj y= } zt � d| � W Y d }~dS d }~ww t |d t j �S )N��dialectz2Exception attempting to detect unicode returns: %rFr )r �select�compilera �cursor�_cursor_execute�fetchone�closer � DBAPIErrorr r) � isinstance� text_type)�test� statementro �row�de)�cast_tora � parametersrF r rH � check_unicodez s$ � ���� z<DefaultDialect._check_unicode_returns.<locals>.check_unicodez'test plain returns'�< z'test unicode returns'c s h | ]}� |��qS r r )�.0rv )r| r rH � <setcomp>� s z8DefaultDialect._check_unicode_returns.<locals>.<setcomp>TF� conditional)r �py2k�supports_unicode_statements� binary_typeru r, �execute_sequence_formatr �cast�literal_column�sqltypes�VARCHAR�Unicode� issuperset)rF ra �additional_tests�tests�resultsr )rz r| ra r{ rF rH rc o s, ��� z%DefaultDialect._check_unicode_returnsc C sz t jr | js t j}nt j}|j�� }z%|�|t� t� d�� d�g�j | d��� t |jd d t j�W |�� S |�� w )Nz'x'� some_labelrk r )r r� r� r� ru ra ro �executer rm r� �labelrn rt � descriptionrr )rF ra rz ro r r rH re � 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`. )r� � adapt_type�colspecs)rF �typeobjr r rH �type_descriptor� s z DefaultDialect.type_descriptorc K s$ t j�|�}|j||||fi |��S rJ )r � Inspector� from_engine� reflecttable)rF ra �table�include_columns�exclude_columns� resolve_fks�opts�inspr r rH r� � 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(). �constrained_columnsr )�get_primary_keys)rF �conn� table_namer �kwr r rH �get_pk_constraint� s ����z DefaultDialect.get_pk_constraintc C s& t |�| jkrt�d|| jf ��d S )Nz7Identifier '%s' exceeds maximum length of %d characters)�lenr= r �IdentifierError)rF �identr r rH �validate_identifier� s ���z"DefaultDialect.validate_identifierc O s | j j|i |��S rJ )r. �connect)rF �cargs�cparamsr r rH r� � s zDefaultDialect.connectc C s |� � }|�|j� g |gS rJ )�translate_connect_args�update�query)rF rX r� r r rH �create_connect_args� s z"DefaultDialect.create_connect_argsc sf d|v r|d �t �|d���fdd��}d|v r1t�|d �� � |_t �|d�� fdd��}d S d S )N�isolation_level�engine_connectc s |s �� | � � d S d S rJ )�_set_connection_isolation�ra �branch)r� rF r rH � set_isolation� s �zBDefaultDialect.set_engine_execution_options.<locals>.set_isolation�schema_translate_mapc s � | _ d S rJ )�schema_for_objectr� )�getterr rH �set_schema_translate_map� � zMDefaultDialect.set_engine_execution_options.<locals>.set_schema_translate_map)r � listens_forr �_schema_getterr� )rF �enginer� r� r� r )r� r� rF rH � set_engine_execution_options� s �z+DefaultDialect.set_engine_execution_optionsc C s< d|v r | � ||d � d|v r t�|d �}||_d S d S )Nr� r� )r� r r� r� )rF ra r� r� r r rH � set_connection_execution_options s �z/DefaultDialect.set_connection_execution_optionsc C s6 |� � r t�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) �in_transactionr r) �set_isolation_levelra �_connection_record�finalize_callback�append�reset_isolation_level)rF ra �levelr r rH r� s � �z(DefaultDialect._set_connection_isolationc C � d S rJ r �rF �dbapi_connectionr r rH �do_begin � zDefaultDialect.do_beginc C � |� � d S rJ )�rollbackr� r r rH � do_rollback � zDefaultDialect.do_rollbackc C r� rJ )�commitr� r r rH � do_commit r� zDefaultDialect.do_commitc C r� rJ )rr r� r r rH �do_close r� zDefaultDialect.do_closec C s t t�dg�j| d��S )Nr rk )�strr rm rn rM r r rH � _dialect_specific_select_one" s z+DefaultDialect._dialect_specific_select_onec C sp d }z|� � }z|�| j� W |�� W dS |�� w | jjy7 } z| �|||�r2W Y d }~dS � d }~ww )NFT)ro r� r� rr r. �Error� is_disconnect)rF r� ro �errr r rH �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�randintrM r r rH � create_xid6 s zDefaultDialect.create_xidc C � |� t�|�� d S rJ )r� r �SavepointClause�rF ra r* r r rH � do_savepoint? � zDefaultDialect.do_savepointc C r� rJ )r� r �RollbackToSavepointClauser� r r rH �do_rollback_to_savepointB r� z'DefaultDialect.do_rollback_to_savepointc C r� rJ )r� r �ReleaseSavepointClauser� r r rH �do_release_savepointE r� z#DefaultDialect.do_release_savepointc C � |� ||� d S rJ )� executemany�rF ro rw r{ �contextr r rH �do_executemanyH � z DefaultDialect.do_executemanyc C r� rJ �r� r� r r rH � do_executeK r� zDefaultDialect.do_executec C s |� |� d S rJ r� )rF ro rw r� r r rH �do_execute_no_paramsN � z#DefaultDialect.do_execute_no_paramsc C ri )NFr )rF �era ro r r rH r� Q r� z DefaultDialect.is_disconnectc C s | � || j� d S rJ )r� rb )rF � dbapi_connr r rH r� T s z$DefaultDialect.reset_isolation_levelc C st |d u rd S t jrt|t�r|�| j�}|�� }|�� }||kr"|S ||kr.| j� |�s.|S ||kr8t |dd�S |S )NT)�quote) r r� rt r� �decoder+ �lower�upperr3 �_requires_quotesr �rF r* � name_lower� name_upperr r rH �normalize_nameZ s$ �� z DefaultDialect.normalize_namec C sh |d u rd S |� � }|�� }||kr|S ||kr | j�|�s |}tjr2| js.|�| j�}|S t |�}|S rJ ) r� r� r3 r� r r� �supports_unicode_binds�encoder+ �unicoder� r r rH �denormalize_namew s$ �� �z DefaultDialect.denormalize_name) Fr NNNNTNr! NNrJ )e�__name__� __module__� __qualname__�__doc__r � SQLCompiler�statement_compiler� DDLCompiler� ddl_compiler�GenericTypeCompilerr4 �IdentifierPreparerr2 �supports_alter�supports_comments�inline_comments�default_sequence_base�tupler� �supports_views�supports_sequences�sequences_optional�"preexecute_autoincrement_sequences�postfetch_lastrowidr1 r5 �cte_follows_insert�supports_native_enumr6 �#non_native_boolean_check_constraint� supports_simple_order_by_label�tuple_in_valuesr � immutabledict� bool_or_str�asint�asbool�engine_config_types�supports_native_decimal�py3kr� r� rd r? r* r= r< �max_index_name_lengthrS � supports_sane_multi_rowcountr� r0 �supports_default_values�supports_empty_insert�supports_multivalues_insert� supports_server_side_cursorsr\ �construct_arguments�requires_name_normalize�reflection_options� dbapi_exception_translation_map�deprecated_paramsrI �memoized_propertyrN �propertyrR rT � classmethodrY rh rj rf rc re 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 rH r , s� ��)� �J , 7 r c @ s< e Zd ZejZejZejZ ej Z dZ dZ dZdZdZdZdS )�StrCompileDialectTFN)r� r r r �StrSQLCompilerr r r �StrSQLTypeCompilerr4 r r2 r r r r1 r6 r r r r rH r- � 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@� Z5 dPdAdB�Z6dCdD� Z7dZ8 dQdFdG�Z9dHdI� Z:dJdK� Z;dLdM� Z<dNdO� Z=dS )R�DefaultExecutionContextFNc C s� | � | �}||_||_|j|_| |_}d|_|j|_|jr+t|j�|_|j� |j� |j s>t � |�|_ |�|j �d |_n t � |� |_|_ |�� |_g |_|jrZ|�� g|_|S i g|_|S )z8Initialize execution context for a DDLElement construct.Tr )�__new__�root_connection�_dbapi_connectionrl �compiled�isddl�execution_options�_execution_options�dictr� r� r ru �unicode_statementrD rw � create_cursorro �compiled_parametersr, r� r{ )rW rl ra r� � compiled_ddlrF r4 r r rH � _init_ddl� s* �z!DefaultExecutionContext._init_ddlc s@ | � | �}||_||_|j|_� |_� jsJ �� j�|j�|_� j � j � j f|_ t �� �|_�js;|j�|jj�|_n|j|_� j|_� j|_� j|_� j|_|sX� �� g|_n� fdd�t|�D �|_t |�dk|_ |� � |_ |jsy|jsy|jr�d|_ t!� jj"�|_#t!� j$o�� jj" �|_%|jj&s�|jj'r�|j r�|�(� n|�)� � j*�� j+r�t,���|�-� ��}n� j.r�|jj/}g }� j.r�|jD ](�g }|D ]} | �v r�|�0�| �| �� q�|�0�| � q�|�0��1|�� q�n,�j } |jD ]$�| �rt,���fdd��D ��}n t,��fdd��D ��}|�0|� q��1|�|_2|S )z6Initialize execution context for a Compiled construct.c s g | ] \}}� j ||d ��qS ))� _group_number)�construct_params)r~ �grp�m)r4 r rH � <listcomp> s ��z:DefaultExecutionContext._init_compiled.<locals>.<listcomp>r Tc 3 s> � | ]}�� |�d |�v r�| � | �n� | fV qdS )r N�rD �r~ �key)�compiled_paramsrl r r rH � <genexpr>1 s � ��� �z9DefaultExecutionContext._init_compiled.<locals>.<genexpr>c 3 s4 � | ]}||�v r�| � | �n� | fV qd S rJ r rD )rF r r rH rG ; s � ��� �)3r1 r2 r3 rl r4 � can_executer6 �unionr7 �_result_columns�_ordered_columns�_textual_ordered_columns�result_column_structr ru r9 r� r� r+ rw �isinsert�isupdate�isdelete� isplaintext�is_textr? r; � enumerater� r� r: ro �is_crud�bool� _returning�_is_explicit_returning� returning�_is_implicit_returning�insert_prefetch�update_prefetch� _process_executemany_defaults� _process_executesingle_defaults�_bind_processors� contains_expanding_parametersr8 �_expand_in_parametersr, � positiontupr� r� r{ ) rW rl ra r� r4 r{ rF ra �paramrE r� r )r4 rF rl r rH �_init_compiled� s� �� � � � � � � z&DefaultExecutionContext._init_compiledc s� �j rt�d���jjr�jjrtd��i �_�jd }|jr"g }nd}i �i }|jr/�jj n�jj D ]�� �jj � }|j r� �v rG|� �no|� � �}|seg �|� < �j� |jr]|jn|jg��� < nQt|d ttf�r�� fdd�t|d�D � �|� < �jjr�dnd d ���fd d �t|�D �� �� < n � fd d�t|d�D � �|� < d ��fdd ��D ���� < |��� ��� �fdd ��D �� |jr�|�dd � �D �� dd� �D ��j� < q3|jr�|�� � q3�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�q qS )r z%s_%s_%s)rS )r~ �i� tuple_element�j�value�r* r rH rB � s ���zADefaultExecutionContext._expand_in_parameters.<locals>.<listcomp>r zVALUES � �, c 3 s: � | ]\� �d d� � ���fdd�t��D �� V qdS )z(%s)rj c 3 s8 � | ]\}}�j jd �� t�� | d i V qdS )r* r N)r4 � bindtemplater� )r~ rf rg )rd rF � to_updatere r rH rG � s � ����� �zJDefaultExecutionContext._expand_in_parameters.<locals>.<genexpr>.<genexpr>N)�joinrS �r~ )rF rl )rd re rH rG � s � ��� �z@DefaultExecutionContext._expand_in_parameters.<locals>.<genexpr>c s g | ] \}}d � |f |f�qS )z%s_%sr )r~ rd rg rh r rH rB � s ��c 3 s$ � | ] \}}� j jd |i V qdS )r* N)r4 rk �r~ rE rg rM r rH rG � s � � �c 3 s( � | ]\}}� �v r|�� fV qd S rJ r ro )r* r r rH rG � s � ��c s s � | ]\}}|V qd S rJ r )r~ r* rg r r rH rG � s � c S s g | ]\}}|�qS r r )r~ � expand_keyrg r r rH rB � s �c s � | � d� S )Nr )�group)rA )�replacement_expressionsr rH �process_expanding� r� zHDefaultExecutionContext._expand_in_parameters.<locals>.process_expandingz\[EXPANDING_(\S+)\]) r� r �InvalidRequestErrorr4 r, �_numeric_bindsr] �_expanded_parametersr; ra �binds� expanding�pop�visit_empty_set_expr�_expanding_in_types�typert r �listrS rl r rm r� �extendr� �re�subrw ) rF r4 r rF ra �to_update_sets� parameter�valuesrs r )r* r rr rF rl rH r` K s� �� �� ���� � � � � � �� � �z-DefaultExecutionContext._expand_in_parametersc s | � | �}||_||_|j|_d|_|j|_|s(|jjr#� �� g|_ n7i g|_ n2t |d � j�r4||_ n&t |d t �rP� j rB||_ n� fdd�|D �pMi g|_ n � fdd�|D �|_ t |�dk|_� j svt |tj�rv||_� �|�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 rC )r~ �k)�drl r rH � <dictcomp>� s zFDefaultExecutionContext._init_statement.<locals>.<listcomp>.<dictcomp>r rn rk )r� rH rB � � ��z;DefaultExecutionContext._init_statement.<locals>.<listcomp>c s g | ]}� � |��qS r )r� )r~ �prk r rH rB � s �r )r1 r2 r3 rl rR r7 r6 r, r� r{ rt r8 r� r� r� r ru r9 rD rw r: ro )rW rl ra r� rw r{ rF r rk rH �_init_statement� s@ �� �� z'DefaultExecutionContext._init_statementc C s4 | � | �}||_||_|j|_|j|_|�� |_|S )z;Initialize execution context for a ColumnDefault construct.)r1 r2 r3 rl r7 r6 r: ro )rW rl ra r� rF r r rH � _init_default� s z%DefaultExecutionContext._init_defaultc C � | j jS rJ )r2 r� rM r r rH r� � rO z DefaultExecutionContext.enginec C r� rJ )r4 � postfetchrM r r rH �postfetch_cols� rO z&DefaultExecutionContext.postfetch_colsc C s | j r| jjS | jr| jjS dS )Nr )rN r4 rZ rO r[ rM r r rH � prefetch_cols� s z%DefaultExecutionContext.prefetch_colsc C s | j j d S rJ )r4 rX rM r r rH �returning_cols s z&DefaultExecutionContext.returning_colsc C s | j �dd�S )N� no_parametersF)r6 �getrM r r rH r� rZ z%DefaultExecutionContext.no_parametersc C s< | j �d| j r| jrtjpd�}|tju r | �| j�S |S )N� autocommitF)r6 r� r4 rw r �PARSE_AUTOCOMMIT�should_autocommit_textr9 )rF r� r r rH �should_autocommit s ��� z)DefaultExecutionContext.should_autocommitc C s� | j }t|tj�r| jjs| j�|�d }| jjr | j�� }ni }|j | j ||| d� | j � � d }|durI|� | j| j j d d �}|rI||�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 )r2 rt r ru rl r� rD r, r� rp ro rq �_cached_result_processorr� )rF �stmt�type_r� �default_params�r�procr r rH �_execute_scalar s$ �� �z'DefaultExecutionContext._execute_scalarc C s | j �� S rJ )r2 �_branchrM r r rH ra > r� z"DefaultExecutionContext.connectionc C s t �|�S rJ )�AUTOCOMMIT_REGEXP�match)rF rw r r rH r� B s z.DefaultExecutionContext.should_autocommit_textc C sx | j jsdS | j jr3| j�dd�o0| jr t| jjtj �p0| j p't| jjtj �o0| jo0t � | j�}|S | j�dd�}|S )NF�stream_resultsT) rl r$ �server_side_cursorsr6 r� r4 rt rw r � Selectable� TextClause�SERVER_SIDE_CURSOR_REr� )rF �use_server_sider r rH �_use_server_side_cursorE s2 � �� ��� ���z/DefaultExecutionContext._use_server_side_cursorc C s& | � � r d| _| �� S d| _| j�� S )NTF)r� �_is_server_side�create_server_side_cursorr3 ro rM r r rH r: e s z%DefaultExecutionContext.create_cursorc C s t � �rJ )r] rM r r rH r� m s z1DefaultExecutionContext.create_server_side_cursorc C r� rJ r rM r r rH �pre_execp r� z DefaultExecutionContext.pre_execc C r� rJ r rM r r rH � post_execs r� 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� rl )rF r� �colname�coltyper r rH �get_result_processorv s z,DefaultExecutionContext.get_result_processorc C r� )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. )ro � lastrowidrM r r rH � get_lastrowid� s z%DefaultExecutionContext.get_lastrowidc C r� rJ r )rF r� r r rH �handle_dbapi_exception� r� z.DefaultExecutionContext.handle_dbapi_exceptionc C s | j rt�| �S t�| �S rJ )r� r �BufferedRowResultProxy� ResultProxyrM r r rH �get_result_proxy� s z(DefaultExecutionContext.get_result_proxyc C r� rJ )ro �rowcountrM r r rH r� � rO z DefaultExecutionContext.rowcountc C r� rJ )rl rS rM r r rH rS � � z.DefaultExecutionContext.supports_sane_rowcountc C r� rJ )rl r rM r r rH r � r� z4DefaultExecutionContext.supports_sane_multi_rowcountc C s� | j r | js | js| jjs| jjr| �� n| js | �� | � � }| j rH| jr<|� � }|| _ | � |� |� � d |_|S | jsF|� � d |_|S | jr^| jr^|� � }|| _ |� � d |_|S |jd u rj|j |� � |S rJ )rN r� rY r4 �inlinerl r � _setup_ins_pk_from_lastrowid�_setup_ins_pk_from_emptyr� rq �returned_defaults�%_setup_ins_pk_from_implicit_returning� _soft_close� _metadatarW rO r� )rF r rx r r rH �_setup_crud_result_proxy� sB ��� � � �z0DefaultExecutionContext._setup_crud_result_proxyc s� | j jd �| j jj}| jd �| �� ��d ur?|j� � d ur/� j�| j d �}|d ur/|���� ���fdd�|j D �| _ d S ��fdd�|j D �| _ d S )Nr r c s( g | ]}|� u r �n�� �|�d ��qS rJ �r� �r~ �c�� autoinc_colrF � key_getterr� r rH rB � s ���zHDefaultExecutionContext._setup_ins_pk_from_lastrowid.<locals>.<listcomp>c � g | ] }� � �|�d ��qS rJ r� r� �rF r� r rH rB � s ��) r4 � _key_getters_for_crud_columnrw r� r; r� �_autoincrement_columnr| r� rl � primary_key�inserted_primary_key)rF r� r� r r� rH r� � s$ � � �z4DefaultExecutionContext._setup_ins_pk_from_lastrowidc s< | j jd �| j jj}| jd � � �fdd�|jD �| _d S )Nr r c r� rJ r� r� r� r rH rB � s �zDDefaultExecutionContext._setup_ins_pk_from_empty.<locals>.<listcomp>)r4 r� rw r� r; r� r� )rF r� r r� rH r� � s �z0DefaultExecutionContext._setup_ins_pk_from_emptyc s\ �d u r d | _ d S | jjd �| jjj}| jd � �fdd�� �fdd�|jD �D �| _ d S )Nr r c s$ g | ]\}}|d u r� | n|�qS rJ r )r~ �colrg )rx r rH rB s ��zQDefaultExecutionContext._setup_ins_pk_from_implicit_returning.<locals>.<listcomp>c s g | ] }|� � �|�d �f�qS rJ r� )r~ r� r� r rH rB r� )r� r4 r� rw r� r; r� )rF rx r� r )rF r� rx rH r� � s � �z=DefaultExecutionContext._setup_ins_pk_from_implicit_returningc C s | j s| jo t| jj�S rJ )rN rO rU r4 r� rM r r rH �lastrow_has_defaults s �z,DefaultExecutionContext.lastrow_has_defaultsc sB t | jd�sdS i }| jjD ]5}|j�| j�}t|�}|�| jj�� � dur?|r0� |vr?||vr?|r:� |v s:||v r?� ||< qd||< q| jjrV| jj � || j | j | j | � | jjr�g }| jjD ]*} | jj| }|�|d�� � dur�| | jv r�|�� gt| j| � � q`|�� � q`z | j j|� W dS ty� } z| j�| ddd| � W Y d} ~ dS d} ~ ww i } | jj�� D ];\}} |�|d�� � dur�|r�|�| | �} | jjs�| j� | �d } | | jv r�| � � fdd�| j| D �� q�� | | < q�z | j jdi | �� W dS t�y } z| j�| ddd| � W Y d} ~ dS d} ~ ww )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 rJ r )r~ rp ��dbtyper rH rG [ s � � �z:DefaultExecutionContext.set_input_sizes.<locals>.<genexpr>r ) �hasattrr4 r� r| �_unwrapped_dialect_implrl �get_dbapi_typer. � _has_events�dispatch�do_setinputsizesro rw r{ r, ra rw r� rv r~ r� r� � setinputsizes� BaseExceptionr2 �_handle_dbapi_exception�itemsr� rD r� ) rF � translate� include_types� exclude_types� inputsizes� bindparam� dialect_impl�dialect_impl_cls�positional_inputsizesrE r� �keyword_inputsizesr r� rH �set_input_sizes s� �� � � � ��� �� ���z'DefaultExecutionContext.set_input_sizesc C s| |j r | �||�S |jr|| _|�| �S |jr;| j}|js%t� |j|�}n|j}t� |g�j |d�}|� |di �� � S |jS )N)�bindr )� is_sequence� fire_sequence� is_callable�current_column�arg�is_clause_elementra � _arg_is_typedr � type_coercerm rn �_execute_compiled�scalar)rF �columnr r� r� � default_argr� r r rH � _exec_defaulth s z%DefaultExecutionContext._exec_defaultTc s� z| j �| j}W n ty t�d��w |rS| jrS| jjjrS|j r1|j d � |j j �|j i}n |j �|j i}d� | jjj d �� }|�� �fdd�|D �� |S �S )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 rD ��indexr{ r rH rG � s � �zADefaultExecutionContext.get_current_parameters.<locals>.<genexpr>)�current_parametersr� �AttributeErrorr rt rN r4 rw �_has_multi_parameters�_is_multiparam_columnr� �originalrE r{ �keysr� )rF �isolate_multiinsert_groupsr� r� r� r r� rH �get_current_parameters� s2 ����� �z.DefaultExecutionContext.get_current_parametersc C � |j d u rd S | �||j |j�S rJ )r r� r| �rF r� r r rH �get_insert_default� � z*DefaultExecutionContext.get_insert_defaultc C r� rJ )�onupdater� r| r� r r rH �get_update_default� r� z*DefaultExecutionContext.get_update_defaultc C s� | j jd }i }| j j}| j j}|D ]}|jr!|jjr!|jj||< q|D ]}|jr3|jjr3|jj||< q$| jD ]?}|| _ |D ]}||v rI|| }n| � |�}|d urX||||�< q>|D ]}||v rf|| }n| � |�}|d uru||||�< q[q7| ` d S )Nr ) r4 r� rZ r[ r � is_scalarr� r� r; r� r� r )rF r� �scalar_defaultsrZ r[ r� rb �valr r rH r\ � s< � � � ��z5DefaultExecutionContext._process_executemany_defaultsc C s� | j jd }| jd | _}| j jD ]!}|jr$|jjs$|jjr$|jj}n| � |�}|d ur3||||�< q| j j D ]}| � |�}|d urI||||�< q8| `d S )Nr r ) r4 r� r; r� rZ r r� r r� r� r[ r )rF r� r; r� r r r rH r] � s" � � �z7DefaultExecutionContext._process_executesingle_defaults)NNN)T)>r� r r rN rO rP rT rR r5 r� r4 rw rM r� rY rW �_translate_colnamer r rv r, r= rc r` r� r� r* r� r� r� r� r� r� r� r+ ra r� r� r: r� r� r� r� r� r� r� r� rS r r� r� r� r� r� r� r� r� r� r� r r\ r] r r r rH r0 � s� un / ! ' �X : %r0 )!r rB r� r rK ri r r r r r r r r r� r �sqlr r r � sql.elementsr rn �I�UNICODEr� r� �Dialectr r- �ExecutionContextr0 �execution_ctx_clsr r r rH �<module> sH � f |
Save
Cancel