Merge pull request #11 from bunq/feature/bunq/tinker_python#10_automatic_money_request

Request money on Sandbox if balance zero. (#10)

Author: OGKevin

tinker/libs/bunq_lib.py

@@ -3,6 +3,7 @@
 from os.path import isfile
 from os import remove
 import socket
+from time import sleep
 
 import requests
 from bunq.sdk.client import Pagination
@@ -14,6 +15,7 @@
 from bunq.sdk.model.generated import endpoint
 from bunq.sdk.model.generated.object_ import Pointer, Amount, NotificationFilter
 
+
 NOTIFICATION_DELIVERY_METHOD_URL = 'URL'
 
 NOTIFICATION_CATEGORY_MUTATION = 'MUTATION'
@@ -31,9 +33,16 @@ class BunqLib(object):
 
     _DEFAULT_COUNT = 10
     _POINTER_TYPE_EMAIL = 'EMAIL'
-    _CURRENCY_EURL = 'EUR'
+    _CURRENCY_EUR = 'EUR'
     _DEVICE_DESCRIPTION = "python tinker"
 
+    _REQUEST_SPENDING_MONEY_AMOUNT = '500.0'
+    _REQUEST_SPENDING_MONEY_RECIPIENT = 'sugardaddy@bunq.com'
+    _REQUEST_SPENDING_MONEY_DESCRIPTION = 'Requesting some spending money.'
+    _REQUEST_SPENDING_MONEY_WAIT_TIME_SECONDS = 1
+
+    _ZERO_BALANCE = 0.0
+
     def __init__(self, env):
         """
         :type env: ApiEnvironmentType
@@ -43,6 +52,7 @@ def __init__(self, env):
         self.env = env
         self.setup_context()
         self.setup_current_user()
+        self.__request_spending_money_if_needed()
 
     def setup_context(self, reset_config_if_needed=True):
         if isfile(self.determine_bunq_conf_filename()):
@@ -161,7 +171,7 @@ def make_payment(self, amount_string, description, recipient):
         """
 
         endpoint.Payment.create(
-            amount=Amount(amount_string, self._CURRENCY_EURL),
+            amount=Amount(amount_string, self._CURRENCY_EUR),
             counterparty_alias=Pointer(self._POINTER_TYPE_EMAIL, recipient),
             description=description
         )
@@ -174,7 +184,7 @@ def make_request(self, amount_string, description, recipient):
         """
 
         endpoint.RequestInquiry.create(
-            amount_inquired=Amount(amount_string, self._CURRENCY_EURL),
+            amount_inquired=Amount(amount_string, self._CURRENCY_EUR),
             counterparty_alias=Pointer(self._POINTER_TYPE_EMAIL, recipient),
             description=description,
             allow_bunqme=True
@@ -249,3 +259,17 @@ def generate_new_sandbox_user(self):
                 json.dumps(response_json["Response"][0]["ApiKey"]))
 
         raise BunqException(self._ERROR_COULD_NOT_CREATE_NEW_SANDBOX_USER)
+
+    def __request_spending_money_if_needed(self):
+        if self.__should_request_spending_money():
+            endpoint.RequestInquiry.create(
+                amount_inquired=Amount(self._REQUEST_SPENDING_MONEY_AMOUNT, self._CURRENCY_EUR),
+                counterparty_alias=Pointer(self._POINTER_TYPE_EMAIL, self._REQUEST_SPENDING_MONEY_RECIPIENT),
+                description=self._REQUEST_SPENDING_MONEY_DESCRIPTION,
+                allow_bunqme=False
+            )
+            sleep(self._REQUEST_SPENDING_MONEY_WAIT_TIME_SECONDS)
+
+    def __should_request_spending_money(self):
+        return self.env == ApiEnvironmentType.SANDBOX \
+                and float(BunqContext.user_context().primary_monetary_account.balance.value) <= self._ZERO_BALANCE