mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-12 01:00:29 +08:00
143 lines
5.3 KiB
Python
143 lines
5.3 KiB
Python
# ========================================================================== #
|
|
# #
|
|
# KVMD - The main PiKVM daemon. #
|
|
# #
|
|
# Copyright (C) 2018-2022 Maxim Devaev <mdevaev@gmail.com> #
|
|
# #
|
|
# This program is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
|
|
# #
|
|
# ========================================================================== #
|
|
|
|
|
|
import os
|
|
import asyncio
|
|
|
|
from typing import Callable
|
|
from typing import AsyncGenerator
|
|
from typing import TypeVar
|
|
|
|
from ....logging import get_logger
|
|
|
|
from .... import env
|
|
from .... import tools
|
|
from .... import aiofs
|
|
from .... import aioproc
|
|
|
|
from .base import BaseInfoSubmanager
|
|
|
|
|
|
# =====
|
|
_RetvalT = TypeVar("_RetvalT")
|
|
|
|
|
|
# =====
|
|
class HwInfoSubmanager(BaseInfoSubmanager):
|
|
def __init__(
|
|
self,
|
|
vcgencmd_cmd: list[str],
|
|
state_poll: float,
|
|
) -> None:
|
|
|
|
self.__vcgencmd_cmd = vcgencmd_cmd
|
|
self.__state_poll = state_poll
|
|
|
|
self.__dt_cache: dict[str, str] = {}
|
|
|
|
async def get_state(self) -> dict:
|
|
(model, cpu_temp, throttling) = await asyncio.gather(
|
|
self.__read_dt_file("model"),
|
|
self.__get_cpu_temp(),
|
|
self.__get_throttling(),
|
|
)
|
|
return {
|
|
"platform": {
|
|
"type": "rpi",
|
|
"base": model,
|
|
"serial": "0000000000000000",
|
|
},
|
|
"health": {
|
|
"temp": {
|
|
"cpu": cpu_temp,
|
|
},
|
|
"throttling": throttling,
|
|
},
|
|
}
|
|
|
|
async def poll_state(self) -> AsyncGenerator[dict, None]:
|
|
prev_state: dict = {}
|
|
while True:
|
|
state = await self.get_state()
|
|
if state != prev_state:
|
|
yield state
|
|
prev_state = state
|
|
await asyncio.sleep(self.__state_poll)
|
|
|
|
# =====
|
|
|
|
async def __read_dt_file(self, name: str) -> (str | None):
|
|
if name not in self.__dt_cache:
|
|
path = os.path.join(f"{env.PROCFS_PREFIX}/proc/device-tree", name)
|
|
try:
|
|
self.__dt_cache[name] = (await aiofs.read(path)).strip(" \t\r\n\0")
|
|
except Exception as err:
|
|
get_logger(0).error("Can't read DT %s from %s: %s", name, path, err)
|
|
return None
|
|
return self.__dt_cache[name]
|
|
|
|
async def __get_cpu_temp(self) -> (float | None):
|
|
temp_path = f"{env.SYSFS_PREFIX}/sys/class/thermal/thermal_zone0/temp"
|
|
try:
|
|
return int((await aiofs.read(temp_path)).strip()) / 1000
|
|
except Exception as err:
|
|
return None
|
|
|
|
async def __get_throttling(self) -> (dict | None):
|
|
# https://www.raspberrypi.org/forums/viewtopic.php?f=63&t=147781&start=50#p972790
|
|
flags = await self.__parse_vcgencmd(
|
|
arg="get_throttled",
|
|
parser=(lambda text: int(text.split("=")[-1].strip(), 16)),
|
|
)
|
|
if flags is not None:
|
|
return {
|
|
"raw_flags": flags,
|
|
"parsed_flags": {
|
|
"undervoltage": {
|
|
"now": bool(flags & (1 << 0)),
|
|
"past": bool(flags & (1 << 16)),
|
|
},
|
|
"freq_capped": {
|
|
"now": bool(flags & (1 << 1)),
|
|
"past": bool(flags & (1 << 17)),
|
|
},
|
|
"throttled": {
|
|
"now": bool(flags & (1 << 2)),
|
|
"past": bool(flags & (1 << 18)),
|
|
},
|
|
},
|
|
}
|
|
return None
|
|
|
|
async def __parse_vcgencmd(self, arg: str, parser: Callable[[str], _RetvalT]) -> (_RetvalT | None):
|
|
cmd = [*self.__vcgencmd_cmd, arg]
|
|
try:
|
|
text = (await aioproc.read_process(cmd, err_to_null=True))[1]
|
|
except Exception:
|
|
get_logger(0).exception("Error while executing: %s", tools.cmdfmt(cmd))
|
|
return None
|
|
try:
|
|
return parser(text)
|
|
except Exception as err:
|
|
get_logger(0).error("Can't parse [ %s ] output: %r: %s", tools.cmdfmt(cmd), text, tools.efmt(err))
|
|
return None
|