quantplay 2.0.141__tar.gz → 2.0.142__tar.gz

{quantplay-2.0.141 → quantplay-2.0.142}/PKG-INFO

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

quantplay-2.0.142/quantplay/broker/breeze/breeze_utils.py

@@ -0,0 +1,90 @@
+import binascii
+import time
+import traceback
+
+import pyotp
+from retrying import retry  # type: ignore
+from selenium.common.exceptions import WebDriverException
+
+from quantplay.exception.exceptions import (
+    InvalidArgumentException,
+    RetryableException,
+    TokenException,
+    WrongLibrarySetup,
+    retry_exception,
+)
+from quantplay.utils.selenium_utils import Selenium
+
+
+class BreezeUtils:
+    user_id_xpath = '//*[@id="txtuid"]'
+    password_xpath = '//*[@id="txtPass"]'
+    i_agree_xpath = '//*[@id="chkssTnc"]'
+
+    login_xpath = '//*[@id="btnSubmit"]'
+    totp_xpath = '//*[@id="pnlOTP"]/div[2]/div[2]/div[3]/div/div[{}]/input'
+    submit_xpath = '//*[@id="Button1"]'
+
+    @staticmethod
+    @retry(
+        wait_exponential_multiplier=1000,
+        wait_exponential_max=10000,
+        stop_max_attempt_number=3,
+        retry_on_exception=retry_exception,
+    )
+    def get_session_code(api_key: str, user_id: str, password: str, totp: str):
+        try:
+            driver = Selenium().get_browser()
+
+            icici_direct_url = (
+                f"https://api.icicidirect.com/apiuser/login?api_key={api_key}"
+            )
+
+            driver.get(icici_direct_url)
+            time.sleep(2)
+
+            user_id_element = driver.find_element("xpath", BreezeUtils.user_id_xpath)
+            password_element = driver.find_element("xpath", BreezeUtils.password_xpath)
+            i_agree_element = driver.find_element("xpath", BreezeUtils.i_agree_xpath)
+            login_element = driver.find_element("xpath", BreezeUtils.login_xpath)
+
+            user_id_element.send_keys(user_id)
+            password_element.send_keys(password)
+            i_agree_element.click()
+            time.sleep(0.5)
+            login_element.click()
+            time.sleep(2)
+            totp_code = pyotp.TOTP(str(totp)).now()
+            for i in range(1, 7):
+                entry = totp_code[i - 1]
+                totp_pin = driver.find_element(
+                    "xpath", BreezeUtils.totp_xpath.replace("{}", str(i))
+                )
+                totp_pin.send_keys(entry)
+            submit_element = driver.find_element("xpath", BreezeUtils.submit_xpath)
+            submit_element.click()
+
+            time.sleep(2)
+
+            url = driver.current_url
+            print("got icici redirect url {}".format(url))
+            session_token = url.split("apisession=")[1].split("&")[0]
+
+            driver.close()
+
+            return session_token
+
+        except binascii.Error:
+            raise InvalidArgumentException("Invalid TOTP key provided")
+
+        except (InvalidArgumentException, TokenException) as e:
+            raise e
+
+        except WebDriverException:
+            raise WrongLibrarySetup(
+                "ICICI login failed, please contact support team or generate token from Dashboard page"
+            )
+
+        except Exception as e:
+            traceback.print_exc()
+            raise RetryableException(str(e))

{quantplay-2.0.141 → quantplay-2.0.142}/quantplay/broker/broker_factory.py

@@ -262,14 +262,24 @@ class BrokerFactory:
             broker_data["user_token"] = broker_client.user_token
 
         elif broker == Broker.ICICI:
