add schedule payout

Author: corentinalcoy

.gitignore

@@ -3,4 +3,6 @@
 test.py
 kkiapay.egg-info
 dist
-build
+build
+venv
+.idea

kkiapay/core.py

@@ -1,12 +1,22 @@
 import json
-import requests
 from collections import namedtuple
 
+import requests
 
-class Kkiapay:
+from kkiapay.exceptions import KKiapayAlogrithmException, KKiapayDestinationTypeException
 
+
+class Kkiapay:
     BASE_URL = "https://api.kkiapay.me"
     SANDBOX_URL = "https://api-sandbox.kkiapay.me"
+    ALGORITHMS = {"1": 'rate', "2": 'roof'}
+
+    RATE_FREQUENCIES = {"1": '3d', "2": '1w', "3": '1m'}
+    default_roof_amount = "50000"
+    DESTINATION_TYPES = {
+        "1": 'MOBILE_MONEY',
+        "2": 'BANK_ACCOUNT',
+    }
 
     def __init__(self, public_key, private_key, secret, sandbox=False):
         self.secret = secret
@@ -45,7 +55,29 @@ def refund_transaction(self, transaction_id):
             ),
         )
 
-    def setup_payout(self, options):
-        self.url += "/merchant/payouts/schedule"
-        r = requests.post(self.url, data=options, headers=self.headers)
-        return r.text
+    def setup_payout(self, algorithm: str, destination: str, destination_type: str, roof_amount: str = None,
+                     send_notification: bool = True, rate_frequency: str = None, country_code: str = "229"):
+        options = {
+            "destination": country_code + destination if destination_type == "1" else destination,
+            "send_notification": 1 if send_notification else 0,
+        }
+
+        if destination_type in self.DESTINATION_TYPES.keys():
+            options["destination_type"] = self.DESTINATION_TYPES[destination_type]
+        else:
+            raise KKiapayDestinationTypeException(self.DESTINATION_TYPES)
+
+        if algorithm in self.ALGORITHMS.keys():
+            options['algorithm'] = self.ALGORITHMS[algorithm]
+        else:
+            raise KKiapayAlogrithmException(self.ALGORITHMS)
+        if algorithm == "2":
+            options["roof_amount"] = roof_amount if roof_amount is not None else self.default_roof_amount
+        else:
+            options["rate_frequency"] = rate_frequency if rate_frequency in self.RATE_FREQUENCIES.values() else \
+                self.RATE_FREQUENCIES["1"]
+        r = requests.post(self.url + '/merchant/payouts/schedule', data=options, headers=self.headers)
+        return {
+            'response': r.json(),
+            'status_code': r.status_code
+        }

kkiapay/exceptions.py

@@ -1,2 +1,16 @@
 class KkiapayException(Exception):
     pass
+
+
+class KKiapayAlogrithmException(Exception):
+    def __init__(self, algorithms):
+        Exception.__init__(self,
+                           "algorithm must be {}".format(
+                               " or ".join(" if it's ".join(algorithm) for algorithm in algorithms.items())))
+
+
+class KKiapayDestinationTypeException(Exception):
+    def __init__(self, algorithms):
+        Exception.__init__(self,
+                           "destination_type must be {}".format(
+                               " or ".join(" if it's ".join(algorithm) for algorithm in algorithms.items())))