quantplay 2.0.165__tar.gz → 2.0.166__tar.gz

{quantplay-2.0.165 → quantplay-2.0.166}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantplay
-Version: 2.0.165
+Version: 2.0.166
 Summary: This python package will be stored in AWS CodeArtifact
 Home-page: 
 Author: 

{quantplay-2.0.165 → quantplay-2.0.166}/quantplay/broker/broker_factory.py

@@ -188,6 +188,7 @@ class BrokerFactory:
                 consumer_secret=broker_data["consumer_secret"],
                 mobilenumber=broker_data["mobilenumber"],
                 password=broker_data["password"],
+                otp=broker_data.get("otp"),
                 load_instrument=False,
             )
 
@@ -386,6 +387,7 @@ class BrokerFactory:
         elif broker == Broker.KOTAK:
             broker_client = Kotak(
                 configuration=broker_data["configuration"],
+                otp=broker_data.get("otp"),
                 load_instrument=load_instrument,
             )
         elif broker == Broker.ZERODHA:

{quantplay-2.0.165 → quantplay-2.0.166}/quantplay/broker/kotak.py

@@ -45,12 +45,13 @@ class Kotak(Broker):
         consumer_secret: str | None = None,
         mobilenumber: str | None = None,
         password: str | None = None,
+        otp: str | None = None,
         configuration: dict[str, str] | None = None,
         load_instrument: bool = True,
     ) -> None:
         super().__init__()
 
-        self.configuration: dict[str, str] = {
+        self.configuration: dict[str, str | None] = {
             "fin_key": "X6Nk8cQhUgGmJ2vBdWw4sfzrz4L5En",
         }
 
@@ -60,15 +61,18 @@ class Kotak(Broker):
                 "bearer_token": configuration["bearer_token"],
                 "view_token": configuration["view_token"],
                 "sid": configuration["sid"],
-                "edit_token": configuration["edit_token"],
-                "edit_sid": configuration["edit_sid"],
-                "edit_rid": configuration["edit_rid"],
-                "serverId": configuration["serverId"],
+                "edit_token": configuration.get("edit_token"),
+                "edit_sid": configuration.get("edit_sid"),
+                "edit_rid": configuration.get("edit_rid"),
+                "serverId": configuration.get("serverId"),
             }
 
         elif consumer_key and consumer_secret and mobilenumber and password:
             self.login(consumer_key, consumer_secret, mobilenumber, password)
 
+        if otp:
+            self.verify_2fa(otp)
+
         if load_instrument:
             self.load_instrument()
 
@@ -145,7 +149,14 @@ class Kotak(Broker):
 
         _generate_otp = json.loads(generate_otp_resp.text)
 
-    def verify_2fa(self, otp: int) -> None:
+    def verify_2fa(self, otp: str) -> None:
+        if (
+            self.configuration["bearer_token"] is None
+            or self.configuration["sid"] is None
+            or self.configuration["view_token"] is None
+        ):
+            raise InvalidArgumentException("Missing Params")
+
         login_resp = requests.post(
             url=f"{PROD_BASE_URL}login/1.0/login/v2/validate",
             headers={
@@ -154,7 +165,7 @@ class Kotak(Broker):
                 "Auth": self.configuration["view_token"],
                 "Content-Type": "application/json",
             },
-            data=json.dumps({"userId": self.kotakUserId, "otp": str(otp)}),
+            data=json.dumps({"userId": self.kotakUserId, "otp": otp}),
         )
 
         edit_token_resp = json.loads(login_resp.text)
@@ -580,7 +591,7 @@ class Kotak(Broker):
         url = f"{PROD_BASE_URL}{url}"
 
         request_headers = {
-            "Authorization": "Bearer " + self.configuration["bearer_token"],
+            "Authorization": f'Bearer {self.configuration["bearer_token"]}',
             "Sid": self.configuration["edit_sid"],
             "Auth": self.configuration["edit_token"],
             "neo-fin-key": self.configuration["fin_key"],

{quantplay-2.0.165 → quantplay-2.0.166}/quantplay.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantplay
-Version: 2.0.165
+Version: 2.0.166
 Summary: This python package will be stored in AWS CodeArtifact
 Home-page: 
 Author: 

{quantplay-2.0.165 → quantplay-2.0.166}/setup.py

@@ -21,7 +21,7 @@ requirements = [
 setup(
     name="quantplay",
     long_description=Path("README.md").read_text(),
-    version="2.0.165",
+    version="2.0.166",
     setup_requires=["pytest-runner"],
     install_requires=requirements,
     tests_require=[],