diff options
author | mike bayer <mike_mp@zzzcomputing.com> | 2022-06-29 17:41:16 +0000 |
---|---|---|
committer | Gerrit Code Review <gerrit@ci3.zzzcomputing.com> | 2022-06-29 17:41:16 +0000 |
commit | b35bbdb13be9a359fc3cfae00e2b6fdca63dea29 (patch) | |
tree | 372ff30a2093cab38dd0797dd6cb60513f025e59 /lib/sqlalchemy | |
parent | 4270b26afc7d26ec5324b455e02282bea3b96c03 (diff) | |
parent | f848140083c316b81456c8c3426ad3fd04caf6fa (diff) | |
download | sqlalchemy-b35bbdb13be9a359fc3cfae00e2b6fdca63dea29.tar.gz |
Merge "Change setinputsizes behavior for mssql+pyodbc" into main
Diffstat (limited to 'lib/sqlalchemy')
-rw-r--r-- | lib/sqlalchemy/connectors/pyodbc.py | 9 | ||||
-rw-r--r-- | lib/sqlalchemy/dialects/mssql/pyodbc.py | 64 | ||||
-rw-r--r-- | lib/sqlalchemy/sql/sqltypes.py | 2 |
3 files changed, 67 insertions, 8 deletions
diff --git a/lib/sqlalchemy/connectors/pyodbc.py b/lib/sqlalchemy/connectors/pyodbc.py index e24acda75..7c5e62fae 100644 --- a/lib/sqlalchemy/connectors/pyodbc.py +++ b/lib/sqlalchemy/connectors/pyodbc.py @@ -42,6 +42,8 @@ class PyODBCConnector(Connector): supports_native_decimal = True default_paramstyle = "named" + fast_executemany = False + # for non-DSN connections, this *may* be used to # hold the desired driver name pyodbc_driver_name: Optional[str] = None @@ -203,6 +205,13 @@ class PyODBCConnector(Connector): # parameter were not passed to the dialect, or if no types were # specified in list_of_tuples + # as of #8177 for 2.0 we assume use_setinputsizes=True and only + # omit the setinputsizes calls for .executemany() with + # fast_executemany=True + + if context.executemany and self.fast_executemany: + return + cursor.setinputsizes( [ (dbtype, None, None) diff --git a/lib/sqlalchemy/dialects/mssql/pyodbc.py b/lib/sqlalchemy/dialects/mssql/pyodbc.py index 6d64fdc3e..9f73ed28c 100644 --- a/lib/sqlalchemy/dialects/mssql/pyodbc.py +++ b/lib/sqlalchemy/dialects/mssql/pyodbc.py @@ -291,12 +291,10 @@ driver in order to use this flag:: Setinputsizes Support ----------------------- -The pyodbc ``cursor.setinputsizes()`` method can be used if necessary. To -enable this hook, pass ``use_setinputsizes=True`` to :func:`_sa.create_engine`:: +As of version 2.0, the pyodbc ``cursor.setinputsizes()`` method is used by +default except for .executemany() calls when fast_executemany=True. - engine = create_engine("mssql+pyodbc://...", use_setinputsizes=True) - -The behavior of the hook can then be customized, as may be necessary +The behavior of setinputsizes can be customized, as may be necessary particularly if fast_executemany is in use, via the :meth:`.DialectEvents.do_setinputsizes` hook. See that method for usage examples. @@ -304,6 +302,9 @@ examples. .. versionchanged:: 1.4.1 The pyodbc dialects will not use setinputsizes unless ``use_setinputsizes=True`` is passed. +.. versionchanged:: 2.0 The mssql+pyodbc dialect now defaults to using + setinputsizes except for .executemany() calls when fast_executemany=True. + """ # noqa @@ -313,11 +314,16 @@ import re import struct from .base import _MSDateTime +from .base import _MSUnicode +from .base import _MSUnicodeText from .base import BINARY from .base import DATETIMEOFFSET from .base import MSDialect from .base import MSExecutionContext from .base import VARBINARY +from .json import JSON as _MSJson +from .json import JSONIndexType as _MSJsonIndexType +from .json import JSONPathType as _MSJsonPathType from ... import exc from ... import types as sqltypes from ... import util @@ -466,6 +472,36 @@ class _BINARY_pyodbc(_ms_binary_pyodbc, BINARY): pass +class _String_pyodbc(sqltypes.String): + def get_dbapi_type(self, dbapi): + return dbapi.SQL_VARCHAR + + +class _Unicode_pyodbc(_MSUnicode): + def get_dbapi_type(self, dbapi): + return dbapi.SQL_WVARCHAR + + +class _UnicodeText_pyodbc(_MSUnicodeText): + def get_dbapi_type(self, dbapi): + return dbapi.SQL_WVARCHAR + + +class _JSON_pyodbc(_MSJson): + def get_dbapi_type(self, dbapi): + return dbapi.SQL_WVARCHAR + + +class _JSONIndexType_pyodbc(_MSJsonIndexType): + def get_dbapi_type(self, dbapi): + return dbapi.SQL_WVARCHAR + + +class _JSONPathType_pyodbc(_MSJsonPathType): + def get_dbapi_type(self, dbapi): + return dbapi.SQL_WVARCHAR + + class MSExecutionContext_pyodbc(MSExecutionContext): _embedded_scope_identity = False @@ -541,11 +577,25 @@ class MSDialect_pyodbc(PyODBCConnector, MSDialect): VARBINARY: _VARBINARY_pyodbc, sqltypes.VARBINARY: _VARBINARY_pyodbc, sqltypes.LargeBinary: _VARBINARY_pyodbc, + sqltypes.String: _String_pyodbc, + sqltypes.Unicode: _Unicode_pyodbc, + sqltypes.UnicodeText: _UnicodeText_pyodbc, + sqltypes.JSON: _JSON_pyodbc, + sqltypes.JSON.JSONIndexType: _JSONIndexType_pyodbc, + sqltypes.JSON.JSONPathType: _JSONPathType_pyodbc, + # this excludes Enum from the string/VARCHAR thing for now + # it looks like Enum's adaptation doesn't really support the + # String type itself having a dialect-level impl + sqltypes.Enum: sqltypes.Enum, }, ) - def __init__(self, fast_executemany=False, **params): - super(MSDialect_pyodbc, self).__init__(**params) + def __init__( + self, fast_executemany=False, use_setinputsizes=True, **params + ): + super(MSDialect_pyodbc, self).__init__( + use_setinputsizes=use_setinputsizes, **params + ) self.use_scope_identity = ( self.use_scope_identity and self.dbapi diff --git a/lib/sqlalchemy/sql/sqltypes.py b/lib/sqlalchemy/sql/sqltypes.py index 6adc6ca77..c67614070 100644 --- a/lib/sqlalchemy/sql/sqltypes.py +++ b/lib/sqlalchemy/sql/sqltypes.py @@ -2482,7 +2482,7 @@ class JSON(Indexable, TypeEngine[Any]): .. versionadded:: 1.3.11 """ - return self._binary_w_type(String(), "as_string") + return self._binary_w_type(Unicode(), "as_string") def as_integer(self): """Cast an indexed value as integer. |