관리-도구
편집 파일: pyodbc.cpython-38.pyc
U -?�f�i � @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z G dd� d�Z G dd� de ej �Z!G dd� de ej"�Z#G dd � d �Z$G d d � d �Z%G d d � d e%e�Z&G d!d"� d"e%e �Z'G d#d$� d$e$e �Z(G d%d&� d&e$e �Z)G d'd(� d(ej*�Z+G d)d*� d*e�Z,G d+d,� d,e�Z-G d-d.� d.e�Z.G d/d0� d0e�Z/G d1d2� d2e�Z0G d3d4� d4e �Z1G d5d6� d6ee �Z2e2Z3dS )7ay6 .. dialect:: mssql+pyodbc :name: PyODBC :dbapi: pyodbc :connectstring: mssql+pyodbc://<username>:<password>@<dsnname> :url: https://pypi.org/project/pyodbc/ Connecting to PyODBC -------------------- The URL here is to be translated to PyODBC connection strings, as detailed in `ConnectionStrings <https://code.google.com/p/pyodbc/wiki/ConnectionStrings>`_. DSN Connections ^^^^^^^^^^^^^^^ A DSN connection in ODBC means that a pre-existing ODBC datasource is configured on the client machine. The application then specifies the name of this datasource, which encompasses details such as the specific ODBC driver in use as well as the network address of the database. Assuming a datasource is configured on the client, a basic DSN-based connection looks like:: engine = create_engine("mssql+pyodbc://scott:tiger@some_dsn") Which above, will pass the following connection string to PyODBC:: DSN=some_dsn;UID=scott;PWD=tiger If the username and password are omitted, the DSN form will also add the ``Trusted_Connection=yes`` directive to the ODBC string. Hostname Connections ^^^^^^^^^^^^^^^^^^^^ Hostname-based connections are also supported by pyodbc. These are often easier to use than a DSN and have the additional advantage that the specific database name to connect towards may be specified locally in the URL, rather than it being fixed as part of a datasource configuration. When using a hostname connection, the driver name must also be specified in the query parameters of the URL. As these names usually have spaces in them, the name must be URL encoded which means using plus signs for spaces:: engine = create_engine("mssql+pyodbc://scott:tiger@myhost:port/databasename?driver=ODBC+Driver+17+for+SQL+Server") The ``driver`` keyword is significant to the pyodbc dialect and must be specified in lowercase. Any other names passed in the query string are passed through in the pyodbc connect string, such as ``authentication``, ``TrustServerCertificate``, etc. Multiple keyword arguments must be separated by an ampersand (``&``); these will be translated to semicolons when the pyodbc connect string is generated internally:: e = create_engine( "mssql+pyodbc://scott:tiger@mssql2017:1433/test?" "driver=ODBC+Driver+18+for+SQL+Server&TrustServerCertificate=yes" "&authentication=ActiveDirectoryIntegrated" ) The equivalent URL can be constructed using :class:`_sa.engine.URL`:: from sqlalchemy.engine import URL connection_url = URL.create( "mssql+pyodbc", username="scott", password="tiger", host="mssql2017", port=1433, database="test", query={ "driver": "ODBC Driver 18 for SQL Server", "TrustServerCertificate": "yes", "authentication": "ActiveDirectoryIntegrated", }, ) Pass through exact Pyodbc string ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ A PyODBC connection string can also be sent in pyodbc's format directly, as specified in `the PyODBC documentation <https://github.com/mkleehammer/pyodbc/wiki/Connecting-to-databases>`_, using the parameter ``odbc_connect``. A :class:`_sa.engine.URL` object can help make this easier:: from sqlalchemy.engine import URL connection_string = "DRIVER={SQL Server Native Client 10.0};SERVER=dagger;DATABASE=test;UID=user;PWD=password" connection_url = URL.create("mssql+pyodbc", query={"odbc_connect": connection_string}) engine = create_engine(connection_url) .. _mssql_pyodbc_access_tokens: Connecting to databases with access tokens ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Some database servers are set up to only accept access tokens for login. For example, SQL Server allows the use of Azure Active Directory tokens to connect to databases. This requires creating a credential object using the ``azure-identity`` library. More information about the authentication step can be found in `Microsoft's documentation <https://docs.microsoft.com/en-us/azure/developer/python/azure-sdk-authenticate?tabs=bash>`_. After getting an engine, the credentials need to be sent to ``pyodbc.connect`` each time a connection is requested. One way to do this is to set up an event listener on the engine that adds the credential token to the dialect's connect call. This is discussed more generally in :ref:`engines_dynamic_tokens`. For SQL Server in particular, this is passed as an ODBC connection attribute with a data structure `described by Microsoft <https://docs.microsoft.com/en-us/sql/connect/odbc/using-azure-active-directory#authenticating-with-an-access-token>`_. The following code snippet will create an engine that connects to an Azure SQL database using Azure credentials:: import struct from sqlalchemy import create_engine, event from sqlalchemy.engine.url import URL from azure import identity SQL_COPT_SS_ACCESS_TOKEN = 1256 # Connection option for access tokens, as defined in msodbcsql.h TOKEN_URL = "https://database.windows.net/" # The token URL for any Azure SQL database connection_string = "mssql+pyodbc://@my-server.database.windows.net/myDb?driver=ODBC+Driver+17+for+SQL+Server" engine = create_engine(connection_string) azure_credentials = identity.DefaultAzureCredential() @event.listens_for(engine, "do_connect") def provide_token(dialect, conn_rec, cargs, cparams): # remove the "Trusted_Connection" parameter that SQLAlchemy adds cargs[0] = cargs[0].replace(";Trusted_Connection=Yes", "") # create token credential raw_token = azure_credentials.get_token(TOKEN_URL).token.encode("utf-16-le") token_struct = struct.pack(f"<I{len(raw_token)}s", len(raw_token), raw_token) # apply it to keyword arguments cparams["attrs_before"] = {SQL_COPT_SS_ACCESS_TOKEN: token_struct} .. tip:: The ``Trusted_Connection`` token is currently added by the SQLAlchemy pyodbc dialect when no username or password is present. This needs to be removed per Microsoft's `documentation for Azure access tokens <https://docs.microsoft.com/en-us/sql/connect/odbc/using-azure-active-directory#authenticating-with-an-access-token>`_, stating that a connection string when using an access token must not contain ``UID``, ``PWD``, ``Authentication`` or ``Trusted_Connection`` parameters. .. _azure_synapse_ignore_no_transaction_on_rollback: Avoiding transaction-related exceptions on Azure Synapse Analytics ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Azure Synapse Analytics has a significant difference in its transaction handling compared to plain SQL Server; in some cases an error within a Synapse transaction can cause it to be arbitrarily terminated on the server side, which then causes the DBAPI ``.rollback()`` method (as well as ``.commit()``) to fail. The issue prevents the usual DBAPI contract of allowing ``.rollback()`` to pass silently if no transaction is present as the driver does not expect this condition. The symptom of this failure is an exception with a message resembling 'No corresponding transaction found. (111214)' when attempting to emit a ``.rollback()`` after an operation had a failure of some kind. This specific case can be handled by passing ``ignore_no_transaction_on_rollback=True`` to the SQL Server dialect via the :func:`_sa.create_engine` function as follows:: engine = create_engine(connection_url, ignore_no_transaction_on_rollback=True) Using the above parameter, the dialect will catch ``ProgrammingError`` exceptions raised during ``connection.rollback()`` and emit a warning if the error message contains code ``111214``, however will not raise an exception. .. versionadded:: 1.4.40 Added the ``ignore_no_transaction_on_rollback=True`` parameter. Enable autocommit for Azure SQL Data Warehouse (DW) connections ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Azure SQL Data Warehouse does not support transactions, and that can cause problems with SQLAlchemy's "autobegin" (and implicit commit/rollback) behavior. We can avoid these problems by enabling autocommit at both the pyodbc and engine levels:: connection_url = sa.engine.URL.create( "mssql+pyodbc", username="scott", password="tiger", host="dw.azure.example.com", database="mydb", query={ "driver": "ODBC Driver 17 for SQL Server", "autocommit": "True", }, ) engine = create_engine(connection_url).execution_options( isolation_level="AUTOCOMMIT" ) Avoiding sending large string parameters as TEXT/NTEXT ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ By default, for historical reasons, Microsoft's ODBC drivers for SQL Server send long string parameters (greater than 4000 SBCS characters or 2000 Unicode characters) as TEXT/NTEXT values. TEXT and NTEXT have been deprecated for many years and are starting to cause compatibility issues with newer versions of SQL_Server/Azure. For example, see `this issue <https://github.com/mkleehammer/pyodbc/issues/835>`_. Starting with ODBC Driver 18 for SQL Server we can override the legacy behavior and pass long strings as varchar(max)/nvarchar(max) using the ``LongAsMax=Yes`` connection string parameter:: connection_url = sa.engine.URL.create( "mssql+pyodbc", username="scott", password="tiger", host="mssqlserver.example.com", database="mydb", query={ "driver": "ODBC Driver 18 for SQL Server", "LongAsMax": "Yes", }, ) Pyodbc Pooling / connection close behavior ------------------------------------------ PyODBC uses internal `pooling <https://github.com/mkleehammer/pyodbc/wiki/The-pyodbc-Module#pooling>`_ by default, which means connections will be longer lived than they are within SQLAlchemy itself. As SQLAlchemy has its own pooling behavior, it is often preferable to disable this behavior. This behavior can only be disabled globally at the PyODBC module level, **before** any connections are made:: import pyodbc pyodbc.pooling = False # don't use the engine before pooling is set to False engine = create_engine("mssql+pyodbc://user:pass@dsn") If this variable is left at its default value of ``True``, **the application will continue to maintain active database connections**, even when the SQLAlchemy engine itself fully discards a connection or if the engine is disposed. .. seealso:: `pooling <https://github.com/mkleehammer/pyodbc/wiki/The-pyodbc-Module#pooling>`_ - in the PyODBC documentation. Driver / Unicode Support ------------------------- PyODBC works best with Microsoft ODBC drivers, particularly in the area of Unicode support on both Python 2 and Python 3. Using the FreeTDS ODBC drivers on Linux or OSX with PyODBC is **not** recommended; there have been historically many Unicode-related issues in this area, including before Microsoft offered ODBC drivers for Linux and OSX. Now that Microsoft offers drivers for all platforms, for PyODBC support these are recommended. FreeTDS remains relevant for non-ODBC drivers such as pymssql where it works very well. Rowcount Support ---------------- Previous limitations with the SQLAlchemy ORM's "versioned rows" feature with Pyodbc have been resolved as of SQLAlchemy 2.0.5. See the notes at :ref:`mssql_rowcount_versioning`. .. _mssql_pyodbc_fastexecutemany: Fast Executemany Mode --------------------- The PyODBC driver includes support for a "fast executemany" mode of execution which greatly reduces round trips for a DBAPI ``executemany()`` call when using Microsoft ODBC drivers, for **limited size batches that fit in memory**. The feature is enabled by setting the attribute ``.fast_executemany`` on the DBAPI cursor when an executemany call is to be used. The SQLAlchemy PyODBC SQL Server dialect supports this parameter by passing the ``fast_executemany`` parameter to :func:`_sa.create_engine` , when using the **Microsoft ODBC driver only**:: engine = create_engine( "mssql+pyodbc://scott:tiger@mssql2017:1433/test?driver=ODBC+Driver+17+for+SQL+Server", fast_executemany=True) .. versionchanged:: 2.0.9 - the ``fast_executemany`` parameter now has its intended effect of this PyODBC feature taking effect for all INSERT statements that are executed with multiple parameter sets, which don't include RETURNING. Previously, SQLAlchemy 2.0's :term:`insertmanyvalues` feature would cause ``fast_executemany`` to not be used in most cases even if specified. .. versionadded:: 1.3 .. seealso:: `fast executemany <https://github.com/mkleehammer/pyodbc/wiki/Features-beyond-the-DB-API#fast_executemany>`_ - on github .. _mssql_pyodbc_setinputsizes: Setinputsizes Support ----------------------- As of version 2.0, the pyodbc ``cursor.setinputsizes()`` method is used for all statement executions, except for ``cursor.executemany()`` calls when fast_executemany=True where it is not supported (assuming :ref:`insertmanyvalues <engine_insertmanyvalues>` is kept enabled, "fastexecutemany" will not take place for INSERT statements in any case). The use of ``cursor.setinputsizes()`` can be disabled by passing ``use_setinputsizes=False`` to :func:`_sa.create_engine`. When ``use_setinputsizes`` is left at its default of ``True``, the specific per-type symbols passed to ``cursor.setinputsizes()`` can be programmatically customized using the :meth:`.DialectEvents.do_setinputsizes` hook. See that method for usage examples. .. versionchanged:: 2.0 The mssql+pyodbc dialect now defaults to using ``use_setinputsizes=True`` for all statement executions with the exception of cursor.executemany() calls when fast_executemany=True. The behavior can be turned off by passing ``use_setinputsizes=False`` to :func:`_sa.create_engine`. � N� )� _MSDateTime)� _MSUnicode)�_MSUnicodeText)�BINARY)�DATETIMEOFFSET)� MSDialect)�MSExecutionContext)� VARBINARY)�JSON)� JSONIndexType)� JSONPathType� )�exc)�types)�util)�PyODBCConnector)�cursorc s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�_ms_numeric_pyodbcz�Turns Decimals with adjusted() < 0 or > 7 into strings. The routines here are needed for older pyodbc versions as well as current mxODBC versions. c s( t � �|��|js�S � �fdd�}|S )Nc sR � j r>t| tj�r>| �� }|dk r,� �| �S |dkr>� �| �S �rJ�| �S | S d S )Nr � )Z asdecimal� isinstance�decimal�Decimal�adjusted�_small_dec_to_string�_large_dec_to_string)�valuer ��selfZ super_process� �P/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/mssql/pyodbc.py�process� s z2_ms_numeric_pyodbc.bind_processor.<locals>.process)�super�bind_processor�_need_decimal_fix�r �dialectr! �� __class__r r r# { s z!_ms_numeric_pyodbc.bind_processorc C sB d|dk rdpddt |�� �d d�dd� |�� d D ��f S ) Nz%s0.%s%sr �-� �0r c S s g | ] }t |��qS r ��str)�.0Znintr r r � <listcomp>� s z;_ms_numeric_pyodbc._small_dec_to_string.<locals>.<listcomp>)�absr �join�as_tuple)r r r r r r � s �z'_ms_numeric_pyodbc._small_dec_to_stringc C s |� � d }dt|�krXd|dk r&dp(dd�dd� |D ��d |�� t|�d f }n�t|�d |�� kr�d |dk rzdp|dd�d d� |D �d|�� d � �d�d d� |D �|�� d d � �f }n8d |dk r�dp�dd�dd� |D �d|�� d � �f }|S )Nr �Ez%s%s%sr r) r* c S s g | ] }t |��qS r r, �r. �sr r r r/ � s z;_ms_numeric_pyodbc._large_dec_to_string.<locals>.<listcomp>r+ z%s%s.%sc S s g | ] }t |��qS r r, r4 r r r r/ � s c S s g | ] }t |��qS r r, r4 r r r r/ � s z%s%sc S s g | ] }t |��qS r r, r4 r r r r/ � s )r2 r- r1 r �len)r r �_int�resultr r r r � s$ �""�"�z'_ms_numeric_pyodbc._large_dec_to_string)�__name__� __module__� __qualname__�__doc__r# r r � __classcell__r r r' r r s s r c @ s e Zd ZdS )�_MSNumeric_pyodbcN�r9 r: r; r r r r r> � s r> c @ s e Zd ZdS )�_MSFloat_pyodbcNr? r r r r r@ � s r@ c @ s e Zd ZdZdd� ZdS )�_ms_binary_pyodbcz�Wraps binary values in dialect-specific Binary wrapper. If the value is null, return a pyodbc-specific BinaryNull object to prevent pyODBC [and FreeTDS] from defaulting binary NULL types to SQLWCHAR and causing implicit conversion errors. c s( �j d krd S �j j� � �fdd�}|S )Nc s | d k r� | �S �j jS d S �N)�dbapiZ BinaryNull)r �Z DBAPIBinaryr& r r r! � s z1_ms_binary_pyodbc.bind_processor.<locals>.process)rC �Binaryr% r rD r r# � s z _ms_binary_pyodbc.bind_processorN)r9 r: r; r< r# r r r r rA � s rA c @ s e Zd ZdZdZdd� ZdS )�_ODBCDateTimeBindProcessorz6Add bind processors to handle datetimeoffset behaviorsFc s � fdd�}|S )Nc sP | d kr d S t | t�r| S | jr,� js0� js0| S | �d�}t�dd|�}|S d S )Nz%Y-%m-%d %H:%M:%S.%f %zz([\+\-]\d{2})([\d\.]+)$z\1:\2)r r- �tzinfo�timezone�has_tz�strftime�re�sub)r Z dto_string�r r r r! � s �z:_ODBCDateTimeBindProcessor.bind_processor.<locals>.processr r% r rM r r# � s z)_ODBCDateTimeBindProcessor.bind_processorN)r9 r: r; r< rI r# r r r r rF � s rF c @ s e Zd ZdS )� _ODBCDateTimeNr? r r r r rN � s rN c @ s e Zd ZdZdS )�_ODBCDATETIMEOFFSETTN)r9 r: r; rI r r r r rO � s rO c @ s e Zd ZdS )�_VARBINARY_pyodbcNr? r r r r rP � s rP c @ s e Zd ZdS )�_BINARY_pyodbcNr? r r r r rQ � s rQ c @ s e Zd Zdd� ZdS )�_String_pyodbcc C s* | j dks| j dkr |jddfS |jS d S �N)N�maxi� r )�lengthZ SQL_VARCHAR�r rC r r r �get_dbapi_type s z _String_pyodbc.get_dbapi_typeN�r9 r: r; rW r r r r rR � s rR c @ s e Zd Zdd� ZdS )�_Unicode_pyodbcc C s* | j dks| j dkr |jddfS |jS d S rS �rU � SQL_WVARCHARrV r r r rW s z _Unicode_pyodbc.get_dbapi_typeNrX r r r r rY s rY c @ s e Zd Zdd� ZdS )�_UnicodeText_pyodbcc C s* | j dks| j dkr |jddfS |jS d S rS rZ rV r r r rW s z"_UnicodeText_pyodbc.get_dbapi_typeNrX r r r r r\ s r\ c @ s e Zd Zdd� ZdS )� _JSON_pyodbcc C s |j ddfS �Nr �r[ rV r r r rW s z_JSON_pyodbc.get_dbapi_typeNrX r r r r r] s r] c @ s e Zd Zdd� ZdS )�_JSONIndexType_pyodbcc C s |j S rB r_ rV r r r rW s z$_JSONIndexType_pyodbc.get_dbapi_typeNrX r r r r r` s r` c @ s e Zd Zdd� ZdS )�_JSONPathType_pyodbcc C s |j S rB r_ rV r r r rW " s z#_JSONPathType_pyodbc.get_dbapi_typeNrX r r r r ra ! s ra c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�MSExecutionContext_pyodbcFc s> t � �� | jr:| jjr:t| jd �r:d| _| jd7 _dS )a� where appropriate, issue "select scope_identity()" in the same statement. Background on why "scope_identity()" is preferable to "@@identity": https://msdn.microsoft.com/en-us/library/ms190315.aspx Background on why we attempt to embed "scope_identity()" into the same statement as the INSERT: https://code.google.com/p/pyodbc/wiki/FAQs#How_do_I_retrieve_autogenerated/identity_values? r Tz; select scope_identity()N) r" �pre_execZ_select_lastrowidr& �use_scope_identityr6 � parameters�_embedded_scope_identity� statementrM r' r r rc ) s �� �z"MSExecutionContext_pyodbc.pre_execc s| | j rnz| j�� }W n$ | jjjk r8 | j�� Y qX |sJ| j�� q|d }qVqt|d �| _t j | _ n t � � � d S r^ )rf r Zfetchallr& rC �Error�nextset�intZ _lastrowid�_cursorZ_NO_CURSOR_DMLZcursor_fetch_strategyr" � post_exec)r �rows�rowr' r r rl C s z#MSExecutionContext_pyodbc.post_exec)r9 r: r; rf rc rl r= r r r' r rb & s rb c ! s� e Zd ZdZdZeZe�e j e j e e jeeee jeeeeee jee jee jee je e j e e j e e j j!e"e j j#e$e j%e j%i�Z d� fdd� Z&� fdd�Z'� fdd�Z(d d � Z)d� fd d � Z*� fdd�Z+� Z,S )�MSDialect_pyodbcTFc sZ t � jf d|i|�� | jo.| jo.t| jjd�| _| joB| �� dk | _|| _|rVd| _ d S )N�use_setinputsizesri )� r � F) r" �__init__rd rC �hasattrZCursorZ_dbapi_versionr$ �fast_executemanyZ!use_insertmanyvalues_wo_returning)r ru rp �paramsr' r r rs � s � �zMSDialect_pyodbc.__init__c s� z|� d��� }W n" tjk r4 t� �|� Y S X g }t�d�}|�|�D ],}z|� t |�� W qN t k rx Y qNX qNt |�S d S )Nz8SELECT CAST(SERVERPROPERTY('ProductVersion') AS VARCHAR)z[.\-]) Zexec_driver_sqlZscalarr Z DBAPIErrorr" �_get_server_version_inforK �compile�split�appendrj � ValueError�tuple)r � connection�raw�version�r�nr' r r rw � s � z)MSDialect_pyodbc._get_server_version_infoc s t � �� �� �fdd�}|S )Nc s �d k r�| � � � | � d S rB )�_setup_timestampoffset_type)�conn�r Zsuper_r r � on_connect� s z/MSDialect_pyodbc.on_connect.<locals>.on_connect)r" r� )r r� r' r� r r� � s zMSDialect_pyodbc.on_connectc C s dd� }d}|� ||� d S )Nc S s\ t �d| �}t�|d |d |d |d |d |d |d d t�tj|d |d d ���S ) Nz<6hI2hr r rq r � � � i� r rr )�hours�minutes)�struct�unpack�datetimerH � timedelta)Z dto_value�tupr r r �_handle_datetimeoffset� s ��zLMSDialect_pyodbc._setup_timestampoffset_type.<locals>._handle_datetimeoffsetie���)Zadd_output_converter)r r} r� Zodbc_SQL_SS_TIMESTAMPOFFSETr r r r� � s �z,MSDialect_pyodbc._setup_timestampoffset_typeNc s$ | j r d|_ t� j||||d� d S )NT)�context)ru r" �do_executemany)r r rg re r� r' r r r� � s z MSDialect_pyodbc.do_executemanyc s4 t || jj�r$|jd }|dkr$dS t� �|||�S )Nr > �08007�01002�HY010�01000�08003�08001�HYT00�08S02�08S01�10054T)r rC rh �argsr" � is_disconnect)r �er} r �coder' r r r� � s z MSDialect_pyodbc.is_disconnect)FT)N)-r9 r: r; Zsupports_statement_cacheZ supports_sane_rowcount_returningrb Zexecution_ctx_clsr Z update_copyr Zcolspecs�sqltypes�Numericr> �Floatr@ r rQ �DateTimerN r rO r rP Z LargeBinary�StringrR ZUnicoderY Z UnicodeTextr\ r r] r r` r ra �Enumrs rw r� r� r� r� r= r r r' r ro ` sZ �� � ro )4r< r� r rK r� �baser r r r r r r r �jsonr Z_MSJsonr Z_MSJsonIndexTyper Z_MSJsonPathTyper* r r r� r Zconnectors.pyodbcr Zenginer rk r r� r> r� r@ rA rF rN rO rP rQ r� rR rY r\ r] r` ra rb ro r& r r r r �<module> sR U > :