wiz-trader 0.9.0__py3-none-any.whl → 0.11.0__py3-none-any.whl

wiz_trader/__init__.py

@@ -3,6 +3,6 @@
 from .quotes import QuotesClient
 from .apis import WizzerClient
 
-__version__ = "0.9.0"
+__version__ = "0.11.0"
 
 __all__ = ["QuotesClient", "WizzerClient"]

wiz_trader/apis/client.py

@@ -23,7 +23,77 @@ class WizzerClient:
     log_level (str): Logging level. Options: "error", "info", "debug".
     strategy_id (str): Default strategy ID to use if not provided in methods.
   """
+  
+  # Constants
+  # Transaction types
+  TRANSACTION_TYPE_BUY = "BUY"
+  TRANSACTION_TYPE_SELL = "SELL"
+
+  # Product types
+  PRODUCT_CNC = "CNC"         # Cash and Carry
+  PRODUCT_MIS = "MIS"         # Margin Intraday Square-off
+  PRODUCT_NRML = "NRML"       # Normal / Overnight Futures and Options
+
+  # Order types
+  ORDER_TYPE_MARKET = "MARKET"
+  ORDER_TYPE_LIMIT = "LIMIT"
+  ORDER_TYPE_SL = "SL"        # Stop Loss
+  ORDER_TYPE_SLM = "SL-M"     # Stop Loss Market
+
+  # Validity types
+  VALIDITY_DAY = "DAY"
+  VALIDITY_IOC = "IOC"        # Immediate or Cancel
+  VALIDITY_GTT = "GTT"        # Good Till Triggered
+
+  # Variety types
+  VARIETY_REGULAR = "REGULAR"
+  VARIETY_AMO = "AMO"         # After Market Order
+  VARIETY_BO = "BO"           # Bracket Order
+  VARIETY_CO = "CO"           # Cover Order
+
+  # Exchanges
+  EXCHANGE_NSE = "NSE"        # National Stock Exchange
+  EXCHANGE_BSE = "BSE"        # Bombay Stock Exchange
+  EXCHANGE_WZR = "WZR"        # Wizzer Exchange (for baskets)
+
+  # Segments
+  SEGMENT_NSE_CM = "NSECM"    # NSE Cash Market
+  SEGMENT_BSE_CM = "BSECM"    # BSE Cash Market
+  SEGMENT_NSE_FO = "NSEFO"    # NSE Futures and Options
+  SEGMENT_WZREQ = "WZREQ"     # Wizzer Basket Segment
 
+  # URIs to various API endpoints
+  _routes = {
+    # Order related endpoints
+    "order.place": "/orders",
+    "order.modify": "/orders/{order_id}",
+    "order.cancel": "/orders/{order_id}",
+    "order.info": "/orders/{order_id}",
+    
+    # Basket order endpoints
+    "basket.order.place": "/orders/basket",
+    "basket.order.exit": "/orders/basket/exit",
+    "basket.order.modify": "/orders/basket/{order_id}",
+    
+    # Portfolio and position management
+    "portfolio.positions": "/portfolios/positions",
+    "portfolio.positions.exit.all": "/portfolios/positions/exit/all",
+    "portfolio.positions.exit.strategy": "/portfolios/positions/exit/strategies/{strategy_id}",
+    "portfolio.holdings": "/portfolios/holdings",
+    
+    # Basket management
+    "basket.create": "/baskets",
+    "basket.list": "/baskets",
+    "basket.info": "/baskets/{basket_id}",
+    "basket.instruments": "/baskets/{basket_id}/instruments",
+    "basket.rebalance": "/baskets/rebalance",
+    
+    # Data hub endpoints
+    "datahub.indices": "/datahub/indices",
+    "datahub.index.components": "/datahub/index/components",
+    "datahub.historical.ohlcv": "/datahub/historical/ohlcv",
+  }
+  
   def __init__(
     self,
     base_url: Optional[str] = None,
@@ -90,7 +160,6 @@ class WizzerClient:
     Returns:
       List[Dict[str, Any]]: List of index information.
     """
-    endpoint = "/datahub/indices"
     params = {}
     
     if trading_symbol:
@@ -99,7 +168,7 @@ class WizzerClient:
       params["exchange"] = exchange
 
     logger.debug("Fetching indices with params: %s", params)
-    response = self._make_request("GET", endpoint, params=params)
+    response = self._make_request("GET", self._routes["datahub.indices"], params=params)
     return response
 
   def get_index_components(self, trading_symbol: str, exchange: str) -> List[Dict[str, Any]]:
