import os from dotenv import load_dotenv from sqlalchemy import create_engine, Column, Integer, String, ForeignKey, Boolean, DateTime from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker load_dotenv() SQLALCHEMY_DATABASE_URL = os.getenv("SQLALCHEMY_DATABASE_URL") engine = create_engine( SQLALCHEMY_DATABASE_URL, pool_size=10, max_overflow=20, pool_timeout=30, pool_recycle=1800, ) Base = declarative_base() Base.metadata.create_all(bind=engine) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) def get_db(): db = SessionLocal() try: yield db finally: db.close()