quantplay 2.0.20__tar.gz → 2.0.22__tar.gz

{quantplay-2.0.20 → quantplay-2.0.22}/PKG-INFO

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

{quantplay-2.0.20 → quantplay-2.0.22}/pyproject.toml

@@ -6,6 +6,7 @@ line-length = 90
 
 [tool.pyright]
 typeCheckingMode = "standard"
-pythonVersion = "3.10"
+pythonVersion = "3.12"
 venvPath = "."
 venv = ".venv"
+reportUntypedFunctionDecorator="none"

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/aliceblue.py

@@ -3,13 +3,13 @@ import copy
 import pickle
 from queue import Queue
 import traceback
-from typing import Callable, Dict, Literal
+from typing import Any, Callable, Dict, Literal
 
 import polars as pl
 from pya3 import Aliceblue as Alice
 from pya3 import OrderType as AliceOrderType
 from pya3 import ProductType, TransactionType
-from retrying import retry
+from retrying import retry  # type: ignore
 
 from quantplay.broker.generics.broker import Broker
 from quantplay.exception.exceptions import (
@@ -22,7 +22,7 @@ from quantplay.exception.exceptions import (
 from quantplay.exception.exceptions import TokenException as QuantplayTokenException
 from quantplay.model.broker import ModifyOrderRequest, UserBrokerProfileResponse
 from quantplay.model.order_event import OrderUpdateEvent
-from quantplay.utils.constant import Constants, OrderType, timeit
+from quantplay.utils.constant import Constants, OrderType
 from quantplay.utils.pickle_utils import InstrumentData
 from quantplay.wrapper.aws.s3 import S3Utils
 
@@ -30,7 +30,6 @@ logger = Constants.logger
 
 
 class Aliceblue(Broker):
-    @timeit(MetricName="Aliceblue:__init__")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -99,7 +98,6 @@ class Aliceblue(Broker):
             return symbol
         return self.quantplay_symbol_map[symbol]
 
-    @timeit(MetricName="Aliceblue:load_instrument")
     def load_instrument(self, file_name: str | None = None):
         try:
             instrument_data_instance = InstrumentData.get_instance()
@@ -298,7 +296,6 @@ class Aliceblue(Broker):
 
         return response
 
-    @timeit(MetricName="Aliceblue:holdings")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -352,7 +349,6 @@ class Aliceblue(Broker):
 
         return holdings_df[list(self.holidings_schema.keys())].cast(self.holidings_schema)
 
-    @timeit(MetricName="Aliceblue:positions")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -570,7 +566,8 @@ class Aliceblue(Broker):
 
         return margins
 
-    def invoke_aliceblue_api(self, fn: Callable, *args, **kwargs) -> Dict:
+    # TODO
+    def invoke_aliceblue_api(self, fn: Callable, *args, **kwargs) -> Any:
         try:
             response = fn(*args, **kwargs)
             if response is None:

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/angelone.py

@@ -10,7 +10,7 @@ from typing import Callable, Dict
 import polars as pl
 import pyotp
 from requests.exceptions import ConnectionError, ConnectTimeout
-from retrying import retry
+from retrying import retry  # type: ignore
 from SmartApi import SmartConnect
 
 from quantplay.broker.generics.broker import Broker
@@ -26,7 +26,7 @@ from quantplay.exception.exceptions import (
 from quantplay.exception.exceptions import TokenException as QuantplayTokenException
 from quantplay.model.broker import ModifyOrderRequest, UserBrokerProfileResponse
 from quantplay.model.order_event import OrderUpdateEvent
-from quantplay.utils.constant import Constants, OrderType, timeit
+from quantplay.utils.constant import Constants, OrderType
 from quantplay.utils.exchange import Market as MarketConstants
 from quantplay.utils.pickle_utils import InstrumentData
 from quantplay.wrapper.aws.s3 import S3Utils
@@ -36,7 +36,6 @@ class AngelOne(Broker):
     order_sl = "STOPLOSS_LIMIT"
     order_slm = "STOPLOSS_MARKET"
 
-    @timeit(MetricName="Angelone:init")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -108,7 +107,6 @@ class AngelOne(Broker):
         if "errorCode" in response and response["errorCode"] == "AG8001":
             raise TokenException(f"{self.user_id}: Invalid Token")
 
-    @timeit(MetricName="Angelone:load_instrument")
     def load_instrument(self, file_name: str | None = None) -> None:
         try:
             instrument_data_instance = InstrumentData.get_instance()
@@ -164,7 +162,6 @@ class AngelOne(Broker):
             "Product {} not supported for trading".format(product)
         )
 
-    @timeit(MetricName="Angelone:get_ltp")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -195,7 +192,6 @@ class AngelOne(Broker):
 
         return response["data"]["ltp"]
 
-    @timeit(MetricName="Angelone:place_order")
     def place_order(
         self,
         tradingsymbol=None,
@@ -255,7 +251,6 @@ class AngelOne(Broker):
             return "NORMAL"
         return variety
 
-    @timeit(MetricName="Angelone:modify_order")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -312,11 +307,9 @@ class AngelOne(Broker):
             )
             raise
 
-    @timeit(MetricName="Angelone:cancel_order")
     def cancel_order(self, order_id, variety="NORMAL"):
         self.wrapper.cancelOrder(order_id=order_id, variety=variety)
 
-    @timeit(MetricName="Angelone:holdings")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=15000,
@@ -354,7 +347,6 @@ class AngelOne(Broker):
 
         return holdings
 
-    @timeit(MetricName="Angelone:positions")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=15000,
@@ -435,7 +427,6 @@ class AngelOne(Broker):
             self.positions_schema
         )
 
-    @timeit(MetricName="Angelone:orders")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -543,7 +534,6 @@ class AngelOne(Broker):
                 traceback.print_exc()
                 raise ServiceException("Unknown error while fetching order book [{}]")
 
-    @timeit(MetricName="Angelone:profile")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -569,7 +559,6 @@ class AngelOne(Broker):
 
         return response
 
-    @timeit(MetricName="Angelone:margins")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/auto_login/aliceblue.py

@@ -3,7 +3,7 @@ import time
 import traceback
 
 import pyotp
-from retrying import retry
+from retrying import retry  # type: ignore
 from selenium.common.exceptions import NoSuchElementException, WebDriverException
 
 from quantplay.exception.exceptions import (
@@ -12,11 +12,12 @@ from quantplay.exception.exceptions import (
     RetryableException,
 )
 from quantplay.utils.selenium_utils import Selenium
+from selenium.webdriver import Chrome
 
 
 class AliceblueLogin:
     @staticmethod
-    def check_error(page_source):
+    def check_error(page_source: str):
         for error_message in [
             "User profile not found",
             "Invalid username or password",
@@ -33,36 +34,36 @@ class AliceblueLogin:
             raise InvalidArgumentException("Invalid API Key")
 
     @staticmethod
-    def click_on_login(driver):
+    def click_on_login(driver: Chrome):
         login = '//*[@id="login_card_inital"]/div[6]/button'
         login = driver.find_element("xpath", login)
         login.click()
 
     @staticmethod
-    def enter_user_id(driver, user_id: str):
+    def enter_user_id(driver: Chrome, user_id: str):
         time.sleep(0.5)
         user_id_xpath = '//*[@id="userid_inp"]'
-        driver.find_element("xpath", user_id_xpath).send_keys(user_id)
+        driver.find_element("xpath", user_id_xpath).send_keys(user_id)  # type: ignore
 
         time.sleep(0.5)
         next_xpath = '//*[@id="userId_btn_label"]'
         driver.find_element("xpath", next_xpath).click()
 
     @staticmethod
-    def enter_password(driver, password: str):
+    def enter_password(driver: Chrome, password: str):
         time.sleep(0.5)
         password_xpath = '//*[@id="password_inp"]'
-        driver.find_element("xpath", password_xpath).send_keys(password)
+        driver.find_element("xpath", password_xpath).send_keys(password)  # type: ignore
 
         time.sleep(0.5)
         next_xpath = '//*[@id="password_btn"]'
         driver.find_element("xpath", next_xpath).click()
 
     @staticmethod
-    def enter_totp(driver, totp: str):
+    def enter_totp(driver: Chrome, totp: str):
         time.sleep(1.5)
         totp_xpath = '//*[@id="totp_otp_inp"]'
-        driver.find_element("xpath", totp_xpath).send_keys(totp)
+        driver.find_element("xpath", totp_xpath).send_keys(totp)  # type: ignore
 
         # time.sleep(0.5)
         # next_xpath = '//*[@id="totp_btn"]'

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/five_paisa.py

@@ -5,7 +5,7 @@ import traceback
 import polars as pl
 import pyotp
 from py5paisa import FivePaisaClient
-from retrying import retry
+from retrying import retry  # type: ignore
 
 from quantplay.broker.generics.broker import Broker
 from quantplay.exception.exceptions import (
@@ -17,7 +17,7 @@ from quantplay.exception.exceptions import (
     retry_exception,
 )
 from quantplay.model.broker import UserBrokerProfileResponse
-from quantplay.utils.constant import Constants, OrderStatus, timeit
+from quantplay.utils.constant import Constants, OrderStatus
 
 logger = Constants.logger
 
@@ -29,7 +29,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="5Paisa:__init__")
     def __init__(
         self,
         app_source: str | None = None,
@@ -122,7 +121,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="5Paisa:profile")
     def profile(self):
         response: UserBrokerProfileResponse = {
             "user_id": self.client.client_code,
@@ -136,7 +134,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="5Paisa:get_ltp")
     def get_ltp(self, exchange, tradingsymbol: str) -> float:
         tradingsymbol = self.get_symbol(tradingsymbol, exchange)
         exchange_name = self.get_exchange(exchange)
@@ -153,16 +150,16 @@ class FivePaisa(Broker):
 
         return data.with_columns(
             pl.when((pl.col("Exch").eq("N")) & (pl.col("ExchType").eq("D")))
-            .then("NFO")
+            .then(pl.lit("NFO"))
             .alias("exchange"),
             pl.when((pl.col("Exch").eq("N")) & (pl.col("ExchType").eq("C")))
-            .then("NFO")
+            .then(pl.lit("NFO"))
             .alias("exchange"),
             pl.when((pl.col("Exch").eq("B")) & (pl.col("ExchType").eq("D")))
-            .then("NFO")
+            .then(pl.lit("NFO"))
             .alias("exchange"),
             pl.when((pl.col("Exch").eq("B")) & (pl.col("ExchType").eq("C")))
-            .then("NFO")
+            .then(pl.lit("NFO"))
             .alias("exchange"),
         )
 
@@ -172,7 +169,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="5Paisa:holdings")
     def holdings(self):
         holdings = self.client.holdings()
         if holdings is None:
@@ -219,7 +215,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="5Paisa:positions")
     def positions(self, drop_cnc: bool = True):
         positions = self.client.positions()
 
@@ -279,7 +274,7 @@ class FivePaisa(Broker):
                 (pl.col("exchange").is_in(["NFO", "BFO"]))
                 & (pl.col("product").ne("NRML"))
             )
