Update database management

flanabot/main.py

@@ -1,15 +1,14 @@
+import asyncio
 import os
 
 import flanautils
 
-os.environ |= flanautils.find_environment_variables('../.env')
-
-import asyncio
 from flanabot.bots.flana_disc_bot import FlanaDiscBot
 from flanabot.bots.flana_tele_bot import FlanaTeleBot
 
 
 async def main():
+    os.environ |= flanautils.find_environment_variables('../.env')
     flana_disc_bot = FlanaDiscBot()
     flana_tele_bot = FlanaTeleBot()
 

flanabot/models/punishment.py

@@ -4,7 +4,7 @@ import datetime
 from dataclasses import dataclass
 from typing import Any, Callable
 
-from multibot.models import Platform, PunishmentBase, db
+from multibot.models import Platform, PunishmentBase
 
 from flanabot import constants
 from flanabot.models.message import Message
@@ -12,7 +12,7 @@ from flanabot.models.message import Message
 
 @dataclass(eq=False)
 class Punishment(PunishmentBase):
-    collection = db.punishment
+    collection_name = 'punishment'
 
     level: int = 0