格式化代码

This commit is contained in:
liuwei
2025-06-04 17:52:41 +08:00
parent cd2f8b9679
commit 94e0c16e7b
9 changed files with 148 additions and 110 deletions

View File

@@ -4,6 +4,7 @@ from functools import wraps
# 创建认证蓝图
auth_bp = Blueprint('auth', __name__)
# 登录检查装饰器
def login_required(f):
@wraps(f)
@@ -11,8 +12,10 @@ def login_required(f):
if not session.get('logged_in'):
return redirect(url_for('auth.login'))
return f(*args, **kwargs)
return decorated_function
# 登录页面
@auth_bp.route('/login', methods=['GET', 'POST'])
def login():
@@ -32,6 +35,7 @@ def login():
return render_template('login.html', error=error)
# 登出
@auth_bp.route('/logout')
def logout():

View File

@@ -7,8 +7,10 @@ from loguru import logger
# 创建联系人管理蓝图
contacts_bp = Blueprint('contacts', __name__, url_prefix='/contacts')
def send_message_in_thread(func, *args, **kwargs):
"""在独立线程中发送消息"""
def run():
loop = None
try:
@@ -59,6 +61,7 @@ def send_message_in_thread(func, *args, **kwargs):
thread.daemon = True # 设置为守护线程,这样主程序退出时线程会自动结束
thread.start()
# 联系人管理页面
@contacts_bp.route('/')
@login_required

View File

@@ -4,11 +4,13 @@ from loguru import logger
file_browser_bp = Blueprint('file_browser', __name__)
@file_browser_bp.route('/file_browser')
def file_browser():
"""文件浏览器页面"""
return render_template('file_browser.html')
@file_browser_bp.route('/api/list_files')
def list_files():
"""获取指定目录下的文件列表"""
@@ -79,6 +81,7 @@ def list_files():
logger.error(f"列出文件失败: {e}")
return jsonify({"success": False, "message": str(e)})
@file_browser_bp.route('/api/download_file')
def download_file():
"""下载指定文件"""

View File

@@ -4,6 +4,7 @@ from .auth import login_required
# 创建主页蓝图
main_bp = Blueprint('main', __name__)
@main_bp.route('/')
@login_required
def index():

View File

@@ -18,6 +18,7 @@ LOG = logger
def robot_management():
return render_template('plugins_manage.html')
@plugin_routes.route('/api/plugins', methods=['GET'])
@login_required
def get_plugins():
@@ -51,6 +52,7 @@ def get_plugins():
LOG.error(f"获取插件列表失败: {str(e)}", exc_info=True)
return jsonify({"success": False, "message": f"获取插件列表失败: {str(e)}"})
@plugin_routes.route('/api/plugins/info', methods=['GET'])
@login_required
def get_plugin_info():
@@ -91,6 +93,7 @@ def get_plugin_info():
LOG.error(f"获取插件详情失败: {str(e)}", exc_info=True)
return jsonify({"success": False, "message": f"获取插件详情失败: {str(e)}"})
@plugin_routes.route('/api/plugins/enable', methods=['POST'])
@login_required
def enable_plugin():
@@ -112,6 +115,7 @@ def enable_plugin():
LOG.error(f"启用插件失败: {str(e)}", exc_info=True)
return jsonify({"success": False, "message": f"启用插件失败: {str(e)}"})
@plugin_routes.route('/api/plugins/disable', methods=['POST'])
@login_required
def disable_plugin():
@@ -132,6 +136,7 @@ def disable_plugin():
LOG.error(f"禁用插件失败: {str(e)}", exc_info=True)
return jsonify({"success": False, "message": f"禁用插件失败: {str(e)}"})
@plugin_routes.route('/api/plugins/reload', methods=['POST'])
@login_required
def reload_plugin():
@@ -199,6 +204,7 @@ def get_raw_plugin_config():
LOG.error(f"获取插件配置文件失败: {str(e)}", exc_info=True)
return jsonify({"success": False, "message": f"获取配置文件失败: {str(e)}"})
@plugin_routes.route('/api/plugins/config/update', methods=['POST'])
@login_required
def update_plugin_config():

View File

@@ -7,27 +7,32 @@ from flask import current_app
# 创建统计数据蓝图
stats_bp = Blueprint('stats', __name__)
# 页面路由
@stats_bp.route('/plugins')
@login_required
def plugins():
return render_template('plugins.html')
@stats_bp.route('/users')
@login_required
def users_page():
return render_template('users.html')
@stats_bp.route('/groups')
@login_required
def groups():
return render_template('groups.html')
@stats_bp.route('/errors')
@login_required
def errors():
return render_template('errors.html')
# API路由
@stats_bp.route('/api/user_stats')
@login_required
@@ -49,6 +54,7 @@ def api_user_stats():
logger.error(f"获取用户统计失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@stats_bp.route('/api/group_stats')
@login_required
def api_group_stats():
@@ -69,6 +75,7 @@ def api_group_stats():
logger.error(f"获取群组统计失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@stats_bp.route('/api/plugin_stats')
@login_required
def api_plugin_stats():
@@ -81,6 +88,7 @@ def api_plugin_stats():
logger.error(f"获取插件统计失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@stats_bp.route('/api/error_logs')
@login_required
def api_error_logs():
@@ -94,6 +102,7 @@ def api_error_logs():
logger.error(f"获取错误日志失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@stats_bp.route('/api/dashboard_summary')
@login_required
def api_dashboard_summary():
@@ -118,6 +127,7 @@ def api_dashboard_summary():
logger.error(f"获取仪表盘摘要数据出错: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@stats_bp.route('/api/plugin_trend')
@login_required
def api_plugin_trend():
@@ -131,6 +141,7 @@ def api_plugin_trend():
logger.error(f"获取插件趋势失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@stats_bp.route('/api/error_detail/<int:error_id>')
@login_required
def api_error_detail(error_id):

View File

@@ -13,20 +13,27 @@ system_bp = Blueprint('system', __name__)
# 记录应用启动时间
APP_START_TIME = time.time()
@system_bp.route('/api_docs')
@login_required
def api_docs():
return render_template('api_docs.html')
@system_bp.route('/system_status')
@login_required
def system_status():
return render_template('system_status.html')
# 页面路由
@system_bp.route('/wx_logs')
@login_required
def wx_logs():
return render_template('wx_logs.html')
# API路由
@system_bp.route('/api/system_info')
@login_required
@@ -50,6 +57,7 @@ def api_system_info():
logger.error(f"获取系统信息失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
@system_bp.route('/api/wx_logs')
@login_required
def api_wx_logs():
@@ -95,6 +103,7 @@ def api_wx_logs():
logger.error(f"获取微信日志失败: {e}")
return jsonify({"success": False, "error": str(e)}), 500
# 在现有路由下添加
@system_bp.route('/api/current_user_info', methods=['GET'])
@login_required

View File

@@ -3,15 +3,16 @@ import uuid
from flask import Blueprint, jsonify, request, current_app, render_template
from .auth import login_required
virtual_group_bp = Blueprint('virtual_group', __name__)
# 添加虚拟群组管理视图
@virtual_group_bp.route('/')
@login_required
def virtual_group():
return render_template('virtual_group_management.html')
@virtual_group_bp.route('/api/virtual_groups', methods=['GET'])
@login_required
def get_virtual_groups():