sqlalchemy.py 426 B

123456789101112131415161718
  1. from sqlalchemy import create_engine
  2. from sqlalchemy.ext.declarative import declarative_base
  3. from sqlalchemy.orm import sessionmaker
  4. from config.settings import POSTGRES_URL
  5. engine = create_engine(POSTGRES_URL)
  6. SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
  7. Base = declarative_base()
  8. # Dependency
  9. def get_db():
  10. db = SessionLocal()
  11. try:
  12. yield db
  13. finally:
  14. db.close()