16 lines
440 B
Python
16 lines
440 B
Python
from chain_service.database.database import Database
|
|
from chain_service.database.models.chain import Chain
|
|
|
|
|
|
class ChainRepository:
|
|
|
|
def __init__(self, database: Database):
|
|
self.collection = database.get_collection("chains")
|
|
|
|
async def upsert(self, chain: Chain) -> Chain:
|
|
await self.collection.replace_one(
|
|
{"_id": chain.id}, chain.model_dump(by_alias=True), upsert=True
|
|
)
|
|
|
|
return chain
|