get_wallet_balance_if_non_negative

1 files changed, 6 insertions, 2 deletions

diff --git a/jb/flow/assignment_tasks.py b/jb/flow/assignment_tasks.py
index a41d7cf..b678f44 100644
--- a/jb/flow/assignment_tasks.py
+++ b/jb/flow/assignment_tasks.py
@@ -23,7 +23,7 @@ from jb.managers.thl import (
     user_cashout_request,
     manage_pending_cashout,
     get_user_blocked_or_not_exists,
-    get_wallet_balance,
+    get_wallet_balance_if_non_negative,
 )
 from jb.models.assignment import Assignment
 from jb.models.definitions import AssignmentStatus
@@ -458,7 +458,11 @@ def issue_worker_payment(
     #   then approve it, send the amt bonus, then complete it
     amt_assignment_id = assignment.amt_assignment_id
     hit = hm.get_from_amt_id(amt_hit_id=assignment.amt_hit_id)
-    wallet_balance = get_wallet_balance(amt_worker_id=assignment.amt_worker_id)
+    wallet_balance = get_wallet_balance_if_non_negative(
+        amt_worker_id=assignment.amt_worker_id
+    )
+    if not wallet_balance:
+        return None
     amount = round_payment(amount=wallet_balance)
     if not amount:
         return None