Merge pull request #259 from godatadriven/support-timespine-custom-gran

Supporting custom granularity in time_spine
godatadriven · Sep 25, 2024 · 84cfa7a · 84cfa7a
2 parents 5628bab + baea3df
commit 84cfa7a
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):