fix(chalice): fixed cards-schema validators for the new Pydantic

tahayk · Aug 13, 2024 · f9ea222 · f9ea222
1 parent 02a2a51
commit f9ea222
Showing 1 changed file with 36 additions and 36 deletions.
diff --git a/api/schemas/schemas.py b/api/schemas/schemas.py
@@ -1151,10 +1151,10 @@ def __enforce_default(cls, values):
         values["metricValue"] = []
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfTimeseries(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfTimeseries(self.metric_of)
+        return self
 
 
 class CardTable(__CardSchema):
@@ -1170,10 +1170,10 @@ def __enforce_default(cls, values):
             values["metricValue"] = []
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfTable(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfTable(self.metric_of)
+        return self
 
     @model_validator(mode="after")
     def __validator(self):
@@ -1200,10 +1200,10 @@ def __enforce_default(cls, values):
             values["series"] = [values["series"][0]]
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfFunnels(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfFunnels(self.metric_of)
+        return self
 
 
 class CardErrors(__CardSchema):
@@ -1217,10 +1217,10 @@ def __enforce_default(cls, values):
         values["series"] = []
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfErrors(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfErrors(self.metric_of)
+        return self
 
 
 class CardPerformance(__CardSchema):
@@ -1234,10 +1234,10 @@ def __enforce_default(cls, values):
         values["series"] = []
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfPerformance(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfPerformance(self.metric_of)
+        return self
 
 
 class CardResources(__CardSchema):
@@ -1251,10 +1251,10 @@ def __enforce_default(cls, values):
         values["series"] = []
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfResources(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfResources(self.metric_of)
+        return self
 
 
 class CardWebVital(__CardSchema):
@@ -1268,10 +1268,10 @@ def __enforce_default(cls, values):
         values["series"] = []
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfWebVitals(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfWebVitals(self.metric_of)
+        return self
 
 
 class CardHeatMap(__CardSchema):
@@ -1284,10 +1284,10 @@ class CardHeatMap(__CardSchema):
     def __enforce_default(cls, values):
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfHeatMap(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfHeatMap(self.metric_of)
+        return self
 
 
 class MetricOfInsights(str, Enum):
@@ -1305,10 +1305,10 @@ def __enforce_default(cls, values):
         values["view_type"] = MetricOtherViewType.LIST_CHART
         return values
 
-    # @model_validator(mode="after")
-    # def __transform(self):
-    #     self.metric_of = MetricOfInsights(self.metric_of)
-    #     return self
+    @model_validator(mode="after")
+    def __transform(self):
+        self.metric_of = MetricOfInsights(self.metric_of)
+        return self
 
     @model_validator(mode="after")
     def restrictions(self):