From 4e9e94faee69ead57813d60be3a039d5e7cafd87 Mon Sep 17 00:00:00 2001
From: Adhidarma Hadiwinoto <gua@adhisimon.org>
Date: Sat, 12 Mar 2016 13:38:55 +0700
Subject: [PATCH] update balance setelah terima uang xltunai

---
 main.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index a767d36..23d80e4 100755
--- a/main.py
+++ b/main.py
@@ -164,6 +164,7 @@ def handleSms(sms):
     if sms.text.find('Anda terima uang XLTunai') >= 0:
         LAST_SN = xltunai.getSNFromReceiveTransferMessage(sms.text)
         logger.info('Override LAST_SN: {0}'.format(LAST_SN))
+        checkBalance(modem)
         return
     
     if sms.text.find('Kirim uang ke ') == 0:
@@ -639,9 +640,16 @@ def checkBalance(modem, do_not_update = False, return_result = False):
         if response.sessionActive:
             response.cancel()
             
-        if _BALANCE != 0 and _BALANCE < MIN_BALANCE:
-            logger.info('Disabling pull, balance {0} < {1}'.format(_BALANCE, MIN_BALANCE))
-            disablePull()
+        if _BALANCE != 0:
+            if BALANCE < MIN_BALANCE:
+                
+                logger.info('Disabling pull, balance {0} < {1}'.format(_BALANCE, MIN_BALANCE))
+                disablePull()
+                
+            else:
+                
+                logger.info('Enabling pull, balance {0} > {1}'.format(_BALANCE, MIN_BALANCE))
+                enablePull()
             
     except:
         logger.warning('Error when requesting BALANCE by USSD')
-- 
1.9.0