jm_job_info.py 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. import time
  2. from typing import List
  3. from app import models, schemas
  4. from sqlalchemy.orm import Session
  5. def create_jm_job_info(db: Session, jm_job_info: models.JmJobInfo):
  6. db.add(jm_job_info)
  7. db.commit()
  8. db.refresh(jm_job_info)
  9. return jm_job_info
  10. def get_jm_job_infos(db: Session):
  11. res: List[models.JmJobInfo] = db.query(models.JmJobInfo)\
  12. .filter(models.JmJobInfo.delete_status != 0)\
  13. .order_by(models.JmJobInfo.id.desc()).all()
  14. return res
  15. def get_jm_job_info(db: Session, jm_job_id: int):
  16. item = db.query(models.JmJobInfo)\
  17. .filter(models.JmJobInfo.id == jm_job_id)\
  18. .filter(models.JmJobInfo.delete_status != 0).first()
  19. if not item:
  20. raise Exception('未找到该定时任务')
  21. return item
  22. def update_jm_job_info(db: Session, item: models.JmJobInfo):
  23. db.commit()
  24. db.flush()
  25. db.refresh(item)
  26. return item
  27. def delete_jm_job_info(db: Session, jm_job_id: int):
  28. jm_job_info = db.query(models.JmJobInfo)\
  29. .filter(models.JmJobInfo.id == jm_job_id).first()
  30. if not jm_job_info:
  31. raise Exception('未找到该定时任务')
  32. if jm_job_info.status == 1:
  33. raise Exception('该任务未停用,不能删除')
  34. jm_job_info.delete_status = 0
  35. db.commit()
  36. db.flush()
  37. db.refresh(jm_job_info)
  38. return jm_job_info
  39. def update_jm_job_status(db: Session, jm_job_id: int, status: int):
  40. jm_job_info = db.query(models.JmJobInfo)\
  41. .filter(models.JmJobInfo.id == jm_job_id)\
  42. .filter(models.JmJobInfo.delete_status != 0).first()
  43. if not jm_job_info:
  44. raise Exception('未找到该定时任务')
  45. jm_job_info.status = status
  46. db.commit()
  47. db.flush()
  48. db.refresh(jm_job_info)
  49. return jm_job_info