67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
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_account(db, account.account)
|
|
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,
|
|
account=request.account,
|
|
section=request.section,
|
|
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="账号不存在"
|
|
) |