diff --git a/app/api/account.py b/app/api/account.py index 68d53bb..e118fbd 100644 --- a/app/api/account.py +++ b/app/api/account.py @@ -6,7 +6,7 @@ from ..core.response_code import ResponseCode, ResponseMessage from ..schemas.account import ( AccountCreate, AccountUpdate, AccountResponse, AccountListRequest, AccountGetRequest, AccountUpdateRequest, AccountDeleteRequest, - AccountApiResponse, AccountListResponse + AccountApiResponse, AccountListResponse,AccountGetRequestYH ) from ..services.account import AccountService @@ -69,6 +69,31 @@ def get_account(request: AccountGetRequest, db: Session = Depends(get_db)): data=accounts ) +# 宇恒一号特定查询接口 +@router.post("/get/yh", response_model=AccountListResponse) +def get_account(request: AccountGetRequestYH, db: Session = Depends(get_db)): + """根据多种条件查询账号""" + accounts = AccountService.search_accounts( + db, + project_name=request.project_name, + yh_id=request.yh_id + ) + if not accounts: + return AccountListResponse( + code=ResponseCode.ACCOUNT_NOT_FOUND, + message=ResponseMessage.ACCOUNT_NOT_FOUND, + total=0, + data=[] + ) + return AccountListResponse( + code=ResponseCode.SUCCESS, + message="查询成功", + total=len(accounts), + data=accounts + ) + + + @router.post("/update", response_model=AccountApiResponse) def update_account(request: AccountUpdateRequest, db: Session = Depends(get_db)): """更新账号""" diff --git a/app/schemas/account.py b/app/schemas/account.py index da1da1e..8cd8089 100644 --- a/app/schemas/account.py +++ b/app/schemas/account.py @@ -66,6 +66,11 @@ class AccountGetRequest(BaseModel): yh_id: Optional[str] = None cl_name: Optional[str] = None +# 宇恒一号特定查询模型 +class AccountGetRequestYH(BaseModel): + project_name: Optional[str] = None + yh_id: str + class AccountUpdateRequest(BaseModel): account_id: int account_data: AccountUpdate