task_dao.py 655 B

12345678910111213141516171819202122232425
  1. from sqlalchemy.orm import Session
  2. from app.model.task_model import Task
  3. class TaskDAO:
  4. def __init__(self, db_session: Session):
  5. self.db_session = db_session
  6. def get_all(self):
  7. return self.db_session.query(Task).all()
  8. def get_by_id(self, task_id: int):
  9. return self.db_session.query(Task).get(task_id)
  10. def create(self, task: Task):
  11. self.db_session.add(task)
  12. self.db_session.commit()
  13. return task
  14. def update(self, task: Task):
  15. self.db_session.commit()
  16. return task
  17. def delete(self, task: Task):
  18. self.db_session.delete(task)
  19. self.db_session.commit()