From d0af0c38d7b59013e35eca4e2679a509b56f5980 Mon Sep 17 00:00:00 2001
From: stuppie
Date: Sat, 11 Apr 2026 10:43:53 -0600
Subject: get_wallet_balance_if_non_negative

---
 jb/flow/assignment_tasks.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'jb/flow/assignment_tasks.py')

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
-- 
cgit v1.2.3