Merge pull request #1775 from PolicyEngine/fix/419-check-for-malforme…

…d-country_id-in-get_analysis Check for invalid/malformed country ID in get_analysis endpoint
PolicyEngine · Sep 20, 2024 · 62b9c9f · 62b9c9f
2 parents 1476132 + c0e2118
commit 62b9c9f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/changelog_entry.yaml b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: minor
+  changes:
+    added:
+    - Added check for invalid country in get_analysis endpoint
diff --git a/policyengine_api/endpoints/analysis.py b/policyengine_api/endpoints/analysis.py
@@ -1,4 +1,5 @@
 import os
+from policyengine_api.country import validate_country
 from policyengine_api.data import local_database
 from policyengine_api.utils import hash_object
 import time
@@ -79,6 +80,9 @@ def trigger_policy_analysis(prompt: str, prompt_id: int):
 
 
 def get_analysis(country_id: str, prompt_id=None):
+    invalid_country = validate_country(country_id)
+    if invalid_country:
+        return invalid_country
     try:
         prompt = flask.request.json.get("prompt")
     except: