from sqlalchemy.orm import Session from app.crud.base import CRUDBase from app.models.ml_models_path.early_start import EarlyStartDTRModelPath from app.schemas.model_path.early_start import EarlyStartDTRModelPathCreate, EarlyStartDTRModelPathUpdate class CRUDModelPathEarlyStartDTR( CRUDBase[EarlyStartDTRModelPath, EarlyStartDTRModelPathCreate, EarlyStartDTRModelPathUpdate] ): def get_path_by_device(self, db: Session, device_id: str) -> EarlyStartDTRModelPath: return db.query(self.model).filter(EarlyStartDTRModelPath.device_id == device_id).first() model_path_early_start_dtr = CRUDModelPathEarlyStartDTR(EarlyStartDTRModelPath)