from typing import List
from sqlalchemy.orm import Session

from data.models import TTSData
from utils.logger import Logger

log = Logger()


class TTSDataRepo:
    def __init__(self, db: Session) -> None:
        self.db = db

    def get_range(self, start_id: int, end_id: int) -> List[TTSData]:
        return (
            self.db.query(TTSData)
            .filter(TTSData.id >= start_id, TTSData.id <= end_id)
            .all()
        )