格式化代码
This commit is contained in:
@@ -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():
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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():
|
||||||
"""下载指定文件"""
|
"""下载指定文件"""
|
||||||
|
|||||||
@@ -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():
|
||||||
|
|||||||
@@ -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():
|
||||||
|
|||||||
@@ -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):
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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():
|
||||||
|
|||||||
Reference in New Issue
Block a user