81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
# 日志推送模块 · HTTP POST到主服务器
|
||
# HLDP://zhuyuan-agent/log-pusher
|
||
|
||
import json
|
||
import urllib.request
|
||
from datetime import datetime
|
||
|
||
|
||
class LogPusher:
|
||
"""向主服务器推送操作日志、日记、GPU指标、训练进度"""
|
||
|
||
def __init__(self, base_url: str, api_key: str, hostname: str = "3090-server"):
|
||
self.base_url = base_url.rstrip("/")
|
||
self.api_key = api_key
|
||
self.hostname = hostname
|
||
self._headers = {
|
||
"Authorization": f"Bearer {api_key}",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
def _post(self, path: str, data: dict, timeout: int = 15) -> bool:
|
||
"""POST请求到主服务器"""
|
||
url = f"{self.base_url}{path}"
|
||
try:
|
||
req = urllib.request.Request(
|
||
url,
|
||
data=json.dumps(data).encode("utf-8"),
|
||
headers=self._headers,
|
||
method="POST"
|
||
)
|
||
resp = urllib.request.urlopen(req, timeout=timeout)
|
||
result = json.loads(resp.read())
|
||
return result.get("ok", False)
|
||
except Exception as e:
|
||
print(f"[PUSH ERROR] {path}: {e}")
|
||
return False
|
||
|
||
def push_gpu(self, gpu_data: dict) -> bool:
|
||
"""推送GPU指标"""
|
||
data = {
|
||
"hostname": self.hostname,
|
||
"gpus": gpu_data.get("gpus", [])
|
||
}
|
||
return self._post("/api/gpu/status", data)
|
||
|
||
def push_training(self, training_data: dict) -> bool:
|
||
"""推送训练进度"""
|
||
return self._post("/api/training/status", training_data)
|
||
|
||
def push_log(self, level: str, message: str, category: str = "agent") -> bool:
|
||
"""推送操作日志"""
|
||
return self._post("/api/agent/log", {
|
||
"level": level,
|
||
"message": message,
|
||
"category": category
|
||
})
|
||
|
||
def push_diary(self, entry_type: str, title: str, description: str = "") -> bool:
|
||
"""推送日记条目"""
|
||
return self._post("/api/agent/diary", {
|
||
"type": entry_type,
|
||
"title": title,
|
||
"description": description
|
||
})
|
||
|
||
def log_info(self, msg: str):
|
||
"""快捷：info级别日志"""
|
||
self.push_log("info", msg)
|
||
|
||
def log_success(self, msg: str):
|
||
"""快捷：success级别日志"""
|
||
self.push_log("success", msg)
|
||
|
||
def log_warn(self, msg: str):
|
||
"""快捷：warn级别日志"""
|
||
self.push_log("warn", msg)
|
||
|
||
def log_error(self, msg: str):
|
||
"""快捷：error级别日志"""
|
||
self.push_log("error", msg)
|