phzhik/chain-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

catch wrong uuid convertion

Browse Source
This commit is contained in:
Robert 2024-02-28 13:04:48 +07:00
parent da00232fe7
commit 86a18ed76d
No known key found for this signature in database
GPG Key ID: F631C7FD957D5F22
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
chain_service/repositories/chain.py
View File

@ -1,3 +1,5 @@
from loguru import logger
import pymongo import pymongo
from uuid import UUID from uuid import UUID
from typing import List from typing import List
@ -24,6 +26,10 @@ class ChainRepository:
return TypeAdapter(List[Chain]).validate_python(chains) return TypeAdapter(List[Chain]).validate_python(chains)
async def get_by_id(self, chain_id: str) -> Chain | None: async def get_by_id(self, chain_id: str) -> Chain | None:
query = {"_id": UUID(chain_id)} try:
chain = await self.collection.find_one(query) query = {"_id": UUID(chain_id)}
return Chain.model_validate(chain) if chain else None chain = await self.collection.find_one(query)
return Chain.model_validate(chain) if chain else None
except ValueError:
logger.error(f"Cannot convert {chain_id} to UUID")