格式化代码

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__) auth_bp = Blueprint('auth', __name__)
# 登录检查装饰器 # 登录检查装饰器
def login_required(f): def login_required(f):
@wraps(f) @wraps(f)
@@ -11,8 +12,10 @@ def login_required(f):
if not session.get('logged_in'): if not session.get('logged_in'):
return redirect(url_for('auth.login')) return redirect(url_for('auth.login'))
return f(*args, **kwargs) return f(*args, **kwargs)
return decorated_function return decorated_function
# 登录页面 # 登录页面
@auth_bp.route('/login', methods=['GET', 'POST']) @auth_bp.route('/login', methods=['GET', 'POST'])
def login(): def login():
@@ -32,6 +35,7 @@ def login():
return render_template('login.html', error=error) return render_template('login.html', error=error)
# 登出 # 登出
@auth_bp.route('/logout') @auth_bp.route('/logout')
def logout(): def logout():

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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