125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
|
"""
|
|||
|
|
文件上传路由
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
from datetime import datetime
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, UploadFile, File, Form, HTTPException, status
|
|||
|
|
from pydantic import BaseModel
|
|||
|
|
|
|||
|
|
from app.core.config import settings
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
router = APIRouter()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UploadResponse(BaseModel):
|
|||
|
|
"""上传响应模型"""
|
|||
|
|
success: bool
|
|||
|
|
filename: str
|
|||
|
|
file_type: str
|
|||
|
|
original_filename: str
|
|||
|
|
task_description: str
|
|||
|
|
message: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UploadImageResponse(BaseModel):
|
|||
|
|
"""上传图片响应模型"""
|
|||
|
|
success: bool
|
|||
|
|
filename: str
|
|||
|
|
file_type: str
|
|||
|
|
original_filename: str
|
|||
|
|
original_image: str
|
|||
|
|
task_description: str
|
|||
|
|
message: str
|
|||
|
|
|
|||
|
|
|
|||
|
|
def allowed_file(filename: str) -> bool:
|
|||
|
|
"""检查文件是否被允许"""
|
|||
|
|
if '.' not in filename:
|
|||
|
|
return False
|
|||
|
|
ext = filename.rsplit('.', 1)[1].lower()
|
|||
|
|
return ext in settings.ALLOWED_EXTENSIONS
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/upload", response_model=UploadResponse, summary="上传CSV或图片文件")
|
|||
|
|
async def upload_file(
|
|||
|
|
file: UploadFile = File(...),
|
|||
|
|
task_description: str = Form(default="时间序列数据分析")
|
|||
|
|
) -> dict:
|
|||
|
|
"""
|
|||
|
|
上传数据文件（CSV 或图片）
|
|||
|
|
|
|||
|
|
- **file**: CSV 或图片文件 (PNG, JPG, BMP, TIFF)
|
|||
|
|
- **task_description**: 分析任务描述
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
logger.info(f"=== 上传请求开始 ===")
|
|||
|
|
logger.info(f"文件名: {file.filename}")
|
|||
|
|
logger.info(f"任务描述: {task_description}")
|
|||
|
|
|
|||
|
|
# 检查文件名
|
|||
|
|
if not file.filename:
|
|||
|
|
logger.error("文件名为空")
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|||
|
|
detail="没有选择文件"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 检查文件类型
|
|||
|
|
if not allowed_file(file.filename):
|
|||
|
|
logger.error(f"不支持的文件类型: {file.filename}")
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|||
|
|
detail=f"不支持的文件类型。允许的类型: {', '.join(settings.ALLOWED_EXTENSIONS)}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取文件扩展名
|
|||
|
|
file_ext = file.filename.rsplit('.', 1)[1].lower()
|
|||
|
|
|
|||
|
|
# 生成文件名
|
|||
|
|
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
|||
|
|
new_filename = f"upload_{timestamp}_{file.filename}"
|
|||
|
|
|
|||
|
|
# 保存文件
|
|||
|
|
file_path = settings.get_upload_path(new_filename)
|
|||
|
|
logger.info(f"保存文件到: {file_path}")
|
|||
|
|
|
|||
|
|
content = await file.read()
|
|||
|
|
with open(file_path, 'wb') as f:
|
|||
|
|
f.write(content)
|
|||
|
|
|
|||
|
|
logger.info(f"文件保存成功，大小: {len(content)} bytes")
|
|||
|
|
|
|||
|
|
# 处理不同的文件类型
|
|||
|
|
if file_ext == 'csv':
|
|||
|
|
logger.info("处理 CSV 文件")
|
|||
|
|
return {
|
|||
|
|
"success": True,
|
|||
|
|
"filename": new_filename,
|
|||
|
|
"file_type": "csv",
|
|||
|
|
"original_filename": file.filename,
|
|||
|
|
"task_description": task_description
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
logger.warning(f"不支持的文件类型: {file_ext}")
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|||
|
|
detail=f"目前只支持 CSV 文件。您上传的是: {file_ext}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except HTTPException:
|
|||
|
|
raise
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"上传处理异常: {str(e)}", exc_info=True)
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|||
|
|
detail=str(e)
|
|||
|
|
)
|