from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session from typing import List from ..core.database import get_db from ..schemas.account import ( AccountCreate, AccountUpdate, AccountResponse, AccountListRequest, AccountGetRequest, AccountUpdateRequest, AccountDeleteRequest ) from ..services.account import AccountService router = APIRouter(prefix="/accounts", tags=["账号管理"]) @router.post("/create", response_model=AccountResponse, status_code=status.HTTP_201_CREATED) def create_account(account: AccountCreate, db: Session = Depends(get_db)): """创建账号""" # 检查账号是否已存在 existing_account = AccountService.get_account_by_username(db, account.username) if existing_account: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail="用户名已存在" ) return AccountService.create_account(db, account) @router.post("/list", response_model=List[AccountResponse]) def get_accounts(request: AccountListRequest, db: Session = Depends(get_db)): """获取账号列表""" return AccountService.get_accounts(db, skip=request.skip, limit=request.limit) @router.post("/get", response_model=List[AccountResponse]) def get_account(request: AccountGetRequest, db: Session = Depends(get_db)): """根据多种条件查询账号""" accounts = AccountService.search_accounts( db, account_id=request.account_id, username=request.username, project_name=request.project_name, status=request.status, today_updated=request.today_updated ) if not accounts: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="未找到符合条件的账号" ) return accounts @router.post("/update", response_model=AccountResponse) def update_account(request: AccountUpdateRequest, db: Session = Depends(get_db)): """更新账号""" account = AccountService.update_account(db, request.account_id, request.account_data) if not account: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="账号不存在" ) return account @router.post("/delete", status_code=status.HTTP_204_NO_CONTENT) def delete_account(request: AccountDeleteRequest, db: Session = Depends(get_db)): """删除账号""" if not AccountService.delete_account(db, request.account_id): raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="账号不存在" )