Fix API compatibility and add user/role/permission and asset import/export

backend_new/app/api/v1/auth.py

@@ -0,0 +1,147 @@
+"""
+认证相关API路由
+"""
+from datetime import datetime
+from fastapi import APIRouter, Depends, HTTPException, status, Header
+from fastapi.security import HTTPBearer
+from sqlalchemy.ext.asyncio import AsyncSession
+from app.core.deps import get_db, get_current_user
+from app.schemas.user import (
+    LoginRequest,
+    LoginResponse,
+    RefreshTokenRequest,
+    RefreshTokenResponse,
+    ChangePasswordRequest,
+)
+from app.services.auth_service import auth_service
+from app.models.user import User
+from app.core.response import success_response
+from app.core.config import settings
+from jose import jwt
+import logging
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter()
+security = HTTPBearer()
+
+
+@router.post("/login", response_model=LoginResponse)
+async def login(
+    credentials: LoginRequest,
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    用户登录
+
+    - **username**: 用户名
+    - **password**: 密码
+    - **captcha**: 验证码
+    - **captcha_key**: 验证码UUID
+    """
+    logger.info(f"登录请求 - 用户名: {credentials.username}, "
+                f"验证码: {credentials.captcha}, 验证码Key: {credentials.captcha_key}")
+
+    try:
+        result = await auth_service.login(
+            db=db,
+            username=credentials.username,
+            password=credentials.password,
+            captcha=credentials.captcha,
+            captcha_key=credentials.captcha_key
+        )
+        logger.info(f"登录成功 - 用户名: {credentials.username}")
+        return success_response(data=result)
+    except Exception as e:
+        logger.error(f"登录失败 - 用户名: {credentials.username}, 错误: {str(e)}", exc_info=True)
+        raise
+
+
+@router.post("/refresh", response_model=RefreshTokenResponse)
+async def refresh_token(
+    token_request: RefreshTokenRequest,
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    刷新访问令牌
+
+    - **refresh_token**: 刷新令牌
+    """
+    result = await auth_service.refresh_token(
+        db=db,
+        refresh_token=token_request.refresh_token
+    )
+    return success_response(data=result)
+
+
+@router.post("/logout")
+async def logout(
+    current_user: User = Depends(get_current_user),
+    authorization: str = Header(...),
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    用户登出
+    """
+    from app.utils.redis_client import redis_client
+
+    # 提取Token
+    token = authorization.replace("Bearer ", "")
+
+    # 获取Token剩余有效期
+    try:
+        payload = jwt.decode(token, settings.SECRET_KEY, algorithms=[settings.ALGORITHM])
+        exp = payload.get("exp")
+
+        if exp:
+            # 计算剩余秒数
+            remaining_time = int(exp) - int(datetime.utcnow().timestamp())
+
+            if remaining_time > 0:
+                # 将Token加入黑名单
+                await redis_client.setex(
+                    f"blacklist:{token}",
+                    remaining_time,
+                    "1"
+                )
+    except Exception as e:
+        logger.error(f"Token黑名单添加失败: {str(e)}")
+
+    return success_response(message="登出成功")
+
+
+@router.put("/change-password")
+async def change_password(
+    password_data: ChangePasswordRequest,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    修改密码
+
+    - **old_password**: 旧密码
+    - **new_password**: 新密码
+    - **confirm_password**: 确认密码
+    """
+    await auth_service.change_password(
+        db=db,
+        user=current_user,
+        old_password=password_data.old_password,
+        new_password=password_data.new_password
+    )
+    return success_response(message="密码修改成功")
+
+
+@router.get("/captcha")
+async def get_captcha():
+    """
+    获取验证码
+
+    返回验证码图片和captcha_key
+    """
+    captcha_data = await auth_service._generate_captcha()
+
+    return success_response(data={
+        "captcha_key": captcha_data["captcha_key"],
+        "captcha_image": captcha_data["captcha_base64"]
+    })