Getting table comment is broken if it's not supported (dungdm93#11)

youngwookim · May 2, 2021 · ffafff2 · ffafff2
1 parent b8e9aff
commit ffafff2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/sqlalchemy_trino/dialect.py b/sqlalchemy_trino/dialect.py
@@ -222,6 +222,7 @@ def get_table_comment(self, connection: Connection,
                 error.NOT_FOUND,
                 error.COLUMN_NOT_FOUND,
                 error.TABLE_NOT_FOUND,
+                error.NOT_SUPPORTED
             ):
                 return dict(text=None)
             raise

diff --git a/sqlalchemy_trino/error.py b/sqlalchemy_trino/error.py
@@ -8,6 +8,7 @@
 TABLE_NOT_FOUND = 'TABLE_NOT_FOUND'
 SCHEMA_NOT_FOUND = 'SCHEMA_NOT_FOUND'
 CATALOG_NOT_FOUND = 'CATALOG_NOT_FOUND'
+NOT_SUPPORTED = 'NOT_SUPPORTED'
 
 MISSING_TABLE = 'MISSING_TABLE'
 MISSING_COLUMN_NAME = 'MISSING_COLUMN_NAME'