Supporting custom granularity in time_spine

godatadriven · Sep 25, 2024 · baea3df · baea3df
1 parent 5628bab
commit baea3df
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/dbt_bouncer/dbt_cloud/manifest_latest.py b/src/dbt_bouncer/dbt_cloud/manifest_latest.py
@@ -13,11 +13,20 @@
 from pydantic import ConfigDict, Field, constr
 
 
+class CustomGranularity(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    name: str
+    column_name: Optional[str] = None
+
+
 class TimeSpine(BaseParserModel):
     model_config = ConfigDict(
-        extra="forbid",
+        extra='forbid',
     )
     standard_granularity_column: str
+    custom_granularities: Optional[List[CustomGranularity]] = None
 
 
 class Metadata(BaseParserModel):