Merge pull request #16 from verkada/jackkeane/fix-query-parsing

Fix bug where query parameter string could get parsed as number
verkada · Oct 16, 2024 · 48384d0 · 48384d0
2 parents 3822689 + 5443a6f
commit 48384d0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/flask_pydantic_spec/flask_backend.py b/flask_pydantic_spec/flask_backend.py
@@ -151,7 +151,9 @@ def request_validation(
                     {} if request.get_data() == b"" else request.get_json(force=True)
                 )
         elif request.content_type and "multipart/form-data" in request.content_type:
-            parsed_body = parse_multi_dict(request.form) if request.form else {}
+            parsed_body = (
+                parse_multi_dict(request.form, parse_json=True) if request.form else {}
+            )
         else:
             parsed_body = request.get_data() or {}
         req_headers: Optional[Headers] = request.headers or None

diff --git a/flask_pydantic_spec/utils.py b/flask_pydantic_spec/utils.py
@@ -197,14 +197,16 @@ def default_after_handler(
         )
 
 
-def parse_multi_dict(input: MultiDict) -> Dict[str, Any]:
+def parse_multi_dict(input: MultiDict, parse_json: bool = False) -> Dict[str, Any]:
     result = {}
     for key, value in input.to_dict(flat=False).items():
-        if len(value) == 1:
+        if len(value) == 1 and parse_json:
             try:
                 value_to_use = json.loads(value[0])
             except (TypeError, JSONDecodeError):
                 value_to_use = value[0]
+        elif len(value) == 1:
+            value_to_use = value[0]
         else:
             value_to_use = value
         result[key] = value_to_use