import json import sys from datetime import datetime from typing import Any class OutputFormatter: def __init__(self, format_type: str = "text", quiet: bool = False): self.format_type = format_type self.quiet = quiet def output(self, data: Any, message_type: str = "response"): if self.quiet and message_type not in ["error", "result"]: return if self.format_type == "json": self._output_json(data, message_type) elif self.format_type == "structured": self._output_structured(data, message_type) else: self._output_text(data, message_type) def _output_json(self, data: Any, message_type: str): output = { "type": message_type, "timestamp": datetime.now().isoformat(), "data": data, } print(json.dumps(output, indent=2)) def _output_structured(self, data: Any, message_type: str): if isinstance(data, dict): for key, value in data.items(): print(f"{key}: {value}") elif isinstance(data, list): for item in data: print(f"- {item}") else: print(data) def _output_text(self, data: Any, message_type: str): if isinstance(data, (dict, list)): print(json.dumps(data, indent=2)) else: print(data) def error(self, message: str): if self.format_type == "json": self._output_json({"error": message}, "error") else: print(f"Error: {message}", file=sys.stderr) def success(self, message: str): if not self.quiet: if self.format_type == "json": self._output_json({"success": message}, "success") else: print(message) def info(self, message: str): if not self.quiet: if self.format_type == "json": self._output_json({"info": message}, "info") else: print(message) def result(self, data: Any): self.output(data, "result")