fix: database quoting (#299)

Author: Ryuk-me

dcs_core/core/datasource/sql_datasource.py

@@ -137,6 +137,14 @@ def qualified_table_name(self, table_name: str) -> str:
             return f"[{self.schema_name}].[{table_name}]"
         return f"[{table_name}]"
 
+    def quote_database(self, database: str) -> str:
+        """
+        Quote the database name
+        :param database: name of the database
+        :return: quoted database name
+        """
+        return f'"{database}"'
+
     def quote_column(self, column: str) -> str:
         """
         Quote the column name

dcs_core/integrations/databases/mssql.py

@@ -150,7 +150,7 @@ def query_get_table_columns(
         :return: RawColumnInfo object containing column information
         """
         schema = schema or self.schema_name
-        database = self.database
+        database = self.quote_database(self.database)
         query = f"SELECT column_name, data_type, datetime_precision, numeric_precision, numeric_scale, collation_name, character_maximum_length FROM {database}.information_schema.columns WHERE table_name = '{table}' AND table_schema = '{schema}'"
 
         rows = self.fetchall(query)

dcs_core/integrations/databases/postgres.py

@@ -82,9 +82,9 @@ def query_get_table_names(
         """
 
         schema = schema or self.schema_name
-        database = self.database
+        database = self.quote_database(self.database)
         query = (
-            f'SELECT table_name FROM "{database}".information_schema.tables '
+            f'SELECT table_name FROM {database}.information_schema.tables '
             f"WHERE table_schema = '{schema}' AND table_type = 'BASE TABLE'"
         )
         result = self.fetchall(query)
@@ -103,7 +103,8 @@ def query_get_table_columns(
         schema = schema or self.schema_name
         info_schema_path = ["information_schema", "columns"]
         if self.database:
-            info_schema_path.insert(0, self.database)
+            database = self.quote_database(self.database)
+            info_schema_path.insert(0, database)
         query = (
             f"SELECT column_name, data_type, datetime_precision, "
             f"CASE WHEN data_type = 'numeric' "