-            .then("CNC")
+            .then(pl.lit("CNC"))
             .otherwise(pl.col("product"))
             .alias("product"),
         )
@@ -295,7 +290,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="Upstox:orders")
     def orders(self, tag: str | None = None, add_ltp: bool = True):
         orders = self.client.order_book()
         if orders is None:
@@ -335,13 +329,13 @@ class FivePaisa(Broker):
         orders = (
             orders.with_columns(
                 pl.when(pl.col("AtMarket") == "Y")
-                .then("MARKET")
-                .otherwise("LIMIT")
+                .then(pl.lit("MARKET"))
+                .otherwise(pl.lit("LIMIT"))
                 .alias("order_type"),
             )
             .with_columns(
                 pl.when((pl.col("order_type") == "LIMIT") & (pl.col("WithSL") == "Y"))
-                .then("SL")
+                .then(pl.lit("SL"))
                 .otherwise(pl.col("order_type"))
                 .alias("order_type"),
                 pl.col("transaction_type")
@@ -378,7 +372,7 @@ class FivePaisa(Broker):
                     pl.col("product")
                     == "CNC" & (pl.col("exchange").is_in(["NFO", "BFO"]))
                 )
-                .then("NRML")
+                .then(pl.lit("NRML"))
                 .otherwise(pl.col("product"))
                 .alias("product"),
             )