-            broker_client = ICICI(
-                api_key=broker_data["api_key"],
-                api_secret=broker_data["api_secret"],
-                session_token=broker_data["session_token"],
-                load_instrument=False,
-            )
-
-            # TODO
+            if "password" in broker_data:
+                broker_client = ICICI(
+                    api_key=broker_data["api_key"],
+                    api_secret=broker_data["api_secret"],
+                    user_id=broker_data["user_id"],
+                    password=broker_data["password"],
+                    totp=broker_data["totp"],
+                    session_token=broker_data["session_token"],
+                    load_instrument=False,
+                )
+            else:
+                broker_client = ICICI(
+                    api_key=broker_data["api_key"],
+                    api_secret=broker_data["api_secret"],
+                    session_token=broker_data["session_token"],
+                    load_instrument=False,
+                )
+            broker_data["session_token"] = broker_client.session_token
 
         elif broker == Broker.FLATTRADE:
             broker_client = FlatTrade(

{quantplay-2.0.141 → quantplay-2.0.142}/quantplay/broker/icici_direct.py

@@ -2,6 +2,8 @@ from collections.abc import Callable
 from datetime import datetime, date, timedelta
 from typing import Any, Literal, Dict
 from breeze_connect.breeze_connect import BreezeConnect  # type: ignore
+
+from quantplay.broker.breeze.breeze_utils import BreezeUtils
 from quantplay.exception import QuantplayOrderPlacementException, InvalidArgumentException
 
 from quantplay.broker.generics.broker import Broker
@@ -52,15 +54,36 @@ class ICICI(Broker):
         api_key: str | None = None,
         api_secret: str | None = None,
         session_token: str | None = None,
+        user_id: str | None = None,
+        password: str | None = None,
+        totp: str | None = None,
         load_instrument: bool = True,
     ) -> None:
-        if api_key and api_secret and session_token:
-            self.wrapper = BreezeConnect(api_key)  # type:ignore
-            self.wrapper.generate_session(api_secret, session_token)  # type: ignore
+        if api_key and user_id and password and totp:
+            self.set_session_token(api_key, user_id, password, totp)
+        else:
+            self.session_token = session_token
 
+        if api_key and api_secret and self.session_token:
+            self.wrapper = BreezeConnect(api_key)  # type:ignore
+            self.wrapper.generate_session(api_secret, self.session_token)  # type: ignore
+        else:
+            raise InvalidArgumentException(
+                "ICICI token generation failed, missing arguments"
+            )
         if load_instrument:
             self.load_instrument()
 
+    def set_session_token(
+        self,
+        api_key: str,
+        user_id: str,
+        password: str,
+        totp: str,
+    ):
+        session_token = BreezeUtils.get_session_code(api_key, user_id, password, totp)
+        self.session_token = session_token
+
     def load_instrument(self, file_name: str | None = None) -> None:
         try:
             self.symbol_data = InstrumentData.get_instance().load_data(  # type: ignore

{quantplay-2.0.141 → quantplay-2.0.142}/quantplay/broker/xts_utils/Connect.py

@@ -12,7 +12,7 @@ from typing import Any, Literal
 
 import requests
 import requests.adapters
-import urllib3
+import urllib3  # type:ignore
 from six.moves.urllib.parse import urljoin  # type:ignore
 
 from quantplay.broker.xts_utils import Exception as ex

{quantplay-2.0.141 → quantplay-2.0.142}/quantplay.egg-info/PKG-INFO

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

{quantplay-2.0.141 → quantplay-2.0.142}/quantplay.egg-info/SOURCES.txt

@@ -28,6 +28,8 @@ quantplay/broker/xts.py
 quantplay/broker/zerodha.py
 quantplay/broker/auto_login/__init__.py
 quantplay/broker/auto_login/aliceblue.py
+quantplay/broker/breeze/__init__.py
+quantplay/broker/breeze/breeze_utils.py
 quantplay/broker/finvasia_utils/__init__.py
 quantplay/broker/finvasia_utils/fa_noren.py
 quantplay/broker/ft_utils/__init__.py

{quantplay-2.0.141 → quantplay-2.0.142}/setup.py

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