mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
229 lines
6.9 KiB
Python
229 lines
6.9 KiB
Python
import asyncio
|
|
import os
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from typing import Dict, Optional, Union
|
|
|
|
import psutil
|
|
import ujson as json
|
|
from fastapi import APIRouter
|
|
|
|
from configs.path_config import (
|
|
DATA_PATH,
|
|
FONT_PATH,
|
|
IMAGE_PATH,
|
|
LOG_PATH,
|
|
RECORD_PATH,
|
|
TEMP_PATH,
|
|
TEXT_PATH,
|
|
)
|
|
from services.log import logger
|
|
from utils.http_utils import AsyncHttpx
|
|
|
|
from ..models.model import (
|
|
Result,
|
|
SystemFolderSize,
|
|
SystemNetwork,
|
|
SystemResult,
|
|
SystemStatus,
|
|
SystemStatusList,
|
|
)
|
|
from ..utils import authentication
|
|
|
|
CPU_DATA_PATH = DATA_PATH / "system" / "cpu.json"
|
|
MEMORY_DATA_PATH = DATA_PATH / "system" / "memory.json"
|
|
DISK_DATA_PATH = DATA_PATH / "system" / "disk.json"
|
|
CPU_DATA_PATH.parent.mkdir(exist_ok=True, parents=True)
|
|
cpu_data = {"data": []}
|
|
memory_data = {"data": []}
|
|
disk_data = {"data": []}
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/system", dependencies=[authentication()])
|
|
async def _() -> Result:
|
|
return await get_system_data()
|
|
|
|
|
|
@router.get("/status", dependencies=[authentication()])
|
|
async def _() -> Result:
|
|
return Result.ok(
|
|
await asyncio.get_event_loop().run_in_executor(None, _get_system_status),
|
|
)
|
|
|
|
|
|
@router.get("/system/disk", dependencies=[authentication()])
|
|
async def _(type_: Optional[str] = None) -> Result:
|
|
return Result.ok(
|
|
data=await asyncio.get_event_loop().run_in_executor(
|
|
None, _get_system_disk, type_
|
|
),
|
|
)
|
|
|
|
|
|
@router.get("/system/statusList", dependencies=[authentication()])
|
|
async def _() -> Result:
|
|
global cpu_data, memory_data, disk_data
|
|
await asyncio.get_event_loop().run_in_executor(None, _get_system_status)
|
|
cpu_rst = cpu_data["data"][-10:] if len(cpu_data["data"]) > 10 else cpu_data["data"]
|
|
memory_rst = (
|
|
memory_data["data"][-10:]
|
|
if len(memory_data["data"]) > 10
|
|
else memory_data["data"]
|
|
)
|
|
disk_rst = (
|
|
disk_data["data"][-10:] if len(disk_data["data"]) > 10 else disk_data["data"]
|
|
)
|
|
return Result.ok(
|
|
SystemStatusList(
|
|
cpu_data=cpu_rst,
|
|
memory_data=memory_rst,
|
|
disk_data=disk_rst,
|
|
),
|
|
)
|
|
|
|
|
|
async def get_system_data():
|
|
"""
|
|
说明:
|
|
获取系统信息，资源文件大小，网络状态等
|
|
"""
|
|
baidu = 200
|
|
google = 200
|
|
try:
|
|
await AsyncHttpx.get("https://www.baidu.com/", timeout=5)
|
|
except Exception as e:
|
|
logger.warning(f"访问BaiDu失败... {type(e)}: {e}")
|
|
baidu = 404
|
|
try:
|
|
await AsyncHttpx.get("https://www.google.com/", timeout=5)
|
|
except Exception as e:
|
|
logger.warning(f"访问Google失败... {type(e)}: {e}")
|
|
google = 404
|
|
network = SystemNetwork(baidu=baidu, google=google)
|
|
disk = await asyncio.get_event_loop().run_in_executor(None, _get_system_disk, None)
|
|
status = await asyncio.get_event_loop().run_in_executor(None, _get_system_status)
|
|
return Result.ok(
|
|
SystemResult(
|
|
status=status,
|
|
network=network,
|
|
disk=disk, # type: ignore
|
|
check_time=datetime.now().replace(microsecond=0),
|
|
),
|
|
)
|
|
|
|
|
|
def _get_system_status() -> SystemStatus:
|
|
"""
|
|
说明:
|
|
获取系统信息等
|
|
"""
|
|
cpu = psutil.cpu_percent()
|
|
memory = psutil.virtual_memory().percent
|
|
disk = psutil.disk_usage("/").percent
|
|
save_system_data(cpu, memory, disk)
|
|
return SystemStatus(
|
|
cpu=cpu,
|
|
memory=memory,
|
|
disk=disk,
|
|
check_time=datetime.now().replace(microsecond=0),
|
|
)
|
|
|
|
|
|
def _get_system_disk(
|
|
type_: Optional[str],
|
|
) -> Union[SystemFolderSize, Dict[str, Union[float, datetime]]]:
|
|
"""
|
|
说明:
|
|
获取资源文件大小等
|
|
"""
|
|
if not type_:
|
|
disk = SystemFolderSize(
|
|
font_dir_size=_get_dir_size(FONT_PATH) / 1024 / 1024,
|
|
image_dir_size=_get_dir_size(IMAGE_PATH) / 1024 / 1024,
|
|
text_dir_size=_get_dir_size(TEXT_PATH) / 1024 / 1024,
|
|
record_dir_size=_get_dir_size(RECORD_PATH) / 1024 / 1024,
|
|
temp_dir_size=_get_dir_size(TEMP_PATH) / 1024 / 102,
|
|
data_dir_size=_get_dir_size(DATA_PATH) / 1024 / 1024,
|
|
log_dir_size=_get_dir_size(LOG_PATH) / 1024 / 1024,
|
|
check_time=datetime.now().replace(microsecond=0),
|
|
)
|
|
return disk
|
|
else:
|
|
if type_ == "image":
|
|
dir_path = IMAGE_PATH
|
|
elif type_ == "font":
|
|
dir_path = FONT_PATH
|
|
elif type_ == "text":
|
|
dir_path = TEXT_PATH
|
|
elif type_ == "record":
|
|
dir_path = RECORD_PATH
|
|
elif type_ == "data":
|
|
dir_path = DATA_PATH
|
|
elif type_ == "temp":
|
|
dir_path = TEMP_PATH
|
|
else:
|
|
dir_path = LOG_PATH
|
|
dir_map = {}
|
|
other_file_size = 0
|
|
for file in os.listdir(dir_path):
|
|
file = Path(dir_path / file)
|
|
if file.is_dir():
|
|
dir_map[file.name] = _get_dir_size(file) / 1024 / 1024
|
|
else:
|
|
other_file_size += os.path.getsize(file) / 1024 / 1024
|
|
dir_map["其他文件"] = other_file_size
|
|
dir_map["check_time"] = datetime.now().replace(microsecond=0)
|
|
return dir_map
|
|
|
|
|
|
def _get_dir_size(dir_path: Path) -> float:
|
|
"""
|
|
说明:
|
|
获取文件夹大小
|
|
参数:
|
|
:param dir_path: 文件夹路径
|
|
"""
|
|
size = 0
|
|
for root, dirs, files in os.walk(dir_path):
|
|
size += sum([os.path.getsize(os.path.join(root, name)) for name in files])
|
|
return size
|
|
|
|
|
|
def save_system_data(cpu: float, memory: float, disk: float):
|
|
"""
|
|
说明:
|
|
保存一些系统信息
|
|
参数:
|
|
:param cpu: cpu
|
|
:param memory: memory
|
|
:param disk: disk
|
|
"""
|
|
global cpu_data, memory_data, disk_data
|
|
if CPU_DATA_PATH.exists() and not cpu_data["data"]:
|
|
with open(CPU_DATA_PATH, "r") as f:
|
|
cpu_data = json.load(f)
|
|
if MEMORY_DATA_PATH.exists() and not memory_data["data"]:
|
|
with open(MEMORY_DATA_PATH, "r") as f:
|
|
memory_data = json.load(f)
|
|
if DISK_DATA_PATH.exists() and not disk_data["data"]:
|
|
with open(DISK_DATA_PATH, "r") as f:
|
|
disk_data = json.load(f)
|
|
now = str(datetime.now().time().replace(microsecond=0))
|
|
cpu_data["data"].append({"time": now, "data": cpu})
|
|
memory_data["data"].append({"time": now, "data": memory})
|
|
disk_data["data"].append({"time": now, "data": disk})
|
|
if len(cpu_data["data"]) > 50:
|
|
cpu_data["data"] = cpu_data["data"][-50:]
|
|
if len(memory_data["data"]) > 50:
|
|
memory_data["data"] = memory_data["data"][-50:]
|
|
if len(disk_data["data"]) > 50:
|
|
disk_data["data"] = disk_data["data"][-50:]
|
|
with open(CPU_DATA_PATH, "w") as f:
|
|
json.dump(cpu_data, f, indent=4, ensure_ascii=False)
|
|
with open(MEMORY_DATA_PATH, "w") as f:
|
|
json.dump(memory_data, f, indent=4, ensure_ascii=False)
|
|
with open(DISK_DATA_PATH, "w") as f:
|
|
json.dump(disk_data, f, indent=4, ensure_ascii=False)
|