yamon 0.1.0__py3-none-any.whl

yamon/__init__.py

@@ -0,0 +1,2 @@
+# Backend package
+

yamon/api/__init__.py

@@ -0,0 +1,2 @@
+# API package
+

yamon/api/metrics.py

@@ -0,0 +1,55 @@
+"""Metrics API endpoints"""
+
+from fastapi import APIRouter
+try:
+    from yamon.collectors.collector import MetricsCollector
+    from yamon.history import MetricsHistory
+except ImportError:
+    from collectors.collector import MetricsCollector
+    from history import MetricsHistory
+from typing import Optional
+
+router = APIRouter()
+collector = MetricsCollector()
+history = MetricsHistory(max_size=120)
+
+@router.get("/metrics")
+async def get_metrics():
+    """获取当前系统指标"""
+    metrics = collector.collect()
+    history.add_metrics(metrics)
+    return {
+        "cpu_percent": metrics.cpu_percent,
+        "cpu_per_core": metrics.cpu_per_core,
+        "cpu_count": metrics.cpu_count,
+        "memory_percent": metrics.memory_percent,
+        "memory_total": metrics.memory_total,
+        "memory_used": metrics.memory_used,
+        "memory_available": metrics.memory_available,
+        "network_sent_rate": metrics.network_sent_rate,
+        "network_recv_rate": metrics.network_recv_rate,
+        "cpu_power": metrics.cpu_power,
+        "gpu_power": metrics.gpu_power,
+        "ane_power": metrics.ane_power,
+        "system_power": metrics.system_power,
+        "gpu_usage": metrics.gpu_usage,
+        "gpu_freq_mhz": metrics.gpu_freq_mhz,
+        "ane_usage": metrics.ane_usage,
+    }
+
+@router.get("/history")
+async def get_history():
+    """获取历史数据"""
+    return {
+        "cpu_percent": history.cpu_percent.get_values(),
+        "memory_percent": history.memory_percent.get_values(),
+        "network_sent_rate": history.network_sent_rate.get_values(),
+        "network_recv_rate": history.network_recv_rate.get_values(),
+        "cpu_power": history.cpu_power.get_values(),
+        "gpu_power": history.gpu_power.get_values(),
+        "ane_power": history.ane_power.get_values(),
+        "system_power": history.system_power.get_values(),
+        "gpu_usage": history.gpu_usage.get_values(),
+        "ane_usage": history.ane_usage.get_values(),
+    }
+

yamon/api/websocket.py

