From a8a4a801ad5e826970e3374f06b81e5efa818a15 Mon Sep 17 00:00:00 2001
From: jenish shah <jenish.shah@iifl.com>
Date: Sat, 20 Jan 2024 14:59:53 +0530
Subject: [PATCH] basket margin

---
 py5paisa/py5paisa.py | 17 ++++++++++++++++-
 py5paisa/urlconst.py |  1 +
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/py5paisa/py5paisa.py b/py5paisa/py5paisa.py
index 5ccfde23..9d7e22ce 100644
--- a/py5paisa/py5paisa.py
+++ b/py5paisa/py5paisa.py
@@ -155,7 +155,10 @@ def order_request(self, req_type) -> None:
         try:
             self.payload["body"]["ClientCode"] = self.client_code
             self.payload["head"]["key"] = self.USER_KEY
-            HEADERS["Authorization"] = f'Bearer {self.access_token}'
+            if self.access_token != "":
+                    HEADERS["Authorization"] = f'Bearer {self.Jwt_token}'
+            else:
+                HEADERS["Authorization"] = f'Bearer {self.access_token}'
             HEADERS["5Paisa-API-Uid"] = self.APIUID
             if req_type == "OP":
                 url = self.ORDER_PLACEMENT_ROUTE
@@ -260,6 +263,8 @@ def order_request(self, req_type) -> None:
                 url = self.GETVTTORDER_ROUTE
             elif req_type == "HVTT":
                 url = self.HISVTTORDER_ROUTE
+            elif req_type == "BMC":
+                url = self.BASKETMARGIN_ROUTE
             else:
                 raise Exception("Invalid request type!")
             res = self.session.post(url, json=self.payload,
@@ -692,6 +697,7 @@ def set_url(self):
             self.MODVTTORDER_ROUTE = MODVTTORDER_ROUTE
             self.GETVTTORDER_ROUTE = GETVTTORDER_ROUTE
             self.HISVTTORDER_ROUTE = HISVTTORDER_ROUTE
+            self.BASKETMARGIN_ROUTE = BASKETMARGIN_ROUTE
         except Exception as e:
             log_response(e)
 
@@ -947,5 +953,14 @@ def vtt_order(self, order_type, **order):
         except KeyError:
             # Handle unknown order_type if needed
             pass
+        except Exception as e:
+            log_response(e)
+
+    def basket_margin(self, BasketID: str,CoverPositions : str):
+        try:
+            if self.client_code != None:
+                self.payload["body"]["BasketID"] = BasketID
+                self.payload["body"]["CoverPositions"] = CoverPositions
+            return self.order_request("BMC")
         except Exception as e:
             log_response(e)
\ No newline at end of file
diff --git a/py5paisa/urlconst.py b/py5paisa/urlconst.py
index 5bab3b5f..0e5dbfbe 100644
--- a/py5paisa/urlconst.py
+++ b/py5paisa/urlconst.py
@@ -64,3 +64,4 @@
 MODVTTORDER_ROUTE=f'{BaseUrl}ModifyVTTOrder'
 GETVTTORDER_ROUTE=f'{BaseUrl}GetVTTOrder'
 HISVTTORDER_ROUTE=f'{BaseUrl}GetHistoryVTTOrder'
+BASKETMARGIN_ROUTE= f'{BaseUrl}PreOrdMarginCalculationBulkVendor'
\ No newline at end of file