Regenerate manifest_v12.py

Signed-off-by: Yu Ishikawa <[email protected]>
yu-iskw · Jan 14, 2025 · 9ccbfcb · 9ccbfcb
1 parent 4e8a728
commit 9ccbfcb
Showing 1 changed file with 104 additions and 34 deletions.
diff --git a/dbt_artifacts_parser/parsers/manifest/manifest_v12.py b/dbt_artifacts_parser/parsers/manifest/manifest_v12.py
@@ -1623,7 +1623,21 @@ class OffsetWindow(BaseParserModel):
         extra='forbid',
     )
     count: int
-    granularity: str
+    granularity: Granularity
+
+
+class OffsetToGrain(Enum):
+    nanosecond = 'nanosecond'
+    microsecond = 'microsecond'
+    millisecond = 'millisecond'
+    second = 'second'
+    minute = 'minute'
+    hour = 'hour'
+    day = 'day'
+    week = 'week'
+    month = 'month'
+    quarter = 'quarter'
+    year = 'year'
 
 
 class Numerator(BaseParserModel):
@@ -1634,7 +1648,7 @@ class Numerator(BaseParserModel):
     filter: Optional[Filter2] = None
     alias: Optional[str] = None
     offset_window: Optional[OffsetWindow] = None
-    offset_to_grain: Optional[str] = None
+    offset_to_grain: Optional[OffsetToGrain] = None
 
 
 class Filter3(BaseParserModel):
@@ -1644,15 +1658,23 @@ class Filter3(BaseParserModel):
     where_filters: List[WhereFilter]
 
 
