| 12345678910111213141516171819202122 |
- from sqlalchemy import create_engine
- from sqlalchemy.orm import sessionmaker, declarative_base
- from app.core.config import settings
- # SQLAlchemy 2.0 style
- engine = create_engine(
- settings.DATABASE_URI,
- pool_pre_ping=True,
- echo=False
- )
- SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
- Base = declarative_base()
- def get_db():
- db = SessionLocal()
- try:
- yield db
- finally:
- db.close()
|