import json
from datetime import datetime
from prospector.formatters.base import Formatter
__all__ = ("JsonFormatter",)
class JsonFormatter(Formatter):
def render(self, summary=True, messages=True, profile=False):
output = {}
if summary:
# we need to slightly change the types and format
# of a few of the items in the summary to make
# them play nice with JSON formatting
munged = {}
for key, value in self.summary.items():
if isinstance(value, datetime):
munged[key] = str(value)
else:
munged[key] = value
output["summary"] = munged
if profile:
output["profile"] = self.profile.as_dict()
if messages:
output["messages"] = [self._message_to_dict(m) for m in self.messages]
return json.dumps(output, indent=2)
|