set access token and remove payload before new request

OpenApi-5p · Mar 27, 2024 · 033a3ed · 033a3ed
1 parent edef647
commit 033a3ed
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/py5paisa/py5paisa.py b/py5paisa/py5paisa.py
@@ -273,7 +273,7 @@ def order_request(self, req_type) -> None:
                 raise Exception("Invalid request type!")
             res = self.session.post(url, json=self.payload,
                                     headers=HEADERS).json()
-
+            self.payload={}
             if req_type == "MS":
                 log_response(res["head"]["statusDescription"])
             else:
@@ -738,6 +738,7 @@ def get_access_token(self, request_token=None):
                 url = ACCESS_TOKEN_ROUTE
 
                 res = self.session.post(url, json=self.payload).json()
+                self.payload={}
                 message = res["body"]["Message"]
 
                 if message == "Success":
@@ -760,6 +761,7 @@ def get_request_token(self, client_code, totp, pin):
             url = GET_REQUEST_TOKEN_ROUTE
 
             res = self.session.post(url, json=self.payload).json()
+            self.payload={}
             message = res["body"]["Status"]
 
             if message == 0:
@@ -982,4 +984,10 @@ def place_order_bulk(self, **order):
             # Handle unknown order_type if needed
             pass
         except Exception as e:
-            log_response(e)
+            log_response(e)
+
+
+    def set_access_token(self,accessToken,clientCode):
+        self.access_token=accessToken
+        self.client_code=clientCode
+        self.Jwt_token = self.access_token