init: RAG 知识库服务初始提交
- 后端 API（Flask + Gunicorn） - RAG 引擎（混合检索 + 云端 Reranker + 引用溯源） - 文档解析（MinerU + 多格式支持） - Docker 生产部署配置 - 排除前端项目、敏感配置、模型文件
This commit is contained in:
lacerate551
225
api/image_routes.py
Normal file
225
api/image_routes.py
Normal file
@@ -0,0 +1,225 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
图片服务 API
|
||||
|
||||
路由:
|
||||
- GET /images/<image_id> - 获取图片
|
||||
- GET /images/<image_id>/info - 获取图片信息
|
||||
- GET /images/list - 列出所有图片
|
||||
"""
|
||||
|
||||
import os
|
||||
from flask import Blueprint, send_file, jsonify, current_app
|
||||
|
||||
image_bp = Blueprint('images', __name__)
|
||||
|
||||
|
||||
def get_images_base_path():
|
||||
"""获取图片存储路径（扁平化）"""
|
||||
# 使用扁平化路径 .data/images
|
||||
data_images_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), ".data", "images")
|
||||
if os.path.exists(data_images_path):
|
||||
return data_images_path
|
||||
|
||||
# 回退到 documents/images
|
||||
try:
|
||||
from config import DOCUMENTS_PATH
|
||||
return os.path.join(DOCUMENTS_PATH, "images")
|
||||
except ImportError:
|
||||
return os.path.join(os.path.dirname(os.path.dirname(__file__)), "documents", "images")
|
||||
|
||||
|
||||
@image_bp.route('/images/<image_id>', methods=['GET'])
|
||||
def get_image(image_id: str):
|
||||
"""
|
||||
获取图片
|
||||
|
||||
Args:
|
||||
image_id: 图片 ID（支持多种格式：xxx.jpg, images/xxx.jpg, /images/xxx.jpg）
|
||||
|
||||
Returns:
|
||||
图片文件
|
||||
"""
|
||||
# 统一提取文件名（处理各种路径格式）
|
||||
image_id = os.path.basename(image_id)
|
||||
|
||||
# 去掉扩展名（后面会重新匹配）
|
||||
image_id = os.path.splitext(image_id)[0]
|
||||
|
||||
# 安全检查：防止路径遍历攻击
|
||||
if '..' in image_id or '/' in image_id or '\\' in image_id:
|
||||
return jsonify({"error": "无效的图片 ID"}), 400
|
||||
|
||||
images_path = get_images_base_path()
|
||||
|
||||
# 支持多种格式
|
||||
supported_formats = ['.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp']
|
||||
|
||||
for ext in supported_formats:
|
||||
image_path = os.path.join(images_path, f"{image_id}{ext}")
|
||||
if os.path.exists(image_path):
|
||||
try:
|
||||
mimetype = {
|
||||
'.png': 'image/png',
|
||||
'.jpg': 'image/jpeg',
|
||||
'.jpeg': 'image/jpeg',
|
||||
'.gif': 'image/gif',
|
||||
'.bmp': 'image/bmp',
|
||||
'.webp': 'image/webp'
|
||||
}.get(ext, 'image/octet-stream')
|
||||
|
||||
return send_file(image_path, mimetype=mimetype)
|
||||
except Exception as e:
|
||||
return jsonify({"error": f"读取图片失败: {str(e)}"}), 500
|
||||
|
||||
return jsonify({"error": "图片不存在", "image_id": image_id}), 404
|
||||
|
||||
|
||||
@image_bp.route('/images/<image_id>/info', methods=['GET'])
|
||||
def get_image_info(image_id: str):
|
||||
"""
|
||||
获取图片元信息
|
||||
|
||||
Args:
|
||||
image_id: 图片 ID
|
||||
|
||||
Returns:
|
||||
图片元信息（宽度、高度、格式等）
|
||||
"""
|
||||
# 统一提取文件名
|
||||
image_id = os.path.basename(image_id)
|
||||
image_id = os.path.splitext(image_id)[0]
|
||||
|
||||
# 安全检查
|
||||
if '..' in image_id or '/' in image_id or '\\' in image_id:
|
||||
return jsonify({"error": "无效的图片 ID"}), 400
|
||||
|
||||
images_path = get_images_base_path()
|
||||
supported_formats = ['.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp']
|
||||
|
||||
for ext in supported_formats:
|
||||
image_path = os.path.join(images_path, f"{image_id}{ext}")
|
||||
if os.path.exists(image_path):
|
||||
try:
|
||||
# 使用 PIL 获取图片信息
|
||||
from PIL import Image
|
||||
|
||||
with Image.open(image_path) as img:
|
||||
return jsonify({
|
||||
"image_id": image_id,
|
||||
"width": img.width,
|
||||
"height": img.height,
|
||||
"format": img.format,
|
||||
"mode": img.mode,
|
||||
"size_bytes": os.path.getsize(image_path),
|
||||
"url": f"/images/{image_id}"
|
||||
})
|
||||
except ImportError:
|
||||
# PIL 未安装，返回基本信息
|
||||
return jsonify({
|
||||
"image_id": image_id,
|
||||
"size_bytes": os.path.getsize(image_path),
|
||||
"url": f"/images/{image_id}"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": f"读取图片信息失败: {str(e)}"}), 500
|
||||
|
||||
return jsonify({"error": "图片不存在", "image_id": image_id}), 404
|
||||
|
||||
|
||||
@image_bp.route('/images/list', methods=['GET'])
|
||||
def list_images():
|
||||
"""
|
||||
列出所有图片
|
||||
|
||||
Query Parameters:
|
||||
limit: 最大返回数量（默认 50）
|
||||
offset: 偏移量（默认 0）
|
||||
|
||||
Returns:
|
||||
图片列表
|
||||
"""
|
||||
from flask import request
|
||||
|
||||
limit = request.args.get('limit', 50, type=int)
|
||||
offset = request.args.get('offset', 0, type=int)
|
||||
|
||||
images_path = get_images_base_path()
|
||||
|
||||
if not os.path.exists(images_path):
|
||||
return jsonify({"images": [], "total": 0})
|
||||
|
||||
supported_extensions = {'.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp'}
|
||||
images = []
|
||||
|
||||
try:
|
||||
for filename in os.listdir(images_path):
|
||||
ext = os.path.splitext(filename)[1].lower()
|
||||
if ext in supported_extensions:
|
||||
image_id = os.path.splitext(filename)[0]
|
||||
filepath = os.path.join(images_path, filename)
|
||||
images.append({
|
||||
"image_id": image_id,
|
||||
"url": f"/images/{image_id}",
|
||||
"size_bytes": os.path.getsize(filepath)
|
||||
})
|
||||
|
||||
# 排序
|
||||
images.sort(key=lambda x: x['image_id'])
|
||||
|
||||
# 分页
|
||||
total = len(images)
|
||||
images = images[offset:offset + limit]
|
||||
|
||||
return jsonify({
|
||||
"images": images,
|
||||
"total": total,
|
||||
"limit": limit,
|
||||
"offset": offset
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({"error": f"列出图片失败: {str(e)}"}), 500
|
||||
|
||||
|
||||
@image_bp.route('/images/stats', methods=['GET'])
|
||||
def image_stats():
|
||||
"""
|
||||
获取图片统计信息
|
||||
|
||||
Returns:
|
||||
图片总数、总大小等统计信息
|
||||
"""
|
||||
images_path = get_images_base_path()
|
||||
|
||||
if not os.path.exists(images_path):
|
||||
return jsonify({
|
||||
"total_images": 0,
|
||||
"total_size_bytes": 0,
|
||||
"supported_formats": ['.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp']
|
||||
})
|
||||
|
||||
supported_extensions = {'.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp'}
|
||||
total_count = 0
|
||||
total_size = 0
|
||||
format_counts = {}
|
||||
|
||||
try:
|
||||
for filename in os.listdir(images_path):
|
||||
ext = os.path.splitext(filename)[1].lower()
|
||||
if ext in supported_extensions:
|
||||
total_count += 1
|
||||
filepath = os.path.join(images_path, filename)
|
||||
total_size += os.path.getsize(filepath)
|
||||
format_counts[ext] = format_counts.get(ext, 0) + 1
|
||||
|
||||
return jsonify({
|
||||
"total_images": total_count,
|
||||
"total_size_bytes": total_size,
|
||||
"total_size_mb": round(total_size / (1024 * 1024), 2),
|
||||
"format_counts": format_counts,
|
||||
"supported_formats": list(supported_extensions)
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({"error": f"获取统计信息失败: {str(e)}"}), 500
|
||||
Reference in New Issue
Block a user