operation_util.py 752 B

12345678910111213141516171819202122232425
  1. import time
  2. from app.constants import OperationStatus
  3. from app.models import db
  4. def handle_operation_success(new_operation, start_time, owner_id=None):
  5. new_operation.duration = time.time() - start_time
  6. new_operation.status = OperationStatus.SUCCESS
  7. new_operation.owner_id = owner_id
  8. db.session.add(new_operation)
  9. db.session.commit()
  10. return new_operation
  11. def handle_operation_failure(new_operation, start_time, failure_message, owner_id=None):
  12. new_operation.duration = time.time() - start_time
  13. new_operation.failure_message = failure_message
  14. new_operation.status = OperationStatus.FAILURE
  15. new_operation.owner_id = owner_id
  16. db.session.add(new_operation)
  17. db.session.commit()
  18. return new_operation