diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2017-08-29 14:31:43 -0400 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2017-08-29 14:32:35 -0400 |
| commit | 56950a78db62d880a2f2c240c8c5873af86310c2 (patch) | |
| tree | f04e099dada18965e03c59ad6151cc144a392823 /lib | |
| parent | ec4f567f31856a92bd91144112cd29df356a8ca8 (diff) | |
| download | sqlalchemy-56950a78db62d880a2f2c240c8c5873af86310c2.tar.gz | |
Implement AUTOCOMMIT for pyodbc, pymssql
In prep for CI coverage for SQL Server, allow AUTOCOMMIT
isolation level to work
Change-Id: I850b977e75f53385986f2c181be4e4412dd3b3f4
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/sqlalchemy/connectors/pyodbc.py | 14 | ||||
| -rw-r--r-- | lib/sqlalchemy/dialects/mssql/base.py | 2 | ||||
| -rw-r--r-- | lib/sqlalchemy/dialects/mssql/pymssql.py | 8 |
3 files changed, 24 insertions, 0 deletions
diff --git a/lib/sqlalchemy/connectors/pyodbc.py b/lib/sqlalchemy/connectors/pyodbc.py index ee8445dae..65fe37212 100644 --- a/lib/sqlalchemy/connectors/pyodbc.py +++ b/lib/sqlalchemy/connectors/pyodbc.py @@ -194,3 +194,17 @@ class PyODBCConnector(Connector): except ValueError: version.append(n) return tuple(version) + + def set_isolation_level(self, connection, level): + # adjust for ConnectionFairy being present + # allows attribute set e.g. "connection.autocommit = True" + # to work properly + if hasattr(connection, 'connection'): + connection = connection.connection + + if level == 'AUTOCOMMIT': + connection.autocommit = True + else: + connection.autocommit = False + super(PyODBCConnector, self).set_isolation_level(connection, + level) diff --git a/lib/sqlalchemy/dialects/mssql/base.py b/lib/sqlalchemy/dialects/mssql/base.py index 00c43bba6..05381d671 100644 --- a/lib/sqlalchemy/dialects/mssql/base.py +++ b/lib/sqlalchemy/dialects/mssql/base.py @@ -250,6 +250,7 @@ To set using per-connection execution options:: Valid values for ``isolation_level`` include: +* ``AUTOCOMMIT`` - pyodbc / pymssql-specific * ``READ COMMITTED`` * ``READ UNCOMMITTED`` * ``REPEATABLE READ`` @@ -259,6 +260,7 @@ Valid values for ``isolation_level`` include: .. versionadded:: 1.1 support for isolation level setting on Microsoft SQL Server. +.. versionadded:: 1.2 added AUTOCOMMIT isolation level setting Nullability ----------- diff --git a/lib/sqlalchemy/dialects/mssql/pymssql.py b/lib/sqlalchemy/dialects/mssql/pymssql.py index 57ca8ab3d..cd800471d 100644 --- a/lib/sqlalchemy/dialects/mssql/pymssql.py +++ b/lib/sqlalchemy/dialects/mssql/pymssql.py @@ -94,4 +94,12 @@ class MSDialect_pymssql(MSDialect): else: return False + def set_isolation_level(self, connection, level): + if level == 'AUTOCOMMIT': + connection.autocommit(True) + else: + connection.autocommit(False) + super(MSDialect_pymssql, self).set_isolation_level(connection, + level) + dialect = MSDialect_pymssql |