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

tahayk · Jun 21, 2024 · 52b4a80 · 52b4a80
1 parent 057338d
commit 52b4a80
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api/schemas/schemas.py b/api/schemas/schemas.py
@@ -1151,7 +1151,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