@@ -113,14 +182,13 @@ class WizzerClient:
     Returns:
        List[Dict[str, Any]]: List of component stocks in the index.
     """
-    endpoint = "/datahub/index/components"
     params = {
       "tradingSymbol": trading_symbol,
       "exchange": exchange
     }
 
     logger.debug("Fetching index components with params: %s", params)
-    response = self._make_request("GET", endpoint, params=params)
+    response = self._make_request("GET", self._routes["datahub.index.components"], params=params)
     return response
 
   def get_historical_ohlcv(
@@ -144,7 +212,6 @@ class WizzerClient:
     Returns:
       List[Dict[str, Any]]: Historical data for requested instruments.
     """
-    endpoint = "/datahub/historical/ohlcv"
     data = {
       "instruments": instruments,
       "startDate": start_date,
@@ -154,7 +221,7 @@ class WizzerClient:
     }
 
     logger.debug("Fetching historical OHLCV with data: %s", data)
-    response = self._make_request("POST", endpoint, json=data)
+    response = self._make_request("POST", self._routes["datahub.historical.ohlcv"], json=data)
     return response
 
   # ===== ORDER MANAGEMENT METHODS =====
@@ -165,13 +232,13 @@ class WizzerClient:
     trading_symbol: str,
     transaction_type: str,
     quantity: int,
-    order_type: str = "MARKET",
-    product: str = "CNC",
+    order_type: str = None,
+    product: str = None,
     price: float = 0,
     trigger_price: float = 0,
     disclosed_qty: int = 0,
-    validity: str = "DAY",
-    variety: str = "REGULAR",
+    validity: str = None,
+    variety: str = None,
     stoploss: float = 0,
     target: float = 0,
     segment: Optional[str] = None,
@@ -187,13 +254,13 @@ class WizzerClient:
       trading_symbol (str): Symbol of the instrument.
       transaction_type (str): "BUY" or "SELL".
       quantity (int): Number of shares to trade.
-      order_type (str, optional): Order type (e.g., "MARKET", "LIMIT"). Defaults to "MARKET".
-      product (str, optional): Product code (e.g., "CNC" for delivery). Defaults to "CNC".
+      order_type (str, optional): Order type (e.g., "MARKET", "LIMIT"). Defaults to MARKET.
+      product (str, optional): Product code (e.g., "CNC" for delivery). Defaults to CNC.
       price (float, optional): Price for limit orders. Defaults to 0.
       trigger_price (float, optional): Trigger price for stop orders. Defaults to 0.
       disclosed_qty (int, optional): Disclosed quantity. Defaults to 0.
-      validity (str, optional): Order validity (e.g., "DAY", "IOC"). Defaults to "DAY".
-      variety (str, optional): Order variety. Defaults to "REGULAR".
+      validity (str, optional): Order validity (e.g., "DAY", "IOC"). Defaults to DAY.
+      variety (str, optional): Order variety. Defaults to REGULAR.
       stoploss (float, optional): Stop loss price. Defaults to 0.
       target (float, optional): Target price. Defaults to 0.
       segment (Optional[str], optional): Market segment. If None, determined from exchange.
@@ -204,11 +271,27 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Order response containing orderId.
     """
-    endpoint = "/orders"
+    endpoint = self._routes["order.place"]
+    
+    # Set default values from constants if not provided
+    if order_type is None:
+      order_type = self.ORDER_TYPE_MARKET
+    if product is None:
+      product = self.PRODUCT_CNC
+    if validity is None:
+      validity = self.VALIDITY_DAY
+    if variety is None:
+      variety = self.VARIETY_REGULAR
     
     # Determine segment if not provided
     if not segment:
       segment = f"{exchange}CM"
+      # If exchange is NSE, use the NSE_CM constant
+      if exchange == self.EXCHANGE_NSE:
+        segment = self.SEGMENT_NSE_CM
+      # If exchange is BSE, use the BSE_CM constant
+      elif exchange == self.EXCHANGE_BSE:
+        segment = self.SEGMENT_BSE_CM
     
     # Get strategy information
     strategy_info = self._get_strategy(strategy)
@@ -236,7 +319,7 @@ class WizzerClient:
       data["exchangeToken"] = exchange_token
         
     logger.debug("Placing order: %s", data)
-    return self._make_request("POST", endpoint, json=data)
+    return self._make_request("POST", self._routes["order.place"], json=data)
     
   def modify_order(
     self,
@@ -253,7 +336,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Order response containing orderId.
     """
