Python Observability Data Helper for Beginners

import json
from datetime import datetime
from collections import defaultdict


class ObservabilityDataHelper:
    """Helper for exploring basic observability data patterns."""

    def __init__(self):
        self.events = []
        self.metrics = defaultdict(list)

    def log_event(self, service, level, message):
        """Record an event with timestamp."""
        self.events.append({
            "timestamp": datetime.utcnow().isoformat(),
            "service": service,
            "level": level,
            "message": message
        })

    def record_metric(self, name, value):
        """Record a metric sample."""
        self.metrics[name].append(value)

    def summarize(self):
        """Return a simple summary for SRE review."""
        levels = defaultdict(int)
        for event in self.events:
            levels[event["level"]] += 1

        metrics_summary = {}
        for name, values in self.metrics.items():
            if values:
                metrics_summary[name] = {
                    "count": len(values),
                    "min": min(values),
                    "max": max(values),
                    "avg": round(sum(values) / len(values), 2)
                }

        return {
            "total_events": len(self.events),
            "events_by_level": dict(levels),
            "metrics_summary": metrics_summary
        }

    def export_json(self):
        """Export observability data as JSON."""
        return json.dumps({
            "events": self.events,
            "metrics": {k: v for k, v in self.metrics.items()}
        }, indent=2)


if __name__ == "__main__":
    helper = ObservabilityDataHelper()
    helper.log_event("api-gateway", "error", "Timeout connecting to auth service")
    helper.log_event("api-gateway", "info", "Request completed successfully")
    helper.log_event("auth-service", "warn", "Token refresh rate high")
    helper.record_metric("request_latency_ms", 120)
    helper.record_metric("request_latency_ms", 95)
    helper.record_metric("request_latency_ms", 210)
    helper.record_metric("error_rate", 0.03)

    print(json.dumps(helper.summarize(), indent=2))
    print(helper.export_json())