From 1852b809a6c728f15503b625451cb09b7c22feb0 Mon Sep 17 00:00:00 2001
From: kompotkot <sumarokovsergei@yandex.ru>
Date: Fri, 16 Aug 2024 12:52:50 +0000
Subject: [PATCH] Return chain_id with registered contract

---
 engineapi/engineapi/contracts_actions.py | 1 +
 engineapi/engineapi/data.py              | 1 +
 engineapi/engineapi/routes/metatx.py     | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/engineapi/engineapi/contracts_actions.py b/engineapi/engineapi/contracts_actions.py
index 8715471b..bd01e8ce 100644
--- a/engineapi/engineapi/contracts_actions.py
+++ b/engineapi/engineapi/contracts_actions.py
@@ -112,6 +112,7 @@ def parse_registered_contract_response(
     return data.RegisteredContractResponse(
         id=obj[0].id,
         blockchain=obj[1].name,
+        chain_id=obj[1].chain_id,
         address=obj[0].address,
         metatx_requester_id=obj[0].metatx_requester_id,
         title=obj[0].title,
diff --git a/engineapi/engineapi/data.py b/engineapi/engineapi/data.py
index 71e3b37d..1c1d0f62 100644
--- a/engineapi/engineapi/data.py
+++ b/engineapi/engineapi/data.py
@@ -247,6 +247,7 @@ class UpdateContractRequest(BaseModel):
 class RegisteredContractResponse(BaseModel):
     id: UUID
     blockchain: Optional[str] = None
+    chain_id: Optional[int] = None
     address: str
     metatx_requester_id: UUID
     title: Optional[str] = None
diff --git a/engineapi/engineapi/routes/metatx.py b/engineapi/engineapi/routes/metatx.py
index 4ae631f7..0cb66918 100644
--- a/engineapi/engineapi/routes/metatx.py
+++ b/engineapi/engineapi/routes/metatx.py
@@ -139,7 +139,7 @@ async def get_registered_contract_route(
     contract_id: UUID = Path(...),
     user_authorization: Tuple[BugoutUser, UUID] = Depends(request_user_auth),
     db_session: Session = Depends(db.yield_db_read_only_session),
-) -> List[data.RegisteredContractResponse]:
+) -> data.RegisteredContractResponse:
     """
     Get the contract by ID.
     """