171 lines
5.7 KiB
Python
171 lines
5.7 KiB
Python
import os
|
|
import re
|
|
import sys
|
|
import time
|
|
import psutil
|
|
import platform
|
|
|
|
from datetime import datetime
|
|
|
|
from flask import Blueprint, render_template
|
|
|
|
from applications.common.utils.http import table_api, success_api
|
|
from applications.common.utils.rights import authorize
|
|
from applications.common.utils.cache import cache_auto_internal
|
|
|
|
|
|
bp = Blueprint('adminMonitor', __name__, url_prefix='/monitor')
|
|
|
|
|
|
def get_disk_partitions_list():
|
|
disk_partitions_list = []
|
|
# 判断是否在容器中
|
|
if not os.path.exists('/.dockerenv'):
|
|
disk_partitions = psutil.disk_partitions()
|
|
for i in disk_partitions:
|
|
try:
|
|
a = psutil.disk_usage(i.device)
|
|
except PermissionError:
|
|
continue
|
|
|
|
disk_partitions_dict = {
|
|
'device': i.device,
|
|
'fstype': i.fstype,
|
|
'total': a.total,
|
|
'used': a.used,
|
|
'free': a.free,
|
|
'percent': a.percent
|
|
}
|
|
disk_partitions_list.append(disk_partitions_dict)
|
|
else:
|
|
try:
|
|
usage = psutil.disk_usage('/')
|
|
except PermissionError:
|
|
pass
|
|
|
|
disk_partitions_list.append({
|
|
'device': '/', # 设备名称（根文件系统）
|
|
'fstype': psutil.disk_partitions()[0].fstype, # 文件系统类型
|
|
'total': usage.total, # 总容量（字节）
|
|
'used': usage.used, # 已用空间（字节）
|
|
'free': usage.free, # 可用空间（字节）
|
|
'percent': usage.percent # 使用百分比
|
|
})
|
|
|
|
return disk_partitions_list
|
|
|
|
def get_basic_info():
|
|
# 主机名称
|
|
hostname = platform.node()
|
|
# 系统版本
|
|
system_version = platform.platform()
|
|
# Python 版本
|
|
python_version = platform.python_version()
|
|
|
|
# 开机时间
|
|
boot_time = datetime.fromtimestamp(psutil.boot_time()).replace(microsecond=0)
|
|
up_time = datetime.now().replace(microsecond=0) - boot_time
|
|
up_time_list = re.split(r':', str(up_time))
|
|
up_time_format = "{} 小时 {} 分钟 {} 秒".format(up_time_list[0], up_time_list[1], up_time_list[2])
|
|
up_time_format = up_time_format.replace("days,", "天")
|
|
|
|
return {
|
|
'hostname': hostname,
|
|
'system_version': system_version,
|
|
'python_version': python_version,
|
|
'boot_time': boot_time,
|
|
'up_time_format': up_time_format
|
|
}
|
|
|
|
# 系统监控
|
|
@bp.get('/')
|
|
@authorize("system:monitor:main")
|
|
def main():
|
|
|
|
|
|
# 当前时间
|
|
time_now = time.strftime('%H:%M:%S ', time.localtime(time.time()))
|
|
return render_template(
|
|
'system/monitor.html',
|
|
time_now=time_now,
|
|
**get_basic_info()
|
|
)
|
|
|
|
|
|
# 图表 api
|
|
@bp.get('/polling')
|
|
@authorize("system:monitor:main")
|
|
def ajax_polling():
|
|
# 获取 CPU 核心数
|
|
cpu_count = cache_auto_internal('cpu_count', lambda: psutil.cpu_count(), expired=999999999)
|
|
|
|
# 获取 CPU 使用率
|
|
cpus_percent = cache_auto_internal('cpus_percent',
|
|
lambda: psutil.cpu_percent(interval=1, percpu=False),
|
|
expired=5)
|
|
|
|
# 每个 CPU 的使用率
|
|
cpu_percent_per_core = cache_auto_internal('cpu_percent_per_core',
|
|
lambda: list(enumerate(psutil.cpu_percent(interval=1, percpu=True))),
|
|
expired=5)
|
|
|
|
# 获取空闲率、等待率
|
|
cpu_times_percent = cache_auto_internal('cpu_times_percent',
|
|
lambda: psutil.cpu_times_percent(interval=1, percpu=False),
|
|
expired=5)
|
|
|
|
# 内存信息
|
|
memory_information = cache_auto_internal('memory_information',
|
|
psutil.virtual_memory,
|
|
expired=5)
|
|
|
|
# 硬盘信息
|
|
disk_partitions_list = cache_auto_internal('disk_partitions_list',
|
|
get_disk_partitions_list,
|
|
expired=5)
|
|
|
|
# 系统信息
|
|
basic_info = cache_auto_internal('basic_info',
|
|
get_basic_info,
|
|
expired=5)
|
|
|
|
memory_usage = memory_information.percent
|
|
memory_used = memory_information.used
|
|
memory_total = memory_information.total
|
|
memory_free = memory_information.free
|
|
|
|
cpu_idle_percent = cpu_times_percent.idle
|
|
if hasattr(cpu_times_percent, 'iowait'):
|
|
cpu_wait_percent = cpu_times_percent.iowait
|
|
else:
|
|
cpu_wait_percent = "-"
|
|
|
|
return table_api(msg="请求成功",
|
|
count=0,
|
|
data={
|
|
'cpu_count': cpu_count,
|
|
'cpus_percent': cpus_percent,
|
|
'cpu_idle_percent': cpu_idle_percent,
|
|
'cpu_wait_percent': cpu_wait_percent,
|
|
'cpu_percent_per_core': cpu_percent_per_core,
|
|
'memory_used': memory_used,
|
|
'memory_total': memory_total,
|
|
'memory_free': memory_free,
|
|
'memory_usage': memory_usage,
|
|
'disk_partitions_list': disk_partitions_list,
|
|
'time_now': time.strftime('%H:%M:%S', time.localtime(time.time())),
|
|
'basic_info': basic_info
|
|
})
|
|
|
|
|
|
# 关闭程序
|
|
@bp.get('/kill')
|
|
@authorize("system:monitor:main")
|
|
def kill():
|
|
# 注：若是多 worker 则不生效
|
|
return success_api(msg="关闭命令已发送，请修改代码以生效。")
|
|
for proc in psutil.process_iter():
|
|
if proc.pid == os.getpid():
|
|
proc.kill()
|
|
sys.exit(1)
|