diff options
Diffstat (limited to 'jb/views/tasks.py')
| -rw-r--r-- | jb/views/tasks.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/jb/views/tasks.py b/jb/views/tasks.py index 7176b29..313295f 100644 --- a/jb/views/tasks.py +++ b/jb/views/tasks.py @@ -18,18 +18,23 @@ def process_request(request: Request) -> None: amt_assignment_id = request.query_params.get("assignmentId", None) if amt_assignment_id == "ASSIGNMENT_ID_NOT_AVAILABLE": raise ValueError("shouldn't happen") + amt_hit_id = request.query_params.get("hitId", None) amt_worker_id = request.query_params.get("workerId", None) print(f"process_request: {amt_assignment_id=} {amt_worker_id=} {amt_hit_id=}") assert amt_worker_id and amt_hit_id and amt_assignment_id # Check that the HIT is still valid - hit = HM.get_from_amt_id(amt_hit_id=amt_hit_id) + hit = HM.get_from_amt_id_if_exists(amt_hit_id=amt_hit_id) + if not hit: + raise ValueError(f"Hit {amt_hit_id} not found in DB") + _ = check_hit_status(amt_hit_id=amt_hit_id, amt_hit_type_id=hit.amt_hit_type_id) emit_assignment_event( status=AssignmentStatus.Accepted, amt_hit_type_id=hit.amt_hit_type_id, ) + # I think it won't be assignable anymore? idk # assert hit_status == HitStatus.Assignable, f"hit {amt_hit_id} {hit_status=}. Expected Assignable" @@ -53,7 +58,7 @@ def process_request(request: Request) -> None: assert assignment_stub.amt_worker_id == amt_worker_id assert assignment_stub.amt_assignment_id == amt_assignment_id assert assignment_stub.created_at > ( - datetime.now(tz=timezone.utc) - timedelta(minutes=90) + datetime.now(tz=timezone.utc) - timedelta(minutes=90) ) return None |