Add File
This commit is contained in:
947
src/landppt/api/image_api.py
Normal file
947
src/landppt/api/image_api.py
Normal file
@@ -0,0 +1,947 @@
|
||||
"""
|
||||
图片服务API路由
|
||||
"""
|
||||
|
||||
from typing import Dict, Any, List, Optional
|
||||
from fastapi import APIRouter, HTTPException, Depends, Request, UploadFile, File, Form
|
||||
from fastapi.responses import FileResponse, StreamingResponse
|
||||
import io
|
||||
import zipfile
|
||||
import time
|
||||
from pathlib import Path
|
||||
from typing import List
|
||||
from pydantic import BaseModel
|
||||
from pydantic import BaseModel
|
||||
import logging
|
||||
import os
|
||||
import asyncio
|
||||
import zipfile
|
||||
import io
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
from ..services.image.image_service import get_image_service
|
||||
from ..services.image.config.image_config import get_image_config
|
||||
from ..auth.middleware import get_current_user_required
|
||||
from ..database.models import User
|
||||
from ..utils.thread_pool import run_blocking_io, to_thread
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
class ImageGenerationRequest(BaseModel):
|
||||
prompt: str
|
||||
provider: Optional[str] = None
|
||||
size: Optional[str] = None
|
||||
quality: Optional[str] = None
|
||||
style: Optional[str] = None
|
||||
|
||||
|
||||
class ImageSuggestionRequest(BaseModel):
|
||||
slide_title: str
|
||||
slide_content: str
|
||||
scenario: str
|
||||
topic: str
|
||||
|
||||
|
||||
@router.get("/api/image/status")
|
||||
async def get_image_service_status(
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""获取图片服务状态"""
|
||||
try:
|
||||
image_config = get_image_config()
|
||||
config = image_config.get_config()
|
||||
|
||||
# 检查可用的提供者
|
||||
available_providers = []
|
||||
|
||||
# 检查DALL-E
|
||||
if config.get('dalle', {}).get('api_key'):
|
||||
available_providers.append('dalle')
|
||||
|
||||
# 检查Stable Diffusion
|
||||
if config.get('stable_diffusion', {}).get('api_key'):
|
||||
available_providers.append('stable_diffusion')
|
||||
|
||||
# 检查SiliconFlow
|
||||
if config.get('siliconflow', {}).get('api_key'):
|
||||
available_providers.append('siliconflow')
|
||||
|
||||
# 检查Pollinations（免费服务，总是可用）
|
||||
available_providers.append('pollinations')
|
||||
|
||||
# 检查搜索服务
|
||||
search_providers = []
|
||||
if config.get('unsplash', {}).get('access_key'):
|
||||
search_providers.append('unsplash')
|
||||
if config.get('pixabay', {}).get('api_key'):
|
||||
search_providers.append('pixabay')
|
||||
|
||||
# 检查缓存目录
|
||||
cache_dir = Path(config.get('cache', {}).get('base_dir', 'temp/images_cache'))
|
||||
cache_info = {
|
||||
'directory': str(cache_dir),
|
||||
'exists': cache_dir.exists(),
|
||||
'size': '0 MB',
|
||||
'file_count': 0
|
||||
}
|
||||
|
||||
if cache_dir.exists():
|
||||
try:
|
||||
files = list(cache_dir.rglob('*'))
|
||||
cache_info['file_count'] = len([f for f in files if f.is_file()])
|
||||
|
||||
total_size = sum(f.stat().st_size for f in files if f.is_file())
|
||||
cache_info['size'] = f"{total_size / (1024 * 1024):.1f} MB"
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to get cache info: {e}")
|
||||
|
||||
return {
|
||||
"status": "ok" if available_providers else "no_providers",
|
||||
"available_providers": available_providers,
|
||||
"search_providers": search_providers,
|
||||
"cache_info": cache_info,
|
||||
"message": f"Found {len(available_providers)} image generation providers and {len(search_providers)} search providers"
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get image service status: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to get image service status: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/test")
|
||||
async def test_image_service(
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""测试图片服务"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
image_config = get_image_config()
|
||||
config = image_config.get_config()
|
||||
|
||||
test_results = {
|
||||
"providers": {},
|
||||
"cache_info": {}
|
||||
}
|
||||
|
||||
# 测试DALL-E
|
||||
if config.get('dalle', {}).get('api_key'):
|
||||
try:
|
||||
# 这里可以添加实际的DALL-E测试逻辑
|
||||
test_results["providers"]["dalle"] = {
|
||||
"available": True,
|
||||
"message": "DALL-E API密钥已配置"
|
||||
}
|
||||
except Exception as e:
|
||||
test_results["providers"]["dalle"] = {
|
||||
"available": False,
|
||||
"message": f"DALL-E测试失败: {str(e)}"
|
||||
}
|
||||
else:
|
||||
test_results["providers"]["dalle"] = {
|
||||
"available": False,
|
||||
"message": "DALL-E API密钥未配置"
|
||||
}
|
||||
|
||||
# 测试Stable Diffusion
|
||||
if config.get('stable_diffusion', {}).get('api_key'):
|
||||
try:
|
||||
test_results["providers"]["stable_diffusion"] = {
|
||||
"available": True,
|
||||
"message": "Stable Diffusion API密钥已配置"
|
||||
}
|
||||
except Exception as e:
|
||||
test_results["providers"]["stable_diffusion"] = {
|
||||
"available": False,
|
||||
"message": f"Stable Diffusion测试失败: {str(e)}"
|
||||
}
|
||||
else:
|
||||
test_results["providers"]["stable_diffusion"] = {
|
||||
"available": False,
|
||||
"message": "Stable Diffusion API密钥未配置"
|
||||
}
|
||||
|
||||
# 测试SiliconFlow
|
||||
if config.get('siliconflow', {}).get('api_key'):
|
||||
try:
|
||||
test_results["providers"]["siliconflow"] = {
|
||||
"available": True,
|
||||
"message": "SiliconFlow API密钥已配置"
|
||||
}
|
||||
except Exception as e:
|
||||
test_results["providers"]["siliconflow"] = {
|
||||
"available": False,
|
||||
"message": f"SiliconFlow测试失败: {str(e)}"
|
||||
}
|
||||
else:
|
||||
test_results["providers"]["siliconflow"] = {
|
||||
"available": False,
|
||||
"message": "SiliconFlow API密钥未配置"
|
||||
}
|
||||
|
||||
# 测试Pollinations（免费服务）
|
||||
try:
|
||||
test_results["providers"]["pollinations"] = {
|
||||
"available": True,
|
||||
"message": "Pollinations服务可用（免费服务）"
|
||||
}
|
||||
except Exception as e:
|
||||
test_results["providers"]["pollinations"] = {
|
||||
"available": False,
|
||||
"message": f"Pollinations测试失败: {str(e)}"
|
||||
}
|
||||
|
||||
# 测试缓存
|
||||
cache_dir = Path(config.get('cache', {}).get('base_dir', 'temp/images_cache'))
|
||||
try:
|
||||
cache_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
files = list(cache_dir.rglob('*'))
|
||||
file_count = len([f for f in files if f.is_file()])
|
||||
total_size = sum(f.stat().st_size for f in files if f.is_file())
|
||||
|
||||
test_results["cache_info"] = {
|
||||
"directory": str(cache_dir),
|
||||
"file_count": file_count,
|
||||
"size": f"{total_size / (1024 * 1024):.1f} MB",
|
||||
"writable": os.access(cache_dir, os.W_OK)
|
||||
}
|
||||
except Exception as e:
|
||||
test_results["cache_info"] = {
|
||||
"directory": str(cache_dir),
|
||||
"error": f"缓存目录测试失败: {str(e)}"
|
||||
}
|
||||
|
||||
return test_results
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Image service test failed: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Image service test failed: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/cache/clear")
|
||||
async def clear_image_cache(
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""清理图片缓存"""
|
||||
try:
|
||||
image_config = get_image_config()
|
||||
config = image_config.get_config()
|
||||
|
||||
cache_dir = Path(config.get('cache', {}).get('base_dir', 'temp/images_cache'))
|
||||
|
||||
if not cache_dir.exists():
|
||||
return {
|
||||
"success": True,
|
||||
"deleted_files": 0,
|
||||
"freed_space": "0 MB",
|
||||
"message": "缓存目录不存在"
|
||||
}
|
||||
|
||||
# 在线程池中执行文件删除操作
|
||||
result = await run_blocking_io(_clear_cache_sync, cache_dir)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"deleted_files": result["deleted_count"],
|
||||
"freed_space": f"{result['freed_space_mb']:.1f} MB",
|
||||
"message": f"成功清理了 {result['deleted_count']} 个文件"
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to clear image cache: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to clear image cache: {str(e)}")
|
||||
|
||||
|
||||
def _clear_cache_sync(cache_dir: Path) -> Dict[str, Any]:
|
||||
"""同步清理缓存（在线程池中运行）"""
|
||||
# 统计删除前的信息
|
||||
files = list(cache_dir.rglob('*'))
|
||||
files_to_delete = [f for f in files if f.is_file()]
|
||||
total_size_before = sum(f.stat().st_size for f in files_to_delete)
|
||||
|
||||
# 删除文件
|
||||
deleted_count = 0
|
||||
for file_path in files_to_delete:
|
||||
try:
|
||||
file_path.unlink()
|
||||
deleted_count += 1
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to delete {file_path}: {e}")
|
||||
|
||||
# 删除空目录
|
||||
for dir_path in sorted([f for f in files if f.is_dir()], reverse=True):
|
||||
try:
|
||||
if not any(dir_path.iterdir()): # 如果目录为空
|
||||
dir_path.rmdir()
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to remove directory {dir_path}: {e}")
|
||||
|
||||
freed_space_mb = total_size_before / (1024 * 1024)
|
||||
|
||||
return {
|
||||
"deleted_count": deleted_count,
|
||||
"freed_space_mb": freed_space_mb
|
||||
}
|
||||
|
||||
|
||||
@router.post("/api/image/generate")
|
||||
async def generate_image(
|
||||
request: ImageGenerationRequest,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""生成图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 创建图片生成请求对象
|
||||
from ..services.image.models import ImageGenerationRequest as ServiceImageGenerationRequest, ImageProvider
|
||||
|
||||
# 解析尺寸
|
||||
width, height = 1024, 1024
|
||||
if request.size:
|
||||
try:
|
||||
width, height = map(int, request.size.split('x'))
|
||||
except ValueError:
|
||||
width, height = 1024, 1024
|
||||
|
||||
# 解析提供者
|
||||
provider = ImageProvider.DALLE
|
||||
if request.provider:
|
||||
try:
|
||||
provider = ImageProvider(request.provider)
|
||||
except ValueError:
|
||||
provider = ImageProvider.DALLE
|
||||
|
||||
service_request = ServiceImageGenerationRequest(
|
||||
prompt=request.prompt,
|
||||
provider=provider,
|
||||
width=width,
|
||||
height=height,
|
||||
quality=request.quality or "standard",
|
||||
style=request.style
|
||||
)
|
||||
|
||||
# 生成图片
|
||||
result = await image_service.generate_image(service_request)
|
||||
|
||||
if result.success:
|
||||
# 返回通过本地图床服务可访问的图片URL（绝对地址）
|
||||
if result.image_info:
|
||||
from ..services.url_service import build_image_url
|
||||
image_url = build_image_url(result.image_info.image_id)
|
||||
else:
|
||||
image_url = None
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"image_path": image_url,
|
||||
"image_id": result.image_info.image_id if result.image_info else None,
|
||||
"message": result.message
|
||||
}
|
||||
else:
|
||||
return {
|
||||
"success": False,
|
||||
"message": result.message,
|
||||
"error_code": result.error_code
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Image generation failed: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Image generation failed: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/suggest")
|
||||
async def suggest_images(
|
||||
request: ImageSuggestionRequest,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""为幻灯片建议图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 获取图片建议
|
||||
suggestions = await image_service.suggest_images_for_ppt_slide(
|
||||
slide_title=request.slide_title,
|
||||
slide_content=request.slide_content,
|
||||
scenario=request.scenario,
|
||||
topic=request.topic
|
||||
)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"suggestions": suggestions,
|
||||
"message": "图片建议生成成功"
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Image suggestion failed: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Image suggestion failed: {str(e)}")
|
||||
|
||||
|
||||
# 图库管理API
|
||||
@router.get("/api/image/gallery/stats")
|
||||
async def get_gallery_stats(
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""获取图库统计信息"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
stats = await image_service.get_cache_stats()
|
||||
|
||||
# 按来源分类统计
|
||||
cache_stats = {
|
||||
'ai_generated': 0,
|
||||
'web_search': 0,
|
||||
'local_storage': 0,
|
||||
'cache_size': '0 MB'
|
||||
}
|
||||
|
||||
if 'categories' in stats:
|
||||
for category, count in stats['categories'].items():
|
||||
if category in cache_stats:
|
||||
cache_stats[category] = count
|
||||
|
||||
if 'total_size_mb' in stats:
|
||||
cache_stats['cache_size'] = f"{stats['total_size_mb']:.1f} MB"
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"stats": cache_stats
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get gallery stats: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to get gallery stats: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/api/image/gallery/list")
|
||||
async def list_gallery_images(
|
||||
page: int = 1,
|
||||
per_page: int = 20,
|
||||
category: str = "",
|
||||
search: str = "",
|
||||
sort: str = "created_desc",
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""获取图库图片列表"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 构建搜索参数
|
||||
search_params = {
|
||||
'page': page,
|
||||
'per_page': per_page,
|
||||
'category': category if category else None,
|
||||
'search': search if search else None,
|
||||
'sort': sort
|
||||
}
|
||||
|
||||
# 获取图片列表（这里需要实现相应的服务方法）
|
||||
result = await image_service.list_cached_images(**search_params)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"images": result['images'],
|
||||
"pagination": {
|
||||
"current_page": page,
|
||||
"per_page": per_page,
|
||||
"total_count": result['total_count'],
|
||||
"total_pages": (result['total_count'] + per_page - 1) // per_page,
|
||||
"has_prev": page > 1,
|
||||
"has_next": page * per_page < result['total_count']
|
||||
}
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to list gallery images: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to list gallery images: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/api/image/detail/{image_id}")
|
||||
async def get_image_detail(
|
||||
image_id: str,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""获取图片详细信息"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
image_info = await image_service.get_image(image_id)
|
||||
|
||||
if not image_info:
|
||||
raise HTTPException(status_code=404, detail="Image not found")
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"image": {
|
||||
"image_id": image_info.image_id,
|
||||
"title": image_info.title,
|
||||
"description": image_info.description,
|
||||
"tags": ','.join([tag.name for tag in image_info.tags]) if image_info.tags else '',
|
||||
"category": image_info.source_type.value,
|
||||
"filename": image_info.filename,
|
||||
"file_size": image_info.metadata.file_size,
|
||||
"width": image_info.metadata.width,
|
||||
"height": image_info.metadata.height,
|
||||
"format": image_info.metadata.format.value,
|
||||
"source_type": image_info.source_type.value,
|
||||
"provider": image_info.provider.value,
|
||||
"created_at": image_info.created_at,
|
||||
"access_count": getattr(image_info, 'access_count', 0)
|
||||
}
|
||||
}
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get image detail: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to get image detail: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/api/image/{image_id}/info")
|
||||
async def get_image_info(
|
||||
image_id: str,
|
||||
request: Request,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""获取图片信息，包括绝对URL"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
image_info = await image_service.get_image(image_id)
|
||||
|
||||
if not image_info:
|
||||
raise HTTPException(status_code=404, detail="Image not found")
|
||||
|
||||
# 构建绝对URL
|
||||
from ..services.url_service import build_image_url
|
||||
absolute_url = build_image_url(image_id)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"image_info": {
|
||||
"image_id": image_info.image_id,
|
||||
"title": image_info.title,
|
||||
"filename": image_info.filename,
|
||||
"absolute_url": absolute_url,
|
||||
"file_size": image_info.metadata.file_size,
|
||||
"width": image_info.metadata.width,
|
||||
"height": image_info.metadata.height,
|
||||
"format": image_info.metadata.format.value,
|
||||
"source_type": image_info.source_type.value,
|
||||
"provider": image_info.provider.value,
|
||||
"created_at": image_info.created_at
|
||||
}
|
||||
}
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get image info: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to get image info: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/api/image/view/{image_id}")
|
||||
async def view_image(
|
||||
image_id: str
|
||||
):
|
||||
"""查看图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
image_info = await image_service.get_image(image_id)
|
||||
|
||||
if not image_info or not image_info.local_path:
|
||||
raise HTTPException(status_code=404, detail="Image not found")
|
||||
|
||||
image_path = Path(image_info.local_path)
|
||||
if not image_path.exists():
|
||||
raise HTTPException(status_code=404, detail="Image file not found")
|
||||
|
||||
return FileResponse(
|
||||
path=str(image_path),
|
||||
media_type=f"image/{image_info.metadata.format.value}",
|
||||
filename=image_info.filename
|
||||
)
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to view image: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to view image: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/api/image/thumbnail/{image_id}")
|
||||
async def get_image_thumbnail(
|
||||
image_id: str
|
||||
):
|
||||
"""获取图片缩略图"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 尝试获取缩略图
|
||||
thumbnail_path = await image_service.get_thumbnail(image_id)
|
||||
|
||||
if thumbnail_path and Path(thumbnail_path).exists():
|
||||
return FileResponse(
|
||||
path=str(thumbnail_path),
|
||||
media_type="image/jpeg"
|
||||
)
|
||||
|
||||
# 如果没有缩略图，返回原图
|
||||
image_info = await image_service.get_image(image_id)
|
||||
if image_info and image_info.local_path and Path(image_info.local_path).exists():
|
||||
return FileResponse(
|
||||
path=str(image_info.local_path),
|
||||
media_type=f"image/{image_info.metadata.format.value}"
|
||||
)
|
||||
|
||||
raise HTTPException(status_code=404, detail="Thumbnail not found")
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get thumbnail: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to get thumbnail: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/api/image/download/{image_id}")
|
||||
async def download_image(
|
||||
image_id: str,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""下载单张图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
image_info = await image_service.get_image(image_id)
|
||||
|
||||
if not image_info or not image_info.local_path:
|
||||
raise HTTPException(status_code=404, detail="Image not found")
|
||||
|
||||
image_path = Path(image_info.local_path)
|
||||
if not image_path.exists():
|
||||
raise HTTPException(status_code=404, detail="Image file not found")
|
||||
|
||||
return FileResponse(
|
||||
path=str(image_path),
|
||||
media_type=f"image/{image_info.metadata.format.value}",
|
||||
filename=image_info.filename,
|
||||
headers={"Content-Disposition": f"attachment; filename=\"{image_info.filename}\""}
|
||||
)
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to download image: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to download image: {str(e)}")
|
||||
|
||||
|
||||
class BatchDeleteRequest(BaseModel):
|
||||
image_ids: List[str]
|
||||
|
||||
|
||||
class BatchDownloadRequest(BaseModel):
|
||||
image_ids: List[str]
|
||||
|
||||
|
||||
@router.delete("/api/image/delete/{image_id}")
|
||||
async def delete_single_image(
|
||||
image_id: str,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""删除单张图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 删除图片
|
||||
result = await image_service.delete_image(image_id)
|
||||
|
||||
if result:
|
||||
return {
|
||||
"success": True,
|
||||
"message": "图片删除成功"
|
||||
}
|
||||
else:
|
||||
raise HTTPException(status_code=404, detail="Image not found")
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to delete image: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to delete image: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/gallery/batch-delete")
|
||||
async def batch_delete_images(
|
||||
request: BatchDeleteRequest,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""批量删除图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
deleted_count = 0
|
||||
failed_ids = []
|
||||
|
||||
for image_id in request.image_ids:
|
||||
try:
|
||||
result = await image_service.delete_image(image_id)
|
||||
if result:
|
||||
deleted_count += 1
|
||||
else:
|
||||
failed_ids.append(image_id)
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to delete image {image_id}: {e}")
|
||||
failed_ids.append(image_id)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"deleted_count": deleted_count,
|
||||
"failed_count": len(failed_ids),
|
||||
"failed_ids": failed_ids,
|
||||
"message": f"成功删除 {deleted_count} 张图片"
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Batch delete failed: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Batch delete failed: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/gallery/batch-download")
|
||||
async def batch_download_images(
|
||||
request: BatchDownloadRequest,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""批量下载图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 获取所有图片信息
|
||||
image_infos = []
|
||||
for image_id in request.image_ids:
|
||||
try:
|
||||
image_info = await image_service.get_image(image_id)
|
||||
if image_info and image_info.local_path:
|
||||
image_path = Path(image_info.local_path)
|
||||
if image_path.exists():
|
||||
image_infos.append({
|
||||
'path': str(image_path),
|
||||
'filename': image_info.filename
|
||||
})
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to get image {image_id}: {e}")
|
||||
continue
|
||||
|
||||
# 在线程池中创建ZIP文件
|
||||
zip_data = await run_blocking_io(_create_zip_sync, image_infos)
|
||||
|
||||
# 生成文件名
|
||||
timestamp = int(time.time())
|
||||
filename = f"images_{timestamp}.zip"
|
||||
|
||||
return StreamingResponse(
|
||||
io.BytesIO(zip_data),
|
||||
media_type="application/zip",
|
||||
headers={"Content-Disposition": f"attachment; filename=\"{filename}\""}
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Batch download failed: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Batch download failed: {str(e)}")
|
||||
|
||||
|
||||
def _create_zip_sync(image_infos: List[Dict[str, str]]) -> bytes:
|
||||
"""同步创建ZIP文件（在线程池中运行）"""
|
||||
zip_buffer = io.BytesIO()
|
||||
|
||||
with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
|
||||
for image_info in image_infos:
|
||||
try:
|
||||
zip_file.write(image_info['path'], image_info['filename'])
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to add {image_info['filename']} to zip: {e}")
|
||||
continue
|
||||
|
||||
zip_buffer.seek(0)
|
||||
return zip_buffer.read()
|
||||
|
||||
|
||||
@router.post("/api/image/upload")
|
||||
async def upload_image(
|
||||
file: UploadFile = File(...),
|
||||
title: str = Form(""),
|
||||
description: str = Form(""),
|
||||
category: str = Form("local_storage"),
|
||||
tags: str = Form(""),
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""上传图片"""
|
||||
try:
|
||||
# 验证文件类型
|
||||
if not file.content_type or not file.content_type.startswith('image/'):
|
||||
raise HTTPException(status_code=400, detail="Invalid file type")
|
||||
|
||||
# 在线程池中读取文件数据
|
||||
file_data = await file.read()
|
||||
|
||||
# 创建上传请求
|
||||
from ..services.image.models import ImageUploadRequest
|
||||
upload_request = ImageUploadRequest(
|
||||
filename=file.filename,
|
||||
file_size=len(file_data),
|
||||
title=title if title else file.filename.split('.')[0],
|
||||
description=description,
|
||||
category=category,
|
||||
tags=[tag.strip() for tag in tags.split(',') if tag.strip()] if tags else [],
|
||||
content_type=file.content_type
|
||||
)
|
||||
|
||||
# 上传图片
|
||||
image_service = get_image_service()
|
||||
result = await image_service.upload_image(upload_request, file_data)
|
||||
|
||||
if result.success:
|
||||
return {
|
||||
"success": True,
|
||||
"image_id": result.image_info.image_id,
|
||||
"message": "图片上传成功"
|
||||
}
|
||||
else:
|
||||
raise HTTPException(status_code=400, detail=result.message)
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Image upload failed: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Image upload failed: {str(e)}")
|
||||
|
||||
|
||||
class ImageUpdateRequest(BaseModel):
|
||||
"""图片信息更新请求"""
|
||||
title: Optional[str] = None
|
||||
description: Optional[str] = None
|
||||
tags: Optional[str] = None
|
||||
category: Optional[str] = None
|
||||
|
||||
|
||||
@router.put("/api/image/{image_id}/update")
|
||||
async def update_image_info(
|
||||
image_id: str,
|
||||
request: ImageUpdateRequest,
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""更新图片信息"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
image_info = await image_service.get_image(image_id)
|
||||
|
||||
if not image_info:
|
||||
raise HTTPException(status_code=404, detail="Image not found")
|
||||
|
||||
# 更新图片信息
|
||||
if request.title is not None:
|
||||
image_info.title = request.title
|
||||
|
||||
if request.description is not None:
|
||||
image_info.description = request.description
|
||||
|
||||
if request.tags is not None:
|
||||
# 清除现有标签
|
||||
image_info.tags = []
|
||||
# 添加新标签
|
||||
if request.tags.strip():
|
||||
tag_names = [tag.strip() for tag in request.tags.split(',') if tag.strip()]
|
||||
for tag_name in tag_names:
|
||||
image_info.add_tag(tag_name)
|
||||
|
||||
if request.category is not None:
|
||||
# 更新分类（通过source_type）
|
||||
from ..services.image.models import ImageSourceType
|
||||
try:
|
||||
image_info.source_type = ImageSourceType(request.category)
|
||||
except ValueError:
|
||||
# 如果分类无效，保持原有分类
|
||||
pass
|
||||
|
||||
# 更新时间戳
|
||||
import time
|
||||
image_info.updated_at = time.time()
|
||||
|
||||
# 保存更新后的图片信息
|
||||
# 通过重新保存元数据来更新信息
|
||||
cache_key = image_service.cache_manager._generate_cache_key(image_info)
|
||||
await image_service.cache_manager._save_image_metadata(cache_key, image_info)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"message": "图片信息已更新",
|
||||
"image": {
|
||||
"image_id": image_info.image_id,
|
||||
"title": image_info.title,
|
||||
"description": image_info.description,
|
||||
"tags": ','.join([tag.name for tag in image_info.tags]) if image_info.tags else '',
|
||||
"category": image_info.source_type.value
|
||||
}
|
||||
}
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to update image info: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to update image info: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/gallery/deduplicate")
|
||||
async def deduplicate_gallery(
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""去重图库中的重复图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 执行去重操作
|
||||
result = await image_service.deduplicate_cache()
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"message": f"已去重 {result['duplicates_removed']} 张重复图片",
|
||||
"duplicates_removed": result['duplicates_removed']
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to deduplicate gallery: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to deduplicate gallery: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/api/image/gallery/clear-all")
|
||||
async def clear_all_images(
|
||||
user: User = Depends(get_current_user_required)
|
||||
):
|
||||
"""清空图床 - 删除所有图片"""
|
||||
try:
|
||||
image_service = get_image_service()
|
||||
|
||||
# 获取所有图片的统计信息
|
||||
stats = await image_service.get_cache_stats()
|
||||
total_images = stats.get('total_entries', 0)
|
||||
|
||||
if total_images == 0:
|
||||
return {
|
||||
"success": True,
|
||||
"deleted_count": 0,
|
||||
"message": "图床已经是空的"
|
||||
}
|
||||
|
||||
# 清空所有缓存
|
||||
deleted_count = await image_service.clear_all_cache()
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"deleted_count": deleted_count,
|
||||
"message": f"成功清空图床，删除了 {deleted_count} 张图片"
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to clear all images: {e}")
|
||||
raise HTTPException(status_code=500, detail=f"Failed to clear all images: {str(e)}")
|
||||
Reference in New Issue
Block a user