SQLAlchemy 2.0.48__py3-none-any.whl → 2.0.50__py3-none-any.whl

sqlalchemy/__init__.py

@@ -269,7 +269,7 @@ from .types import Uuid as Uuid
 from .types import VARBINARY as VARBINARY
 from .types import VARCHAR as VARCHAR
 
-__version__ = "2.0.48"
+__version__ = "2.0.50"
 
 
 def __go(lcls: Any) -> None:

sqlalchemy/connectors/aioodbc.py

@@ -20,7 +20,6 @@ from .. import util
 from ..util.concurrency import await_fallback
 from ..util.concurrency import await_only
 
-
 if TYPE_CHECKING:
     from ..engine.interfaces import ConnectArgsType
     from ..engine.url import URL
@@ -36,6 +35,14 @@ class AsyncAdapt_aioodbc_cursor(AsyncAdapt_dbapi_cursor):
         # how it's supposed to work
         # return self.await_(self._cursor.setinputsizes(*inputsizes))
 
+    @property
+    def fast_executemany(self):
+        return self._cursor._impl.fast_executemany
+
+    @fast_executemany.setter
+    def fast_executemany(self, value):
+        self._cursor._impl.fast_executemany = value
+
 
 class AsyncAdapt_aioodbc_ss_cursor(
     AsyncAdapt_aioodbc_cursor, AsyncAdapt_dbapi_ss_cursor
@@ -125,6 +132,7 @@ class AsyncAdapt_aioodbc_dbapi:
             "ProgrammingError",
             "InternalError",
             "NotSupportedError",
+            "SQL_DRIVER_NAME",
             "NUMBER",
             "STRING",
             "DATETIME",
@@ -133,6 +141,7 @@ class AsyncAdapt_aioodbc_dbapi:
             "BinaryNull",
             "SQL_VARCHAR",
             "SQL_WVARCHAR",
+            "SQL_DECIMAL",
         ):
             setattr(self, name, getattr(self.pyodbc, name))
 

sqlalchemy/dialects/_typing.py

@@ -18,7 +18,6 @@ from ..sql.schema import Column
 from ..sql.schema import ColumnCollectionConstraint
 from ..sql.schema import Index
 
