hello-agents/Co-creation-projects/Apricity-InnocoreAI/api/routes/users.py

"""
用户相关API路由
"""

from fastapi import APIRouter, HTTPException
from typing import Dict, Any, Optional
from pydantic import BaseModel
import logging
import uuid

# from ...core.database import db_manager
# 临时注释，避免相对导入错误
db_manager = None

logger = logging.getLogger(__name__)
router = APIRouter()

# Pydantic模型
class UserCreateRequest(BaseModel):
    email: str
    profile: Optional[Dict[str, Any]] = {}

class UserUpdateRequest(BaseModel):
    profile: Optional[Dict[str, Any]] = {}

class UserResponse(BaseModel):
    id: str
    email: str
    profile: Dict[str, Any]
    created_at: str

@router.post("/", response_model=UserResponse)
async def create_user(request: UserCreateRequest):
    """创建用户"""
    try:
        user_id = await db_manager.create_user(
            email=request.email,
            profile=request.profile
        )
        
        user = await db_manager.get_user(user_id)
        
        return UserResponse(
            id=user["id"],
            email=user["email"],
            profile=user["profile"],
            created_at=user["created_at"].isoformat() if user["created_at"] else ""
        )
        
    except Exception as e:
        logger.error(f"创建用户失败: {str(e)}")
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/{user_id}", response_model=UserResponse)
async def get_user(user_id: str):
    """获取用户信息"""
    try:
        user = await db_manager.get_user(user_id)
        if not user:
            raise HTTPException(status_code=404, detail="用户不存在")
        
        return UserResponse(
            id=user["id"],
            email=user["email"],
            profile=user["profile"],
            created_at=user["created_at"].isoformat() if user["created_at"] else ""
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"获取用户信息失败: {str(e)}")
        raise HTTPException(status_code=500, detail=str(e))

@router.put("/{user_id}", response_model=Dict[str, Any])
async def update_user(user_id: str, request: UserUpdateRequest):
    """更新用户信息"""
    try:
        success = await db_manager.update_user_profile(
            user_id=user_id,
            profile=request.profile
        )
        
        if success:
            return {"success": True, "message": "用户信息已更新"}
        else:
            raise HTTPException(status_code=404, detail="用户不存在")
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"更新用户信息失败: {str(e)}")
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/{user_id}/profile")
async def get_user_profile(user_id: str):
    """获取用户配置"""
    try:
        user = await db_manager.get_user(user_id)
        if not user:
            raise HTTPException(status_code=404, detail="用户不存在")
        
        return {
            "success": True,
            "profile": user.get("profile", {})
        }
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"获取用户配置失败: {str(e)}")
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/{user_id}/profile")
async def update_user_profile(user_id: str, profile: Dict[str, Any]):
    """更新用户配置"""
    try:
        success = await db_manager.update_user_profile(
            user_id=user_id,
            profile=profile
        )
        
        if success:
            return {"success": True, "message": "用户配置已更新"}
        else:
            raise HTTPException(status_code=404, detail="用户不存在")
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"更新用户配置失败: {str(e)}")
        raise HTTPException(status_code=500, detail=str(e))