From d0030acb98157b4b92c372a5ba26cb7c534e79a4 Mon Sep 17 00:00:00 2001 From: stuppie Date: Tue, 21 Apr 2026 21:25:17 -0600 Subject: add an unreject script --- jb/flow/assignment_tasks.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'jb/flow/assignment_tasks.py') diff --git a/jb/flow/assignment_tasks.py b/jb/flow/assignment_tasks.py index 6bda96b..326c87c 100644 --- a/jb/flow/assignment_tasks.py +++ b/jb/flow/assignment_tasks.py @@ -216,7 +216,6 @@ def handle_assignment_w_no_work( assignment = approve_assignment( amtm=amtm, am=am, - hm=hm, amt_assignment_id=amt_assignment_id, msg=NO_WORK_APPROVAL_MESSAGE, amt_hit_type_id=hit.amt_hit_type_id, @@ -274,23 +273,28 @@ def reject_assignment( def approve_assignment( amtm: AMTManager, am: AssignmentManager, - hm: HitManager, amt_assignment_id: str, msg: str, amt_hit_type_id: str, + override_rejection: bool = False, ) -> Assignment: # Approve in AMT, update db res = amtm.approve_assignment_if_possible( - amt_assignment_id=amt_assignment_id, msg=msg + amt_assignment_id=amt_assignment_id, + msg=msg, + override_rejection=override_rejection, ) if res is None: - # We failed to approve this assignment. Will this happen? + # We failed to approve this assignment. Cannot distinguish between + # failed b/c assignment is already approved, or it is not possible. emit_error_event( event_type="failed_to_approve_assignment", amt_hit_type_id=amt_hit_type_id, ) - raise Exception(f"Failed to approve assignment: {amt_assignment_id}") + # The assignment might already be approved, the error msg is useless, so + # keep going. + # raise Exception(f"Failed to approve assignment: {amt_assignment_id}") # We just approved this assignment, get it from amazon again assignment = amtm.get_assignment(amt_assignment_id=amt_assignment_id) @@ -391,7 +395,6 @@ def handle_assignment_w_work( assignment = approve_assignment( amtm=amtm, am=am, - hm=hm, amt_assignment_id=amt_assignment_id, msg=APPROVAL_MESSAGE, amt_hit_type_id=hit.amt_hit_type_id, -- cgit v1.2.3