-
 _OnConflictConstraintT = Union[str, ColumnCollectionConstraint, Index, None]
 _OnConflictIndexElementsT = Optional[
     Iterable[Union[Column[Any], str, roles.DDLConstraintColumnRole]]

sqlalchemy/dialects/mssql/__init__.py

@@ -45,7 +45,6 @@ from .base import VARCHAR
 from .base import XML
 from ...sql import try_cast
 
-
 base.dialect = dialect = pyodbc.dialect
 
 

sqlalchemy/dialects/mssql/base.py

@@ -3220,8 +3220,7 @@ class MSDialect(default.DefaultDialect):
 
             view_name = f"sys.{row[0]}"
 
-            cursor.execute(
-                """
+            cursor.execute("""
                     SELECT CASE transaction_isolation_level
                     WHEN 0 THEN NULL
                     WHEN 1 THEN 'READ UNCOMMITTED'
@@ -3232,10 +3231,7 @@ class MSDialect(default.DefaultDialect):
                     AS TRANSACTION_ISOLATION_LEVEL
                     FROM {}
                     where session_id = @@SPID
-                """.format(
-                    view_name
-                )
-            )
+                """.format(view_name))
         except self.dbapi.Error as err:
             raise NotImplementedError(
                 "Can't fetch isolation level;  encountered error {} when "
@@ -3254,13 +3250,6 @@ class MSDialect(default.DefaultDialect):
         self._setup_supports_comments(connection)
 
     def _setup_version_attributes(self):
-        if self.server_version_info[0] not in list(range(8, 17)):
-            util.warn(
-                "Unrecognized server version info '%s'.  Some SQL Server "
-                "features may not function properly."
-                % ".".join(str(x) for x in self.server_version_info)
-            )
-
         if self.server_version_info >= MS_2008_VERSION:
             self.supports_multivalues_insert = True
         else:
@@ -3439,8 +3428,7 @@ class MSDialect(default.DefaultDialect):
             else "NULL as filter_definition"
         )
         rp = connection.execution_options(future_result=True).execute(
-            sql.text(
-                f"""
+            sql.text(f"""
 select
     ind.index_id,
     ind.is_unique,
@@ -3460,8 +3448,7 @@ where
     and ind.type != 0
 order by
     ind.name
-                """
-            )
+                """)
             .bindparams(
                 sql.bindparam("tabname", tablename, ischema.CoerceUnicode()),
                 sql.bindparam("schname", owner, ischema.CoerceUnicode()),
@@ -3489,8 +3476,7 @@ order by
                 do["mssql_where"] = row["filter_definition"]
 
         rp = connection.execution_options(future_result=True).execute(
-            sql.text(
-                """
+            sql.text("""
 select
     ind_col.index_id,
     col.name,
@@ -3510,8 +3496,7 @@ where
 order by
     ind_col.index_id,
     ind_col.key_ordinal
-            """
-            )
+            """)
             .bindparams(
                 sql.bindparam("tabname", tablename, ischema.CoerceUnicode()),
                 sql.bindparam("schname", owner, ischema.CoerceUnicode()),
@@ -3641,6 +3626,7 @@ order by
     def get_columns(self, connection, tablename, dbname, owner, schema, **kw):
         sys_columns = ischema.sys_columns
         sys_types = ischema.sys_types
+        sys_base_types = ischema.sys_types.alias("base_types")
         sys_default_constraints = ischema.sys_default_constraints
         computed_cols = ischema.computed_columns
         identity_cols = ischema.identity_columns
@@ -3682,6 +3668,7 @@ order by
             sql.select(
                 sys_columns.c.name,
                 sys_types.c.name,
+                sys_base_types.c.name.label("base_type"),
                 sys_columns.c.is_nullable,
                 sys_columns.c.max_length,
                 sys_columns.c.precision,
@@ -3701,6 +3688,15 @@ order by
                 onclause=sys_columns.c.user_type_id
                 == sys_types.c.user_type_id,
             )
+            .outerjoin(
+                sys_base_types,
+                onclause=sql.and_(
+                    sys_types.c.system_type_id
+                    == sys_base_types.c.system_type_id,
+                    sys_base_types.c.user_type_id
+                    == sys_base_types.c.system_type_id,
+                ),
+            )
             .outerjoin(
                 sys_default_constraints,
                 sql.and_(
@@ -3747,6 +3743,7 @@ order by
         for row in c.mappings():
             name = row[sys_columns.c.name]
             type_ = row[sys_types.c.name]
+            base_type = row["base_type"]
             nullable = row[sys_columns.c.is_nullable] == 1
             maxlen = row[sys_columns.c.max_length]
             numericprec = row[sys_columns.c.precision]
@@ -3760,7 +3757,17 @@ order by
             identity_increment = row[identity_cols.c.increment_value]
             comment = row[extended_properties.c.value]
 
+            # Try to resolve the user type first (e.g., "sysname"),
+            # then fall back to the base type (e.g., "nvarchar").
+            # base_type may be None for CLR types (geography, geometry,
+            # hierarchyid) which have no corresponding base type.
             coltype = self.ischema_names.get(type_, None)
+            if (
+                coltype is None
+                and base_type is not None
+                and base_type != type_
+            ):
+                coltype = self.ischema_names.get(base_type, None)
 
             kwargs = {}
 
@@ -3788,10 +3795,16 @@ order by
                     kwargs["collation"] = collation
 
             if coltype is None:
-                util.warn(
-                    "Did not recognize type '%s' of column '%s'"
-                    % (type_, name)
-                )
+                if base_type is not None and base_type != type_:
+                    util.warn(
+                        "Did not recognize type '%s' (user type) or '%s' "
+                        "(base type) of column '%s'" % (type_, base_type, name)
+                    )
+                else:
+                    util.warn(
+                        "Did not recognize type '%s' of column '%s'"
+                        % (type_, name)
+                    )
                 coltype = sqltypes.NULLTYPE
             else:
                 if issubclass(coltype, sqltypes.Numeric):
@@ -3909,8 +3922,7 @@ order by
     ):
         # Foreign key constraints
         s = (
-            text(
-                """\
+            text("""\
 WITH fk_info AS (
     SELECT
         ischema_ref_con.constraint_schema,
@@ -4010,8 +4022,7 @@ index_info AS (
 
     ORDER BY fk_info.constraint_schema, fk_info.constraint_name,
         fk_info.ordinal_position
-"""
-            )
+""")
             .bindparams(
                 sql.bindparam("tablename", tablename, ischema.CoerceUnicode()),
                 sql.bindparam("owner", owner, ischema.CoerceUnicode()),

sqlalchemy/dialects/mssql/information_schema.py

@@ -20,7 +20,6 @@ from ...types import String
 from ...types import TypeDecorator
 from ...types import Unicode
 
-
 ischema = MetaData()
 
 

sqlalchemy/dialects/mssql/pymssql.py

@@ -22,6 +22,7 @@ pymssql is a Python module that provides a Python DBAPI interface around
 
 
 """  # noqa
+
 import re
 
 from .base import MSDialect

sqlalchemy/dialects/mssql/pyodbc.py

@@ -359,7 +359,6 @@ hook. See that method for usage examples.
 
 """  # noqa
 
-
 import datetime
 import decimal
 import re

sqlalchemy/dialects/mysql/aiomysql.py

@@ -28,6 +28,7 @@ This dialect should normally be used only with the
     )
 
 """  # noqa
+
 from __future__ import annotations
 
 from types import ModuleType
@@ -38,6 +39,7 @@ from typing import Tuple
 from typing import TYPE_CHECKING
 from typing import Union
 
+from .pymysql import _connection_ping_reconnects_true
 from .pymysql import MySQLDialect_pymysql
 from ... import pool
 from ... import util
@@ -46,6 +48,7 @@ from ...connectors.asyncio import AsyncAdapt_dbapi_cursor
 from ...connectors.asyncio import AsyncAdapt_dbapi_module
 from ...connectors.asyncio import AsyncAdapt_dbapi_ss_cursor
 from ...connectors.asyncio import AsyncAdapt_terminate
+from ...util import langhelpers
 from ...util.concurrency import await_fallback
 from ...util.concurrency import await_only
 
@@ -91,9 +94,12 @@ class AsyncAdapt_aiomysql_connection(
     _cursor_cls = AsyncAdapt_aiomysql_cursor
     _ss_cursor_cls = AsyncAdapt_aiomysql_ss_cursor
 
-    def ping(self, reconnect: bool) -> None:
+    def ping(self, reconnect: bool = False) -> None:
         assert not reconnect
-        self.await_(self._connection.ping(reconnect))
+        if self.dbapi._send_false_to_ping:
+            self.await_(self._connection.ping(reconnect=False))
+        else:
+            self.await_(self._connection.ping())
 
     def character_set_name(self) -> Optional[str]:
         return self._connection.character_set_name()  # type: ignore[no-any-return]  # noqa: E501
@@ -170,6 +176,24 @@ class AsyncAdapt_aiomysql_dbapi(AsyncAdapt_dbapi_module):
                 await_only(creator_fn(*arg, **kw)),
             )
 
+    @langhelpers.memoized_property
+    def _send_false_to_ping(self) -> bool:
+        """determine if aiomysql has deprecated, changed the default of,
+        or removed the 'reconnect' argument of connection.ping().
+
+        See #13306 and #10492
+
+        """  # noqa: E501
+
+        try:
+            Connection = __import__(
+                "aiomysql.connection"
+            ).connection.Connection
+        except (ImportError, AttributeError):
+            return True
+        else:
+            return _connection_ping_reconnects_true(Connection)
+
     def _init_cursors_subclasses(
         self,
     ) -> Tuple[AsyncIODBAPICursor, AsyncIODBAPICursor]:

sqlalchemy/dialects/mysql/asyncmy.py

@@ -26,6 +26,7 @@ This dialect should normally be used only with the
     )
 
 """  # noqa
+
 from __future__ import annotations
 
 from types import ModuleType
@@ -35,6 +36,7 @@ from typing import Optional
 from typing import TYPE_CHECKING
 from typing import Union
 
+from .pymysql import _connection_ping_reconnects_true
 from .pymysql import MySQLDialect_pymysql
 from ... import pool
 from ... import util
@@ -43,6 +45,7 @@ from ...connectors.asyncio import AsyncAdapt_dbapi_cursor
 from ...connectors.asyncio import AsyncAdapt_dbapi_module
 from ...connectors.asyncio import AsyncAdapt_dbapi_ss_cursor
 from ...connectors.asyncio import AsyncAdapt_terminate
+from ...util import langhelpers
 from ...util.concurrency import await_fallback
 from ...util.concurrency import await_only
 
@@ -86,18 +89,21 @@ class AsyncAdapt_asyncmy_connection(
         if isinstance(error, AttributeError):
             raise self.dbapi.InternalError(
                 "network operation failed due to asyncmy attribute error"
-            )
+            ) from error
 
         raise error
 
-    def ping(self, reconnect: bool) -> None:
+    def ping(self, reconnect: bool = False) -> None:
         assert not reconnect
         return self.await_(self._do_ping())
 
     async def _do_ping(self) -> None:
         try:
             async with self._execute_mutex:
-                await self._connection.ping(False)
+                if self.dbapi._send_false_to_ping:
+                    await self._connection.ping(reconnect=False)
+                else:
+                    await self._connection.ping()
         except Exception as error:
             self._handle_exception(error)
 
@@ -171,6 +177,22 @@ class AsyncAdapt_asyncmy_dbapi(AsyncAdapt_dbapi_module):
                 await_only(creator_fn(*arg, **kw)),
             )
 
+    @langhelpers.memoized_property
+    def _send_false_to_ping(self) -> bool:
+        """determine if asyncmy has deprecated, changed the default of,
+        or removed the 'reconnect' argument of connection.ping().
+
+        See #13306 and #10492
+
+        """  # noqa: E501
+
+        try:
+            Connection = __import__("asyncmy.connection").connection.Connection
+        except (ImportError, AttributeError):
+            return True
+        else:
+            return _connection_ping_reconnects_true(Connection)
+
 
 class MySQLDialect_asyncmy(MySQLDialect_pymysql):
     driver = "asyncmy"

sqlalchemy/dialects/mysql/base.py

@@ -1063,6 +1063,7 @@ output:
     )
 
 """  # noqa
+
 from __future__ import annotations
 
 from collections import defaultdict
@@ -2764,6 +2765,7 @@ class MySQLDialect(default.DefaultDialect):
     # i.e. first connect
     _backslash_escapes = True
     _server_ansiquotes = False
+    _casing = 0
 
     server_version_info: Tuple[int, ...]
     identifier_preparer: MySQLIdentifierPreparer
@@ -3186,9 +3188,9 @@ class MySQLDialect(default.DefaultDialect):
             self._is_mysql and self.server_version_info >= (8, 0, 1)
         )
 
-        self._needs_correct_for_88718_96365 = (
-            not self.is_mariadb and self.server_version_info >= (8,)
-        )
+        self._needs_correct_for_88718_96365 = self.server_version_info >= (
+            8,
+        ) and (self.server_version_info < (8, 0, 14) or self._casing == 2)
 
         self.delete_returning = (
             self.is_mariadb and self.server_version_info >= (10, 0, 5)
@@ -3253,7 +3255,7 @@ class MySQLDialect(default.DefaultDialect):
     def _is_mariadb_102(self) -> bool:
         return (
             self.is_mariadb
-            and self._mariadb_normalized_version_info  # type:ignore[operator]
+            and self._mariadb_normalized_version_info  # type: ignore[operator]
             > (
                 10,
                 2,

sqlalchemy/dialects/mysql/cymysql.py

@@ -20,6 +20,7 @@ r"""
     dialects are mysqlclient and PyMySQL.
 
 """  # noqa
+
 from __future__ import annotations
 
 from typing import Any

sqlalchemy/dialects/mysql/dml.py

@@ -28,7 +28,6 @@ from ...sql.expression import alias
 from ...sql.selectable import NamedFromClause
 from ...util.typing import Self
 
-
 __all__ = ("Insert", "insert")
 
 

sqlalchemy/dialects/mysql/mariadbconnector.py

@@ -27,6 +27,7 @@ be ``mysqldb``. ``mariadb+mariadbconnector://`` is required to use this driver.
 .. mariadb: https://github.com/mariadb-corporation/mariadb-connector-python
 
 """  # noqa
+
 from __future__ import annotations
 
 import re

sqlalchemy/dialects/mysql/mysqlconnector.py

@@ -45,6 +45,7 @@ charset/collation will allow connectivity.
 
 
 """  # noqa
+
 from __future__ import annotations
 
 import re
@@ -110,8 +111,8 @@ class IdentifierPreparerCommon_mysqlconnector:
 
     def _escape_identifier(self, value: str) -> str:
         value = value.replace(
-            self.escape_quote,  # type:ignore[attr-defined]
-            self.escape_to_quote,  # type:ignore[attr-defined]
+            self.escape_quote,  # type: ignore[attr-defined]
+            self.escape_to_quote,  # type: ignore[attr-defined]
         )
         return value
 

sqlalchemy/dialects/mysql/mysqldb.py

@@ -84,6 +84,7 @@ Server Side Cursors
 The mysqldb dialect supports server-side cursors. See :ref:`mysql_ss_cursors`.
 
 """
+
 from __future__ import annotations
 
 import re

sqlalchemy/dialects/mysql/pymysql.py

@@ -47,11 +47,13 @@ and targets 100% compatibility.   Most behavioral notes for MySQL-python apply
 to the pymysql driver as well.
 
 """  # noqa
+
 from __future__ import annotations
 
 from typing import Any
 from typing import Dict
 from typing import Optional
+from typing import Type
 from typing import TYPE_CHECKING
 from typing import Union
 
@@ -69,6 +71,31 @@ if TYPE_CHECKING:
     from ...engine.url import URL
 
 
+def _connection_ping_reconnects_true(connection_cls: Type[Any]) -> bool:
+    """Given a Connection class like pymysql.Connection, aiomysql.Connection,
+    asyncmy.Connection, inspect the ping() method and determine if it
+    has a "reconnect" parameter that either defaults to True, or is positional.
+
+    a return value of True here means that when we call ``connection.ping()``,
+    we **must** pass `reconnect=False`.  a return value of False means that
+    we should call ``connection.ping()`` with **no arguments**.
+
+    This routine originates from issue #10492 for pymysql, however arg
+    signature mismatches in aiomysql/asyncmy tracked by issue #13306
+    necessitated a more open ended function.
+
+    """
+    insp = langhelpers.get_callable_argspec(connection_cls.ping)
+    try:
+        reconnect_arg = insp.args[1]
+    except IndexError:
+        return False
+    else:
+        return reconnect_arg == "reconnect" and (
+            not insp.defaults or insp.defaults[0] is not False
+        )
+
+
 class MySQLDialect_pymysql(MySQLDialect_mysqldb):
     driver = "pymysql"
     supports_statement_cache = True
@@ -97,6 +124,8 @@ class MySQLDialect_pymysql(MySQLDialect_mysqldb):
         https://github.com/PyMySQL/mysqlclient/discussions/651#discussioncomment-7308971
         for background.
 
+        Revised as part of #13306
+
         """  # noqa: E501
 
         try:
@@ -106,15 +135,7 @@ class MySQLDialect_pymysql(MySQLDialect_mysqldb):
         except (ImportError, AttributeError):
             return True
         else:
-            insp = langhelpers.get_callable_argspec(Connection.ping)
-            try:
-                reconnect_arg = insp.args[1]
-            except IndexError:
-                return False
-            else:
-                return reconnect_arg == "reconnect" and (
-                    not insp.defaults or insp.defaults[0] is not False
-                )
+            return _connection_ping_reconnects_true(Connection)
 
     def do_ping(self, dbapi_connection: DBAPIConnection) -> Literal[True]:
         if self._send_false_to_ping:

sqlalchemy/dialects/mysql/pyodbc.py

@@ -42,6 +42,7 @@ Pass through exact pyodbc connection string::
     connection_uri = "mysql+pyodbc:///?odbc_connect=%s" % params
 
 """  # noqa
+
 from __future__ import annotations
 
 import datetime

sqlalchemy/dialects/mysql/reflection.py

@@ -371,7 +371,7 @@ class MySQLTableDefinitionParser:
 
         buffer = []
         for row in columns:
-            (name, col_type, nullable, default, extra) = (
+            name, col_type, nullable, default, extra = (
                 row[i] for i in (0, 1, 2, 4, 5)
             )
 
@@ -704,7 +704,7 @@ def cleanup_text(raw_text: str) -> str:
     if "\\" in raw_text:
         raw_text = re.sub(
             _control_char_regexp,
-            lambda s: _control_char_map[s[0]],  # type: ignore[index]
+            lambda s: _control_char_map[s[0]],  # type: ignore[unused-ignore,index]  # noqa: E501
             raw_text,
         )
     return raw_text.replace("''", "'")

sqlalchemy/dialects/oracle/base.py

@@ -2778,6 +2778,7 @@ class OracleDialect(default.DefaultDialect):
                 all_cols.c.column_name,
                 all_cols.c.data_type,
                 all_cols.c.char_length,
+                all_cols.c.data_length,
                 all_cols.c.data_precision,
                 all_cols.c.data_scale,
                 all_cols.c.nullable,
@@ -2896,6 +2897,9 @@ class OracleDialect(default.DefaultDialect):
             elif coltype in ("VARCHAR2", "NVARCHAR2", "CHAR", "NCHAR"):
                 char_length = maybe_int(row_dict["char_length"])
                 coltype = self.ischema_names.get(coltype)(char_length)
+            elif coltype == "RAW":
+                data_length = maybe_int(row_dict["data_length"])
+                coltype = RAW(data_length)
             elif "WITH TIME ZONE" in coltype:
                 coltype = TIMESTAMP(timezone=True)
             elif "WITH LOCAL TIME ZONE" in coltype:

sqlalchemy/dialects/oracle/cx_oracle.py

@@ -489,6 +489,7 @@ SQLAlchemy type (or a subclass of such).
    as better integration of outputtypehandlers.
 
 """  # noqa
+
 from __future__ import annotations
 
 import decimal

sqlalchemy/dialects/oracle/oracledb.py

@@ -588,6 +588,7 @@ SQLAlchemy type (or a subclass of such).
 .. versionadded:: 2.0.0 added support for the python-oracledb driver.
 
 """  # noqa
+
 from __future__ import annotations
 
 import collections

sqlalchemy/dialects/oracle/vector.py

@@ -313,12 +313,10 @@ class VECTOR(types.TypeEngine):
                 )
 
             else:
-                raise TypeError(
-                    """
+                raise TypeError("""
                     Invalid input for VECTOR: expected a list, an array.array,
                     or a SparseVector object.
-                    """
-                )
+                    """)
 
         return process
 

sqlalchemy/dialects/postgresql/__init__.py

@@ -89,7 +89,6 @@ from .types import TIMESTAMP
 from .types import TSQUERY
 from .types import TSVECTOR
 
-
 # Alias psycopg also as psycopg_async
 psycopg_async = type(
     "psycopg_async", (ModuleType,), {"dialect": psycopg.dialect_async}