Add to_dict method on StatsError

locustio · Oct 11, 2023 · 6e25463 · 6e25463
1 parent 2c281ff
commit 6e25463
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/locust/stats.py b/locust/stats.py
@@ -765,6 +765,14 @@ def _getattr(obj: "StatsError", key: str, default: Optional[Any]) -> Optional[An
     def unserialize(cls, data: StatsErrorDict) -> "StatsError":
         return cls(data["method"], data["name"], data["error"], data["occurrences"])
 
+    def to_dict(self, escape_string_values=False):
+        return {
+            "method": escape(self.method),
+            "name": escape(self.name),
+            "error": escape(self.parse_error(self.error)),
+            "occurrences": self.occurrences,
+        }
+
 
 def avg(values: List[float | int]) -> float:
     return sum(values, 0.0) / max(len(values), 1)