catch wrong uuid convertion

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

View File

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