import os
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from dotenv import load_dotenv

load_dotenv()

class Database:

    def __init__(self):
        self.user = os.getenv("DB_USER")
        self.password = os.getenv("DB_PASSWORD")
        self.server = os.getenv("DB_HOST")
        self.port = os.getenv("DB_PORT")
        self.database =  os.getenv("DB_NAME")
        self.engine = self.getconnection()

    def getconnection(self):
        return create_engine(
            "mysql+pymysql://{0}:{1}@{2}/{3}".format(self.user, self.password, self.server, self.database)
        )
    
    def setConnection(self):
        session = sessionmaker(bind=self.engine)
        return session()