Spaces:
Sleeping
Sleeping
| import datetime | |
| import json | |
| import threading | |
| from typing import Dict, Any, List, Optional | |
| from .analytics_utils import fetch_metrics, export_metrics, log_error, save_report | |
| class AnalyticsAgent: | |
| def __init__(self, config_path: str = "configs/analytics_config.json"): | |
| with open(config_path, "r") as f: | |
| self.config = json.load(f) | |
| self.metrics_cache = {} | |
| self.error_log = [] | |
| self.lock = threading.Lock() | |
| def get_live_metrics(self, spec: Optional[Dict[str, Any]] = None) -> Dict[str, Any]: | |
| """Aggregate real-time metrics on demand (API/button/query).""" | |
| try: | |
| metrics = fetch_metrics(spec or self.config["live_metrics_spec"]) | |
| with self.lock: | |
| self.metrics_cache["live"] = metrics | |
| return metrics | |
| except Exception as e: | |
| log_error(self.error_log, e) | |
| return {"error": str(e)} | |
| def run_end_of_day_report(self) -> Dict[str, Any]: | |
| """Aggregate metrics, save, notify/report at end of day (scheduled).""" | |
| try: | |
| report = fetch_metrics(self.config["end_of_day_spec"]) | |
| save_report(report, period="day") | |
| with self.lock: | |
| self.metrics_cache["end_of_day"] = report | |
| return report | |
| except Exception as e: | |
| log_error(self.error_log, e) | |
| return {"error": str(e)} | |
| def batch_export(self, period_days: int) -> Dict[str, Any]: | |
| """Export summary for past period (7/30/90 days).""" | |
| try: | |
| exported = export_metrics(period_days) | |
| with self.lock: | |
| self.metrics_cache[f"export_{period_days}"] = exported | |
| return exported | |
| except Exception as e: | |
| log_error(self.error_log, e) | |
| return {"error": str(e)} | |
| def get_errors(self) -> List[str]: | |
| """Return error logs or missing data events.""" | |
| return self.error_log | |
| def expose_endpoint(self, endpoint_type: str = "live") -> Dict[str, Any]: | |
| """Hook for UI/API to query/display current metrics.""" | |
| return self.metrics_cache.get(endpoint_type, {}) | |
| def scheduled_tasks(self): | |
| """Set up periodic triggers for end-of-day rollup/reports.""" | |
| # Implementation for actual scheduler integration needed. | |
| pass | |
| def clear_error_log(self): | |
| self.error_log.clear() | |
| # Usage Example: | |
| # agent = AnalyticsAgent() | |
| # live = agent.get_live_metrics() | |
| # eod = agent.run_end_of_day_report() | |
| # sevenday_export = agent.batch_export(7) | |
| # agent.expose_endpoint() | |
| # errors = agent.get_errors() | |