Added Beanie ODM initalizations

Nydauron · May 30, 2024 · c958452 · c958452
1 parent e2e768b
commit c958452
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/bot.py b/bot.py
@@ -17,6 +17,7 @@
 
 import aiohttp
 import discord
+from beanie import init_beanie
 from discord import app_commands
 from discord.ext import commands
 from rich.logging import RichHandler
@@ -173,6 +174,12 @@ async def setup_hook(self) -> None:
         Called when the bot is being setup. Currently sets up a connection to the
         database and initializes all extensions.
         """
+        await init_beanie(
+            database=self.mongo_database.client["data"],
+            document_models=[
+                # TODO
+            ],
+        )
         extensions = (
             "src.discord.censor",
             "src.discord.ping",

diff --git a/requirements.txt b/requirements.txt
@@ -24,3 +24,4 @@ motor==3.1.1
 rich==13.7.0
 pydantic<2.7,>=2.6
 pydantic-settings==2.2.1
+beanie==1.26.0