304 lines
9.9 KiB
Python
304 lines
9.9 KiB
Python
|
#!/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
|