add: fetch view along with table name

Author: AnshumanX304

dcs_core/integrations/databases/mssql.py

@@ -126,16 +126,21 @@ def quote_column(self, column: str) -> str:
         return f"[{column}]"
 
     def query_get_table_names(
-        self,
-        schema: str | None = None,
+        self, schema: str | None = None, with_view: bool = False
     ) -> List[str]:
         """
         Get the list of tables in the database.
         :param schema: optional schema name
         :return: list of table names
         """
         schema = schema or self.schema_name
-        query = f"SELECT o.name AS table_name FROM sys.objects o JOIN sys.schemas s ON o.schema_id = s.schema_id WHERE o.type = 'U' AND s.name = '{schema}' ORDER BY o.name"
+
+        if with_view:
+            object_types = "IN ('U', 'V')"
+        else:
+            object_types = "= 'U'"
+
+        query = f"SELECT o.name AS table_name FROM sys.objects o JOIN sys.schemas s ON o.schema_id = s.schema_id WHERE o.type {object_types} AND s.name = '{schema}' ORDER BY o.name"
 
         rows = self.fetchall(query)
         res = [row[0] for row in rows] if rows else []

dcs_core/integrations/databases/mysql.py

@@ -91,14 +91,20 @@ def quote_column(self, column: str) -> str:
     def query_get_table_names(
         self,
         schema: str | None = None,
+        with_view: bool = False,
     ) -> List[str]:
         """
         Get the list of tables in the database.
         :param schema: optional schema name
         :return: list of table names
         """
         database = self.database
-        query = f"SELECT TABLES.TABLE_NAME FROM information_schema.tables WHERE TABLES.TABLE_SCHEMA = '{database}' and TABLES.TABLE_TYPE = 'BASE TABLE'"
+        if with_view:
+            table_type_condition = "TABLES.TABLE_TYPE IN ('BASE TABLE', 'VIEW')"
+        else:
+            table_type_condition = "TABLES.TABLE_TYPE = 'BASE TABLE'"
+
+        query = f"SELECT TABLES.TABLE_NAME FROM information_schema.tables WHERE TABLES.TABLE_SCHEMA = '{database}' and {table_type_condition}"
         result = self.fetchall(query)
         return [row[0] for row in result]
 

dcs_core/integrations/databases/oracle.py

@@ -86,14 +86,22 @@ def quote_column(self, column: str) -> str:
     def query_get_table_names(
         self,
         schema: str | None = None,
+        with_view: bool = False,
     ) -> List[str]:
         """
         Get the list of tables in the database.
         :param schema: optional schema name
         :return: list of table names
         """
         schema = schema or self.schema_name
-        query = f"SELECT TABLE_NAME FROM ALL_ALL_TABLES WHERE OWNER = '{schema}'"
+        if with_view:
+            query = (
+                f"SELECT TABLE_NAME FROM ALL_ALL_TABLES WHERE OWNER = '{schema}' "
+                f"UNION "
+                f"SELECT VIEW_NAME AS TABLE_NAME FROM ALL_VIEWS WHERE OWNER = '{schema}'"
+            )
+        else:
+            query = f"SELECT TABLE_NAME FROM ALL_ALL_TABLES WHERE OWNER = '{schema}'"
         result = self.fetchall(query)
         return [row[0] for row in result]
 

dcs_core/integrations/databases/postgres.py

@@ -74,6 +74,7 @@ def quote_column(self, column: str) -> str:
     def query_get_table_names(
         self,
         schema: str | None = None,
+        with_view: bool = False,
     ) -> List[str]:
         """
         Get the list of tables in the database.
@@ -83,9 +84,15 @@ def query_get_table_names(
 
         schema = schema or self.schema_name
         database = self.quote_database(self.database)
+
+        if with_view:
+            table_type_condition = "table_type IN ('BASE TABLE', 'VIEW')"
+        else:
+            table_type_condition = "table_type = 'BASE TABLE'"
+
         query = (
-            f'SELECT table_name FROM {database}.information_schema.tables '
-            f"WHERE table_schema = '{schema}' AND table_type = 'BASE TABLE'"
+            f"SELECT table_name FROM {database}.information_schema.tables "
+            f"WHERE table_schema = '{schema}' AND {table_type_condition}"
         )
         result = self.fetchall(query)
         return [row[0] for row in result]

dcs_core/integrations/databases/sybase.py

@@ -331,6 +331,7 @@ def query_get_table_columns(
     def query_get_table_names(
         self,
         schema: str | None = None,
+        with_view: bool = False,
     ) -> List[str]:
         """
         Get the list of tables in the database.
@@ -339,13 +340,20 @@ def query_get_table_names(
         """
         schema = schema or self.schema_name
         database = self.database
+        if with_view:
+            type_condition = "IN ('U', 'V')"
+        else:
+            type_condition = "= 'U'"
 
         if self.sybase_driver_type.is_iq:
-            query = f"SELECT table_name FROM {database}.SYS.SYSTABLE WHERE creator = USER_ID('{schema}') AND table_type = 'BASE'"
+            table_type_condition = (
+                "table_type IN ('BASE', 'VIEW')" if with_view else "table_type = 'BASE'"
+            )
+            query = f"SELECT table_name FROM {database}.SYS.SYSTABLE WHERE creator = USER_ID('{schema}') AND {table_type_condition}"
         elif self.sybase_driver_type.is_ase:
-            query = f"SELECT name AS table_name FROM {database}..sysobjects WHERE type = 'U' AND uid = USER_ID('{schema}')"
+            query = f"SELECT name AS table_name FROM {database}..sysobjects WHERE type {type_condition} AND uid = USER_ID('{schema}')"
         elif self.sybase_driver_type.is_freetds:
-            query = f"SELECT name AS table_name FROM {database}.dbo.sysobjects WHERE type = 'U' AND uid = USER_ID('{schema}')"
+            query = f"SELECT name AS table_name FROM {database}.dbo.sysobjects WHERE type {type_condition} AND uid = USER_ID('{schema}')"
         else:
             raise ValueError("Unknown Sybase driver type")