cloudpassage/connector

"""Json and key-value formatter"""
import json
from lib.formatter import Formatter


class FormatJson(Formatter):
    def format_event(self, event):
        """Return Halo event, formatted as a json string."""
        formatted_event = "%s\n" % json.dumps(event)
        return formatted_event

    def format_issue(self, issue):
        """Return Halo issue, formatted as a json string."""
        formatted_issue = "%s\n" % json.dumps(issue)
        return formatted_issue


class FormatKv(Formatter):
    def format_event(self, event):
        """Return Halo event, formatted as a key-value string"""
        formatted_event = ""
        for key, value in event.items():
            formatted_event += "%s=\"%s\" " % (key, value)
        formatted_event += "\n"
        return formatted_event

    def format_issue(self, issue):
        """Return Halo issue, formatted as a key-value string"""
        formatted_issue = ""
        for key, value in issue.items():
            formatted_issue += "%s=\"%s\" " % (key, value)
        formatted_issue += "\n"
        return formatted_issue