From edeb01fc534fdb1e8b1ed5d9c61a031650293c4a Mon Sep 17 00:00:00 2001
From: lhx <liy200110@163.com>
Date: Wed, 17 Dec 2025 15:34:30 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=87=E6=81=92=E4=B8=80=E5=8F=B7=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E8=B4=A6=E5=8F=B7=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/account.py     | 27 ++++++++++++++++++++++++++-
 app/schemas/account.py |  5 +++++
 2 files changed, 31 insertions(+), 1 deletion(-)

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