-    endpoint = f"/orders/{order_id}"
+    endpoint = self._routes["order.modify"].format(order_id=order_id)
     
     logger.debug("Modifying order %s with params: %s", order_id, params)
     return self._make_request("PATCH", endpoint, json=params)
@@ -268,7 +351,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Response with the cancelled order ID.
     """
-    endpoint = f"/orders/{order_id}"
+    endpoint = self._routes["order.cancel"].format(order_id=order_id)
     
     logger.debug("Cancelling order: %s", order_id)
     return self._make_request("DELETE", endpoint)
@@ -283,7 +366,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Order details.
     """
-    endpoint = f"/orders/{order_id}"
+    endpoint = self._routes["order.info"].format(order_id=order_id)
     
     logger.debug("Fetching order: %s", order_id)
     return self._make_request("GET", endpoint)
@@ -299,7 +382,7 @@ class WizzerClient:
     Returns:
       List[Dict[str, Any]]: List of positions matching the filter criteria.
     """
-    endpoint = "/portfolios/positions"
+    endpoint = self._routes["portfolio.positions"]
     params = {}
     
     if position_status:
@@ -338,7 +421,7 @@ class WizzerClient:
     Returns:
       List[Dict[str, Any]]: List of holdings.
     """
-    endpoint = "/portfolios/holdings"
+    endpoint = self._routes["portfolio.holdings"]
     params = {"portfolios": portfolios}
     
     logger.debug("Fetching holdings for portfolio: %s", portfolios)
@@ -368,7 +451,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Basket information.
     """
-    endpoint = "/baskets"
+    endpoint = self._routes["basket.create"]
     
     # Set defaults
     if capital is None:
@@ -392,7 +475,7 @@ class WizzerClient:
     Returns:
       List[Dict[str, Any]]: List of baskets.
     """
-    endpoint = "/baskets"
+    endpoint = self._routes["basket.list"]
     
     logger.debug("Fetching baskets")
     return self._make_request("GET", endpoint)
@@ -407,7 +490,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Basket information.
     """
-    endpoint = f"/baskets/{basket_id}"
+    endpoint = self._routes["basket.info"].format(basket_id=basket_id)
     
     logger.debug("Fetching basket: %s", basket_id)
     return self._make_request("GET", endpoint)
@@ -422,7 +505,7 @@ class WizzerClient:
     Returns:
       List[Dict[str, Any]]: List of instruments in the basket.
     """
-    endpoint = f"/baskets/{basket_id}/instruments"
+    endpoint = self._routes["basket.instruments"].format(basket_id=basket_id)
     
     logger.debug("Fetching instruments for basket: %s", basket_id)
     return self._make_request("GET", endpoint)
@@ -433,15 +516,15 @@ class WizzerClient:
     transaction_type: str,
     quantity: float,
     price: float = 0,
-    order_type: str = "MARKET",
-    product: str = "CNC",
-    validity: str = "DAY",
+    order_type: str = None,
+    product: str = None,
+    validity: str = None,
     exchange_token: Optional[int] = None,
     trigger_price: float = 0,
     stoploss: float = 0,
     target: float = 0,
     broker: str = "wizzer",
-    variety: str = "REGULAR",
+    variety: str = None,
     strategy: Optional[Dict[str, str]] = None,
     disclosed_qty: int = 0,
     sl_applied_level: Optional[str] = None
@@ -454,15 +537,15 @@ class WizzerClient:
       transaction_type (str): "BUY" or "SELL".
       quantity (float): Quantity/units of the basket.
       price (float, optional): Price for limit orders. Defaults to 0.
-      order_type (str, optional): Order type. Defaults to "MARKET".
-      product (str, optional): Product code. Defaults to "CNC".
-      validity (str, optional): Order validity. Defaults to "DAY".
+      order_type (str, optional): Order type. Defaults to MARKET.
+      product (str, optional): Product code. Defaults to CNC.
+      validity (str, optional): Order validity. Defaults to DAY.
       exchange_token (Optional[int], optional): Exchange token for the basket.
       trigger_price (float, optional): Trigger price. Defaults to 0.
       stoploss (float, optional): Stop loss price. Defaults to 0.
       target (float, optional): Target price. Defaults to 0.
       broker (str, optional): Broker code. Defaults to "wizzer".
-      variety (str, optional): Order variety. Defaults to "REGULAR".
+      variety (str, optional): Order variety. Defaults to REGULAR.
       strategy (Optional[Dict[str, str]], optional): Strategy information. If None, uses default.
       disclosed_qty (int, optional): Disclosed quantity. Defaults to 0.
       sl_applied_level (Optional[str], optional): Stop loss applied level (e.g., "basket").
