crawler-plugin/app/blueprints/main.py

"""
主页面蓝图：首页、home、图片工作台、品牌页、静态文件、Logo
"""
import os
from flask import Blueprint, send_file

from app_common import (
    get_db,
    _render_html,
    _is_session_user_valid,
    login_required,
    admin_required,
    STATIC_DIR,
    BASE_DIR,
    ASSETS_DIR
)
from flask import redirect, url_for, session
from flask import Flask, request, Response, stream_with_context
import requests

from config import base_url,version,JAVA_API_BASE

main_bp = Blueprint('main', __name__)


@main_bp.route('/')
def index():
    if session.get('user_id') and _is_session_user_valid():
        return redirect(url_for('main.home'))
    return redirect(url_for('auth.login'))


@main_bp.route('/home')
@login_required
def home():
    try:
        conn = get_db()
        with conn.cursor() as cur:
            cur.execute("SELECT username, is_admin FROM users WHERE id = %s", (session['user_id'],))
            row = cur.fetchone()
        conn.close()
        return _render_html('home.html', username=row.get('username', ''), is_admin=bool(row.get('is_admin')), user_id=session.get('user_id'),baseUrl=base_url,version=version)
    except Exception:
        return _render_html('home.html', username=session.get('username', ''), is_admin=False, user_id=session.get('user_id'),baseUrl=base_url,version=version)


@main_bp.route('/image')
@login_required
def wb():
    return _render_html('index.html')


@main_bp.route('/brand')
@login_required
def brand_page():
    return _render_html('brand.html')




@main_bp.route('/static/<path:filename>')
def serve_static(filename):
    """提供 static 目录及子目录下的静态文件访问。"""
    filepath = os.path.normpath(os.path.join(STATIC_DIR, filename))
    static_abs = os.path.abspath(STATIC_DIR)
    file_abs = os.path.abspath(filepath)
    if not file_abs.startswith(static_abs) or not os.path.isfile(file_abs):
        return '', 404
    return send_file(file_abs, as_attachment=False)


@main_bp.route('/assets/<path:filename>')
def serve_assets(filename):
    """提供 static 目录及子目录下的静态文件访问。"""
    filepath = os.path.normpath(os.path.join(ASSETS_DIR, filename))
    static_abs = os.path.abspath(ASSETS_DIR)
    file_abs = os.path.abspath(filepath)
    if not file_abs.startswith(static_abs) or not os.path.isfile(file_abs):
        return '', 404
    return send_file(file_abs, as_attachment=False)


@main_bp.route('/new_web_source/<path:filename>')
def serve_new_web_source(filename):
    """提供 static 目录及子目录下的静态文件访问。"""
    filepath = os.path.normpath(os.path.join("new_web_source", filename))
    static_abs = os.path.abspath("new_web_source")
    file_abs = os.path.abspath(filepath)
    if not file_abs.startswith(static_abs) or not os.path.isfile(file_abs):
        return '', 404
    return send_file(file_abs, as_attachment=False)



@main_bp.route('/logo.jpg', methods=['GET'])
def get_logo_image():
    return send_file(os.path.join(BASE_DIR, "logo.jpg"), mimetype='image/jpeg')


@main_bp.route('/newApi/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS'])
def proxy(path):
    target_url = f"{JAVA_API_BASE}/{path}"

    # 复制请求参数
    params = request.args.to_dict()

    # 处理请求头
    headers = {}
    for key, value in request.headers:
        if key.lower() in ['host', 'content-length', 'connection']:
            continue
        headers[key] = value
    try:
        print("=============================")
        print("target_url：",target_url)
        print("params：",params)
        print("data：",request.get_data())
        print("=============================")
    except Exception as e:
        print("打印失败",e)

    try:
        # 使用流式请求
        req = requests.request(
            method=request.method,
            url=target_url,
            params=params,
            headers=headers,
            data=request.get_data() if request.get_data() else None,
            cookies=request.cookies,
            stream=True,  # 启用流式传输
            timeout=30
        )

        # 流式响应
        def generate():
            for chunk in req.iter_content(chunk_size=8192):
                if chunk:
                    yield chunk

        # 构建响应
        response = Response(stream_with_context(generate()), status=req.status_code)

        # 复制响应头
        for key, value in req.headers.items():
            if key.lower() not in ['content-encoding', 'content-length', 'transfer-encoding', 'connection']:
                response.headers[key] = value

        return response

    except requests.exceptions.Timeout:
        print("后端服务超时")
        return Response("后端服务超时", status=504)
    except requests.exceptions.ConnectionError:
        print("无法连接到后端服务")
        return Response("无法连接到后端服务", status=502)
    except Exception as e:
        print(f"代理请求失败: {e}")
        return Response(f"代理错误: {str(e)}", status=500)