Re Initializing headers (token) in case of JWT

Author: krishna

setu/deeplink.py

@@ -23,12 +23,13 @@ def __init__(self, schemeId, secret, productInstance, authType="JWT", mode="SAND
         self.mode = mode
         self.authType = authType
 
-        # Generate required headers
-        self.headers = generate_setu_headers(
-            self.schemeId, self.secret, self.productInstance, self.url, self.authType
-        )
-
+        
         if self.authType != 'JWT':
+            # Generate required headers
+            self.headers = generate_setu_headers(
+                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+            )
+            # Set the url to v2
             self.url = self.url + '/v2'
 
     # Generate UPI payment link method
@@ -69,6 +70,12 @@ def create_payment_link(
         if validationRules is not None:
             payload.update({"validationRules": validationRules})
 
+        # 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
+            )
+
         # Call API with required parameters
         response = requests.post(
             self.url + path, json=payload, headers=self.headers
@@ -86,6 +93,12 @@ def check_payment_status(
     ):
         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
+            )
+
         # Call API with required parameters
         response = requests.get(
             self.url + path, headers=self.headers
@@ -114,6 +127,12 @@ def trigger_mock_payment(self, amountValue, upiID):
             "type": "UPI",
         }
 
+        # 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
+            )
+
         # Call API with required parameters
         response = requests.post(
             self.url + path, json=payload, headers=self.headers