86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
import json
|
|
import logging
|
|
import time
|
|
from flask import Blueprint, render_template, request, Response, stream_with_context, jsonify
|
|
from lib.log_handler import subscribe, unsubscribe, get_all_logs, query_logs
|
|
|
|
pages_bp = Blueprint('pages', __name__)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@pages_bp.route('/')
|
|
def index():
|
|
return render_template('index.html')
|
|
|
|
|
|
@pages_bp.route('/db-manager')
|
|
def db_manager():
|
|
return render_template('db_manager.html')
|
|
|
|
|
|
@pages_bp.route('/logs')
|
|
def logs_page():
|
|
return render_template('logs.html')
|
|
|
|
|
|
@pages_bp.route('/api/logs/stream')
|
|
def logs_stream():
|
|
sub_id, q = subscribe()
|
|
|
|
def generate():
|
|
try:
|
|
for entry in get_all_logs():
|
|
yield f"data: {json.dumps(entry, ensure_ascii=False)}\n\n"
|
|
while True:
|
|
if q:
|
|
while q:
|
|
entry = q.popleft()
|
|
yield f"data: {json.dumps(entry, ensure_ascii=False)}\n\n"
|
|
else:
|
|
time.sleep(0.1)
|
|
except GeneratorExit:
|
|
pass
|
|
finally:
|
|
unsubscribe(sub_id)
|
|
|
|
return Response(
|
|
stream_with_context(generate()),
|
|
mimetype='text/event-stream',
|
|
headers={
|
|
'Cache-Control': 'no-cache',
|
|
'X-Accel-Buffering': 'no',
|
|
'Connection': 'keep-alive',
|
|
}
|
|
)
|
|
|
|
|
|
@pages_bp.route('/api/logs', methods=['GET'])
|
|
def logs_poll():
|
|
"""SPA 兼容的轮询接口：按 after_id 增量返回，并支持 level/source/q 过滤。"""
|
|
after_id = request.args.get('after_id', 0)
|
|
limit = request.args.get('limit', 200)
|
|
level = request.args.get('level') or None
|
|
source = request.args.get('source') or None
|
|
keyword = request.args.get('q') or None
|
|
result = query_logs(after_id=after_id, level=level, source=source,
|
|
keyword=keyword, limit=limit)
|
|
return jsonify({'success': True, **result})
|
|
|
|
|
|
@pages_bp.route('/api/logs/history')
|
|
def logs_history():
|
|
source = request.args.get('source', 'all')
|
|
level = request.args.get('level', 'all')
|
|
keyword = request.args.get('keyword', '')
|
|
|
|
logs = get_all_logs()
|
|
|
|
if source != 'all':
|
|
logs = [l for l in logs if l.get('source') == source]
|
|
if level != 'all':
|
|
logs = [l for l in logs if l.get('level') == level]
|
|
if keyword:
|
|
logs = [l for l in logs if keyword.lower() in l.get('message', '').lower()]
|
|
|
|
return jsonify({'success': True, 'data': logs})
|