@@ -445,7 +439,6 @@ class FivePaisa(Broker):
             return "D"
         raise InvalidArgumentException(f"exchange {exchange} not supported")
 
-    @timeit(MetricName="FivePaisa:place_order")
     def place_order(
         self,
         tradingsymbol=None,
@@ -514,7 +507,6 @@ class FivePaisa(Broker):
         stop_max_attempt_number=3,
         retry_on_exception=retry_exception,
     )
-    @timeit(MetricName="FivePaisa:margins")
     def margins(self):
         margins = self.client.margin()
         if margins is None:
@@ -527,7 +519,6 @@ class FivePaisa(Broker):
         }
         return margins
 
-    @timeit(MetricName="5Paisa:account_summary")
     def account_summary(self):
         margins = self.margins()
         response = {

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/flattrade.py

@@ -9,11 +9,9 @@ from quantplay.broker.ft_utils.ft_noren import FT_NorenApi
 from quantplay.broker.noren import Noren
 from quantplay.exception.exceptions import InvalidArgumentException, RetryableException
 from quantplay.model.order_event import OrderUpdateEvent
-from quantplay.utils.constant import timeit
 
 
 class FlatTrade(Noren):
-    @timeit(MetricName="Flattrade:__init__")
     def __init__(
         self,
         order_updates: Queue[OrderUpdateEvent] | None = None,

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/ft_utils/flattrade_utils.py

@@ -3,7 +3,7 @@ import time
 import traceback
 
 import pyotp
-from retrying import retry
+from retrying import retry  # type: ignore
 from selenium.common.exceptions import WebDriverException
 
 from quantplay.exception.exceptions import (

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/generics/broker.py

@@ -34,7 +34,7 @@ from quantplay.model.broker import (
 )
 from quantplay.model.generics import ProductType, TransactionType, OrderTypeType
 from quantplay.model.order_event import OrderUpdateEvent
-from quantplay.utils.constant import Constants, timeit
+from quantplay.utils.constant import Constants
 from quantplay.utils.exchange import Market as MarketConstants
 from quantplay.utils.number_utils import NumberUtils
 from quantplay.utils.pickle_utils import InstrumentData, PickleUtils
@@ -163,7 +163,6 @@ class Broker:
         if save_as:
             PickleUtils.save_data(self.symbol_data, save_as)
 
-    @timeit(MetricName="Broker:initialize_broker_symbol_map")
     def initialize_broker_symbol_map(self):
         self.broker_symbol_map = {}
         for a in self.symbol_data:
@@ -183,7 +182,6 @@ class Broker:
         else:
             self.quantplay_symbol_map = {v: k for k, v in self.broker_symbol_map.items()}
 
-    @timeit(MetricName="Broker:load_instrument")
     def load_instrument(self, file_name: str) -> None:
         try:
             instrument_data_instance = InstrumentData.get_instance()
@@ -663,7 +661,6 @@ class Broker:
             self.cancel_order(order_id)
             time.sleep(0.1)
 
-    @timeit(MetricName="Broker:square_off_all")
     def square_off_all(
         self,
         dry_run: bool = True,
@@ -699,8 +696,8 @@ class Broker:
 
         positions = positions.with_columns(
             pl.when(pl.col("quantity") < 0)
-            .then("BUY")
-            .otherwise("SELL")
+            .then(pl.lit("BUY"))
+            .otherwise(pl.lit("SELL"))
             .alias("transaction_type")
         )
 

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/iifl_xts.py

@@ -1,9 +1,7 @@
 from quantplay.broker.xts import XTS
-from quantplay.utils.constant import timeit
 
 
 class IIFL(XTS):
-    @timeit(MetricName="IIFL:__init__")
     def __init__(
         self,
         api_secret: str | None = None,

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/kite_utils.py

@@ -3,7 +3,7 @@ import time
 import traceback
 
 import pyotp
-from retrying import retry
+from retrying import retry  # type: ignore
 from selenium.common.exceptions import WebDriverException
 
 from quantplay.exception.exceptions import (

{quantplay-2.0.20 → quantplay-2.0.22}/quantplay/broker/motilal.py

@@ -10,7 +10,7 @@ import polars as pl
 
 from quantplay.model.broker import ModifyOrderRequest, UserBrokerProfileResponse
 from quantplay.model.order_event import OrderUpdateEvent
-from quantplay.utils.constant import Constants, timeit, OrderType
+from quantplay.utils.constant import Constants, OrderType
 from quantplay.broker.generics.broker import Broker
 from quantplay.exception.exceptions import InvalidArgumentException
 import pyotp
@@ -22,7 +22,7 @@ from quantplay.exception.exceptions import (
     retry_exception,
     RetryableException,
 )
-from retrying import retry
+from retrying import retry  # type: ignore
 from quantplay.wrapper.aws.s3 import S3Utils
 
 
@@ -52,7 +52,6 @@ class Motilal(Broker):
         "browserversion": "105.0",
     }
 
-    @timeit(MetricName="Motilal:__init__")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -150,15 +149,13 @@ class Motilal(Broker):
         self.nfo_exchange = "NSEFO"
         self.exchange_code_map = {"NFO": "NSEFO", "CDS": "NSECD", "BFO": "BSEFO"}
 
-    @timeit(MetricName="Motilal:load_file_by_url")
     def load_file_by_url(self, exchange):
         data_url = "https://openapi.motilaloswal.com/getscripmastercsv?name={}".format(
             exchange
         )
-        df = pd.read_csv(data_url)
+        df = pd.read_csv(data_url)  # type: ignore
         self.instrument_data_by_exchange[exchange] = df
 
-    @timeit(MetricName="Motilal:load_instrument")
     def load_instrument(self, file_name: str | None = None) -> None:
         try:
             self.symbol_data = InstrumentData.get_instance().load_data(  # type: ignore
@@ -266,7 +263,6 @@ class Motilal(Broker):
         Constants.logger.info(response)
         return response
 
-    @timeit(MetricName="Motilal:get_ltp")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -289,7 +285,6 @@ class Motilal(Broker):
         Constants.logger.info("[GET_LTP_RESPONSE] response {}".format(response.json()))
         return response.json()["data"]["ltp"] / 100.0
 
-    @timeit(MetricName="Motilal:modify_price")
     def modify_price(
         self,
         order_id: str,
@@ -319,7 +314,6 @@ class Motilal(Broker):
         print(order)
         self.modify_order(order)  # type: ignore
 
-    @timeit(MetricName="Motilal:modify_order")
     def modify_order(self, order_to_modify: ModifyOrderRequest) -> str:
         order: dict = copy.deepcopy(order_to_modify)  # type:ignore
         data = {
@@ -356,7 +350,6 @@ class Motilal(Broker):
             Constants.logger.error("{}".format(exception_message))
         return order["order_id"]
 
-    @timeit(MetricName="Motilal:cancel_order")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -401,7 +394,6 @@ class Motilal(Broker):
 
         return response["data"]
 
-    @timeit(MetricName="Motilal:margins")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -426,7 +418,6 @@ class Motilal(Broker):
 
         return margins
 
-    @timeit(MetricName="Motilal:place_order")
     def place_order(
         self,
         tradingsymbol=None,
@@ -557,14 +548,12 @@ class Motilal(Broker):
 
         thread.start_new_thread(run, ())
 
-    @timeit(MetricName="Motilal:account_summary")
     def account_summary(self):
         response = self.margins()
         response["pnl"] = float(self.positions()["pnl"].sum())  # type: ignore
 
         return response
 
-    @timeit(MetricName="Motilal:profile")
     def profile(self) -> UserBrokerProfileResponse:
         api_response = requests.post(
             self.get_profile_url,
@@ -583,7 +572,6 @@ class Motilal(Broker):
 
         return response
 
-    @timeit(MetricName="Motilal:holdings")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -636,7 +624,6 @@ class Motilal(Broker):
 
         return holdings_df[list(self.holidings_schema.keys())].cast(self.holidings_schema)
 
-    @timeit(MetricName="Motilal:positions")
     @retry(
         wait_exponential_multiplier=3000,
         wait_exponential_max=10000,
@@ -711,7 +698,6 @@ class Motilal(Broker):
             self.positions_schema
         )
 
-    @timeit(MetricName="Motilal:orders")
     def orders(self, tag: str | None = None, add_ltp: bool = True) -> pl.DataFrame:
         response = (requests.post(self.order_book_url, headers=self.headers)).json()
         if response["status"] == "ERROR":