165 lines
5.4 KiB
Python
165 lines
5.4 KiB
Python
"""
|
||
主页面蓝图：首页、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', user_id=session.get('user_id'))
|
||
|
||
|
||
|
||
|
||
@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
|
||
ignore_url = [f"{JAVA_API_BASE}/api/delete-brand/tasks/batch",
|
||
f"{JAVA_API_BASE}/api/delete-brand/history",
|
||
f"{JAVA_API_BASE}/api/product-risk-resolve/tasks/batch",
|
||
]
|
||
if target_url not in ignore_url:
|
||
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)
|