Use the same select_table_schema and _normalize_table_path as Redshift (#2)

Author: dbeatty10

sqeleton/databases/duckdb.py

@@ -167,13 +167,26 @@ def create_connection(self):
         except ddb.OperationalError as e:
             raise ConnectError(*e.args) from e
 
+    def select_table_schema(self, path: DbPath) -> str:
+        database, schema, table = self._normalize_table_path(path)
+
+        info_schema_path = ["information_schema", "columns"]
+        if database:
+            info_schema_path.insert(0, database)
+
+        return (
+            f"SELECT column_name, data_type, datetime_precision, numeric_precision, numeric_scale FROM {'.'.join(info_schema_path)} "
+            f"WHERE table_name = '{table.lower()}' AND table_schema = '{schema.lower()}'"
+        )
+
     def _normalize_table_path(self, path: DbPath) -> DbPath:
         if len(path) == 1:
-            return self.default_schema, path[0]
+            return None, self.default_schema, path[0]
         elif len(path) == 2:
+            return None, path[0], path[1]
+        elif len(path) == 3:
             return path
-        elif len(path) > 2:
-            # Use only the last two values from the path
-            return path[-2:]
 
-        raise ValueError(f"{self.name}: Bad table path for {self}: '{'.'.join(path)}'. Expected form: schema.table")
+        raise ValueError(
+            f"{self.name}: Bad table path for {self}: '{'.'.join(path)}'. Expected format: table, schema.table, or database.schema.table"
+        )