narinder1231 commited on
Commit
fadcd11
·
1 Parent(s): 8a1e4d6

configure mongo DB

Browse files
Files changed (2) hide show
  1. src/config/__init__.py +2 -1
  2. src/config/_database.py +18 -0
src/config/__init__.py CHANGED
@@ -1,5 +1,6 @@
1
  from ._logger import logger
 
2
 
3
- __all__ = ["logger"]
4
  __version__ = "0.1.0"
5
  __author__ = "Ramanjit Singh"
 
1
  from ._logger import logger
2
+ from ._database import DatabaseConfig
3
 
4
+ __all__ = ["logger", "DatabaseConfig"]
5
  __version__ = "0.1.0"
6
  __author__ = "Ramanjit Singh"
src/config/_database.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from beanie import init_beanie
3
+ from motor.motor_asyncio import AsyncIOMotorClient
4
+
5
+ from src.models import User, Session
6
+
7
+
8
+ class DatabaseConfig:
9
+ def __init__(self):
10
+
11
+ self.MONGO_URI = os.getenv("MONGO_DB_URI")
12
+ self.MONGO_DB_NAME = os.getenv("MONGO_DB_NAME")
13
+ self.client = AsyncIOMotorClient(self.MONGO_URI)
14
+ self.db = self.client[self.MONGO_DB_NAME]
15
+
16
+ async def init_beanie(self):
17
+
18
+ await init_beanie(database=self.db, document_models=[User, Session])