Merge 6533b14 into d4353f0

rycus86 · Jul 10, 2020 · 4b481bd · 4b481bd
2 parents d4353f0 + 6533b14
commit 4b481bd
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/prometheus_flask_exporter/__init__.py b/prometheus_flask_exporter/__init__.py
@@ -1,3 +1,4 @@
+import dataclasses
 import os
 import re
 import sys
@@ -624,6 +625,15 @@ def func(*args, **kwargs):
                             except AttributeError:
                                 break
 
+                        # Breaks dataclass/list of dataclass down to dict/dicts.
+                        if dataclasses.is_dataclass(response):
+                            response = dataclasses.asdict(response[0]), 200
+                        if isinstance(response, tuple):
+                            if dataclasses.is_dataclass(response[0]):
+                                response = dataclasses.asdict(response[0]), response[1]
+                            elif isinstance(response[0], list) and dataclasses.is_dataclass(response[0][0]):
+                                response = [dataclasses.asdict(x) for x in response[0]], response[1]
+
                         if view_func == f:
                             # we are in a request handler method
                             response = make_response(response)