Function get_db

pstext/crud.py:8–13

()

from pstext.database import SessionLocal

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

def get_audio_segment_data(db:Session, audio_id: int):
    return db.query(models.Data).filter(models.Data.audio_id == audio_id).all()