<reponame>vipulgupta2048/kube-hunter
from io import StringIO
from ruamel.yaml import YAML

from kube_hunter.conf import config
from kube_hunter.modules.report.base import BaseReporter


class YAMLReporter(BaseReporter):
    def get_report(self):
        report = super().get_report()
        output = StringIO()
        yaml = YAML()
        yaml.dump(report, output)
        return output.getvalue()
