runabot/RUNA/Database/Mongo/__init__.py

19 lines
596 B
Python
Raw Normal View History

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