db.py

import os
import logging
from dotenv import load_dotenv
import databases
import sqlalchemy

# Load environment variables from .env file
load_dotenv()

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
)
logger = logging.getLogger(__name__)

# Load the DATABASE_URL from environment variables

# Explicitly load the .env file from the current project directory
ENV_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../.env")
load_dotenv(dotenv_path=ENV_PATH)

DATABASE_URL = os.getenv("DATABASE_URI")

if not DATABASE_URL:
    logger.error("DATABASE_URI not found in environment variables.")
    raise ValueError("DATABASE_URI not found in environment variables. Ensure it is set in the .env file.")

# Initialize the database connection and metadata
try:
    database = databases.Database(DATABASE_URL)
    metadata = sqlalchemy.MetaData()
    logger.info("Database connection initialized successfully.")
except Exception as e:
    logger.error("Failed to initialize database connection: %s", e)
    raise

# Helper function to initialize database
async def connect_to_database():
    """
    Connects to the database when the application starts.
    """
    try:
        await database.connect()
        logger.info("Successfully connected to the database.")
    except Exception as e:
        logger.error("Error connecting to the database: %s", e)
        raise

# Helper function to disconnect database
async def disconnect_from_database():
    """
    Disconnects from the database when the application shuts down.
    """
    try:
        await database.disconnect()
        logger.info("Successfully disconnected from the database.")
    except Exception as e:
        logger.error("Error disconnecting from the database: %s", e)
        raise