| 123456789101112131415161718 |
- from sqlalchemy import create_engine
- from sqlalchemy.ext.declarative import declarative_base
- from sqlalchemy.orm import sessionmaker
- from config.settings import POSTGRES_URL
- engine = create_engine(POSTGRES_URL)
- SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
- Base = declarative_base()
- # Dependency
- def get_db():
- db = SessionLocal()
- try:
- yield db
- finally:
- db.close()
|