mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-30 01:00:02 +08:00
重构webui主页与好友/群组api
This commit is contained in:
HibiKier
+205
-205
@@ -1,228 +1,228 @@
|
||||
import asyncio
|
||||
import os
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional, Union
|
||||
# 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
|
||||
# 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 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
|
||||
# from ..base_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": []}
|
||||
# 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 = APIRouter()
|
||||
|
||||
|
||||
@router.get("/system", dependencies=[authentication()])
|
||||
async def _() -> Result:
|
||||
return await get_system_data()
|
||||
# @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("/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/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,
|
||||
),
|
||||
)
|
||||
# @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),
|
||||
),
|
||||
)
|
||||
# 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_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_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 _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)
|
||||
# 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)
|
||||
|
||||
Reference in New Issue
Block a user