mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-21 04:46:26 +00:00
zhangtao
ef7143b2ae
refactor(validator): 优化日期验证器并修复序列化问题 refactor(controller): 移除未使用的导入和接口 refactor(service): 优化类型提示和错误处理 fix(model): 修复JobLogModel字段定义 style(param): 移除未使用的导入和优化时间处理逻辑 fix(service): 修复服务器监控信息返回格式 fix(auth): 增强登录验证和错误处理 fix(dict): 修复字典数据状态更新逻辑 refactor(job): 优化任务服务和日志查询逻辑
127 lines
4.3 KiB
Python
127 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
import platform
|
||
import psutil
|
||
import socket
|
||
import time
|
||
from pathlib import Path
|
||
from typing import List, Dict
|
||
|
||
from app.utils.common_util import bytes2human
|
||
from .schema import (
|
||
CpuInfoSchema,
|
||
MemoryInfoSchema,
|
||
PyInfoSchema,
|
||
ServerMonitorSchema,
|
||
DiskInfoSchema,
|
||
SysInfoSchema
|
||
)
|
||
|
||
|
||
class ServerService:
|
||
"""服务监控模块服务层"""
|
||
|
||
@classmethod
|
||
async def get_server_monitor_info_service(cls) -> Dict:
|
||
"""获取服务器监控信息"""
|
||
return ServerMonitorSchema(
|
||
cpu=cls._get_cpu_info(),
|
||
mem=cls._get_memory_info(),
|
||
sys=cls._get_system_info(),
|
||
py=cls._get_python_info(),
|
||
disks=cls._get_disk_info()
|
||
).model_dump()
|
||
|
||
@classmethod
|
||
def _get_cpu_info(cls) -> CpuInfoSchema:
|
||
"""获取CPU信息"""
|
||
cpu_times = psutil.cpu_times_percent()
|
||
cpu_num=psutil.cpu_count(logical=True)
|
||
if not cpu_num:
|
||
cpu_num = 1
|
||
return CpuInfoSchema(
|
||
cpu_num=cpu_num,
|
||
used=cpu_times.user,
|
||
sys=cpu_times.system,
|
||
free=cpu_times.idle
|
||
)
|
||
|
||
@classmethod
|
||
def _get_memory_info(cls) -> MemoryInfoSchema:
|
||
"""获取内存信息"""
|
||
memory = psutil.virtual_memory()
|
||
return MemoryInfoSchema(
|
||
total=bytes2human(memory.total),
|
||
used=bytes2human(memory.used),
|
||
free=bytes2human(memory.free),
|
||
usage=memory.percent
|
||
)
|
||
|
||
@classmethod
|
||
def _get_system_info(cls) -> SysInfoSchema:
|
||
"""获取系统信息"""
|
||
hostname = socket.gethostname()
|
||
return SysInfoSchema(
|
||
computer_ip=socket.gethostbyname(hostname),
|
||
computer_name=platform.node(),
|
||
os_arch=platform.machine(),
|
||
os_name=platform.platform(),
|
||
user_dir=str(Path.cwd())
|
||
)
|
||
|
||
@classmethod
|
||
def _get_python_info(cls) -> PyInfoSchema:
|
||
"""获取Python解释器信息"""
|
||
current_process = psutil.Process()
|
||
memory = psutil.virtual_memory()
|
||
process_memory = current_process.memory_info()
|
||
|
||
start_time = current_process.create_time()
|
||
run_time = ServerService._calculate_run_time(start_time)
|
||
|
||
return PyInfoSchema(
|
||
name=current_process.name(),
|
||
version=platform.python_version(),
|
||
start_time=time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(start_time)),
|
||
run_time=run_time,
|
||
home=str(Path(current_process.exe())),
|
||
memory_total=bytes2human(memory.available),
|
||
memory_used=bytes2human(process_memory.rss),
|
||
memory_free=bytes2human(memory.available - process_memory.rss),
|
||
memory_usage=round((process_memory.rss / memory.available) * 100, 2)
|
||
)
|
||
|
||
@classmethod
|
||
def _get_disk_info(cls) -> List[DiskInfoSchema]:
|
||
"""获取磁盘信息"""
|
||
disk_info = []
|
||
for partition in psutil.disk_partitions():
|
||
try:
|
||
# 使用mountpoint而不是device来获取磁盘使用情况
|
||
usage = psutil.disk_usage(partition.mountpoint)
|
||
mount_point = str(Path(partition.mountpoint))
|
||
disk_info.append(
|
||
DiskInfoSchema(
|
||
dir_name=mount_point, # 使用mountpoint替代device
|
||
sys_type_name=partition.fstype,
|
||
type_name=f'本地固定磁盘（{mount_point}）',
|
||
total=bytes2human(usage.total),
|
||
used=bytes2human(usage.used),
|
||
free=bytes2human(usage.free),
|
||
usage=usage.percent # 直接使用数字而不是字符串
|
||
)
|
||
)
|
||
except (PermissionError, FileNotFoundError):
|
||
# 明确指定可能的异常
|
||
continue
|
||
return disk_info
|
||
|
||
@classmethod
|
||
def _calculate_run_time(cls,start_time: float) -> str:
|
||
"""计算运行时间"""
|
||
difference = time.time() - start_time
|
||
days = int(difference // (24 * 60 * 60))
|
||
hours = int((difference % (24 * 60 * 60)) // (60 * 60))
|
||
minutes = int((difference % (60 * 60)) // 60)
|
||
return f'{days}天{hours}小时{minutes}分钟'
|