summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2023-02-06 21:56:18 +0000
committerGerrit Code Review <gerrit@bbpush.zzzcomputing.com>2023-02-06 21:56:18 +0000
commit557c4a97a8657b86ee383ce8891d59f1ce3f2ec8 (patch)
treee90d6aa69a466a149a3d32f4f77ba891eb6d61f7 /lib/sqlalchemy
parent066b1d607e670ed00177e05969c4a30f750d1315 (diff)
parent662d6e24526bfd9d358b8a9e1906fb29bf93ef30 (diff)
downloadsqlalchemy-557c4a97a8657b86ee383ce8891d59f1ce3f2ec8.tar.gz
Merge "add error code 1049 for mysql has_table" into main
Diffstat (limited to 'lib/sqlalchemy')
-rw-r--r--lib/sqlalchemy/dialects/mysql/base.py16
-rw-r--r--lib/sqlalchemy/dialects/sqlite/base.py5
-rw-r--r--lib/sqlalchemy/testing/suite/test_reflection.py9
3 files changed, 29 insertions, 1 deletions
diff --git a/lib/sqlalchemy/dialects/mysql/base.py b/lib/sqlalchemy/dialects/mysql/base.py
index 87fdabff5..ebef48a77 100644
--- a/lib/sqlalchemy/dialects/mysql/base.py
+++ b/lib/sqlalchemy/dialects/mysql/base.py
@@ -2714,7 +2714,21 @@ class MySQLDialect(default.DefaultDialect):
) as rs:
return rs.fetchone() is not None
except exc.DBAPIError as e:
- if self._extract_error_code(e.orig) == 1146:
+ # https://dev.mysql.com/doc/mysql-errors/8.0/en/server-error-reference.html # noqa: E501
+ # there are a lot of codes that *may* pop up here at some point
+ # but we continue to be fairly conservative. We include:
+ # 1146: Table '%s.%s' doesn't exist - what every MySQL has emitted
+ # for decades
+ #
+ # mysql 8 suddenly started emitting:
+ # 1049: Unknown database '%s' - for nonexistent schema
+ #
+ # also added:
+ # 1051: Unknown table '%s' - not known to emit
+ #
+ # there's more "doesn't exist" kinds of messages but they are
+ # less clear if mysql 8 would suddenly start using one of those
+ if self._extract_error_code(e.orig) in (1146, 1049, 1051):
return False
raise
diff --git a/lib/sqlalchemy/dialects/sqlite/base.py b/lib/sqlalchemy/dialects/sqlite/base.py
index e46443a74..fcabf5bb8 100644
--- a/lib/sqlalchemy/dialects/sqlite/base.py
+++ b/lib/sqlalchemy/dialects/sqlite/base.py
@@ -2152,6 +2152,11 @@ class SQLiteDialect(default.DefaultDialect):
def has_table(self, connection, table_name, schema=None, **kw):
self._ensure_has_table_connection(connection)
+ if schema is not None and schema not in self.get_schema_names(
+ connection, **kw
+ ):
+ return False
+
info = self._get_table_pragma(
connection, "table_info", table_name, schema=schema
)
diff --git a/lib/sqlalchemy/testing/suite/test_reflection.py b/lib/sqlalchemy/testing/suite/test_reflection.py
index 2550eff4e..8b7cb8cbc 100644
--- a/lib/sqlalchemy/testing/suite/test_reflection.py
+++ b/lib/sqlalchemy/testing/suite/test_reflection.py
@@ -177,6 +177,15 @@ class HasTableTest(OneConnectionTablesTest):
)
)
+ @testing.requires.schemas
+ def test_has_table_nonexistent_schema(self):
+ with config.db.begin() as conn:
+ is_false(
+ config.db.dialect.has_table(
+ conn, "test_table", schema="nonexistent_schema"
+ )
+ )
+
@testing.requires.views
def test_has_table_view(self, connection):
insp = inspect(connection)