early_start.py 801 B

1234567891011121314151617181920212223242526272829
  1. from typing import Optional
  2. from sqlalchemy.orm import Session
  3. from app.crud.base import CRUDBase
  4. from app.models.ml_models_path.early_start import EarlyStartDTRModelPath
  5. from app.schemas.model_path.early_start import (
  6. EarlyStartDTRModelPathCreate,
  7. EarlyStartDTRModelPathUpdate,
  8. )
  9. class CRUDModelPathEarlyStartDTR(
  10. CRUDBase[
  11. EarlyStartDTRModelPath,
  12. EarlyStartDTRModelPathCreate,
  13. EarlyStartDTRModelPathUpdate,
  14. ]
  15. ):
  16. def get_path_by_device(
  17. self, db: Session, device_id: str
  18. ) -> Optional[EarlyStartDTRModelPath]:
  19. return (
  20. db.query(self.model)
  21. .filter(EarlyStartDTRModelPath.device_id == device_id)
  22. .first()
  23. )
  24. model_path_early_start_dtr = CRUDModelPathEarlyStartDTR(EarlyStartDTRModelPath)