fix

app/storage/mongo/__init__.py

@@ -18,3 +18,9 @@ def create_indexes():
     database.get_collection('locks').create_index([
         ('name', 1), 
     ], unique=True)
+
+
+def remove_outdated():
+    client = pymongo.MongoClient(CONNECTION_STRING)
+    while True:
+        

app/storage/mongo/locks.py

@@ -1,22 +1,25 @@
-import bson
 import datetime
-import pydantic
 
 from app.storage.mongo import database
 from app.utils import time
 from bson import codec_options
 
+from pymongo import errors
+
+
+class ConflictException(Exception):
+    pass
+
 
 collection = database.get_collection("locks", codec_options=codec_options.CodecOptions(tz_aware=True))
 
 
-class Lock(pydantic.BaseModel):
-    name: str
-    locked_until: pydantic.AwareDatetime
-
-
-async def acquire(lock: Lock):
-    await collection.insert_one(lock.model_dump())
+async def acquire(name: str, ttl: int):
+    locked_until = time.now() + datetime.timedelta(seconds=ttl)
+    try:
+        await collection.insert_one({'name': name, 'locked_until': locked_until})
+    except errors.DuplicateKeyError:
+        raise ConflictException
 
 
 async def release(name: str):