관리-도구
편집 파일: default.cpython-38.pyc
U -?�f�J � @ s d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl Z ddl m Z ddl m Z ddl m Z ddl m Z dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddlZddlmZ ddlm Z ddlm Z ddl m Z ddl m!Z! ddl m"Z" ddl m#Z# dd l m$Z$ dd l m%Z% dd l&m'Z' dd l&m(Z( d d!lm)Z) d d"lm*Z* d d#lm+Z+ d d$lm,Z, d d%l-m.Z. d d&l-m/Z/ d d'l-m0Z0 d d(l-m1Z1 d d$l-m,Z2 d d)l3m4Z4 d d*l5m6Z6 d d+l7m8Z8 d d,l7m9Z9 d d-l7m:Z: d d.l;m<Z< d d/l=m>Z> d d0l=m?Z? e j�r�dd1l@mAZA dd2l mBZB dd3l mCZC dd4l mDZD dd5l mEZE dd6l mFZF dd7l mGZG dd8l mHZH dd9l mIZI dd:l mJZJ dd;l mKZK dd<l mLZL dd=lMmNZN dd>lOmPZP d d?l)mQZQ d d@l+mRZR d dAl+mSZS d dBl-mTZT d dCl7mUZU d dDl7mVZV d dEl7mWZW d dFlXmYZY d dGlXmZZZ d dHl;m[Z[ d dIl\m]Z] d dJl^m_Z_ d dKl^m`Z` d dLl^maZa e�bdMejcejdB �Zeefe!�\ZgZhZiZjZkG dNdO� dOe#�ZlG dPdQ� dQel�ZmG dRdS� dSe%�Znenel_odS )Ta 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. � )� annotationsN)� perf_counter)�Any)�Callable)�cast)�Dict)�List)�Mapping)�MutableMapping)�MutableSequence)�Optional)�Sequence)�Set)�Tuple)�Type)� TYPE_CHECKING)�Union� )�characteristics)�cursor)� interfaces)� Connection)� CacheStats)� DBAPICursor)�Dialect)� ExecuteStyle)�ExecutionContext)� ObjectKind)� ObjectScope� )�event)�exc)�pool)�util)�compiler)�dml)� expression)�type_api)� is_tuple_type)�_NoArg)� DDLCompiler)� InsertmanyvaluesSentinelOpts)� SQLCompiler)� quoted_name)�Final)�Literal)� ModuleType)�Engine)�ResultFetchStrategy)�_CoreMultiExecuteParams)�_CoreSingleExecuteParams)�_DBAPICursorDescription)�_DBAPIMultiExecuteParams)�_ExecuteOptions)� _MutableCoreSingleExecuteParams)� _ParamStyle)�DBAPIConnection)�IsolationLevel)�Row)�URL)�_ListenerFnType)�Pool)�PoolProxiedConnection)� Executable)�Compiled)�Linting)�ResultColumnsEntry)�DMLState)� UpdateBase)� BindParameter)�Column)�_BindProcessorType)�_ResultProcessorType)� TypeEnginez \s*SELECTc @ s� e Zd ZU dZejZejZej Z ej Z dZ dZdZdZdZdZejjZdZded< dZded< dZeZ dZ dZ dZ dZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2e3�4e5�6� e5�7� d ��Z8e3�4e3j9e3�:d �e3�:d �e3j9e3j9e3j9e3j;d ��Z<d ed < dZ=dZ>dZ?dZ@ded< dZAded< dZBded< dZCded< dZDdZEi ZFded< dZGdZHdZIdZJdZKdZLdZMded< dZNded < eOjPZQd ed < d ZRd ed!< d"ZSdZTdZUdZVdZWdZXdZYded#< dZZdZ[dZ\d$Z]e3j^d%d&d'�ddddddde_j`deaejb�df d(d)d*d+d,ddd-d,d.dd/d0� d1d2��Zcede3�ed3d4�d5d6� ��Zfe3jgd7d8� �Zhe3jgd9d:� �ZidZjdZke3jgd;d<�d=d>��Zle3jgd?d@� �ZmdAdB� Zne3jgdCdD� �Zoe3jgdEdF� �ZpeddGdH� �ZqeddIdJ� �ZresdKdLdM�dNdO��ZtdKdLdM�dPdQ�ZuesdRdS� �ZvdTd<�dUdV�ZwdWdX� ZxdYdZ� Zyd[d\� Zzd]d^� Z{d_d`� Z|d�dadb�Z}dcddd/dde�dfdg�Z~dhdi� Zdjdk� Z�dldm� Z�dnd dodp�dqdr�Z�dcd dods�dtdu�Z�dvdw� Z�dxdy� Z�dzd{� Z�d|d}� Z�d~d� Z�d�d�� Z�d�d�� Z�e3jgd�d�� �Z�d�dd��d�d��Z�d�dd��d�d��Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�� Z�e3j�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�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�dS )��DefaultDialectz!Default implementation of DialectTFNzOptional[Set[Any]]�include_set_input_sizes�exclude_set_input_sizesr )�isolation_levelZ logging_token�debug)� pool_timeoutZechoZ echo_poolZ pool_recycleZ pool_sizeZ max_overflow�future�Mapping[str, Any]�engine_config_types�defaulti' � Optional[int]�#_user_defined_max_identifier_lengthz Optional[str]rO �max_index_name_length�max_constraint_name_lengthz<MutableMapping[Type[TypeEngine[Any]], Type[TypeEngine[Any]]]�colspecs�named�DEFAULT�bool�use_insertmanyvalues�!use_insertmanyvalues_wo_returningr+ �"insertmanyvalues_implicit_sentineli� �int�insertmanyvalues_page_sizei� �default_schema_namezutf-8)�1.4a The :paramref:`_sa.create_engine.empty_in_strategy` keyword is deprecated, and no longer has any effect. All IN expressions are now rendered using the "expanding parameter" strategy which renders a set of boundexpressions, or an "empty set" SELECT, at statement executiontime.)rd z�The :paramref:`_sa.create_engine.server_side_cursors` parameter is deprecated and will be removed in a future release. Please use the :paramref:`_engine.Connection.execution_options.stream_results` parameter.)Zempty_in_strategy�server_side_cursorszOptional[_ParamStyle]zOptional[IsolationLevel]zOptional[ModuleType]z Literal[True]zOptional[bool]zUnion[_NoArg, int]rC r ) � paramstylerO �dbapi�implicit_returning�supports_native_boolean�max_identifier_length� label_lengthrb r^ �compiler_lintingre �kwargsc K s4 | r | j st�d| ��nd| _t| dd�rBt�dd� tjj | _ d| _ d | _ || _ |d k rd|| _n | j d k rz| j j| _n| j| _| jdk| _ | �| �| _|| _t| dd �} | d k r�tttj | j�}n| j}|| � | _| _|d k r�|| _|| _| j�r| j| _ || _ | | _ | d k �r | | _ |t j!k �r0|| _"d S ) Nz/Dialect %s does not support server side cursorsTZuse_setinputsizesFzoThe dialect-level use_setinputsizes attribute is deprecated. Please use bind_typing = BindTyping.SETINPUTSIZES�2.0)Zqmark�format�numericZnumeric_dollar� type_compiler)#� supports_server_side_cursorsr! � ArgumentErrorre �getattrr# Zwarn_deprecatedr � BindTypingZ SETINPUTSIZES� bind_typing� positionalZ_ischemarg rf �default_paramstyle�preparer�identifier_preparer�_on_connect_isolation_levelr r r$ �GenericTypeCompilerrq �type_compiler_clsZtype_compiler_instanceri rW rj rk rl r^ r) �NO_ARGrb )�selfrf rO rg rh ri rj rk rb r^ rl re rm Zlegacy_tt_callableZ tt_callable� r� �I/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/engine/default.py�__init__ sV $� � �� zDefaultDialect.__init__rn z]full_returning is deprecated, please use insert_returning, update_returning, delete_returningc C s | j o| jo| jS �N)�insert_returning�update_returning�delete_returning�r r� r� r� �full_returningu s ��z DefaultDialect.full_returningc C s | j o | jS )a� Default implementation for insert_executemany_returning, if not otherwise overridden by the specific dialect. The default dialect determines "insert_executemany_returning" is available if the dialect in use has opted into using the "use_insertmanyvalues" feature. If they haven't opted into that, then this attribute is False, unless the dialect in question overrides this and provides some other implementation (such as the Oracle dialect). �r� r^ r� r� r� r� � insert_executemany_returning� s z+DefaultDialect.insert_executemany_returningc C s | j o | jS )a� Default implementation for insert_executemany_returning_deterministic_order, if not otherwise overridden by the specific dialect. The default dialect determines "insert_executemany_returning" can have deterministic order only if the dialect in use has opted into using the "use_insertmanyvalues" feature, which implements deterministic ordering using client side sentinel columns only by default. The "insertmanyvalues" feature also features alternate forms that can use server-generated PK values as "sentinels", but those are only used if the :attr:`.Dialect.insertmanyvalues_implicit_sentinel` bitflag enables those alternate SQL forms, which are disabled by default. If the dialect in use hasn't opted into that, then this attribute is False, unless the dialect in question overrides this and provides some other implementation (such as the Oracle dialect). r� r� r� r� r� �4insert_executemany_returning_sort_by_parameter_order� s zCDefaultDialect.insert_executemany_returning_sort_by_parameter_orderr0 ��returnc C s" | j d kr t�d| � d���| j S )N�Dialect z\ does not have a Python DBAPI established and cannot be used for actual database interaction)rg r! �InvalidRequestErrorr� r� r� r� � loaded_dbapi� s �zDefaultDialect.loaded_dbapic C s | j tjjkS r� )rv r ru Z RENDER_CASTSr� r� r� r� �_bind_typing_render_casts� s z(DefaultDialect._bind_typing_render_castsc C s$ t |t�s t�dtt|�f ��d S )Nz�The argument passed to Dialect.has_table() should be a %s, got %s. Additionally, the Dialect.has_table() method is for internal dialect use only; please use ``inspect(some_engine).has_table(<tablename>>)`` for public API use.)� isinstancer r! rs �type)r �argr� r� r� � _ensure_has_table_connection� s ��z+DefaultDialect._ensure_has_table_connectionc C s: | j j�dd �}|d kr2tjd| j| jf dd� t|�S )N�supports_statement_cachea� Dialect %s:%s will not make use of SQL compilation caching as it does not set the 'supports_statement_cache' attribute to ``True``. This can have significant performance implications including some performance degradations in comparison to prior SQLAlchemy versions. Dialect maintainers should seek to set this attribute to True after appropriate development and testing for SQLAlchemy 1.4 caching support. Alternatively, this attribute may be set to False which will disable this warning.Zcprf)�code)� __class__�__dict__�getr# �warn�name�driverr] )r Zsscr� r� r� �_supports_statement_cache� s � �z(DefaultDialect._supports_statement_cachec C s t �� S r� )�weakref�WeakKeyDictionaryr� r� r� r� � _type_memos� s zDefaultDialect._type_memosc C s | j d | j S )N�+)r� r� r� r� r� r� �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 synonymous with ``supports_sane_rowcount``. )�supports_sane_rowcountr� r� r� r� � supports_sane_rowcount_returning� s z/DefaultDialect.supports_sane_rowcount_returningr= z Type[Pool])�urlr� c C s t | dtj�S )NZ poolclass)rt r" Z QueuePool)�clsr� r� r� r� �get_pool_class� s z DefaultDialect.get_pool_classc C s | � |�S r� )r� )r r� r� r� r� �get_dialect_pool_class� s z%DefaultDialect.get_dialect_pool_classc C sD d� | j�d�dd� �}zt|d � W n tk r> Y nX d S )N�.r ���z .provision)�join� __module__�split� __import__� ImportError)r� �packager� r� r� �load_provisioning� s z DefaultDialect.load_provisioningzOptional[_ListenerFnType]c s" � j d k r� fdd�}|S d S d S )Nc s � � | � j� d S r� )� _assert_and_set_isolation_levelr{ )� dbapi_connZconn_recr� r� r� �builtin_connect� s �z:DefaultDialect._builtin_onconnect.<locals>.builtin_connect)r{ )r r� r� r� r� �_builtin_onconnect� s z!DefaultDialect._builtin_onconnectc C s� z| � |�| _W n tk r* d | _Y nX z| �|�| _W n tk rV d | _Y nX z| �|jj�| _W n tk r� d | _Y nX | j s�| � |�}|r�|| _ | j r�| j | j kr�t �d| j | j f ��d S )NzQLabel length of %d is greater than this dialect's maximum identifier length of %d)Z_get_server_version_info�server_version_info�NotImplementedErrorZ_get_default_schema_namerc �get_default_isolation_level� connection�dbapi_connection�default_isolation_levelrW � _check_max_identifier_lengthrj rk r! rs )r r� Zmax_ident_lengthr� r� r� � initialize s> � � � � � ��zDefaultDialect.initializec C s d S r� r� r� r� r� r� � on_connect) s zDefaultDialect.on_connectc C s dS )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� �r r� r� r� r� r� - s z+DefaultDialect._check_max_identifier_lengthc C s | � |�S )a� Given a DBAPI connection, return its isolation level, or a default isolation level if one cannot be retrieved. May be overridden by subclasses in order to provide a "fallback" isolation level for databases that cannot reliably retrieve the actual isolation level. By default, calls the :meth:`_engine.Interfaces.get_isolation_level` method, propagating any exceptions raised. .. versionadded:: 1.3.22 )Zget_isolation_level�r r� r� r� r� r� 9 s z*DefaultDialect.get_default_isolation_levelc 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' Z adapt_typerZ )r Ztypeobjr� r� r� �type_descriptorI s z DefaultDialect.type_descriptorc K sP | j ||fd|i|��s dS | j||fd|i|��D ]}|d |kr4 dS q4dS )N�schemaFr� T)Z has_table� get_indexes)r r� Z table_nameZ index_namer� �kw�idxr� r� r� � has_indexT s ��� zDefaultDialect.has_indexr �str)r� � schema_namer� r� c K s || j |f|�kS r� )Zget_schema_names)r r� r� r� r� r� r� � has_schema_ s zDefaultDialect.has_schemac C s& t |�| jkr"t�d|| jf ��d S )Nz7Identifier '%s' exceeds maximum length of %d characters)�lenrj r! ZIdentifierError)r �identr� r� r� �validate_identifierd s ��z"DefaultDialect.validate_identifierc O s | j j||�S r� )r� �connect)r ZcargsZcparamsr� r� r� r� k s zDefaultDialect.connectc C s |� � }|�|j� g |fS r� )Ztranslate_connect_args�update�query)r r� �optsr� r� r� �create_connect_argso s z"DefaultDialect.create_connect_argsr1 �None)�enginer� r� c sJ t �j����}|rFt��fdd�|D ��� t�|d�� �fdd��}d S )Nc 3 s | ]}|� | fV qd S r� r� ��.0r� �r� r� r� � <genexpr>| s z>DefaultDialect.set_engine_execution_options.<locals>.<genexpr>Zengine_connectc s �� | � � d S r� )� _set_connection_characteristics)r� )r r r� r� � set_connection_characteristics� s �zSDefaultDialect.set_engine_execution_options.<locals>.set_connection_characteristics)�set�connection_characteristics� intersectionr# � immutabledictr Z listens_for)r r� r� �supported_namesr� r� )r r� r r� � set_engine_execution_optionsu s �� z+DefaultDialect.set_engine_execution_options)r� r� r� c s< t | j��� �}|r8t�� fdd�|D ��}| �||� d S )Nc 3 s | ]}|� | fV qd S r� r� r� r� r� r� r� � s zBDefaultDialect.set_connection_execution_options.<locals>.<genexpr>)r� r� r� r# r� r� )r r� r� r� r r� r� r� � set_connection_execution_options� s ��z/DefaultDialect.set_connection_execution_optionsc s� � fdd�|� � D �}|�� rNdd� |D �}|rNt�dd�dd� |D �� ��|jj}|D ]\}}}|�� |||� qZ|jjj � t � � j |�� d S )Nc s g | ]\}}|� j | |f�qS r� )r� )r� r� �valuer� r� r� � <listcomp>� s �zBDefaultDialect._set_connection_characteristics.<locals>.<listcomp>c S s g | ]\}}}|j r||f�qS r� )Z transactional)r� r� �obj�_r� r� r� r� � s �z�This connection has already initialized a SQLAlchemy Transaction() object via begin() or autobegin; %s may not be altered unless rollback() or commit() is called first.�, c s s | ]\}}|V qd S r� r� )r� r� r� r� r� r� r� � s zADefaultDialect._set_connection_characteristics.<locals>.<genexpr>)�itemsZin_transactionr! r� r� r� r� Z set_connection_characteristicZ_connection_recordZfinalize_callback�append� functools�partial�_reset_characteristics) r r� r Zcharacteristic_valuesZ trans_objsr� r� �characteristicr� r� r� r� r� � s0 ���� � �z.DefaultDialect._set_connection_characteristicsc C s$ |D ]}| j | }|�| |� qd S r� )r� Zreset_characteristic)r r r� Zcharacteristic_namer� r� r� r� r� � s �z%DefaultDialect._reset_characteristicsc C s d S r� r� �r r� r� r� r� �do_begin� s zDefaultDialect.do_beginc C s |� � d S r� )�rollbackr� r� r� r� � do_rollback� s zDefaultDialect.do_rollbackc C s |� � d S r� )�commitr� r� r� r� � do_commit� s zDefaultDialect.do_commitc C s | � |� d S r� )�do_closer� r� r� r� � do_terminate� s zDefaultDialect.do_terminatec C s |� � d S r� )�closer� r� r� r� r� � s zDefaultDialect.do_closec C s t t�d�j| d��S )Nr ��dialect)r� r& �select�compiler� r� r� r� � _dialect_specific_select_one� s z+DefaultDialect._dialect_specific_select_oner: )r� r� c C s� z | � |�W S | jjk r� } zl| �||d �}| jrzztj|| |ddd� W n* tjk rx } z |j }W 5 d }~X Y nX |r�W Y � dS � W 5 d }~X Y nX d S )NFT)� is_disconnectZ invalidate_pool_on_disconnectZ is_pre_ping) �do_pingr� �Errorr� � _has_eventsr Z$_handle_dbapi_exception_noconnectionr! ZStatementErrorZconnection_invalidated)r r� �errr� Znew_errr� r� r� �_do_ping_w_event� s$ � z DefaultDialect._do_ping_w_eventc C s, d }|� � }z|�| j� W 5 |�� X dS )NT)r r� �executer� )r r� r r� r� r� r� � 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�randintr� r� r� r� � create_xid� s zDefaultDialect.create_xidc C s |� t�|�� d S r� )r r& ZSavepointClause�r r� r� r� r� r� � do_savepoint� s zDefaultDialect.do_savepointc C s |� t�|�� d S r� )r r& ZRollbackToSavepointClauser r� r� r� �do_rollback_to_savepoint� s z'DefaultDialect.do_rollback_to_savepointc C s |� t�|�� d S r� )r r& ZReleaseSavepointClauser r� r� r� �do_release_savepoint� s z#DefaultDialect.do_release_savepointc # s� t t|�}t t|j�}d� d �d}|j} |j} | d k s:t�t|j�} |j � d�j �} |j rj|j � di �} nd } | r�g }||_ | j}nd}d }|�||| || || �D �]}|V | r�z|�|�}W nD tk �r } z$|j|t�|j�|jd |dd� W 5 d }~X Y nX |d k �st�| j�r�|j�s�| jdk}| j�rZ|�r@t�|�t|t � d�d �� q�| j �sft�| j �srt�| j�|�s�|�r��fd d �t | j |j!� d � �D �� n | j d j"�#�|j!d d ��d}|�r� �fd d�|D ��n(��r�fdd�|D ��ndd� |D ��t$��t$|j%�k�rHt&�'dt$|j%�� dt$��� d���z�fdd �|j(D �}W n> t)k �r� } z t&�'d|j*d �d��|�W 5 d }~X Y nX |�|� q�|�|� q�d S )Nr� Frb �schema_translate_mapT)Z is_sub_execr r� )�keyc s" g | ]\}}|j �� |d ��qS )r )r� �_cached_result_processor)r� �colZ cursor_descr� r� r� r� [ s � �zDDefaultDialect._deliver_insertmanyvalues_batches.<locals>.<listcomp>r c s0 i | ](}t d d� t|� d� � �D ��|�qS )c s s" | ]\}}|r||�n|V qd S r� r� )r� �val�procr� r� r� r� r s �zNDefaultDialect._deliver_insertmanyvalues_batches.<locals>.<dictcomp>.<genexpr>N)�tuple�zip�r� �row)�_composite_sentinel_proc�_nscr� r� � <dictcomp>q s � ��zDDefaultDialect._deliver_insertmanyvalues_batches.<locals>.<dictcomp>c s i | ]}� |d �|�qS �r� r� r )�_scalar_sentinel_procr� r� r { s c S s i | ]}|d |�qS r r� r r� r� r� r s zASentinel-keyed result set did not produce correct number of rows z ; produced zP. Please ensure the sentinel column is fully unique and populated in all cases.c s g | ] }� | �qS r� r� )r� Z sentinel_keys)�rows_by_sentinelr� r� r� � s �zACan't match sentinel values in result set to parameter sets; key aT was not found. There may be a mismatch between the datatype passed to the DBAPI driver vs. that which it returns in a result row. Ensure the given Python value matches the expected result type *exactly*, taking care to not rely upon implicit conversions which may occur such as when using strings in place of UUID or integer values, etc. )+r �DefaultExecutionContextr, �compiled�compiled_parameters�_insertmanyvalues�AssertionErrorr] �effective_returning�execution_optionsr� rb r �_insertmanyvalues_rows�sort_by_parameter_order�!_deliver_insertmanyvalues_batches�fetchall_for_returning� BaseExceptionZ_handle_dbapi_exception�sql_utilZ_long_statementZreplaced_statementZreplaced_parameters�num_sentinel_columnsZ is_downgradedZimplicit_sentinel�extend�sorted�operator� itemgetterZsentinel_param_keys�sentinel_columnsr � descriptionr� r r� �batchr! r� Zsentinel_values�KeyError�args)r r� r � statement� parametersZgeneric_setinputsizes�contextr Z_sentinel_proc_initializedr �imvZ is_returningZ batch_sizer �resultr" Z imv_batch�rows�beZcomposite_sentinelZ ordered_rowsZker� )r r r r r r� r# � s� � � �� � � �� � � � �� � � � z0DefaultDialect._deliver_insertmanyvalues_batchesc C s |� ||� d S r� )� executemany�r r r1 r2 r3 r� r� r� �do_executemany� s z DefaultDialect.do_executemanyc C s |� ||� d S r� �r r9 r� r� r� � do_execute� s zDefaultDialect.do_executec C s |� |� d S r� r; )r r r1 r3 r� r� r� �do_execute_no_params� s z#DefaultDialect.do_execute_no_paramsc C s dS )NFr� )r �er� r r� r� r� r� � s z DefaultDialect.is_disconnectc C sp zt | �|��}W n tk r( Y d S X dd� |D �}||krdtd| j�dtt|��|��� ���t|�S d S )Nc S s g | ]}|� d d��� �qS )r� � )�replace�upper)r� �levelr� r� r� r� � s z@DefaultDialect._gen_allowed_isolation_levels.<locals>.<listcomp>r� zi get_isolation_level_values() method should return names as UPPERCASE using spaces, not underscores; got ) �listZget_isolation_level_valuesr� � ValueErrorr� r) r� � differencer )r r� Z raw_levelsZnormalized_levelsr� r� r� � _gen_allowed_isolation_levels� s � �z,DefaultDialect._gen_allowed_isolation_levelsc C sZ |� dd��� }| �|�}|rJ||krJt�d|�d| j�dd�|�� ���| �||� d S )Nr� r? zInvalid value z1 for isolation_level. Valid isolation levels for z are r� )r@ rA rF r! rs r� r� Zset_isolation_level)r r� rB Z_allowed_isolation_levelsr� r� r� r� � s ��� �z.DefaultDialect._assert_and_set_isolation_levelc C sT | j d k r4| j dks$| j | jks$t�| �|| j � n | jd k sBt�| �|| j� d S )NZ AUTOCOMMIT)r{ r� r r� r� r� r� r� �reset_isolation_level� s ��� ��z$DefaultDialect.reset_isolation_levelc C s\ |d kr d S |� � }|�� }||kr(|S ||kr@| j�|�s@|S ||krTt|dd�S |S d S )NT)�quote)�lowerrA rz �_requires_quotesr- �r r� Z name_lowerZ name_upperr� r� r� �normalize_name� s �� z DefaultDialect.normalize_namec C sD |d kr d S |� � }|�� }||kr(|S ||kr@| j�|�s@|}|S r� )rI rA rz rJ rK r� r� r� �denormalize_name s ��z DefaultDialect.denormalize_namec C s |S r� r� r� r� r� r� �get_driver_connection s z$DefaultDialect.get_driver_connectionc C s t t| �|�jt t|�jk S r� )rt r� �__code__rL )r �methodr� r� r� �_overrides_default s ��z!DefaultDialect._overrides_defaultc k s� g }g } t j|kr*|�| j� | �| j� t j|krL|�| j� | �| j� t j|krb|�| j � |� di �} |r�|t j kr�|t j kr�|} nrg } d|i|�} g } t j |kr�| �|� t j|kr�| �| � | D ]0}z| �||f| �� W q� tk r� Y q�X q�|�r t|�}| D ]�}|�r$||k�r||f}z ||||fd|i|��fV W nN tjk �r� } z|| k�rt|| |< W 5 d }~X Y n tjk �r� Y nX �qd S )N� unreflectabler� )r ZTABLEr� Zget_table_namesZget_temp_table_namesZVIEWZget_view_namesZget_temp_view_namesZMATERIALIZED_VIEWZget_materialized_view_names�popr �ANYr\ r( � TEMPORARYr� r� r! ZUnreflectableTableErrorZNoSuchTableError)r Zsingle_tbl_methodr� �kindr� Z filter_names�scoper� Z names_fnsZtemp_names_fnsrR �namesZname_kwZfns�fn�tabler r� r� r� r� �_default_multi_reflect sf ��� ���� z%DefaultDialect._default_multi_reflectc K s | j | j|f|�S r� )r[ Zget_table_options�r r� r� r� r� r� �get_multi_table_optionsh s ��z&DefaultDialect.get_multi_table_optionsc K s | j | j|f|�S r� )r[ Z get_columnsr\ r� r� r� �get_multi_columnsm s z DefaultDialect.get_multi_columnsc K s | j | j|f|�S r� )r[ Zget_pk_constraintr\ r� r� r� �get_multi_pk_constraintp s ��z&DefaultDialect.get_multi_pk_constraintc K s | j | j|f|�S r� )r[ Zget_foreign_keysr\ r� r� r� �get_multi_foreign_keysu s ��z%DefaultDialect.get_multi_foreign_keysc K s | j | j|f|�S r� )r[ r� r\ r� r� r� �get_multi_indexesz s z DefaultDialect.get_multi_indexesc K s | j | j|f|�S r� )r[ Zget_unique_constraintsr\ r� r� r� � get_multi_unique_constraints} s ��z+DefaultDialect.get_multi_unique_constraintsc K s | j | j|f|�S r� )r[ Zget_check_constraintsr\ r� r� r� �get_multi_check_constraints� s ��z*DefaultDialect.get_multi_check_constraintsc K s | j | j|f|�S r� )r[ Zget_table_commentr\ r� r� r� �get_multi_table_comment� s ��z&DefaultDialect.get_multi_table_comment)N)N)N)N)��__name__r� � __qualname__�__doc__r$ r, �statement_compilerr* � ddl_compilerr| r} �IdentifierPreparerry Zsupports_alterZsupports_commentsZ supports_constraint_commentsZinline_commentsr� Zdiv_is_floordivr ru �NONErv rM �__annotations__rN Zdefault_sequence_baser �execute_sequence_formatZsupports_schemasZsupports_views�supports_sequences�sequences_optional�"preexecute_autoincrement_sequences�supports_identity_columns�postfetch_lastrowidZ favor_returning_over_lastrowidZ#insert_null_pk_still_autoincrementsr� r� Zupdate_returning_multifromZdelete_returning_multifromr� Zcte_follows_insertZsupports_native_enumri Zsupports_native_uuidZreturns_native_bytesZ#non_native_boolean_check_constraint� supports_simple_order_by_labelZtuple_in_valuesr# r� r Z IsolationLevelCharacteristicZLoggingTokenCharacteristicr� ZasintZ bool_or_strZasboolrT Zsupports_native_decimalr� rj rW rO rX rY r� � supports_sane_multi_rowcountrZ rx Zsupports_default_valuesZsupports_default_metavalueZdefault_metavalue_tokenZsupports_empty_insert�supports_multivalues_insertr^ r_ r+ Z NOT_SUPPORTEDr` rb Z insertmanyvalues_max_parametersZsupports_is_distinct_fromrr re Zsupports_for_update_ofr� rc Zrequires_name_normalizeZis_asyncZ has_terminateZ$_legacy_binary_type_literal_encodingZdeprecated_paramsr) r~ ra Z NO_LINTINGr� �property� deprecatedr� �memoized_propertyr� r� � update_executemany_returning� delete_executemany_returningr� r� r� r� r� r� r� � classmethodr� 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� Zmemoized_instancemethodrF r� rG rL rM rN rQ r[ r] r^ r_ r` ra rb rc rd r� r� r� r� rL s sl ���� � ��(R� $ - IrL c @ sP e Zd ZejZejZejZ ej Z dZ dZ dZdZdZdZdZdZdZdZdZdS )�StrCompileDialectTFN)re r� rf r$ ZStrSQLCompilerrh r* ri ZStrSQLTypeCompilerr} rj ry r� r� r� r� rq rn ro rp ri ru rs r� r� r� r� r| � s r| c @ s8 e Zd ZU dZdZdZdZdZdZe j Z de d< dZ de d< dZde d< dZd e d < ejZd e d < ejZdZd e d< dZdZdZdZdZdZ de d< dZ de d< e� � Z de d< e Z!de d< de d< de d< de d < d e d < d e d < d!e d"< d#e d$< e"d%ej�Z#dZ$d&e d'< d(Z%d)e d*< e&dddd d+d,d-�d.d/��Z'e&e(j)fdddd d0d1d2d#d3d,d4� d5d6��Z*e&dddd dd!d,d7�d8d9��Z+e&dddd d,d:�d;d<��Z,dd=�d>d?�Z-e.d@dA� �Z/ej0dBdC� �Z1ej0dDdE� �Z2ej0dFd=�dGdH��Z3ej0dFd=�dIdJ��Z4ej0dKdL� �Z5d�dMdN�Z6ej0dOdP� �Z7dQdR� Z8dSdT� Z9dUdV� Z:dWdX� Z;dYdZ� Z<d[d\� Z=d]d^� Z>d_d`� Z?dadb� Z@dcdd� ZAdedf� ZBejCd)d=�dgdh��ZDe.didj� �ZEdkdl� ZFdmdn� ZGdodp� ZHdqdr� ZIdsdt� ZJej0dudv� �ZKdwdx� ZLdydz� ZMd{d|� ZNd}d~� ZOdd=�d�d��ZPd�d�� ZQd�d�� ZRdZSd�e d�< d�d�d��ZTd�d�� ZUd�d�� ZVd�d�� ZWdS )�r Fr � execute_styleNzOptional[Compiled]r zAOptional[Tuple[List[ResultColumnsEntry], bool, bool, bool, bool]]�result_column_structz Optional[Sequence[Row[Any]]]�returned_default_rowsr7 r zOptional[Executable]�invoked_statementrV � _rowcountz Optional[Callable[[str], str]]�_translate_colnamezMapping[str, List[str]]�_expanded_parametersr �root_connectionr@ �_dbapi_connectionr r� r� �unicode_statementr r z%List[_MutableCoreSingleExecuteParams]r r6 r2 z&Optional[Sequence[BindParameter[Any]]]�extracted_parametersrS z Optional[List[Tuple[Any, ...]]]r! r ra �_num_sentinel_colsr* r )r� r� r� r � compiled_ddlr� c C s� | � | �}||_||_|j|_| |_}d|_||_t|�|_|j rh|j� di �}|j j } | |j|�|_|j|_ |�� |_g |_|jr�|�� g|_n |jg|_|S )zKInitialize execution context for an ExecutableDDLElement construct.Tr )�__new__r� r� r� r �isddlr r� r� r r� ry �_render_schema_translatesr1 � create_cursorr r rw rm r2 �_empty_dict_params) r� r� r� r� r r� r r r �rstr� r� r� � _init_ddl� s2 � � z!DefaultExecutionContext._init_ddlr, r3 rA r ) r� r� r� r r r2 r� r� � cache_hitr� c s� | � | �} || _|| _|j| _�| _|| _� | _| | _|| _� j � j � j � j � j f| _� j | _} � j | _} � j | _} � j| _| s�| s�| �rȈ jj}tr�t|t�s�t�d| _t|j � | _ }t� j � | _ }|r�|j r�d| _!|r�|r�t�|s�|�rȈ j"�r�| �r&| jj#�s&t$�%d| jj&� d���n�| �rV|j'�rV| jj(�sVt$�%d| jj&� d���nr| �rz| jj)�rz� j*�szt$�%d��nN| �r�| jj+�s�t$�%d| jj&� d���n&| �r�| jj,�s�t$�%d| jj&� d���|�s� j-�dd �g| _.nd� �fd d �t/|�D �| _.t0|�d k�rF| j�r>� j*�r>t1j2| _3� j*}|j4d k �rF|j5| _6nt1j7| _3� j8| _9| �:� | _;| jj<�sl| jj=�rt| �>� � j?}|�� j@�s�� jA�r�| jB�r�t$�%d��� �C| j.d �}|j| _9|jD| _EtF|����G|jH� |jI}n� jJ�r�| jjI}nd }� jK�r | j�Ldi �}� jMjN}|| j9|�| _9| j9| _� jJ�rzg }|d k �s@t�| j.D ]*���fdd �|D �}|�O|�P|�� �qF|| _QnZg }� jR�| j.D ]B���r����fdd��D �}n��fdd��D �}|�O|� �q�|| _Q| S )z6Initialize execution context for a Compiled construct.Tr� z] with current server capabilities does not support INSERT..RETURNING when executemany is usedz} with current server capabilities does not support INSERT..RETURNING with deterministic row ordering when executemany is usedznStatement does not have "insertmanyvalues" enabled, can't use INSERT..RETURNING with executemany in this case.z] with current server capabilities does not support UPDATE..RETURNING when executemany is usedz] with current server capabilities does not support DELETE..RETURNING when executemany is usedF)r� � escape_namesc s" g | ]\}}� j |d |�d��qS )F)r� Z _group_numberr� )�construct_params)r� �grp�m)r r� r� r� r� t s ��z:DefaultExecutionContext._init_compiled.<locals>.<listcomp>r NzL'literal_execute' or 'expanding' parameters can't be used with executemany()r r c s, g | ]$}|�kr �| � | �n� | �qS r� r� �r� r ��compiled_params�flattened_processorsr� r� r� � s ��c s6 i | ].}�� ||�|�kr*�| � | �n� | �qS r� )r� r� )r� � escaped_namesr� r� r� r � s � �z:DefaultExecutionContext._init_compiled.<locals>.<dictcomp>c s. i | ]&}||�kr"�| � | �n� | �qS r� r� r� r� r� r� r � s ��)Sr� r� r� r� r� r� r r� r Z_result_columnsZ_ordered_columnsZ_textual_ordered_columnsZ_ad_hoc_textualZ_loose_column_name_matchingr~ �isinsert�isupdate�isdeleteZ isplaintext�is_text� compile_stater1 r r� rF r �is_crudr] Z _returning�_is_explicit_returningrh �_is_implicit_returningZ_supplemental_returning�_is_supplemental_returningZfor_executemanyr� r! r� r� Z_sort_by_parameter_orderr� r^ r ry rz r� r � enumerater� r �INSERTMANYVALUESr} r, r' r� � EXECUTEMANY�stringr� r� r �insert_prefetch�update_prefetch�_process_execute_defaults�_bind_processors�literal_execute_paramsZpost_compile_paramsr8 Z#_process_parameters_for_postcompileZparameter_expansionr� �dictr� � processors� positiontuprw r r� ry r� r� rm r2 �escaped_bind_names)r� r� r� r� r r r2 r� r� r� r �ii�iuZid_Z dml_statementZierZiirr4 r� Zexpanded_stater� r r� Zcore_positional_parametersZl_paramZcore_dict_parametersZd_paramr� )r r� r� r� r� r� �_init_compiled s � � ������������� � ��� � � �� � � z&DefaultExecutionContext._init_compiled)r� r� r� r r1 r2 r� c s� | � | �}||_||_|j|_d|_||_|sP|jjrD� �� g|_q�|j g|_nBt |d � j�rh||_n*t |d t �r~||_n� fdd�|D �|_t |�dkr�t j|_| |_|_|�� |_|S )z8Initialize execution context for a string SQL statement.Tr c s g | ]}� � |��qS r� )rm )r� �pr� r� r� r� s z;DefaultExecutionContext._init_statement.<locals>.<listcomp>r )r� r� r� r� r� r rw rm r2 r� r� r� r� r r� r} r1 r� r� r )r� r� r� r� r r1 r2 r r� r� r� �_init_statement� s, � z'DefaultExecutionContext._init_statement)r� r� r� r r� c C s2 | � | �}||_||_|j|_||_|�� |_|S )z;Initialize execution context for a ColumnDefault construct.)r� r� r� r� r r� r )r� r� r� r� r r r� r� r� � _init_default s z%DefaultExecutionContext._init_defaultr� c C s� | j d krdS t� }| j}| j j}|d k s.t�|tkrDd|| f S |tkrZd|| f S |tkrpd|| f S |tkr�d| j kr�d| j d || f S d|| f S n&|t kr�d| j j | j j || f S d S d S ) Nzraw sqlz no key %.5fszcached since %.4gs agozgenerated in %.5fsZ_cache_disable_reasonz caching disabled (%s) %.5fs zcaching disabled %.5fsz,dialect %s+%s does not support caching %.5fs�unknown)r r r� Z _gen_timer � NO_CACHE_KEY� CACHE_HIT� CACHE_MISS�CACHING_DISABLEDr �NO_DIALECT_SUPPORTr� r� r� )r �now�chZgen_timer� r� r� �_get_cache_stats1 s4 ��z(DefaultExecutionContext._get_cache_statsc C s | j tjtjfkS r� )r} r r� r� r� r� r� r� r8 S s �z#DefaultExecutionContext.executemanyc C s8 | j r| j jS d| jkr,| jj�| jd �S | jjS d S )Nr )r ry r r� rz Z_with_schema_translater� r� r� r� rz Z s �z+DefaultExecutionContext.identifier_preparerc C s | j jS r� )r� r� r� r� r� r� r� e s z DefaultExecutionContext.enginez Optional[Sequence[Column[Any]]]c C s t rt| jt�st�| jjS r� )r r� r r, r � postfetchr� r� r� r� �postfetch_colsi s z&DefaultExecutionContext.postfetch_colsc C s8 t rt| jt�st�| jr"| jjS | jr0| jjS dS d S )Nr� ) r r� r r, r r� r� r� r� r� r� r� r� � prefetch_colso s z%DefaultExecutionContext.prefetch_colsc C s | j �dd�S )N� no_parametersF)r r� r� r� r� r� r� z s z%DefaultExecutionContext.no_parametersc C s� | j }d| jkr0| j�di �}| jj}|||�}|sL| jjrH| j�� }ni }|j| j ||| d� | j � � }|dk r||d }nd}|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 )r3 Nr r ) r� r r� rz r� r� rw rm Z_cursor_executer Zfetchoner r- ) r �stmt�type_r2 �connr r� r �rr r� r� r� �_execute_scalar~ s2 � �z'DefaultExecutionContext._execute_scalarc C s | j S r� )r� r� r� r� r� r� � s z"DefaultExecutionContext.connectionc C sv | j js dS | j jrd| j�dd�o`| jr8t| jjtj �p`| j sPt| jjtj �o`| j o`t � | j �}n| j�dd�}|S )NF�stream_resultsT)r� rr re r r� r r� r1 r& Z SelectableZ TextClauser� �SERVER_SIDE_CURSOR_RE�match)r Zuse_server_sider� r� r� �_use_server_side_cursor� s2 �� ��� � �z/DefaultExecutionContext._use_server_side_cursorc C sF | j jr4| j�dd�s&| j jr4| �� r4d| _| �� S d| _| �� S d S )Nr� FT) r� rr r r� re r� �_is_server_side�create_server_side_cursor�create_default_cursorr� r� r� r� r� � s � ��� z%DefaultExecutionContext.create_cursorc C s |� � S r� )Zfetchall)r r r� r� r� r$ � s z.DefaultExecutionContext.fetchall_for_returningc C s | j �� S r� )r� r r� r� r� r� r� � s z-DefaultExecutionContext.create_default_cursorc C s t � �d S r� �r� r� r� r� r� r� � s z1DefaultExecutionContext.create_server_side_cursorc C s d S r� r� r� r� r� r� �pre_exec� s z DefaultExecutionContext.pre_execc C s t d��d S )Nz,This dialect does not support OUT parametersr� )r rX r� r� r� �get_out_parameter_values� s �z0DefaultExecutionContext.get_out_parameter_valuesc C s d S r� r� r� r� r� r� � 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 r� )r r� ZcolnameZcoltyper� r� r� �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 for an INSERT statement that would need to return the last inserted primary key for those dialects that make use of the lastrowid concept. In these cases, it is called directly after :meth:`.ExecutionContext.post_exec`. )r � lastrowidr� r� r� r� � get_lastrowid� s z%DefaultExecutionContext.get_lastrowidc C s d S r� r� )r r> r� r� r� �handle_dbapi_exception s z.DefaultExecutionContext.handle_dbapi_exceptionc C s | j d k r| j S | jjS d S r� )r� r �rowcountr� r� r� r� r� s z DefaultExecutionContext.rowcountc C s | j d k S r� )r� r� r� r� r� � _has_rowcount s z%DefaultExecutionContext._has_rowcountc C s | j jS r� )r� r� r� r� r� r� r� s z.DefaultExecutionContext.supports_sane_rowcountc C s | j jS r� )r� rt r� r� r� r� rt s z4DefaultExecutionContext.supports_sane_multi_rowcountc C s� | j }| jd kr&|�dd�r&| jj| _| js2| jrD| �� }d }}nl|�dd �}| jp`|�dd�}| j }|r�|t j kr�t � | j| j �}|j p�| jj}|d kr�t j}t �| ||�}| j}|r�| js�tt|�jr�| �|� |j| _|r�|�|�}|S )NZpreserve_rowcountF� yield_perr� )r r� r� r r� r� r� �_setup_dml_or_text_resultr� �cursor_fetch_strategy�_cursor�_DEFAULT_FETCH� BufferedRowCursorFetchStrategy� alternate_cursor_descriptionr- Z_NO_CURSOR_DQL� CursorResultr r� r r, Zhas_out_parameters�_setup_out_parameters� _soft_closedr� )r Zexec_optr5 Zyp�sr�strategy�cursor_descriptionr r� r� r� �_setup_result_proxy s@ ���� � z+DefaultExecutionContext._setup_result_proxyc C s� t t| j�}dd� |j�� D �}i }tdd� |D �| �dd� |D ���D ]P\}}|j}|�| j �}|� | j j �} |� | j | �} | d k r�| |�}|||j < qH||_d S )Nc S s g | ]\}}|j r||f�qS r� )Z isoutparam�r� �paramr� r� r� r� r� L s �zADefaultExecutionContext._setup_out_parameters.<locals>.<listcomp>c S s g | ] \}}|�qS r� r� r� r� r� r� r� T s c S s g | ] \}}|�qS r� r� r� r� r� r� r� V s )r r, r � bind_namesr� r r� r� Z dialect_implr� Zget_dbapi_typer� �result_processorr �out_parameters) r r5 r Zout_bindparamsr� � bindparamZ raw_valuer� Z impl_typeZ dbapi_typer� r� r� r� r� I s, � �� � z-DefaultExecutionContext._setup_out_parametersc C s� t t| j�}| j}| jrP| jtjkr@|jr@t j | j | j |j d�}|jrP| �� | _| jrp|t jkrpt �| j | j�}|t jkr�d }n|j p�| j j}|d kr�t j}n&| jr�| jtjks�t�|d| j � }t �| ||�}| j�r<| j�r*|�� }|| _ | � ||�| _|j j �s t�| j �r |�!|� n|�"� n| j#�s�|�"� n�| j�r�|�� }|�rX|| _ t$|�| _%| j �rv|�!|� n|�"� |j j �s�t�nP|j j �s�| j%d k�r�| j j&| _%|�"� n&| j'�s�| j(�r�| j%d k�r�| j j&| _%|S )N)Zinitial_bufferZalternate_descriptionr ))r r, r r� r� r} r r� r r� Z FullyBufferedCursorFetchStrategyr r! r� rr � _setup_ins_pk_from_lastrowid�inserted_primary_key_rowsr� r� r� r Z_NO_CURSOR_DMLr- r� r r� r� �allr �%_setup_ins_pk_from_implicit_returning� _metadataZ returns_rowsr� �_rewindZ _soft_closer� r� r� r� r� r� )r r r� r� r5 r6 r� r� r� r� e s� ��� � � � � � � z1DefaultExecutionContext._setup_dml_or_text_resultc C s | � � S r� )�_setup_ins_pk_from_emptyr� r� r� r� r� � s z1DefaultExecutionContext.inserted_primary_key_rowsc C s( t t| j�j}| �� }||| jd �gS )Nr )r r, r �+_inserted_primary_key_from_lastrowid_getterr� r )r �getterr� r� r� r� r� � s �z4DefaultExecutionContext._setup_ins_pk_from_lastrowidc s" t t| j�j� � fdd�| jD �S )Nc s g | ]}� d |��qS r� r� )r� r� �r� r� r� r� � s zDDefaultExecutionContext._setup_ins_pk_from_empty.<locals>.<listcomp>)r r, r r� r r� r� r� r� r� � s �z0DefaultExecutionContext._setup_ins_pk_from_emptyc s4 |sg S t t| j�j� | j}� fdd�t||�D �S )Nc s g | ]\}}� ||��qS r� r� )r� r r� r� r� r� r� � s zQDefaultExecutionContext._setup_ins_pk_from_implicit_returning.<locals>.<listcomp>)r r, r Z+_inserted_primary_key_from_returning_getterr r )r r5 r6 r� r� r� r� r� � s � �z=DefaultExecutionContext._setup_ins_pk_from_implicit_returningc C s | j s | jo ttt| j�j�S r� )r� r� r] r r, r r� r� r� r� r� �lastrow_has_defaults� s �z,DefaultExecutionContext.lastrow_has_defaultsz0Optional[List[Tuple[str, Any, TypeEngine[Any]]]]c s� | j s | jrdS tt| j����� }|dkr0dS | j}|jr^t|�}|j � || j | j | j | � �jrl�j�nd�|jr��fdd��jp�dD �}ndd� �j�� D �}g }|D ]�\}� � �jkr�q�|| jk�rFt� j��rt� jj��|� �|�� ���fdd�t| j| �D �� n.|�� d��|�� ��fdd�| j| D �� q�|�� d���dk �rh��||�n|}|� |�� jf� q�|S ) a� Given a cursor and ClauseParameters, prepare arguments in order to 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 set the :attr:`.Dialect.bind_typing` attribute to :attr:`.BindTyping.SETINPUTSIZES`. cx_Oracle is the only DBAPI that requires setinputsizes(), pyodbc offers it as an option. Prior to SQLAlchemy 2.0, the setinputsizes() approach was also used for pg8000 and asyncpg, which has been changed to inline rendering of casts. Nc s g | ]}|� j | f�qS r� )Zbindsr� )r r� r� r� ) s �zDDefaultExecutionContext._prepare_set_input_sizes.<locals>.<listcomp>r� c S s g | ]\}}||f�qS r� r� )r� r� r r� r� r� r� . s �c 3 sD | ]<\}}�d k r �� ||�n|�|� � jj|� fV qd S r� )r� r� �types)r� r� � paramname)r� �dbtypesr� �numr� r� r� < s �� �zCDefaultExecutionContext._prepare_set_input_sizes.<locals>.<genexpr>c 3 s. | ]&}�d k r�� ||�n|�� jfV qd S r� )r� r� )r� r� )r� �dbtyper� r� r� r� L s ���) r� r� r r, r Z_get_set_input_sizes_lookupr� r� r� �dispatchZdo_setinputsizesr r1 r2 r� rw r� r� r� r� r� r( r� r� r� r( r� r� r� )r Z inputsizesr� r� Zgeneric_inputsizesr Z escaped_namer� )r� r r� r� r� r� r� �_prepare_set_input_sizes� sh � �� �� � �� �z0DefaultExecutionContext._prepare_set_input_sizesc C sF |j r| �||�S |jr(|| _|�| �S |jr<| �|||�S |jS d S r� )Z is_sequenceZ fire_sequence� is_callable�current_columnr� Zis_clause_element� _exec_default_clause_element)r �columnrU r� r� r� r� � _exec_defaultg s z%DefaultExecutionContext._exec_defaultc s� |j st�|j|�}n|j}t�|�j| jd�}|�� � |j�|j rh| j� � �fdd�|j p^dD ��}n� �fdd�� D �}| j t |�||d�S )Nr� c s, g | ]$}|�kr �| � | �n� | �qS r� r� r� �r� r� r� r� r� � s ��zHDefaultExecutionContext._exec_default_clause_element.<locals>.<listcomp>r� c s. i | ]&}||�kr"�| � | �n� | �qS r� r� r� r r� r� r � s ��zHDefaultExecutionContext._exec_default_clause_element.<locals>.<dictcomp>)r2 )Z _arg_is_typedr& Z type_coercer� r� r� r� r� r� rw rm r� r� r� )r r rU r� Z default_argr r2 r� r r� r v s( �� � �z4DefaultExecutionContext._exec_default_clause_elementz"Optional[_CoreSingleExecuteParams]�current_parametersTc s� z| j �| j}W n tk r. t�d��Y nX |dk s<t��dk sHt�tdtt| j�j �}|dk sht�|r�t � |�r�|j r�|j r�|jd � |jj�|j i}n|j�|j i}d� |jdk s�t�|j�� }|�� �fdd�|D �� |S �S dS )ai 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:`_schema.Column`, which is usually synonymous with the name. :param isolate_multiinsert_groups=True: indicates that multi-valued INSERT constructs created using :meth:`_expression.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 functionNrE r r c 3 s" | ]}|�d |� f fV qdS )z%s_m%dNr� r� ��indexr2 r� r� r� � s zADefaultExecutionContext.get_current_parameters.<locals>.<genexpr>)r r �AttributeErrorr! r� r r r, r r� r% r� Z_has_multi_parametersZ_is_multiparam_columnr �originalr Z_dict_parameters�keysr� )r Zisolate_multiinsert_groupsr r� �dr r� r r� �get_current_parameters� s@ ! � � ��� �z.DefaultExecutionContext.get_current_parametersc C s$ |j d krd S | �||j |j�S d S r� )rU r r� �r r r� r� r� �get_insert_default� s z*DefaultExecutionContext.get_insert_defaultc C s$ |j d krd S | �||j |j�S d S r� )Zonupdater r� r r� r� r� �get_update_default s z*DefaultExecutionContext.get_update_defaultc s� t t�j�}|j� d}|jr4� �fdd�|jD �}n"|jrR� �fdd�|jD �}ng }�jD ]x}|�_|D ]h\}}\}}} } } | r�|||< |d7 }qj|r�|||< qj| r�|�_|��||< qj| |�} | d k rj| ||< qjq\�`d S )Nr c s g | ]}|� |�|j �jf�qS r� )Z_default_description_tupler �r� �c�Z key_getterr r� r� r� s ��zEDefaultExecutionContext._process_execute_defaults.<locals>.<listcomp>c s g | ]}|� |�|j �jf�qS r� )Z_onupdate_description_tupler r r r� r� r� s ��r ) r r, r Z _within_exec_param_key_getterr� r� r r r ) r r Zsentinel_counterZ prefetch_recsr� r Z param_keyr� Z is_scalarr Z is_sentinel�fallbackr r� r r� r� s@ � � � z1DefaultExecutionContext._process_execute_defaults)N)T)Xre r� rf r� r� r� r� r� r� r ZEXECUTEr} rl r r~ r r# � EMPTY_DICTr 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� rv r8 rx rz r� r� r� r� r� r� r� r� r$ r� r� r� r� r� r� r� r� Znon_memoized_propertyr� r� r� rt r� r� r� r� r� r� r� r� r� r r r r r r r� r� r� r� r� r � s� � , �$ f )" ) * r j' Dr )prg � __future__r r� r* r �re�timer �typingr r r r r r r r r r r r r r r r� � r r r� r �baser r r r r r Z reflectionr r r r! r" r# �sqlr$ r% r& r' r& Z sql._typingr( Zsql.baser) Z sql.compilerr* r+ r, Z sql.elementsr- Z util.typingr. r/ r� r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r r< r� r= r> r? r@ rA rB rC rD Zsql.dmlrE rF rG Z sql.schemarH Z sql.type_apirI rJ rK r� �I�UNICODEr� rC r� r� r� r� r� rL r| r Zexecution_ctx_clsr� r� r� r� �<module> s� � " !