Fallback JSON loading during Task unserializing (#225)

CERT-Polska · Sep 1, 2023 · 0cbb3d0 · 0cbb3d0
1 parent fee4947
commit 0cbb3d0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/karton/core/task.py b/karton/core/task.py
@@ -423,7 +423,11 @@ def unserialize_resources(value: Any) -> Any:
         if parse_resources:
             task_data = json.loads(data, object_hook=unserialize_resources)
         else:
-            task_data = orjson.loads(data)
+            try:
+                task_data = orjson.loads(data)
+            except orjson.JSONDecodeError:
+                # fallback, in case orjson raises exception during loading
+                task_data = json.loads(data, object_hook=unserialize_resources)
 
         # Compatibility with Karton <5.2.0
         headers_persistent_fallback = task_data["payload_persistent"].get(