tradx 0.1.1.2__tar.gz → 0.1.2__tar.gz

{tradx-0.1.1.2 → tradx-0.1.2}/.gitignore

@@ -12,4 +12,5 @@ wheels/
 
 # User-generated files
 .env
-uv.lock
+uv.lock
+uv.lock

{tradx-0.1.1.2 → tradx-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tradx
-Version: 0.1.1.2
+Version: 0.1.2
 Summary: A Package Designed to simplify strategy development on package xts-api-client
 Author-email: "jatin.kumawat" <jatin.kumawat@rmoneyindia.com>
 Requires-Python: >=3.12
@@ -8,7 +8,7 @@ Requires-Dist: aiohttp==3.11.11
 Requires-Dist: pydantic==2.10.4
 Requires-Dist: pytest==8.3.4
 Requires-Dist: shortuuid==1.0.13
-Requires-Dist: xts-api-client==0.1.6
+Requires-Dist: xts-api-client>=0.1.7
 Description-Content-Type: text/markdown
 
 # Changelog

{tradx-0.1.1.2 → tradx-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tradx"
-version = "0.1.1.2"
+version = "0.1.2"
 description = "A Package Designed to simplify strategy development on package xts-api-client"
 readme = "README.md"
 authors = [
@@ -12,7 +12,7 @@ dependencies = [
     "pydantic==2.10.4",
     "pytest==8.3.4",
     "shortuuid==1.0.13",
-    "xts-api-client==0.1.6",
+    "xts-api-client>=0.1.7",
 ]
 
 [build-system]

{tradx-0.1.1.2 → tradx-0.1.2}/src/tradx/baseClass/baseAlgo.py

@@ -1,17 +1,20 @@
 from abc import ABC, abstractmethod
 import shortuuid
-from typing import Any, TYPE_CHECKING, List
+from typing import Any, TYPE_CHECKING, List, Dict
+from datetime import datetime, timedelta
 from tradx.baseClass.order import Order
 from tradx.baseClass.position import Position
+
 import asyncio
 
 
 if TYPE_CHECKING:
-    from tradx import interactiveEngine, marketDataEngine
+    from tradx.marketDataEngine import marketDataEngine
+    from tradx.interactiveEngine import interactiveEngine
+    from tradx.baseClass.touchLineData import TouchLineData
     from tradx.baseClass.orderEvent import OrderEvent
     from tradx.baseClass.tradeEvent import TradeEvent
     from tradx.baseClass.candleData import CandleData
-    from tradx.baseClass.touchLineData import TouchLineData
 
 
 class BaseAlgo(ABC):
@@ -71,7 +74,6 @@ class BaseAlgo(ABC):
         self.order_nos = shortuuid
         self.position_diary: List[Position] = []
         self.order_diary: List[Order] = []
-
         # Registering inside interactive engine
         self.name = interactiveEngine.shortuuid.ShortUUID().random(length=4)
         self.interactiveEngine.strategy_to_id[self.name] = self
@@ -157,7 +159,16 @@ class BaseAlgo(ABC):
         else:
             # Insert the new order by creating an object
             new_order = Order(
-                order.OrderUniqueIdentifier, order.AppOrderID, order.OrderStatus
+                order.OrderUniqueIdentifier,
+                order.AppOrderID,
+                order.ProductType,
+                order.OrderType,
+                order.OrderQuantity,
+                order.OrderDisclosedQuantity,
+                order.OrderPrice,
+                order.OrderStopPrice,
+                order.OrderStatus,
+                order.OrderSide,
             )
             self.order_diary.append(new_order)
             if self.interactiveEngine.user_logger:
@@ -305,7 +316,7 @@ class BaseAlgo(ABC):
 
         qty = 0
         for position in self.position_diary:
-            qty += abs(position.Quantity)
+            qty += position.Quantity
         return qty != 0
 
     async def liquidateIntradayDummy(self) -> None:
@@ -327,7 +338,7 @@ class BaseAlgo(ABC):
             if position.Quantity != 0:
                 _order_no = self.order_no()
                 asyncio.ensure_future(
-                    self.marketDataEngine.dummy_market_order(
+                    self.marketDataEngine.dummy_order(
                         position.ExchangeSegment,
                         position.ExchangeInstrumentID,
                         position.ProductType,
@@ -336,3 +347,122 @@ class BaseAlgo(ABC):
                         self,
                     )
                 )
+
+    async def safe_market_order(
+        self, ExchangeSegment, ExchangeInstrumentID, ProductType, Quantity
+    ):
+        """
+        Places a safe market order by continuously modifying the order until it is filled.
+        Args:
+            ExchangeSegment (str): The segment of the exchange.
+            ExchangeInstrumentID (int): The instrument ID of the exchange.
+            ProductType (str): The type of the product.
+            Quantity (int): The quantity to be ordered. Positive for buy, negative for sell.
+        Returns:
+            None
+        Raises:
+            Exception: If there is an issue with fetching market data or placing/modifying the order.
+        This method performs the following steps:
+        1. Generates a unique identifier for the order.
+        2. Fetches the latest market data for the given instrument.
+        3. Places a limit order with a slight price adjustment.
+        4. Continuously checks the order status until it is filled.
+        5. If the order is not filled, modifies the order with updated market data.
+        """
+        OrderUniqueIdentifier = self.order_no()
+        Data: TouchLineData = await self.marketDataEngine.fetch_ltp(
+            [
+                {
+                    "exchangeSegment": ExchangeSegment,
+                    "exchangeInstrumentID": ExchangeInstrumentID,
+                }
+            ]
+        )[0]
+        await self.interactiveEngine.limit_order(
+            ExchangeSegment,
+            ExchangeInstrumentID,
+            ProductType,
+            Quantity,
+            (Data.AskInfo.Price + 0.05 if Quantity > 0 else Data.BidInfo.Price - 0.05),
+            OrderUniqueIdentifier,
+        )
+
+        await asyncio.sleep(0.1)
+        order = next(
+            (
+                O
+                for O in self.order_diary
+                if O.OrderUniqueIdentifier == OrderUniqueIdentifier
+            ),
+            None,
+        )
+        while order is None or (order is not None and order.OrderStatus != "Filled"):
+            await asyncio.sleep(0.1)
+            order = next(
+                (
+                    O
+                    for O in self.order_diary
+                    if O.OrderUniqueIdentifier == OrderUniqueIdentifier
+                ),
+                None,
+            )
+            if order is not None and order.OrderStatus != "Filled":
+                Data: TouchLineData = await self.marketDataEngine.fetch_ltp(
+                    [
+                        {
+                            "exchangeSegment": ExchangeSegment,
+                            "exchangeInstrumentID": ExchangeInstrumentID,
+                        }
+                    ]
+                )[0]
+                await self.interactiveEngine.xt.modify_order(
+                    order.AppOrderID,
+                    order.ProductType,
+                    order.OrderType,
+                    order.OrderQuantity,
+                    order.OrderDisclosedQuantity,
+                    (
+                        Data.AskInfo.Price + 0.05
+                        if Quantity > 0
+                        else Data.BidInfo.Price - 0.05
+                    ),
+                    order.OrderStopPrice,
+                    order.TimeInForce,
+                    order.OrderUniqueIdentifier,
+                )
+                await asyncio.sleep(0.3)
+
+    async def safeLiquidateIntraday(self) -> None:
+        """
+        Asynchronously liquidates all intraday positions and cancels all open orders.
+        This method performs the following actions:
+        1. Logs the action of canceling open orders and squaring off positions if a user logger is available.
+        2. Iterates through the order diary and cancels any orders with a status of "PendingNew" or "New".
+        3. Iterates through the position diary and places market orders to square off any positions with a non-zero quantity.
+        Returns:
+            None
+        """
+
+        if self.interactiveEngine.user_logger:
+            self.interactiveEngine.user_logger.info(
+                f"Cancel open order and square off position for strategy {self.name}",
+                caller=f"{self.__class__.__name__}.safeLiquidateIntraday",
+            )
+
+        for order in self.order_diary:
+            if order.OrderStatus in ["PendingNew", "New"]:
+                asyncio.ensure_future(
+                    self.interactiveEngine.cancel_order(
+                        order.AppOrderID, order.OrderUniqueIdentifier
+                    )
+                )
+        for position in self.position_diary:
+            if position.Quantity != 0:
+                asyncio.ensure_future(
+                    self.safe_market_order(
+                        position.ExchangeSegment,
+                        position.ExchangeInstrumentID,
+                        position.ProductType,
+                        -1 * position.Quantity,
+                    )
+                )

tradx-0.1.2/src/tradx/baseClass/order.py

@@ -0,0 +1,53 @@
+from pydantic import BaseModel
+from decimal import Decimal
+
+
+class Order(BaseModel):
+    """
+    Order class representing an order in the system.
+    Attributes:
+        OrderUniqueIdentifier (str): A unique identifier for the order.
+        AppOrderID (int): The application-specific order ID.
+        OrderStatus (str): The status of the order. Defaults to an empty string.
+    Methods:
+        __init__(OrderUniqueIdentifier: str, AppOrderID: int, OrderStatus: str = ""):
+            Initializes a new instance of the Order class.
+    """
+
+    OrderUniqueIdentifier: str
+    AppOrderID: int
+    OrderStatus: str
+    ProductType: str
+    OrderType: str
+    OrderQuantity: int
+    OrderDisclosedQuantity: int
+    OrderPrice: Decimal
+    OrderStopPrice: Decimal
+    TimeInForce: str
+    OrderSide: str
+
+    def __init__(
+        self,
+        OrderUniqueIdentifier: str,
+        AppOrderID: int,
+        ProductType: str,
+        OrderType: str,
+        OrderQuantity: int,
+        OrderDisclosedQuantity: int,
+        OrderPrice: Decimal,
+        OrderStopPrice: Decimal,
+        OrderSide: str,
+        OrderStatus: str = "",
+    ):
+        super().__init__(
+            OrderUniqueIdentifier=OrderUniqueIdentifier,
+            AppOrderID=AppOrderID,
+            ProductType=ProductType,
+            OrderType=OrderType,
+            OrderQuantity=OrderQuantity,
+            OrderDisclosedQuantity=OrderDisclosedQuantity,
+            OrderPrice=OrderPrice,
+            OrderStopPrice=OrderStopPrice,
+            OrderStatus=OrderStatus,
+            OrderSide=OrderSide,
+        )

tradx-0.1.1.2/src/tradx/baseClass/order.py

@@ -1,27 +0,0 @@
-from pydantic import BaseModel
-
-
-class Order(BaseModel):
-    """
-    Order class representing an order in the system.
-    Attributes:
-        OrderUniqueIdentifier (str): A unique identifier for the order.
-        AppOrderID (int): The application-specific order ID.
-        OrderStatus (str): The status of the order. Defaults to an empty string.
-    Methods:
-        __init__(OrderUniqueIdentifier: str, AppOrderID: int, OrderStatus: str = ""):
-            Initializes a new instance of the Order class.
-    """
-
-    OrderUniqueIdentifier: str
-    AppOrderID: int
-    OrderStatus: str
-
-    def __init__(
-        self, OrderUniqueIdentifier: str, AppOrderID: int, OrderStatus: str = ""
-    ):
-        super().__init__(
-            OrderUniqueIdentifier=OrderUniqueIdentifier,
-            AppOrderID=AppOrderID,
-            OrderStatus=OrderStatus,
-        )