Add HeXi bot codebase: custom plugins, web frontends, tests
- hexi core: message handling, rate limiting, cooldown, plugin manager - Custom plugins: BF stats, daily check-in, quotes, persona cards, etc. - Community plugins vendored under hexi/plugins with local fixes - Web admin frontends (learning-chat, persona-admin), unified hexi/web - Tests for rate_limit/cooldown/memes/persona; poetry.lock Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
import asyncio
|
||||
from dataclasses import dataclass
|
||||
from typing import cast
|
||||
|
||||
import psutil
|
||||
from cpuinfo import get_cpu_info
|
||||
from nonebot import logger
|
||||
|
||||
from . import first_time_collector, normal_collector, periodic_collector
|
||||
|
||||
|
||||
@dataclass
|
||||
class CpuFreq:
|
||||
current: float | None
|
||||
min: float | None # noqa: A003
|
||||
max: float | None # noqa: A003
|
||||
|
||||
|
||||
@first_time_collector()
|
||||
async def cpu_brand() -> str:
|
||||
try:
|
||||
brand = (
|
||||
cast("str", get_cpu_info().get("brand_raw", ""))
|
||||
.split("@", maxsplit=1)[0]
|
||||
.strip()
|
||||
)
|
||||
if brand.lower().endswith(("cpu", "processor")):
|
||||
brand = brand.rsplit(maxsplit=1)[0].strip()
|
||||
except Exception:
|
||||
logger.exception("Error when getting CPU brand")
|
||||
return "未知型号"
|
||||
else:
|
||||
return brand
|
||||
|
||||
|
||||
@first_time_collector()
|
||||
async def cpu_count_logical() -> int | None:
|
||||
return psutil.cpu_count()
|
||||
|
||||
|
||||
@first_time_collector()
|
||||
async def cpu_count() -> int | None:
|
||||
return psutil.cpu_count(logical=False)
|
||||
|
||||
|
||||
async def get_cpu_percent() -> float:
|
||||
"""实时采样 0.5s,避免 psutil 首次调用返回 0.0"""
|
||||
return await asyncio.to_thread(psutil.cpu_percent, 0.5)
|
||||
|
||||
|
||||
async def get_cpu_percent_periodic() -> float:
|
||||
"""周期采样:两次调用之间的间隔即为采样区间,直接取瞬时差值即可"""
|
||||
return psutil.cpu_percent()
|
||||
|
||||
|
||||
normal_collector("cpu_percent")(get_cpu_percent)
|
||||
periodic_collector("cpu_percent_periodic")(get_cpu_percent_periodic)
|
||||
|
||||
|
||||
async def get_cpu_freq() -> CpuFreq:
|
||||
cpu_freq = psutil.cpu_freq()
|
||||
return CpuFreq(
|
||||
current=getattr(cpu_freq, "current", None),
|
||||
min=getattr(cpu_freq, "min", None),
|
||||
max=getattr(cpu_freq, "max", None),
|
||||
)
|
||||
|
||||
|
||||
normal_collector("cpu_freq")(get_cpu_freq)
|
||||
periodic_collector("cpu_freq_periodic")(get_cpu_freq)
|
||||
Reference in New Issue
Block a user