+class OffsetWindow1(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class Denominator(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
     name: str
     filter: Optional[Filter3] = None
     alias: Optional[str] = None
-    offset_window: Optional[OffsetWindow] = None
-    offset_to_grain: Optional[str] = None
+    offset_window: Optional[OffsetWindow1] = None
+    offset_to_grain: Optional[OffsetToGrain] = None
 
 
 class Window(BaseParserModel):
@@ -1684,15 +1706,23 @@ class Filter4(BaseParserModel):
     where_filters: List[WhereFilter]
 
 
+class OffsetWindow2(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class Metric(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
     name: str
     filter: Optional[Filter4] = None
     alias: Optional[str] = None
-    offset_window: Optional[OffsetWindow] = None
-    offset_to_grain: Optional[str] = None
+    offset_window: Optional[OffsetWindow2] = None
+    offset_to_grain: Optional[OffsetToGrain] = None
 
 
 class Filter5(BaseParserModel):
@@ -1741,7 +1771,7 @@ class Window1(BaseParserModel):
         extra='forbid',
     )
     count: int
-    granularity: str
+    granularity: Granularity
 
 
 class ConstantProperty(BaseParserModel):
@@ -1764,6 +1794,14 @@ class ConversionTypeParams(BaseParserModel):
     constant_properties: Optional[List[ConstantProperty]] = None
 
 
+class Window2(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class PeriodAgg(Enum):
     first = 'first'
     last = 'last'
@@ -1774,8 +1812,8 @@ class CumulativeTypeParams(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
-    window: Optional[Window1] = None
-    grain_to_date: Optional[str] = None
+    window: Optional[Window2] = None
+    grain_to_date: Optional[GrainToDate] = None
     period_agg: Optional[PeriodAgg] = 'first'
 
 
@@ -1820,6 +1858,20 @@ class Metadata1(BaseParserModel):
     file_slice: FileSlice = Field(..., title='FileSlice')
 
 
+class TimeGranularity(Enum):
+    nanosecond = 'nanosecond'
+    microsecond = 'microsecond'
+    millisecond = 'millisecond'
+    second = 'second'
+    minute = 'minute'
+    hour = 'hour'
+    day = 'day'
+    week = 'week'
+    month = 'month'
+    quarter = 'quarter'
+    year = 'year'
+
+
 class Config13(BaseParserModel):
     model_config = ConfigDict(
         extra='allow',
@@ -1847,7 +1899,7 @@ class Metrics(BaseParserModel):
     type_params: TypeParams = Field(..., title='MetricTypeParams')
     filter: Optional[Filter7] = None
     metadata: Optional[Metadata1] = None
-    time_granularity: Optional[str] = None
+    time_granularity: Optional[TimeGranularity] = None
     meta: Optional[Dict[str, Any]] = None
     tags: Optional[List[str]] = None
     config: Optional[Config13] = Field(None, title='MetricConfig')
@@ -3216,15 +3268,23 @@ class Filter10(BaseParserModel):
     where_filters: List[WhereFilter]
 
 
+class OffsetWindow3(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class Numerator1(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
     name: str
     filter: Optional[Filter10] = None
     alias: Optional[str] = None
-    offset_window: Optional[OffsetWindow] = None
-    offset_to_grain: Optional[str] = None
+    offset_window: Optional[OffsetWindow3] = None
+    offset_to_grain: Optional[OffsetToGrain] = None
 
 
 class Filter11(BaseParserModel):
@@ -3234,15 +3294,23 @@ class Filter11(BaseParserModel):
     where_filters: List[WhereFilter]
 
 
+class OffsetWindow4(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class Denominator1(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
     name: str
     filter: Optional[Filter11] = None
     alias: Optional[str] = None
-    offset_window: Optional[OffsetWindow] = None
-    offset_to_grain: Optional[str] = None
+    offset_window: Optional[OffsetWindow4] = None
+    offset_to_grain: Optional[OffsetToGrain] = None
 
 
 class Window3(BaseParserModel):
@@ -3260,15 +3328,23 @@ class Filter12(BaseParserModel):
     where_filters: List[WhereFilter]
 
 
+class OffsetWindow5(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class Metric1(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
     name: str
     filter: Optional[Filter12] = None
     alias: Optional[str] = None
-    offset_window: Optional[OffsetWindow] = None
-    offset_to_grain: Optional[str] = None
+    offset_window: Optional[OffsetWindow5] = None
+    offset_to_grain: Optional[OffsetToGrain] = None
 
 
 class Filter13(BaseParserModel):
@@ -3312,7 +3388,7 @@ class Window4(BaseParserModel):
         extra='forbid',
     )
     count: int
-    granularity: str
+    granularity: Granularity
 
 
 class ConversionTypeParams1(BaseParserModel):
@@ -3327,12 +3403,20 @@ class ConversionTypeParams1(BaseParserModel):
     constant_properties: Optional[List[ConstantProperty]] = None
 
 
+class Window5(BaseParserModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    count: int
+    granularity: Granularity
+
+
 class CumulativeTypeParams1(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',
     )
-    window: Optional[Window4] = None
-    grain_to_date: Optional[str] = None
+    window: Optional[Window5] = None
+    grain_to_date: Optional[GrainToDate] = None
     period_agg: Optional[PeriodAgg] = 'first'
 
 
@@ -3394,7 +3478,7 @@ class Disabled10(BaseParserModel):
     type_params: TypeParams1 = Field(..., title='MetricTypeParams')
     filter: Optional[Filter15] = None
     metadata: Optional[Metadata2] = None
-    time_granularity: Optional[str] = None
+    time_granularity: Optional[TimeGranularity] = None
     meta: Optional[Dict[str, Any]] = None
     tags: Optional[List[str]] = None
     config: Optional[Config27] = Field(None, title='MetricConfig')
@@ -3599,20 +3683,6 @@ class Type25(Enum):
     time = 'time'
 
 
-class TimeGranularity(Enum):
-    nanosecond = 'nanosecond'
-    microsecond = 'microsecond'
-    millisecond = 'millisecond'
-    second = 'second'
-    minute = 'minute'
-    hour = 'hour'
-    day = 'day'
-    week = 'week'
-    month = 'month'
-    quarter = 'quarter'
-    year = 'year'
-
-
 class ValidityParams(BaseParserModel):
     model_config = ConfigDict(
         extra='forbid',