@@ -470,14 +553,24 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Order response containing orderId.
     """
-    endpoint = "/orders/basket"
+    endpoint = self._routes["order.basket"]
+    
+    # Set default values from constants if not provided
+    if order_type is None:
+      order_type = self.ORDER_TYPE_MARKET
+    if product is None:
+      product = self.PRODUCT_CNC
+    if validity is None:
+      validity = self.VALIDITY_DAY
+    if variety is None:
+      variety = self.VARIETY_REGULAR
     
     # Get strategy information
     strategy_info = self._get_strategy(strategy)
     
     data = {
       "tradingSymbol": trading_symbol,
-      "exchange": "WZR",
+      "exchange": self.EXCHANGE_WZR,
       "transactionType": transaction_type,
       "qty": quantity,
       "price": price,
@@ -490,7 +583,7 @@ class WizzerClient:
       "broker": broker,
       "variety": variety,
       "strategy": strategy_info,
-      "segment": "WZREQ",
+      "segment": self.SEGMENT_WZREQ,
       "disclosedQty": disclosed_qty
     }
     
@@ -528,7 +621,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Order response containing orderId.
     """
-    endpoint = "/orders/basket/exit"
+    endpoint = self._routes["order.basket_exit"]
     
     # Build base data
     data = {
@@ -546,17 +639,17 @@ class WizzerClient:
         
     # Set defaults if not in kwargs
     defaults = {
-      "orderType": "MARKET",
-      "product": "CNC", 
-      "validity": "DAY",
+      "orderType": self.ORDER_TYPE_MARKET,
+      "product": self.PRODUCT_CNC, 
+      "validity": self.VALIDITY_DAY,
       "disclosedQty": 0,
       "price": 0,
-      "variety": "REGULAR",
+      "variety": self.VARIETY_REGULAR,
       "stoploss": 0,
       "broker": "wizzer",
       "triggerPrice": 0,
       "target": 0,
-      "segment": "WZREQ"
+      "segment": self.SEGMENT_WZREQ
     }
     
     for key, value in defaults.items():
@@ -581,7 +674,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Order response containing orderId.
     """
-    endpoint = f"/orders/basket/{order_id}"
+    endpoint = self._routes["order.basket_modify"].format(order_id=order_id)
     
     logger.debug("Modifying basket order %s with params: %s", order_id, params)
     return self._make_request("PATCH", endpoint, json=params)
@@ -601,7 +694,7 @@ class WizzerClient:
     Returns:
       Dict[str, Any]: Rebalance response.
     """
-    endpoint = "/baskets/rebalance"
+    endpoint = self._routes["basket.rebalance"]
     
     data = {
       "tradingSymbol": trading_symbol,
@@ -610,6 +703,45 @@ class WizzerClient:
     
     logger.debug("Rebalancing basket %s with instruments: %s", trading_symbol, instruments)
     return self._make_request("POST", endpoint, json=data)
+    
+  def exit_all_positions(self) -> Dict[str, Any]:
+    """
+    Exit all positions across all strategies.
+    
+    This method sends a request to close all open positions for the user.
+    
+    Returns:
+      Dict[str, Any]: Response with summary of success and failure counts.
+    """
+    endpoint = self._routes["portfolio.positions_exit_all"]
+    
+    logger.debug("Exiting all positions")
+    return self._make_request("POST", endpoint)
+    
+  def exit_strategy_positions(self, strategy_id: Optional[str] = None) -> Dict[str, Any]:
+    """
+    Exit all positions for a specific strategy.
+    
+    Args:
+      strategy_id (Optional[str]): ID of the strategy to exit positions for.
+        If None, uses the default strategy ID.
+        
+    Returns:
+      Dict[str, Any]: Response with summary of success and failure counts.
+      
+    Raises:
+      ValueError: If no strategy_id is provided and no default is set.
+    """
+    # Get strategy ID (either from parameter or default)
+    if not strategy_id:
+      if not self.strategy_id:
+        raise ValueError("Strategy ID must be provided either as a parameter or set in .env (WZ__STRATEGY_ID)")
+      strategy_id = self.strategy_id
+      
+    endpoint = self._routes["portfolio.positions_exit_strategies"].format(strategy_id=strategy_id)
+    
+    logger.debug("Exiting all positions for strategy: %s", strategy_id)
+    return self._make_request("POST", endpoint)
 
   def _make_request(
     self,