database.py

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

load_dotenv()

class DatabaseConnector:
    def __init__(self):
        user = os.getenv("DB_USER")
        password = os.getenv("DB_PASS")
        host = os.getenv("DB_HOST")
        db_name = os.getenv("DB_NAME")
        
        self.db_url = f"postgresql://{user}:{password}@{host}:5432/{db_name}"
        
        self.engine = create_engine(self.db_url, echo=True) # echo truth value determines whether database interactions are output to terminal or not
        Session = sessionmaker(bind=self.engine)
        self.session = Session()

    def connect(self):
        Base.metadata.create_all(self.engine)
        print("Database tables verified and created successfully.")

if __name__ == "__main__":
    db = DatabaseConnector()
    db.connect()