import os

from motor.motor_asyncio import AsyncIOMotorClient
from dotenv import load_dotenv, find_dotenv

load_dotenv(find_dotenv())


class MongoDBClient:
    def __init__(self):
        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")]

    def connect(self):
        return self.client()