get tables function modified

PeterDKay · PeterDKay · commit 62c79314c7d0 · 2023-05-22T10:33:02.000+01:00
diff --git a/src/databricks/sqlalchemy/dialect/__init__.py b/src/databricks/sqlalchemy/dialect/__init__.py
@@ -247,12 +247,13 @@ def get_indexes(self, connection, table_name, schema=None, **kw):
     def get_table_names(self, connection, schema=None, **kwargs):
         TABLE_NAME = 1
         catalog = "`" + self.catalog + "`"
+        schema = ("`" + schema + "`") or ("`" + self.schema + "`")
 
         with self.get_driver_connection(
             connection
         )._dbapi_connection.dbapi_connection.cursor() as cur:
             sql_str = "SHOW TABLES FROM {}".format(
-                ".".join([catalog, schema or self.schema])
+                ".".join([catalog, schema])
             )
             data = cur.execute(sql_str).fetchall()
             _tables = [i[TABLE_NAME] for i in data]
