diff options
| author | Federico Caselli <cfederico87@gmail.com> | 2023-02-06 21:56:18 +0000 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@bbpush.zzzcomputing.com> | 2023-02-06 21:56:18 +0000 |
| commit | 557c4a97a8657b86ee383ce8891d59f1ce3f2ec8 (patch) | |
| tree | e90d6aa69a466a149a3d32f4f77ba891eb6d61f7 /lib/sqlalchemy | |
| parent | 066b1d607e670ed00177e05969c4a30f750d1315 (diff) | |
| parent | 662d6e24526bfd9d358b8a9e1906fb29bf93ef30 (diff) | |
| download | sqlalchemy-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.py | 16 | ||||
| -rw-r--r-- | lib/sqlalchemy/dialects/sqlite/base.py | 5 | ||||
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_reflection.py | 9 |
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) |
