feat(chalice): support table-of-URLs showing user-count

tahayk · Jul 1, 2024 · a3ba089 · a3ba089
1 parent a1b4721
commit a3ba089
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api/schemas/schemas.py b/api/schemas/schemas.py
@@ -1184,7 +1184,9 @@ def __transform(cls, values):
 
     @model_validator(mode="after")
     def __validator(cls, values):
-        if values.metric_of not in (MetricOfTable.issues, MetricOfTable.user_browser, MetricOfTable.user_device):
+        if values.metric_of not in (MetricOfTable.issues, MetricOfTable.user_browser,
+                                    MetricOfTable.user_device, MetricOfTable.user_country,
+                                    MetricOfTable.visited_url):
             assert values.metric_format == MetricExtendedFormatType.session_count, \
                 f'metricFormat:{MetricExtendedFormatType.user_count.value} is not supported for this metricOf'
         return values