Merge pull request #13 from SetuHQ/add-additional-info

Support additionalInfo in create link payload

Author: nithinkashyapn

.prettierrc

@@ -49,7 +49,8 @@ link = dl.create_payment_link(
         payeeName=String, # Optional
         expiryDate=String, # Optional
         settlement=Object, # Optional
-        validationRules=Object # Optional
+        validationRules=Object, # Optional
+        additionalInfo=Object # Optional
     )
 print(link)
 ```

README.md

@@ -1,7 +1,2 @@
-certifi==2020.12.5
-chardet==4.0.0
-idna==2.10
-PyJWT>=2.0.1
-requests==2.25.1
-urllib3==1.26.2
-uuid==1.30
+PyJWT==2.3.0
+requests==2.26.0

requirements.txt

@@ -31,14 +31,8 @@ def generate_oAuth_token(clientId, secret, url):
         "clientID": clientId,
         "secret": secret,
     }
-    headers = {
-        "Content-Type": "application/json"
-    }
-    response = requests.post(
-        url,
-        json=payload,
-        headers=headers
-    )
+    headers = {"Content-Type": "application/json"}
+    response = requests.post(url, json=payload, headers=headers)
 
     if response.status_code == 200:
         data = response.json()
@@ -47,11 +41,15 @@ def generate_oAuth_token(clientId, secret, url):
         raise
 
 
-def generate_setu_headers(schemeId, secret, setuProductInstanceID, url,  authType="JWT"):
+def generate_setu_headers(
+    schemeId,
+    secret,
+    setuProductInstanceID,
+    url,
+    authType="JWT",
+):
 
-    headers = {
-        "X-Setu-Product-Instance-ID": setuProductInstanceID
-    }
+    headers = {"X-Setu-Product-Instance-ID": setuProductInstanceID}
 
     if authType == "JWT":
         auth = generate_bearer_JWT(schemeId, secret)

setu/auth.py

@@ -1,11 +1,9 @@
 import requests
-import datetime
 from .auth import generate_setu_headers
 from .errors import handle_setu_errors
 
 
 class URLS:
-
     class Sandbox:
         url = "https://uat.setu.co/api"
 
@@ -14,20 +12,29 @@ class Prod:
 
 
 class Deeplink:
-
-    def __init__(self, schemeId, secret, productInstance, authType="JWT", mode="SANDBOX"):
+    def __init__(
+        self,
+        schemeId,
+        secret,
+        productInstance,
+        authType="JWT",
+        mode="SANDBOX",
+    ):
         self.schemeId = schemeId
         self.secret = secret
         self.productInstance = productInstance
         self.url = URLS.Sandbox.url if mode != "PRODUCTION" else URLS.Prod.url
         self.mode = mode
         self.authType = authType
-        
-        
+
         if self.authType != 'JWT':
             # Generate required headers
             self.headers = generate_setu_headers(
-                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+                self.schemeId,
+                self.secret,
+                self.productInstance,
+                self.url,
+                self.authType,
             )
             # Set the url to v2
             self.url = self.url + '/v2'
@@ -42,7 +49,8 @@ def create_payment_link(
         expiryDate=None,
         payeeName=None,
         settlement=None,
-        validationRules=None
+        validationRules=None,
+        additionalInfo=None,
     ):
 
         path = "/payment-links"
@@ -70,15 +78,24 @@ def create_payment_link(
         if validationRules is not None:
             payload.update({"validationRules": validationRules})
 
+        if additionalInfo is not None:
+            payload.update({"additionalInfo": additionalInfo})
+
         # In case of JWT the token expires often enough to warrant creation of new tokens
         if self.authType == 'JWT':
             self.headers = generate_setu_headers(
-                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+                self.schemeId,
+                self.secret,
+                self.productInstance,
+                self.url,
+                self.authType,
             )
 
         # Call API with required parameters
         response = requests.post(
-            self.url + path, json=payload, headers=self.headers
+            self.url + path,
+            json=payload,
+            headers=self.headers,
         )
 
         # Handle errors
@@ -87,22 +104,21 @@ def create_payment_link(
         return response.json()
 
     # Check status of UPI payment link method
-    def check_payment_status(
-        self,
-        platformBillID
-    ):
+    def check_payment_status(self, platformBillID):
         path = "/payment-links/{}".format(platformBillID)
-        
+
         # In case of JWT the token expires often enough to warrant creation of new tokens
         if self.authType == 'JWT':
             self.headers = generate_setu_headers(
-                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+                self.schemeId,
+                self.secret,
+                self.productInstance,
+                self.url,
+                self.authType,
             )
 
         # Call API with required parameters
-        response = requests.get(
-            self.url + path, headers=self.headers
-        )
+        response = requests.get(self.url + path, headers=self.headers)
 
         return response.json()
 
@@ -130,12 +146,18 @@ def trigger_mock_payment(self, amountValue, upiID):
         # In case of JWT the token expires often enough to warrant creation of new tokens
         if self.authType == 'JWT':
             self.headers = generate_setu_headers(
-                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+                self.schemeId,
+                self.secret,
+                self.productInstance,
+                self.url,
+                self.authType,
             )
 
         # Call API with required parameters
         response = requests.post(
-            self.url + path, json=payload, headers=self.headers
+            self.url + path,
+            json=payload,
+            headers=self.headers,
         )
 
         return response

setu/deeplink.py

@@ -13,7 +13,10 @@
     long_description=README,
     long_description_content_type="text/markdown",
     license='MIT',
-    install_requires=["requests", "uuid", "PyJWT"],
+    install_requires=[
+        "requests==2.26.0",
+        "PyJWT==2.3.0",
+    ],
     packages=['setu'],
-    zip_safe=False
+    zip_safe=False,
 )