2026-1-6
This commit is contained in:
“shengyudong”
303
release/22/image_routes.py
Normal file
303
release/22/image_routes.py
Normal file
@@ -0,0 +1,303 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
图片库管理接口
|
||||
"""
|
||||
|
||||
from flask import Blueprint, request, jsonify
|
||||
import logging
|
||||
from datetime import datetime
|
||||
from auth_utils import require_auth, AuthUtils
|
||||
from database_config import get_db_manager
|
||||
from log_utils import log_create, log_update, log_delete, log_error, log_operation
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 创建蓝图
|
||||
image_bp = Blueprint('image', __name__, url_prefix='/api/images')
|
||||
|
||||
@image_bp.route('/list', methods=['GET'])
|
||||
@require_auth
|
||||
def get_images_list():
|
||||
"""获取图片列表"""
|
||||
try:
|
||||
current_user = AuthUtils.get_current_user()
|
||||
enterprise_id = current_user.get('enterprise_id')
|
||||
|
||||
if not enterprise_id:
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': '无法获取企业ID',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
# 获取查询参数
|
||||
page = int(request.args.get('page', 1))
|
||||
page_size = int(request.args.get('pageSize', 20))
|
||||
keyword = request.args.get('keyword', '').strip()
|
||||
product_id = request.args.get('product_id', '').strip()
|
||||
image_type = request.args.get('type', '').strip()
|
||||
|
||||
# 构建查询条件
|
||||
where_conditions = ["1=1"] # ai_images表没有enterprise_id
|
||||
params = []
|
||||
|
||||
if keyword:
|
||||
where_conditions.append("(image_name LIKE %s OR keywords LIKE %s)")
|
||||
keyword_pattern = f"%{keyword}%"
|
||||
params.extend([keyword_pattern, keyword_pattern])
|
||||
|
||||
if product_id:
|
||||
# 跨表查询，通过ai_image_tags_relation关联
|
||||
where_conditions.append("id IN (SELECT image_id FROM ai_image_tags_relation WHERE product_id = %s)")
|
||||
params.append(product_id)
|
||||
|
||||
if image_type:
|
||||
where_conditions.append("image_type_name LIKE %s")
|
||||
params.append(f"%{image_type}%")
|
||||
|
||||
where_clause = " AND ".join(where_conditions)
|
||||
|
||||
# 计算偏移量
|
||||
offset = (page - 1) * page_size
|
||||
|
||||
db_manager = get_db_manager()
|
||||
|
||||
# 查询总数
|
||||
count_sql = f"""
|
||||
SELECT COUNT(*) as total
|
||||
FROM ai_images
|
||||
WHERE {where_clause}
|
||||
"""
|
||||
count_result = db_manager.execute_query(count_sql, params)
|
||||
total = count_result[0]['total']
|
||||
|
||||
# 查询图片列表
|
||||
sql = f"""
|
||||
SELECT id, image_name, image_url, image_thumb_url, thumbnail_url,
|
||||
image_type_id, image_type_name, department, keywords,
|
||||
size_type, file_size, width, height, status,
|
||||
created_at, updated_at
|
||||
FROM ai_images
|
||||
WHERE {where_clause}
|
||||
ORDER BY created_at DESC
|
||||
LIMIT %s OFFSET %s
|
||||
"""
|
||||
params.extend([page_size, offset])
|
||||
images = db_manager.execute_query(sql, params)
|
||||
|
||||
logger.info(f"获取图片列表成功，总数: {total}")
|
||||
|
||||
return jsonify({
|
||||
'code': 200,
|
||||
'message': 'success',
|
||||
'data': {
|
||||
'total': total,
|
||||
'list': images
|
||||
},
|
||||
'timestamp': int(datetime.now().timestamp() * 1000)
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[获取图片列表] 处理请求时发生错误: {str(e)}", exc_info=True)
|
||||
return jsonify({
|
||||
'code': 500,
|
||||
'message': '服务器内部错误',
|
||||
'data': None
|
||||
}), 500
|
||||
|
||||
@image_bp.route('/upload', methods=['POST'])
|
||||
@require_auth
|
||||
def upload_image():
|
||||
"""上传图片"""
|
||||
try:
|
||||
current_user = AuthUtils.get_current_user()
|
||||
enterprise_id = current_user.get('enterprise_id')
|
||||
|
||||
if not enterprise_id:
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': '无法获取企业ID',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': '请求参数错误',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
# 验证必需字段
|
||||
required_fields = ['image_url', 'image_type_name']
|
||||
for field in required_fields:
|
||||
if not data.get(field):
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': f'缺少必需字段: {field}',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
db_manager = get_db_manager()
|
||||
|
||||
# 插入图片记录
|
||||
sql = """
|
||||
INSERT INTO ai_images
|
||||
(image_name, image_url, image_thumb_url, thumbnail_url, image_type_id, image_type_name,
|
||||
department, keywords, size_type, upload_user_id, status)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
"""
|
||||
image_id = db_manager.execute_insert(sql, (
|
||||
data.get('image_name', ''),
|
||||
data['image_url'],
|
||||
data.get('image_thumb_url', ''),
|
||||
data.get('thumbnail_url', ''),
|
||||
data.get('image_type_id', 0),
|
||||
data['image_type_name'],
|
||||
data.get('department', ''),
|
||||
data.get('keywords', ''),
|
||||
data.get('size_type', 'medical'),
|
||||
current_user.get('user_id', 0),
|
||||
'active'
|
||||
))
|
||||
|
||||
logger.info(f"上传图片成功: ID {image_id}")
|
||||
|
||||
return jsonify({
|
||||
'code': 200,
|
||||
'message': '上传成功',
|
||||
'data': {'id': image_id, 'url': data['image_url']},
|
||||
'timestamp': int(datetime.now().timestamp() * 1000)
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[上传图片] 处理请求时发生错误: {str(e)}", exc_info=True)
|
||||
return jsonify({
|
||||
'code': 500,
|
||||
'message': '服务器内部错误',
|
||||
'data': None
|
||||
}), 500
|
||||
|
||||
@image_bp.route('/batch-upload', methods=['POST'])
|
||||
@require_auth
|
||||
def batch_upload_images():
|
||||
"""批量上传图片"""
|
||||
try:
|
||||
current_user = AuthUtils.get_current_user()
|
||||
enterprise_id = current_user.get('enterprise_id')
|
||||
|
||||
if not enterprise_id:
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': '无法获取企业ID',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
data = request.get_json()
|
||||
if not data or not data.get('images'):
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': '请求参数错误',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
db_manager = get_db_manager()
|
||||
uploaded_ids = []
|
||||
|
||||
for img in data['images']:
|
||||
# 验证必需字段
|
||||
if not img.get('image_url') or not img.get('image_type_name'):
|
||||
continue
|
||||
|
||||
# 插入图片记录
|
||||
sql = """
|
||||
INSERT INTO ai_images
|
||||
(image_name, image_url, image_thumb_url, thumbnail_url, image_type_id, image_type_name,
|
||||
department, keywords, size_type, upload_user_id, status)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
"""
|
||||
image_id = db_manager.execute_insert(sql, (
|
||||
img.get('image_name', ''),
|
||||
img['image_url'],
|
||||
img.get('image_thumb_url', ''),
|
||||
img.get('thumbnail_url', ''),
|
||||
img.get('image_type_id', 0),
|
||||
img['image_type_name'],
|
||||
img.get('department', ''),
|
||||
img.get('keywords', ''),
|
||||
img.get('size_type', 'medical'),
|
||||
current_user.get('user_id', 0),
|
||||
'active'
|
||||
))
|
||||
|
||||
uploaded_ids.append(image_id)
|
||||
|
||||
logger.info(f"批量上传图片成功: {len(uploaded_ids)}张")
|
||||
|
||||
return jsonify({
|
||||
'code': 200,
|
||||
'message': '批量上传成功',
|
||||
'data': {
|
||||
'uploaded_count': len(uploaded_ids),
|
||||
'ids': uploaded_ids
|
||||
},
|
||||
'timestamp': int(datetime.now().timestamp() * 1000)
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[批量上传图片] 处理请求时发生错误: {str(e)}", exc_info=True)
|
||||
return jsonify({
|
||||
'code': 500,
|
||||
'message': '服务器内部错误',
|
||||
'data': None
|
||||
}), 500
|
||||
|
||||
@image_bp.route('/<int:image_id>', methods=['DELETE'])
|
||||
@require_auth
|
||||
def delete_image(image_id):
|
||||
"""删除图片"""
|
||||
try:
|
||||
current_user = AuthUtils.get_current_user()
|
||||
enterprise_id = current_user.get('enterprise_id')
|
||||
|
||||
if not enterprise_id:
|
||||
return jsonify({
|
||||
'code': 400,
|
||||
'message': '无法获取企业ID',
|
||||
'data': None
|
||||
}), 400
|
||||
|
||||
db_manager = get_db_manager()
|
||||
|
||||
# 检查图片是否存在
|
||||
check_sql = "SELECT id FROM ai_images WHERE id = %s"
|
||||
existing = db_manager.execute_query(check_sql, (image_id,))
|
||||
|
||||
if not existing:
|
||||
return jsonify({
|
||||
'code': 404,
|
||||
'message': '图片不存在',
|
||||
'data': None
|
||||
}), 404
|
||||
|
||||
# 删除图片记录（软删除）
|
||||
sql = "UPDATE ai_images SET status = 'deleted' WHERE id = %s"
|
||||
db_manager.execute_update(sql, (image_id,))
|
||||
|
||||
logger.info(f"删除图片成功: ID {image_id}")
|
||||
|
||||
return jsonify({
|
||||
'code': 200,
|
||||
'message': '删除成功',
|
||||
'data': None,
|
||||
'timestamp': int(datetime.now().timestamp() * 1000)
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[删除图片] 处理请求时发生错误: {str(e)}", exc_info=True)
|
||||
return jsonify({
|
||||
'code': 500,
|
||||
'message': '服务器内部错误',
|
||||
'data': None
|
||||
}), 500
|
||||
Reference in New Issue
Block a user