pa2ltb7y5
/
development_of_web_application_systems
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化了代码可读性

Browse Source
main
p6zpjanrb 5 months ago
parent 4bfeb3a474
commit aac4131b11
1 changed files with 77 additions and 89 deletions
Show Stats Download Patch File Download Diff File

166
main.py
Unescape View File

@ -1,49 +1,54 @@
import math
import random
import psutil
import pymysql
import psutil ##使用psutil库获取系统和进程信息
import pymysql ##提供Mysql数据库接口
import pytz
import datetime ##pytz库配合datetime,用于创建本地时间，设置时间戳
from flask import Flask, render_template, jsonify, request, session, redirect, url_for ##导入flask库包括 Flask 主类、用于渲染模板和处理 JSON 的模块，会话管理 session，重定向 redirect，以及日期时间处理 datetime。
from flask import Flask, render_template, jsonify, request, session, redirect, url_for
from datetime import datetime
from sqlalchemy.sql.ddl import CreateTable
##创建flask应用实例
app = Flask(__name__)
# 创建数据库连接
app.secret_key = 'SecretKey' # 设置Secret Key
app.secret_key = 'SecretKey' #设置flask密钥，提高安全性
@app.route("/404") #host:post/404:404页面展示，主要用于划分权限查看
##定义各种路由，以及对应的视图函数，用于处理不同页面的请求和访问
####非管理者查看数据会显示404页面
@app.route("/404")
def index_404():
return render_template('page/404.html',)
##注销页面
@app.route("/logout") #host:post/logout:
####用户注销，重定向到登陆页面，重新登陆
@app.route("/logout")
def logout():
session.clear() # 清除会话数据
session.clear() # 清除会话数据
# session['logged_out'] = True
return redirect(url_for('login'))
##注册页面
####渲染注册界面
@app.route("/register")
def register():
return render_template('page/register.html',)
##登录页面
####渲染登录界面
@app.route("/login")
def login():
return render_template('page/login.html',)
##登录接口，通过Ajax实现异步交互
####API路由，使用GET方法同步登录请求
@app.route("/api_login", methods=['GET'])
def api_login():
username =request.args.get('username')
password = request.args.get('password')
# 连接到数据库
######连接到MySQL数据库
connection = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
try:
with connection.cursor() as cursor:
# 检查用户名和密码是否正确
###### SQL查询相匹配用户
sql = f"SELECT * FROM user WHERE username = '{username}' AND password = '{password}'"
cursor.execute(sql)
result = cursor.fetchone()
######设置会话数据
if result:
session['quanxian'] = result['quanxian']
session['username'] = username
@ -51,29 +56,27 @@ def api_login():
else:
data = {'code': 1, 'msg': '登录失败'}
finally:
connection.close() # 确保数据库连接被关闭
connection.close()
return jsonify(data)
####API路由，同步处理注册请求
@app.route("/api_register", methods=['GET'])
def api_register():
username =request.args.get('username')
password = request.args.get('password')
# 连接到数据库
connection = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
try:
with connection.cursor() as cursor:
# 插入一条数据的 SQL 语句
######执行插入操作，添加新用户
sql = """
INSERT INTO user(username, password,quanxian)
VALUES (%s, %s, %s)
"""
# 替换 %s 为你想要插入的数据
######数据格式：【用户名，密码，用户权限（默认0）】
data = (username,password,0)
# 执行 SQL 语句
cursor.execute(sql, data)
# 提交事务
connection.commit()
#######返回信息
res_data = {
'code': 0,
'msg': '注册成功'
@ -84,25 +87,28 @@ def api_register():
'msg': '注册失败'
}
finally:
connection.close() # 确保数据库连接被关闭
connection.close()
return jsonify(res_data)
###定义默认登录界面
@app.route("/")
def logins():
return render_template('page/login.html',)
###定义主页面
@app.route("/index_echarts")
def index_echarts():
return render_template('index.html',)
##告警
###检查用户的权限，如果是管理员（权限为1），返回告警页面，否则返回404页面
@app.route("/warings")
def warings():
if session['quanxian'] == 1:
return render_template('page/warings.html')
else:
return render_template('page/404.html')
##管理员权限
###管理员页面
@app.route("/admin")
def admin():
username = session['username']
@ -111,10 +117,12 @@ def admin():
}
return render_template('page/index.html', **data)
###日志界面
@app.route("/log")
def log():
return render_template('page/table.html',)
##用户权限
###用户界面，检查用户的权限，如果是管理员（权限为1），返回用户页面，否则返回404页面
@app.route("/user")
def user():
if session['quanxian'] ==1:
@ -122,12 +130,13 @@ def user():
else:
return render_template('page/404.html')
###创建数据库表
@app.route("/install")
def install():
try:
db = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456',database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
with db.cursor() as cursor:
# 创建一个表的 SQL 语句
###### 创建表my_table，包含ID、CPU、内存、磁盘、上行、下行、机器号和时间戳字段
sql = """
CREATE TABLE IF NOT EXISTS my_table (
ID INT AUTO_INCREMENT PRIMARY KEY,
@ -140,6 +149,7 @@ def install():
at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
"""
###### 创建表user，包含ID、用户名、密码、权限和时间戳字段
user_sql = """
CREATE TABLE IF NOT EXISTS user (
ID INT AUTO_INCREMENT PRIMARY KEY,
@ -150,6 +160,7 @@ def install():
at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
"""
###### 创建表warings，包含ID、CPU、内存、磁盘、上行、下行、机器号、消息、IP和时间戳字段
warings = """
CREATE TABLE IF NOT EXISTS warings (
ID INT AUTO_INCREMENT PRIMARY KEY,
@ -164,43 +175,37 @@ def install():
at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
"""
# 执行 SQL 语句
cursor.execute(warings)
# 执行 SQL 语句
cursor.execute(user_sql)
# 执行 SQL 语句
cursor.execute(sql)
# 提交事务
db.commit()
finally:
# 关闭数据库连接
db.close()
return jsonify({'code':0,'msg':'数据库创建成功'})
###使用GET方法处理日志分页查询请求
@app.route("/api_log", methods=['GET'])
def api_log():
# 假设每页显示10条记录
######获取请求参数中的煤业记录数LIMIT_PER_PAGE
LIMIT_PER_PAGE =int(request.args.get('limit'))
# 假设我们要查询第3页的数据
######请求参数的当前页数PAGE
PAGE = int(request.args.get('page'))
# 计算 OFFSET
######计算偏移量OFFSET
OFFSET = (PAGE - 1) * LIMIT_PER_PAGE
# 创建数据库连接
connection = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
try:
with connection.cursor() as cursor:
# 创建一个 SQL 查询，使用 LIMIT 和 OFFSET 进行翻页
######创建一个 SQL 查询，使用 LIMIT 和 OFFSET 进行翻页
sql = "SELECT * FROM my_table LIMIT %s OFFSET %s"
cursor.execute(sql, (LIMIT_PER_PAGE, OFFSET))
# 获取所有记录
######查询并传递参数
results = cursor.fetchall()
# 创建一个 SQL 查询，用于获取表中的行数
sql_count = "SELECT COUNT(*) FROM my_table"
cursor.execute(sql_count)
result_count = cursor.fetchone()
count = result_count['COUNT(*)']
res=[]
######创建结果列表
for row in results:
print(row)
res.append({
@ -214,36 +219,28 @@ def api_log():
'at':row['at']
})
finally:
# 关闭数据库连接
connection.close()
data={'code':0,'msg':'','count':count,'data':res}
return jsonify(data)
###定义一个API路由，使用GET方法处理告警信息分页查询请求，与api_log相比处理了额外的message字段
@app.route("/api_warings", methods=['GET'])
def api_warings():
# 设置页面数据量
LIMIT_PER_PAGE =int(request.args.get('limit'))
# 查询页面数据
PAGE = int(request.args.get('page'))
# 计算 OFFSET
OFFSET = (PAGE - 1) * LIMIT_PER_PAGE
# 创建数据库连接
connection = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
try:
with connection.cursor() as cursor:
# 创建一个 SQL 查询，使用 LIMIT 和 OFFSET 进行翻页
sql = "SELECT * FROM warings LIMIT %s OFFSET %s"
cursor.execute(sql, (LIMIT_PER_PAGE, OFFSET))
# 获取所有记录
results = cursor.fetchall()
# 创建一个 SQL 查询，用于获取表中的行数
sql_count = "SELECT COUNT(*) FROM warings"
cursor.execute(sql_count)
result_count = cursor.fetchone()
count = result_count['COUNT(*)']
res=[]
for row in results:
res.append({
'id':row['ID'],
'cpu':row['CPU'],
@ -256,57 +253,52 @@ def api_warings():
'at':row['at']
})
finally:
# 关闭数据库连接
connection.close()
data={'code':0,'msg':'','count':count,'data':res}
return jsonify(data)
###使用GET方法处理用户信息分页查询请求
@app.route("/api_user", methods=['GET'])
def api_user():
# 假设每页显示10条记录
LIMIT_PER_PAGE =int(request.args.get('limit'))
# 假设我们要查询第3页的数据
PAGE = int(request.args.get('page'))
# 计算 OFFSET
OFFSET = (PAGE - 1) * LIMIT_PER_PAGE
# 创建数据库连接
connection = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
try:
with connection.cursor() as cursor:
# 创建一个 SQL 查询，使用 LIMIT 和 OFFSET 进行翻页
sql = "SELECT * FROM user LIMIT %s OFFSET %s"
cursor.execute(sql, (LIMIT_PER_PAGE, OFFSET))
# 获取所有记录
results = cursor.fetchall()
# 创建一个 SQL 查询，用于获取表中的行数
sql_count = "SELECT COUNT(*) FROM user"
cursor.execute(sql_count)
result_count = cursor.fetchone()
count = result_count['COUNT(*)']
res=[]
######根据用户权限字段，设置权限描述：quanxian==0为普通用户，quanxian==1为管理员
for row in results:
if row['quanxian']==0:
quanxian='普通用户'
else:
quanxian = '超级管理员'
res.append({
'id':row['ID'],
'username':row['username'],
'quanxian':quanxian,
'at':row['at']
})
finally:
# 关闭数据库连接
connection.close()
data={'code':0,'msg':'','count':count,'data':res}
return jsonify(data)
###使用GET方法处理系统资源使用情况查询请求
@app.route("/api_msg")
def api_msg():
###### 获取内存使用情况
total_mem,used_mem,available_memory_gb,memory_percent=neicun()
###### 获取CPU使用率
cpu=get_cpu_percent()
###### 获取磁盘使用情况
used, free,usage_percent= get_disk_usage()
disk_data=[
{
@ -345,35 +337,30 @@ def api_msg():
'name':'未用内存',
'value':available_memory_gb
}]
######插入数据到数据库
try:
db = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456',database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
with db.cursor() as cursor:
# 插入一条数据的 SQL 语句
sql = """
INSERT INTO my_table(CPU, neicun, disk, shang, xia, jiqihao)
VALUES (%s, %s, %s, %s, %s, %s)
"""
data = (int(cpu), memory_percent, usage_percent,shang, xia, '业务2号机')
cursor.execute(sql, data)
# 执行 SQL 语句
######实现告警，这里设置的是CPU负载高于90%，插入到waring表
if 100-int(cpu)>=90:
sql = """
INSERT INTO warings(CPU, neicun, disk, shang, xia, jiqihao,message)
VALUES (%s, %s, %s, %s, %s, %s, %s)
"""
# 替换 %s 为你想要插入的数据
data = (int(cpu), memory_percent, usage_percent, shang, xia, '业务2号机','CPU负载过高')
cursor.execute(sql, data)
db.commit()
print("数据插入成功！")
######捕获并打印数据库操作中任何错误
except pymysql.MySQLError as e:
print(f"发生错误: {e}")
finally:
# 关闭连接
db.close()
data={
'neicun':neicun_data,
@ -383,44 +370,39 @@ def api_msg():
}
return jsonify(data)
###删除用户
@app.route("/delete_username", methods=['GET'])
def delete_username():
# 假设每页显示10条记录，通过设置最大单词读取数，优化数据库性能
id =request.args.get('id')
try:
db = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
with db.cursor() as cursor:
# 插入一条数据的 SQL 语句
sql = f"DELETE FROM user WHERE id = '{id}'"
cursor.execute(sql)
db.commit()
data={
'code':0,
'msg':'用户删除成功'
}
except pymysql.MySQLError as e:
print(f"发生错误: {e}")
data = {
'code': 1,
'msg': f'发生错误: {e}'
}
finally:
# 关闭连接
db.close()
return jsonify(data)
###设置超级管理员
@app.route("/api_Superadmin", methods=['GET'])
def api_Superadmin():
# 假设每页显示10条记录
######通过GET请求参数获取要设置超级管理员的用户ID
id =request.args.get('id')
######尝试连接数据库并更新对应ID的用户权限
try:
db = pymysql.connect(host='192.168.153.165', user='yunwei', password='123456', database='yunwei', charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
with db.cursor() as cursor:
# 插入一条数据的 SQL 语句
sql = f"UPDATE user SET quanxian = 1 WHERE id = '{id}'"
cursor.execute(sql)
db.commit()
@ -428,7 +410,6 @@ def api_Superadmin():
'code':0,
'msg':'用户设置成功'
}
except pymysql.MySQLError as e:
data = {
'code': 1,
@ -439,28 +420,35 @@ def api_Superadmin():
db.close()
return jsonify(data)
###获取资源使用情况的函数，主要借助psutil库
####获取内存总量、已用内存、可用内存和内存使用百分比。
def neicun():
# 获取内存信息
###### 获取内存信息
mem = psutil.virtual_memory()
# 将字节转换为GB
###### 将字节转换为GB
total_memory_gb = mem.total / (1024 ** 3)
used_memory_gb = mem.used / (1024 ** 3)
available_memory_gb = mem.available / (1024 ** 3) # 空闲内存
# 计算空间占比（已用内存占总内存的百分比）
###### 计算空间占比（已用内存占总内存的百分比）
memory_percent = (mem.used / mem.total) * 100
return total_memory_gb,used_memory_gb,available_memory_gb,memory_percent
####获取CPU使用百分比
def get_cpu_percent(interval=1):
return psutil.cpu_percent(interval=interval)
####获取磁盘已用空间、未用空间、总空间和使用百分比
def get_disk_usage(disk_path='/'):
disk_usage = psutil.disk_usage(disk_path)
used = disk_usage.used # 已用空间
free = disk_usage.free # 未用空间
used = disk_usage.used
free = disk_usage.free
total = disk_usage.total
# 计算已用空间与总空间的比例（以百分比表示）
######计算已用空间与总空间的比例（以百分比表示）
usage_percent = (used / total) * 100
return used, free,usage_percent
##设置URL
###启动Flask应用程序，在本地开发环境中运行，监听127.0.0.1地址上的7000端口
if __name__ == '__main__':
app.run(debug=True, host='127.0.0.1', port='7000')

Write Preview
Loading…
Cancel
Save