@@ -0,0 +1,138 @@
+"""WebSocket API for real-time metrics"""
+
+from fastapi import APIRouter, WebSocket, WebSocketDisconnect
+try:
+    from yamon.collectors.collector import MetricsCollector, SystemMetrics
+    from yamon.history import MetricsHistory
+except ImportError:
+    from collectors.collector import MetricsCollector, SystemMetrics
+    from history import MetricsHistory
+import asyncio
+import json
+from typing import Optional
+
+router = APIRouter()
+collector = MetricsCollector()
+history = MetricsHistory(max_size=120)
+
+# 共享的最新指标数据（线程安全）
+_latest_metrics: Optional[SystemMetrics] = None
+_metrics_lock = asyncio.Lock()
+
+# 后台收集任务
+_collection_task: Optional[asyncio.Task] = None
+
+async def _background_collector():
+    """后台任务：持续收集指标数据"""
+    global _latest_metrics
+    while True:
+        try:
+            # 在线程池中运行同步的collect()方法，避免阻塞事件循环
+            metrics = await asyncio.to_thread(collector.collect)
+            async with _metrics_lock:
+                _latest_metrics = metrics
+            history.add_metrics(metrics)
+            # 收集间隔：1秒（1fps）
+            await asyncio.sleep(1.0)
+        except Exception as e:
+            print(f"Background collector error: {e}")
+            await asyncio.sleep(0.5)
+
+async def start_background_collector():
+    """启动后台收集任务"""
+    global _collection_task
+    if _collection_task is None or _collection_task.done():
+        _collection_task = asyncio.create_task(_background_collector())
+
+@router.websocket("/metrics")
+async def websocket_metrics(websocket: WebSocket):
+    """WebSocket 实时推送系统指标"""
+    await websocket.accept()
+    
+    # 确保后台收集任务已启动
+    await start_background_collector()
+    
+    try:
+        while True:
+            # 读取最新数据（不等待收集完成）
+            async with _metrics_lock:
+                metrics = _latest_metrics
+            
+            if metrics is None:
+                # 如果还没有数据，等待一下
+                await asyncio.sleep(0.1)
+                continue
+            
+            # 计算 P 核和 E 核的使用率
+            cpu_count = metrics.cpu_count
+            cpu_per_core = metrics.cpu_per_core
+            
+            # 根据 CPU 核心数判断 P 核和 E 核数量
+            if cpu_count == 8:
+                # M1/M2/M3: 4P + 4E
+                p_core_count = 4
+                e_core_count = 4
+            elif cpu_count == 10:
+                # M1 Pro/Max: 8P + 2E
+                p_core_count = 8
+                e_core_count = 2
+            elif cpu_count == 12:
+                # M2 Pro/Max: 8P + 4E 或 M3 Pro: 6P + 6E
+                # 默认假设 8P + 4E
+                p_core_count = 8
+                e_core_count = 4
+            elif cpu_count == 16:
+                # M3 Max: 12P + 4E
+                p_core_count = 12
+                e_core_count = 4
+            else:
+                # 默认：前一半是 P 核
+                p_core_count = cpu_count // 2
+                e_core_count = cpu_count - p_core_count
+            
+            # 计算 P 核和 E 核的使用率
+            p_cores = cpu_per_core[:p_core_count] if len(cpu_per_core) >= p_core_count else []
+            e_cores = cpu_per_core[p_core_count:] if len(cpu_per_core) > p_core_count else []
+            
+            # 计算 P 核和 E 核对总CPU使用率的贡献百分比
+            # cpu_percent 是所有核心的平均使用率，即 sum(cpu_per_core) / cpu_count
+            # 所以 P核贡献 = sum(p_cores) / cpu_count
+            # E核贡献 = sum(e_cores) / cpu_count
+            # P核贡献 + E核贡献 = cpu_percent
+            if cpu_count > 0:
+                cpu_p_percent = (sum(p_cores) / cpu_count) if p_cores else 0.0
+                cpu_e_percent = (sum(e_cores) / cpu_count) if e_cores else 0.0
+            else:
+                cpu_p_percent = 0.0
+                cpu_e_percent = 0.0
+            
+            # 发送数据
+            await websocket.send_json({
+                "cpu_percent": metrics.cpu_percent,
+                "cpu_per_core": metrics.cpu_per_core,
+                "cpu_count": metrics.cpu_count,
+                "cpu_p_percent": cpu_p_percent,
+                "cpu_e_percent": cpu_e_percent,
+                "memory_percent": metrics.memory_percent,
+                "memory_total": metrics.memory_total,
+                "memory_used": metrics.memory_used,
+                "memory_available": metrics.memory_available,
+                "network_sent_rate": metrics.network_sent_rate,
+                "network_recv_rate": metrics.network_recv_rate,
+                "cpu_power": metrics.cpu_power,
+                "gpu_power": metrics.gpu_power,
+                "ane_power": metrics.ane_power,
+                "system_power": metrics.system_power,
+                "gpu_usage": metrics.gpu_usage,
+                "gpu_freq_mhz": metrics.gpu_freq_mhz,
+                "ane_usage": metrics.ane_usage,
+            })
+            
+            # 等待 1 秒（1fps）
+            await asyncio.sleep(1.0)
+            
+    except WebSocketDisconnect:
+        pass
+    except Exception as e:
+        print(f"WebSocket error: {e}")
+

yamon/cli.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+"""Command-line interface for Yamon"""
+
+import argparse
+import sys
+import uvicorn
+from pathlib import Path
+
+
+def main():
+    """Main entry point for yamon command"""
+    parser = argparse.ArgumentParser(
+        description="Yamon - Visually pleasing, deep system monitoring for macOS"
+    )
+    parser.add_argument(
+        "--host",
+        default="0.0.0.0",
+        help="Host to bind to (default: 0.0.0.0)",
+    )
+    parser.add_argument(
+        "--port",
+        type=int,
+        default=8000,
+        help="Port to bind to (default: 8000)",
+    )
+    parser.add_argument(
+        "--reload",
+        action="store_true",
+        help="Enable auto-reload (development mode)",
+    )
+    
+    args = parser.parse_args()
+    
+    # Import app after parsing args
+    from yamon.main import app
+    
+    uvicorn.run(
+        app,
+        host=args.host,
+        port=args.port,
+        reload=args.reload,
+    )
+
+
+if __name__ == "__main__":
+    main()
+

yamon/collectors/__init__.py

@@ -0,0 +1,2 @@
+# Collectors package
+