71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
 
|
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)
|