Add update_if_needed in base db model

Flagro · Nov 18, 2024 · db16a2d · db16a2d
1 parent a521b2e
commit db16a2d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/bot/rp_bot/db.py b/bot/rp_bot/db.py
@@ -47,6 +47,6 @@ async def create_if_not_exists(self, person: Person, context: Context) -> None:
         for model in self.models:
             await model.create_if_not_exists(person, context)
 
-    async def update_if_needed(self, person: Person) -> None:
+    async def update_if_needed(self, person: Person, context: Context) -> None:
         # TODO: unify update logic
         await self.user_usage.update_usage_if_needed(person)
diff --git a/bot/rp_bot/db_models/base_db_model.py b/bot/rp_bot/db_models/base_db_model.py
@@ -7,5 +7,8 @@ class BaseDBModel:
     def __init__(self, db: AsyncIOMotorDatabase):
         self.db = db
 
-    def create_if_not_exists(self, context: Context, person: Person) -> None:
+    async def create_if_not_exists(self, context: Context, person: Person) -> None:
+        pass
+
+    async def update_if_needed(self, person: Person, context: Context) -> None:
         pass