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