yaml.py 613 B

1234567891011121314151617181920212223242526
  1. import yaml
  2. from prospector.formatters.base import Formatter
  3. __all__ = ("YamlFormatter",)
  4. class YamlFormatter(Formatter):
  5. def render(self, summary=True, messages=True, profile=False):
  6. output = {}
  7. if summary:
  8. output["summary"] = self.summary
  9. if profile:
  10. output["profile"] = self.profile.as_dict()
  11. if messages:
  12. output["messages"] = [self._message_to_dict(m) for m in self.messages]
  13. return yaml.safe_dump(
  14. output,
  15. indent=2,
  16. default_flow_style=False,
  17. allow_unicode=True,
  18. )