sqlspec 0.26.0__py3-none-any.whl → 0.27.0__py3-none-any.whl

sqlspec/adapters/bigquery/data_dictionary.py

@@ -1,10 +1,13 @@
 """BigQuery-specific data dictionary for metadata queries."""
 
-from typing import Optional
+from typing import TYPE_CHECKING, Any, cast
 
 from sqlspec.driver import SyncDataDictionaryBase, SyncDriverAdapterBase, VersionInfo
 from sqlspec.utils.logging import get_logger
 
+if TYPE_CHECKING:
+    from sqlspec.adapters.bigquery.driver import BigQueryDriver
+
 logger = get_logger("adapters.bigquery.data_dictionary")
 
 __all__ = ("BigQuerySyncDataDictionary",)
@@ -13,7 +16,7 @@ __all__ = ("BigQuerySyncDataDictionary",)
 class BigQuerySyncDataDictionary(SyncDataDictionaryBase):
     """BigQuery-specific sync data dictionary."""
 
-    def get_version(self, driver: SyncDriverAdapterBase) -> "Optional[VersionInfo]":
+    def get_version(self, driver: SyncDriverAdapterBase) -> "VersionInfo | None":
         """Get BigQuery version information.
 
         BigQuery is a cloud service without traditional versioning.
@@ -85,6 +88,43 @@ class BigQuerySyncDataDictionary(SyncDataDictionaryBase):
         }
         return type_map.get(type_category, "STRING")
 
+    def get_columns(
+        self, driver: SyncDriverAdapterBase, table: str, schema: "str | None" = None
+    ) -> "list[dict[str, Any]]":
+        """Get column information for a table using INFORMATION_SCHEMA.
+
+        Args:
+            driver: BigQuery driver instance
+            table: Table name to query columns for
+            schema: Schema name (dataset name in BigQuery)
+
+        Returns:
+            List of column metadata dictionaries with keys:
+                - column_name: Name of the column
+                - data_type: BigQuery data type
+                - is_nullable: Whether column allows NULL (YES/NO)
+                - column_default: Default value if any
+        """
+        bigquery_driver = cast("BigQueryDriver", driver)
+
+        if schema:
+            sql = f"""
+                SELECT column_name, data_type, is_nullable, column_default
+                FROM `{schema}.INFORMATION_SCHEMA.COLUMNS`
+                WHERE table_name = '{table}'
+                ORDER BY ordinal_position
+            """
+        else:
+            sql = f"""
+                SELECT column_name, data_type, is_nullable, column_default
+                FROM INFORMATION_SCHEMA.COLUMNS
+                WHERE table_name = '{table}'
+                ORDER BY ordinal_position
+            """
+
+        result = bigquery_driver.execute(sql)
+        return result.data or []
+
     def list_available_features(self) -> "list[str]":
         """List available BigQuery feature flags.