From 8d2c9ae434a9026e9a198120c9a7b6faf0346cd7 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sun, 29 Apr 2007 20:08:55 +0000 Subject: [PATCH] - mysql uses "DESCRIBE [].", catching exceptions if table doesnt exist, in order to determine if a table exists. this supports unicode table names as well as schema names. tested with MySQL5 but should work with 4.1 series as well. (#557) --- CHANGES | 4 ++++ lib/sqlalchemy/databases/mysql.py | 23 ++++++++++++++++++----- test/sql/unicode.py | 4 ---- 3 files changed, 22 insertions(+), 9 deletions(-) diff --git a/CHANGES b/CHANGES index 441be8b036..934733d0b1 100644 --- a/CHANGES +++ b/CHANGES @@ -116,6 +116,10 @@ - mysql - support for SSL arguments given as inline within URL query string, prefixed with "ssl_", courtesy terjeros@gmail.com. + - mysql uses "DESCRIBE [].", catching exceptions + if table doesnt exist, in order to determine if a table exists. + this supports unicode table names as well as schema names. tested + with MySQL5 but should work with 4.1 series as well. (#557) - extensions - big fix to AssociationProxy so that multiple AssociationProxy objects can be associated with a single association collection. diff --git a/lib/sqlalchemy/databases/mysql.py b/lib/sqlalchemy/databases/mysql.py index c33b515b58..0903befe9a 100644 --- a/lib/sqlalchemy/databases/mysql.py +++ b/lib/sqlalchemy/databases/mysql.py @@ -364,11 +364,24 @@ class MySQLDialect(ansisql.ANSIDialect): def has_table(self, connection, table_name, schema=None): # TODO: this does not work for table names that contain multibyte characters. - # i have tried dozens of approaches here with no luck. statements like - # DESCRIBE and SHOW CREATE TABLE work better, but they raise an error when - # the table does not exist. - cursor = connection.execute("show table status like %s", [table_name]) - return bool( not not cursor.rowcount ) + + # http://dev.mysql.com/doc/refman/5.0/en/error-messages-server.html + + # Error: 1146 SQLSTATE: 42S02 (ER_NO_SUCH_TABLE) + # Message: Table '%s.%s' doesn't exist + + # Error: 1046 SQLSTATE: 3D000 (ER_NO_DB_ERROR) + # Message: No database selected + + try: + name = schema and ("%s.%s" % (schema, table_name)) or table_name + connection.execute("DESCRIBE `%s`" % name) + return True + except exceptions.SQLError, e: + if e.orig.args[0] in (1146, 1046): + return False + else: + raise def reflecttable(self, connection, table): # reference: http://dev.mysql.com/doc/refman/5.0/en/name-case-sensitivity.html diff --git a/test/sql/unicode.py b/test/sql/unicode.py index 9dfc75059e..65a7cce0d0 100644 --- a/test/sql/unicode.py +++ b/test/sql/unicode.py @@ -28,10 +28,6 @@ class UnicodeSchemaTest(testbase.PersistTest): def tearDownAll(self): metadata.drop_all() - # has_table() doesnt handle the unicode names on mysql - if testbase.db.name == 'mysql': - t2.drop() - def test_insert(self): t1.insert().execute({u'méil':1, u'éXXm':5}) t2.insert().execute({'a':1, 'b':1}) -- 2.47.2