19 lines
596 B
Python
19 lines
596 B
Python
|
import os
|
||
|
|
||
|
from motor.motor_asyncio import AsyncIOMotorClient
|
||
|
from dotenv import load_dotenv, find_dotenv
|
||
|
|
||
|
load_dotenv(find_dotenv())
|
||
|
|
||
|
|
||
|
class MongoDBClient:
|
||
|
def __init__(self, *args, **kwargs):
|
||
|
super().__init__(*args, **kwargs)
|
||
|
self.client = \
|
||
|
AsyncIOMotorClient(os.getenv("MONGODB_HOST"), int(os.getenv("MONGODB_PORT")),
|
||
|
username=os.getenv("MONGODB_ID"),
|
||
|
password=os.getenv("MONGODB_PASSWORD"))[os.getenv("MONGODB_DATABASE")]
|
||
|
|
||
|
async def connect(self):
|
||
|
return self.client
|