Spaces:
Sleeping
Sleeping
File size: 640 Bytes
34d0dc9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | import os
from dotenv import load_dotenv
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
# Load environment variables
load_dotenv()
# Database Configuration
DATABASE_URL = os.getenv("DATABASE_URL")
# Create Engine
engine = create_engine(DATABASE_URL, pool_size=30, max_overflow=60, pool_timeout=60)
# Session Factory
SessionLocal = sessionmaker(
autocommit=False, autoflush=False, expire_on_commit=False, bind=engine
)
# --- DEPENDENCIES ---
# Dependency to get DB session
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
|