Add "get_view_names" to SQLAlchemy dialect

crate · Dec 18, 2020 · 9c3b47a · 9c3b47a
1 parent af8e256
commit 9c3b47a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/crate/client/sqlalchemy/dialect.py b/src/crate/client/sqlalchemy/dialect.py
@@ -246,6 +246,15 @@ def get_table_names(self, connection, schema=None, **kw):
         )
         return [row[0] for row in cursor.fetchall()]
 
+    @reflection.cache
+    def get_view_names(self, connection, schema=None, **kw):
+        cursor = connection.execute(
+            "SELECT table_name FROM information_schema.views "
+            "ORDER BY table_name ASC, {0} ASC".format(self.schema_column),
+            [schema or self.default_schema_name]
+        )
+        return [row[0] for row in cursor.fetchall()]
+
     @reflection.cache
     def get_columns(self, connection, table_name, schema=None, **kw):
         query = "SELECT column_name, data_type " \