Update database.py

database.py

@@ -1,4 +1,5 @@
 # database.py
+
 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
@@ -8,12 +9,14 @@ from dotenv import load_dotenv
 # Load environment variables from .env file
 load_dotenv()
 
-MYSQL_HOST = os.getenv("MYSQL_HOST")
-MYSQL_PORT = os.getenv("MYSQL_PORT")
-MYSQL_DATABASE = os.getenv("MYSQL_DATABASE")
-MYSQL_USER = os.getenv("MYSQL_USER")
-MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD")
+# Retrieve database connection details from environment variables
+MYSQL_HOST = os.getenv("MYSQL_HOST", "db")  # Default to 'db' if not set
+MYSQL_PORT = os.getenv("MYSQL_PORT", "3306")
+MYSQL_DATABASE = os.getenv("MYSQL_DATABASE", "bbapi")
+MYSQL_USER = os.getenv("MYSQL_USER", "bbapi")
+MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD", "QQQR5dKS8NYGnYcC")
 
+# Construct the database URL
 DATABASE_URL = f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DATABASE}"
 
 # Create SQLAlchemy engine
@@ -26,5 +29,20 @@ engine = create_engine(
 # Create a configured "Session" class
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
-# Base class for models
+# Base class for your ORM models
 Base = declarative_base()
+
+# Dependency to get a database session
+from fastapi import Depends
+from sqlalchemy.orm import Session
+
+def get_db():
+    """
+    Dependency that provides a SQLAlchemy session.
+    Ensures the session is closed after the request is completed.
+    """
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()