120 lines
4.4 KiB
Python
120 lines
4.4 KiB
Python
# _*_ coding : UTF-8 _*_
|
||
# @Time : 2025/02/04 15:25
|
||
# @UpdateTime : 2025/02/04 15:25
|
||
# @Author : sonder
|
||
# @File : server.py
|
||
# @Software : PyCharm
|
||
# @Comment : 本程序
|
||
import os
|
||
import platform
|
||
import socket
|
||
import time
|
||
|
||
import psutil
|
||
from fastapi import APIRouter, Depends, Request
|
||
from fastapi.responses import JSONResponse
|
||
|
||
from annotation.auth import Auth
|
||
from annotation.log import Log
|
||
from config.constant import BusinessType
|
||
from controller.login import LoginController
|
||
from schemas.server import GetServerInfoResponse, CpuInfo, MemoryInfo, SystemInfo, PythonInfo, SystemFiles, \
|
||
GetSystemInfoResult
|
||
from utils.common import bytes2human
|
||
from utils.log import logger
|
||
from utils.response import Response
|
||
|
||
serverAPI = APIRouter(
|
||
prefix="/server",
|
||
dependencies=[Depends(LoginController.get_current_user)]
|
||
)
|
||
|
||
|
||
@serverAPI.get("", response_class=JSONResponse, response_model=GetServerInfoResponse, summary="获取服务器信息")
|
||
@Log(title="获取服务器信息", business_type=BusinessType.SELECT)
|
||
@Auth(permission_list=["server:btn:info"])
|
||
async def get_server_info(request: Request):
|
||
# CPU信息
|
||
# 获取CPU总核心数
|
||
cpu_num = psutil.cpu_count(logical=True)
|
||
cpu_usage_percent = psutil.cpu_times_percent()
|
||
cpu_used = cpu_usage_percent.user
|
||
cpu_sys = cpu_usage_percent.system
|
||
cpu_free = cpu_usage_percent.idle
|
||
cpu = CpuInfo(cpuNum=cpu_num, used=cpu_used, sys=cpu_sys, free=cpu_free)
|
||
|
||
# 内存信息
|
||
memory_info = psutil.virtual_memory()
|
||
memory_total = bytes2human(memory_info.total)
|
||
memory_used = bytes2human(memory_info.used)
|
||
memory_free = bytes2human(memory_info.free)
|
||
memory_usage = memory_info.percent
|
||
mem = MemoryInfo(total=memory_total, used=memory_used, free=memory_free, usage=memory_usage)
|
||
|
||
# 主机信息
|
||
# 获取主机名
|
||
hostname = socket.gethostname()
|
||
# 获取IP
|
||
computer_ip = socket.gethostbyname(hostname)
|
||
os_name = platform.platform()
|
||
computer_name = platform.node()
|
||
os_arch = platform.machine()
|
||
user_dir = os.path.abspath(os.getcwd())
|
||
sys = SystemInfo(
|
||
computerIp=computer_ip, computerName=computer_name, osArch=os_arch, osName=os_name, userDir=user_dir
|
||
)
|
||
|
||
# python解释器信息
|
||
current_pid = os.getpid()
|
||
current_process = psutil.Process(current_pid)
|
||
python_name = current_process.name()
|
||
python_version = platform.python_version()
|
||
python_home = current_process.exe()
|
||
start_time_stamp = current_process.create_time()
|
||
start_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(start_time_stamp))
|
||
current_time_stamp = time.time()
|
||
difference = current_time_stamp - start_time_stamp
|
||
# 将时间差转换为天、小时和分钟数
|
||
days = int(difference // (24 * 60 * 60)) # 每天的秒数
|
||
hours = int((difference % (24 * 60 * 60)) // (60 * 60)) # 每小时的秒数
|
||
minutes = int((difference % (60 * 60)) // 60) # 每分钟的秒数
|
||
run_time = f'{days}天{hours}小时{minutes}分钟'
|
||
# 获取当前Python程序的pid
|
||
pid = os.getpid()
|
||
# 获取该进程的内存信息
|
||
current_process_memory_info = psutil.Process(pid).memory_info()
|
||
py = PythonInfo(
|
||
name=python_name,
|
||
version=python_version,
|
||
startTime=start_time,
|
||
runTime=run_time,
|
||
home=python_home,
|
||
total=bytes2human(memory_info.available),
|
||
used=bytes2human(current_process_memory_info.rss),
|
||
free=bytes2human(memory_info.available - current_process_memory_info.rss),
|
||
usage=round((current_process_memory_info.rss / memory_info.available) * 100, 2),
|
||
)
|
||
|
||
# 磁盘信息
|
||
io = psutil.disk_partitions()
|
||
sys_files = []
|
||
for i in io:
|
||
try:
|
||
o = psutil.disk_usage(i.device)
|
||
disk_data = SystemFiles(
|
||
dirName=i.device,
|
||
sysTypeName=i.fstype,
|
||
typeName='本地固定磁盘（' + i.mountpoint.replace('\\', '') + '）',
|
||
total=bytes2human(o.total),
|
||
used=bytes2human(o.used),
|
||
free=bytes2human(o.free),
|
||
usage=f'{psutil.disk_usage(i.device).percent}%',
|
||
)
|
||
sys_files.append(disk_data)
|
||
except Exception as e:
|
||
logger.error(f"获取磁盘信息失败：{e}")
|
||
continue
|
||
|
||
result = GetSystemInfoResult(cpu=cpu, memory=mem, system=sys, python=py, systemFiles=sys_files)
|
||
return Response.success(data=result)
|