database.py 493 B

1234567891011121314151617181920
  1. # backend/app/database.py
  2. from sqlalchemy import create_engine
  3. from sqlalchemy.ext.declarative import declarative_base
  4. from sqlalchemy.orm import sessionmaker
  5. from .core.config import settings
  6. engine = create_engine(
  7. settings.DATABASE_URL,
  8. connect_args={"check_same_thread": False}
  9. )
  10. SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
  11. Base = declarative_base()
  12. def get_db():
  13. db = SessionLocal()
  14. try:
  15. yield db
  16. finally:
  17. db.close()