관리-도구
편집 파일: oracledb.cpython-38.pyc
U -?�f35 � @ sz d Z ddlmZ ddlZddlZddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd l mZ dd l mZ dd l mZ ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ er�ddlmZ ddlmZ G dd� de j�Z G dd� de j �Z G dd� de�Z G dd � d ee �Z G d d � d e�Z!G d d � d ee!�Z"G d!d"� d"�Z#G d#d$� d$e �Z$G d%d&� d&e �Z%e Z&e%Z'dS )'a� .. dialect:: oracle+oracledb :name: python-oracledb :dbapi: oracledb :connectstring: oracle+oracledb://user:pass@hostname:port[/dbname][?service_name=<service>[&key=value&key=value...]] :url: https://oracle.github.io/python-oracledb/ Description ----------- python-oracledb is released by Oracle to supersede the cx_Oracle driver. It is fully compatible with cx_Oracle and features both a "thin" client mode that requires no dependencies, as well as a "thick" mode that uses the Oracle Client Interface in the same way as cx_Oracle. .. seealso:: :ref:`cx_oracle` - all of cx_Oracle's notes apply to the oracledb driver as well, with the exception that oracledb supports two phase transactions. The SQLAlchemy ``oracledb`` dialect provides both a sync and an async implementation under the same dialect name. The proper version is selected depending on how the engine is created: * calling :func:`_sa.create_engine` with ``oracle+oracledb://...`` will automatically select the sync version, e.g.:: from sqlalchemy import create_engine sync_engine = create_engine("oracle+oracledb://scott:tiger@localhost/?service_name=XEPDB1") * calling :func:`_asyncio.create_async_engine` with ``oracle+oracledb://...`` will automatically select the async version, e.g.:: from sqlalchemy.ext.asyncio import create_async_engine asyncio_engine = create_async_engine("oracle+oracledb://scott:tiger@localhost/?service_name=XEPDB1") The asyncio version of the dialect may also be specified explicitly using the ``oracledb_async`` suffix, as:: from sqlalchemy.ext.asyncio import create_async_engine asyncio_engine = create_async_engine("oracle+oracledb_async://scott:tiger@localhost/?service_name=XEPDB1") .. versionadded:: 2.0.25 added support for the async version of oracledb. Thick mode support ------------------ By default the ``python-oracledb`` is started in thin mode, that does not require oracle client libraries to be installed in the system. The ``python-oracledb`` driver also support a "thick" mode, that behaves similarly to ``cx_oracle`` and requires that Oracle Client Interface (OCI) is installed. To enable this mode, the user may call ``oracledb.init_oracle_client`` manually, or by passing the parameter ``thick_mode=True`` to :func:`_sa.create_engine`. To pass custom arguments to ``init_oracle_client``, like the ``lib_dir`` path, a dict may be passed to this parameter, as in:: engine = sa.create_engine("oracle+oracledb://...", thick_mode={ "lib_dir": "/path/to/oracle/client/lib", "driver_name": "my-app" }) .. seealso:: https://python-oracledb.readthedocs.io/en/latest/api_manual/module.html#oracledb.init_oracle_client Two Phase Transactions Supported -------------------------------- Two phase transactions are fully supported under oracledb. Starting with oracledb 2.3 two phase transactions are supported also in thin mode. APIs for two phase transactions are provided at the Core level via :meth:`_engine.Connection.begin_twophase` and :paramref:`_orm.Session.twophase` for transparent ORM use. .. versionchanged:: 2.0.32 added support for two phase transactions .. versionadded:: 2.0.0 added support for oracledb driver. � )� annotationsN)�Any)� TYPE_CHECKING� )� cx_oracle� )�exc)�pool)�AsyncAdapt_dbapi_connection)�AsyncAdapt_dbapi_cursor)�AsyncAdapt_dbapi_ss_cursor)�#AsyncAdaptFallback_dbapi_connection)�default)�asbool)�await_fallback)� await_only)�AsyncConnection)� AsyncCursorc @ s e Zd ZdS )� OracleExecutionContext_oracledbN)�__name__� __module__� __qualname__� r r �S/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/oracle/oracledb.pyr r s r c s� e Zd ZdZeZdZdZd� fdd� Ze dd� �Z e d d � �Z e d d � �Z d d� Z dd� Zdd� Zddd�Zd dd�Zdd� Z� ZS ) �OracleDialect_oracledbT�oracledb)r Nc sR t � j||||f|� | jd k rN|s.t|t�rNt|t�r<|ni }| jjf |� d S �N)�super�__init__Zdbapi� isinstance�dictZinit_oracle_client)�selfZauto_convert_lobsZcoerce_to_decimal� arraysizeZencoding_errorsZ thick_mode�kwargs�kw�� __class__r r r s �� ��z OracleDialect_oracledb.__init__c C s dd l }|S �Nr )r ��clsr r r r � import_dbapi� s z#OracleDialect_oracledb.import_dbapic C s |j jjS r )� connectionZdbapi_connection�thin)r) r+ r r r � is_thin_mode� s z#OracleDialect_oracledb.is_thin_modec C s t S r )�OracleDialectAsync_oracledb)r) �urlr r r �get_async_dialect_cls� s z,OracleDialect_oracledb.get_async_dialect_clsc C sn d}|d k r:t �d|j�}|r:tdd� |�ddd�D ��}|| _| jdkrj| j| jk rjt�d| j� d ���d S ) N)r r r z(\d+)\.(\d+)(?:\.(\d+))?c s s | ]}|d k rt |�V qd S r )�int)�.0�xr r r � <genexpr>� s z7OracleDialect_oracledb._load_version.<locals>.<genexpr>r � r zoracledb version z and above are supported) �re�match�version�tuple�groupZ oracledb_ver� _min_versionr ZInvalidRequestError)r! Z dbapi_moduler8 �mr r r � _load_version� s �� � �z$OracleDialect_oracledb._load_versionc C s( |j j|� }|j �|� ||j jd< d S )NZ oracledb_xid)r+ �xid� tpc_begin�info)r! r+ r> Zconn_xisr r r �do_begin_twophase� s z(OracleDialect_oracledb.do_begin_twophasec C s |j �� }||jd< d S )N�oracledb_should_commit)r+ � tpc_preparer@ )r! r+ r> � should_commitr r r �do_prepare_twophase� s z*OracleDialect_oracledb.do_prepare_twophaseFc C s& |r|j j|� }nd }|j �|� d S r )r+ r> � tpc_rollback)r! r+ r> � is_prepared�recover�conn_xidr r r �do_rollback_twophase� s z+OracleDialect_oracledb.do_rollback_twophasec C sH d }|s|j �� }n |r*|j j|� }d}n |jd }|rD|j �|� d S )NTrB )r+ rC r> r@ � tpc_commit)r! r+ r> rG rH rI rD r r r �do_commit_twophase� s z)OracleDialect_oracledb.do_commit_twophasec C s dd� |j �� D �S )Nc S s@ g | ]8\}}}|t |t�r"|�� n|t |t�r6|�� n|f�qS r )r �bytes�decode)r2 �fiZgti�bqr r r � <listcomp>� s ��z>OracleDialect_oracledb.do_recover_twophase.<locals>.<listcomp>)r+ � tpc_recover�r! r+ r r r �do_recover_twophase� s �z*OracleDialect_oracledb.do_recover_twophase)TTNNN)TF)TF)r r r �supports_statement_cacher �execution_ctx_clsZdriverr; r � classmethodr* r- r0 r= rA rE rJ rL rT � __classcell__r r r% r r x s4 � � � r c @ s� e Zd ZU ded< dZedd� �Zejdd� �Zdd� Zd d � Z d d d d �d d�Z ddd�dd�Z dd� Z dd� Z dd� Zd d d dd�dd�Zd S ) �AsyncAdapt_oracledb_cursorr �_cursorr c C s | j jS r �rZ �outputtypehandler�r! r r r r\ � s z,AsyncAdapt_oracledb_cursor.outputtypehandlerc C s || j _d S r r[ �r! �valuer r r r\ � s c O s | j j||�S r )rZ �var�r! �argsr# r r r r` � s z AsyncAdapt_oracledb_cursor.varc C s | j �� | j�� d S r )�_rows�clearrZ �closer] r r r re � s z AsyncAdapt_oracledb_cursor.closer �rb r# �returnc O s | j j||�S r )rZ � setinputsizesra r r r rh � s z(AsyncAdapt_oracledb_cursor.setinputsizes)�cursorrg c C s> z |� � W S tk r8 } z| j�|� W 5 d }~X Y nX d S r )� __enter__� ExceptionZ_adapt_connectionZ_handle_exception)r! ri �errorr r r �_aenter_cursor� s z)AsyncAdapt_oracledb_cursor._aenter_cursorc � sZ |d kr | j �|�I d H }n| j �||�I d H }| j jrV| jsVt�| j �� I d H �| _|S r )rZ �execute� description� server_side� collections�dequeZfetchallrc )r! � operation� parameters�resultr r r �_execute_async� s z)AsyncAdapt_oracledb_cursor._execute_asyncc � s | j �||�I d H S r )rZ Z executemany)r! rs Zseq_of_parametersr r r �_executemany_async s z-AsyncAdapt_oracledb_cursor._executemany_asyncc C s | S r r r] r r r rj s z$AsyncAdapt_oracledb_cursor.__enter__�None)�type_r_ � tracebackrg c C s | � � d S r )re )r! ry r_ rz r r r �__exit__ s z#AsyncAdapt_oracledb_cursor.__exit__N)r r r �__annotations__� __slots__�propertyr\ �setterr` re rh rm rv rw rj r{ r r r r rY � s rY c @ s e Zd ZdZdd�dd�ZdS )� AsyncAdapt_oracledb_ss_cursorr rx �rg c C s | j d k r| j �� d | _ d S r )rZ re r] r r r re s z#AsyncAdapt_oracledb_ss_cursor.closeN)r r r r} re r r r r r� s r� c @ s� e Zd ZU ded< dZdZeZdZe dd� �Z e j dd� �Z e d d � �Z e j d d � �Z e d d � �Z e dd� �Zej dd� �Zdd� Zdd� Zdddd�dd�Zdddd�dd�Zdddd�dd �Zdddd�d d �Zdddd�d d �Zdddd�d!d"�ZdS )#� AsyncAdapt_oracledb_connectionr � _connectionr TNc C s | j jS r �r� � autocommitr] r r r r� , s z)AsyncAdapt_oracledb_connection.autocommitc C s || j _d S r r� r^ r r r r� 0 s c C s | j jS r �r� r\ r] r r r r\ 4 s z0AsyncAdapt_oracledb_connection.outputtypehandlerc C s || j _d S r r� r^ r r r r\ 8 s c C s | j jS r )r� r8 r] r r r r8 < s z&AsyncAdapt_oracledb_connection.versionc C s | j jS r �r� � stmtcachesizer] r r r r� @ s z,AsyncAdapt_oracledb_connection.stmtcachesizec C s || j _d S r r� r^ r r r r� D s c C s t | �S r )rY r] r r r ri H s z%AsyncAdapt_oracledb_connection.cursorc C s t | �S r )r� r] r r r � ss_cursorK s z(AsyncAdapt_oracledb_connection.ss_cursorr rf c O s | j j||�S r )r� r> ra r r r r> N s z"AsyncAdapt_oracledb_connection.xidc O s | � | jj||��S r )�await_r� r? ra r r r r? Q s z(AsyncAdapt_oracledb_connection.tpc_beginc O s | � | jj||��S r )r� r� rK ra r r r rK T s z)AsyncAdapt_oracledb_connection.tpc_commitc O s | � | jj||��S r )r� r� rC ra r r r rC W s z*AsyncAdapt_oracledb_connection.tpc_preparec O s | � | jj||��S r )r� r� rR ra r r r rR Z s z*AsyncAdapt_oracledb_connection.tpc_recoverc O s | � | jj||��S r )r� r� rF ra r r r rF ] s z+AsyncAdapt_oracledb_connection.tpc_rollback)r r r r| r} r, rY Z _cursor_clsZ_ss_cursor_clsr~ r� r r\ r8 r� ri r� r> r? rK rC rR rF r r r r r� # s6 r� c @ s e Zd ZdZdS )�&AsyncAdaptFallback_oracledb_connectionr N)r r r r} r r r r r� a s r� c @ s" e Zd Zdd�dd�Zdd� ZdS )�OracledbAdaptDBAPIrx r� c C s2 || _ | j j�� D ]\}}|dkr|| j|< qd S )N�connect)r �__dict__�items)r! r �k�vr r r r h s zOracledbAdaptDBAPI.__init__c O sP |� dd�}|� d| jj�}t|�r8t| t|||���S t| t|||���S d S )N�async_fallbackFZasync_creator_fn)�popr Z connect_asyncr r� r r� r )r! �argr$ r� Z creator_fnr r r r� o s � �zOracledbAdaptDBAPI.connectN)r r r r r� r r r r r� g s r� c @ s$ e Zd ZejjZdd� Zdd� ZdS )�$OracleExecutionContextAsync_oracledbc C s | j �� }| jjr | jj|_|S r )�_dbapi_connectionri �dialectr" �r! �cr r r �create_default_cursor� s z:OracleExecutionContextAsync_oracledb.create_default_cursorc C s | j �� }| jjr | jj|_|S r )r� r� r� r" r� r r r �create_server_side_cursor� s z>OracleExecutionContextAsync_oracledb.create_server_side_cursorN)r r r r ZDefaultExecutionContextZ create_cursorr� r� r r r r r� ~ s r� c @ s@ e Zd ZdZdZdZeZdZe dd� �Z e dd� �Z dd� Z d S ) r. T)r5 c C s dd l }t|�S r' )r r� r( r r r r* � s z(OracleDialectAsync_oracledb.import_dbapic C s& |j �dd�}t|�r tjS tjS d S )Nr� F)�query�getr r Z FallbackAsyncAdaptedQueuePoolZAsyncAdaptedQueuePool)r) r/ r� r r r �get_pool_class� s z*OracleDialectAsync_oracledb.get_pool_classc C s |j S r )r� rS r r r �get_driver_connection� s z1OracleDialectAsync_oracledb.get_driver_connectionN) r r r Zis_asyncZ supports_server_side_cursorsrU r� rV r; rW r* r� r� r r r r r. � s r. )(�__doc__� __future__r rq r6 �typingr r � r Z _cx_oracler r Zconnectors.asyncior r r r Zenginer �utilr r r r r r Z OracleExecutionContext_cx_oracler ZOracleDialect_cx_oracler rY r� r� r� r� r� r. r� Z dialect_asyncr r r r �<module> sL Q �i7 � > �