feat: use public rules endpoint

bento-platform · Jul 16, 2024 · 9f8fb3b · 9f8fb3b
1 parent d9b43f4
commit 9f8fb3b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/bento_beacon/config_files/config.py b/bento_beacon/config_files/config.py
@@ -149,6 +149,7 @@ class Config:
     KATSU_SEARCH_OVERVIEW = "/api/search_overview"
     KATSU_PRIVATE_OVERVIEW = "/api/overview"
     KATSU_PUBLIC_OVERVIEW = "/api/public_overview"
+    KATSU_PUBLIC_RULES = "/api/public_rules"
     KATSU_TIMEOUT = int(os.environ.get("BEACON_KATSU_TIMEOUT", 180))
 
     MAP_EXTRA_PROPERTIES_TO_INFO = os.environ.get("MAP_EXTRA_PROPERTIES_TO_INFO", True)

diff --git a/bento_beacon/utils/katsu_utils.py b/bento_beacon/utils/katsu_utils.py
@@ -278,9 +278,10 @@ def overview_statistics():
 
 
 def katsu_censorship_settings() -> tuple[int | None, int | None]:
-    overview = katsu_get(current_app.config["KATSU_PUBLIC_OVERVIEW"])
-    max_filters = overview.get("max_query_parameters")
-    count_threshold = overview.get("count_threshold")
+    # TODO: should be project-dataset scoped
+    rules = katsu_get(current_app.config["KATSU_PUBLIC_RULES"])
+    max_filters = rules.get("max_query_parameters")
+    count_threshold = rules.get("count_threshold")
     # return even if None
     return max_filters, count_threshold