hts-code-query-system-backend/api/server.py

# _*_ 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)