MemoryBear/api/app/controllers/emotion_config_controller.py

# -*- coding: utf-8 -*-
"""情绪配置控制器模块

本模块提供情绪引擎配置管理的API端点，包括获取和更新配置。

Routes:
    GET /memory/config/emotion - 获取情绪引擎配置
    POST /memory/config/emotion - 更新情绪引擎配置
"""
import uuid

from fastapi import APIRouter, Depends, Query, HTTPException, status
from pydantic import BaseModel, Field
from typing import Optional, Union
from sqlalchemy.orm import Session
from uuid import UUID

from app.core.response_utils import success
from app.dependencies import get_current_user
from app.models.user_model import User
from app.schemas.response_schema import ApiResponse
from app.services.emotion_config_service import EmotionConfigService
from app.core.logging_config import get_api_logger
from app.db import get_db
from app.utils.config_utils import resolve_config_id

# 获取API专用日志器
api_logger = get_api_logger()

router = APIRouter(
    prefix="/memory/emotion",
    tags=["Emotion Config"],
    dependencies=[Depends(get_current_user)]  # 所有路由都需要认证
)

class EmotionConfigQuery(BaseModel):
    """情绪配置查询请求模型"""
    config_id: UUID = Field(..., description="配置ID")

class EmotionConfigUpdate(BaseModel):
    """情绪配置更新请求模型"""
    config_id: Union[uuid.UUID, int, str]= Field(..., description="配置ID")
    emotion_enabled: bool = Field(..., description="是否启用情绪提取")
    emotion_model_id: Optional[str] = Field(None, description="情绪分析专用模型ID")
    emotion_extract_keywords: bool = Field(..., description="是否提取情绪关键词")
    emotion_min_intensity: float = Field(..., ge=0.0, le=1.0, description="最小情绪强度阈值（0.0-1.0）")
    emotion_enable_subject: bool = Field(..., description="是否启用主体分类")

@router.get("/read_config", response_model=ApiResponse)
def get_emotion_config(
    config_id: UUID|int = Query(..., description="配置ID"),
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    """获取情绪引擎配置
    
    查询指定配置ID的情绪相关配置字段。
    
    Args:
        config_id: 配置ID
        
    Returns:
        ApiResponse: 包含情绪配置数据
        
    Example Response:
        {
            "code": 2000,
            "msg": "情绪配置获取成功",
            "data": {
                "config_id": 17,
                "emotion_enabled": true,
                "emotion_model_id": "gpt-4",
                "emotion_extract_keywords": true,
                "emotion_min_intensity": 0.1,
                "emotion_enable_subject": true
            }
        }
    """
    try:
        api_logger.info(
            f"用户 {current_user.username} 请求获取情绪配置",
            extra={"config_id": config_id}
        )
        config_id=resolve_config_id(config_id, db)
        # 初始化服务
        config_service = EmotionConfigService(db)
        
        # 调用服务层
        data = config_service.get_emotion_config(config_id)
        
        api_logger.info(
            "情绪配置获取成功",
            extra={
                "config_id": config_id,
                "emotion_enabled": data.get("emotion_enabled", False)
            }
        )
        
        return success(data=data, msg="情绪配置获取成功")
        
    except ValueError as e:
        api_logger.warning(
            f"获取情绪配置失败: {str(e)}",
            extra={"config_id": config_id}
        )
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=str(e)
        )
    except Exception as e:
        api_logger.error(
            f"获取情绪配置失败: {str(e)}",
            extra={"config_id": config_id},
            exc_info=True
        )
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"获取情绪配置失败: {str(e)}"
        )



@router.post("/updated_config", response_model=ApiResponse)
def update_emotion_config(
    config: EmotionConfigUpdate,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    """更新情绪引擎配置
    
    更新指定配置ID的情绪相关配置字段。
    
    Args:
        config: 配置更新数据（包含config_id）
        
    Returns:
        ApiResponse: 包含更新后的情绪配置数据
        
    Example Request:
        {
            "config_id": 2,
            "emotion_enabled": true,
            "emotion_model_id": "gpt-4",
            "emotion_extract_keywords": true,
            "emotion_min_intensity": 0.1,
            "emotion_enable_subject": true
        }
        
    Example Response:
        {
            "code": 2000,
            "msg": "情绪配置更新成功",
            "data": {
                "config_id": 17,
                "emotion_enabled": true,
                "emotion_model_id": "gpt-4",
                "emotion_extract_keywords": true,
                "emotion_min_intensity": 0.2,
                "emotion_enable_subject": true
            }
        }
    """
    config.config_id=resolve_config_id(config.config_id, db)
    try:
        api_logger.info(
            f"用户 {current_user.username} 请求更新情绪配置",
            extra={
                "config_id": config.config_id,
                "emotion_enabled": config.emotion_enabled,
                "emotion_min_intensity": config.emotion_min_intensity
            }
        )
        
        # 初始化服务
        config_service = EmotionConfigService(db)
        
        # 转换为字典（排除config_id，因为它作为参数传递）
        config_data = config.model_dump(exclude={'config_id'})
        
        # 调用服务层
        data = config_service.update_emotion_config(config.config_id, config_data)
        
        api_logger.info(
            "情绪配置更新成功",
            extra={
                "config_id": config.config_id,
                "emotion_enabled": data.get("emotion_enabled", False)
            }
        )
        
        return success(data=data, msg="情绪配置更新成功")
        
    except ValueError as e:
        api_logger.warning(
            f"更新情绪配置失败: {str(e)}",
            extra={"config_id": config.config_id}
        )
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail=str(e)
        )
    except Exception as e:
        api_logger.error(
            f"更新情绪配置失败: {str(e)}",
            extra={"config_id": config.config_id},
            exc_info=True
        )
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"更新情绪配置失败: {str(e)}"
        )