wiz-trader 0.15.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

wiz_trader/__init__.py

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

wiz_trader/apis/client.py

@@ -1162,36 +1162,69 @@ class WizzerClient:
     return self._make_request("PATCH", endpoint, json=params)
     
   def rebalance_basket(
-    self,
-    trading_symbol: str,
-    instruments: List[str],
-    execution_policy: str
-  ) -> Dict[str, Any]:
-    """
-    Rebalance a basket with new instruments.
-    
-    Args:
-      trading_symbol (str): Basket trading symbol.
-      instruments (List[str]): List of instrument identifiers for the new basket composition.
-      execution_policy (str): Rebalance execution policy. 
-          Options: "full_rebalance", "entry_only", "exit_only".
+        self,
+        trading_symbol: str,
+        instruments: List[str],
+        execution_policy: str,
+        order_type: str = None,
+        product: str = None
+    ) -> Dict[str, Any]:
+        """
+        Rebalance a basket with new instruments.
         
-    Returns:
-      Dict[str, Any]: Rebalance response.
-    """
-    endpoint = self._routes["basket.rebalance"]
-    
-    data = {
-      "tradingSymbol": trading_symbol,
-      "instruments": instruments,
-      "policies": {
-        "execution": execution_policy
-      }
-    }
-    
-    logger.debug("Rebalancing basket %s with instruments: %s, policy: %s", 
-                 trading_symbol, instruments, execution_policy)
-    return self._make_request("POST", endpoint, json=data)
+        Args:
+          trading_symbol (str): Basket trading symbol.
+          instruments (List[str]): List of instrument identifiers for the new basket composition.
+          execution_policy (str): Rebalance execution policy.
+              Options: REBALANCE_FULL, REBALANCE_ENTRY_ONLY, REBALANCE_EXIT_ONLY.
+          order_type (str, optional): Order type to use for rebalance orders.
+              Options: ORDER_TYPE_MARKET, ORDER_TYPE_LIMIT
+          product (str, optional): Product type to use for rebalance orders.
+              Options: PRODUCT_CNC, PRODUCT_MIS
+            
+        Returns:
+          Dict[str, Any]: Rebalance response.
+        """
+        endpoint = "/baskets/rebalance"
+        
+        # Validate execution policy
+        valid_execution_policies = [self.REBALANCE_FULL, self.REBALANCE_ENTRY_ONLY, self.REBALANCE_EXIT_ONLY]
+        if execution_policy not in valid_execution_policies:
+            raise ValueError(f"execution_policy must be one of {valid_execution_policies}")
+        
+        # Build the basic request payload
+        data = {
+            "tradingSymbol": trading_symbol,
+            "instruments": instruments,
+            "policies": {
+                "execution": execution_policy
+            }
+        }
+        
+        # Add order policies if specified
+        if order_type or product:
+            order_policies = {}
+            
+            if order_type:
+                valid_order_types = [self.ORDER_TYPE_MARKET, self.ORDER_TYPE_LIMIT]
+                if order_type not in valid_order_types:
+                    raise ValueError(f"order_type must be one of {valid_order_types}")
+                order_policies["orderType"] = order_type
+                
+            if product:
+                valid_products = [self.PRODUCT_CNC, self.PRODUCT_MIS, self.PRODUCT_NRML]
+                if product not in valid_products:
+                    raise ValueError(f"product must be one of {valid_products}")
+                order_policies["product"] = product
+            
+            if order_policies:
+                data["policies"]["orders"] = order_policies
+        
+        logger.debug("Rebalancing basket %s with instruments: %s, policy: %s, order settings: %s", 
+                    trading_symbol, instruments, execution_policy, 
+                    {"order_type": order_type, "product": product})
+        
+        return self._make_request("POST", endpoint, json=data)
     
   def exit_all_positions(self) -> Dict[str, Any]:
     """

{wiz_trader-0.15.0.dist-info → wiz_trader-0.16.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wiz_trader
-Version: 0.15.0
+Version: 0.16.0
 Summary: A Python SDK for connecting to the Wizzer.
 Home-page: https://bitbucket.org/wizzer-tech/quotes_sdk.git
 Author: Pawan Wagh

wiz_trader-0.16.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+wiz_trader/__init__.py,sha256=ScYHyaz49WDqsMZQVcJLXExQJp_ZOdSeqRvLIiAS-lI,182
+wiz_trader/apis/__init__.py,sha256=ItWKMOl4omiW0g2f-M7WRW3v-dss_ULd9vYnFyIIT9o,132
+wiz_trader/apis/client.py,sha256=JDw0lMCw79LSU9E0YRHt_2FCAzZTZddweB5EsJChTb4,47054
+wiz_trader/quotes/__init__.py,sha256=RF9g9CNP6bVWlmCh_ad8krm3-EWOIuVfLp0-H9fAeEM,108
+wiz_trader/quotes/client.py,sha256=LJeMcQPjJIRxrTIGalWsLYh_XfinDXBP5-4cNS7qCxc,9709
+wiz_trader-0.16.0.dist-info/METADATA,sha256=Xwzf-z6y-ovu7lq-c5K4tlkiyWafGTzUxwgDKJqrJTk,54024
+wiz_trader-0.16.0.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+wiz_trader-0.16.0.dist-info/top_level.txt,sha256=lnYS_g8LlA6ryKYnvY8xIQ6K2K-xzOsd-99AWgnW6VY,11
+wiz_trader-0.16.0.dist-info/RECORD,,

{wiz_trader-0.15.0.dist-info → wiz_trader-0.16.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.4.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

wiz_trader-0.15.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-wiz_trader/__init__.py,sha256=3fwdVbFYiQATRbAy1qYqyHPOW1Z0Nw56KDb1zeKEB14,182
-wiz_trader/apis/__init__.py,sha256=ItWKMOl4omiW0g2f-M7WRW3v-dss_ULd9vYnFyIIT9o,132
-wiz_trader/apis/client.py,sha256=HoMFXfWoO3HdAwPCfOD2_U8nfZJ6iBCB6FnWxzlzl5E,45295
-wiz_trader/quotes/__init__.py,sha256=RF9g9CNP6bVWlmCh_ad8krm3-EWOIuVfLp0-H9fAeEM,108
-wiz_trader/quotes/client.py,sha256=LJeMcQPjJIRxrTIGalWsLYh_XfinDXBP5-4cNS7qCxc,9709
-wiz_trader-0.15.0.dist-info/METADATA,sha256=imCNht3nXlWeViNIFt0_sPyuUv5criLgusNjjkrvUhU,54024
-wiz_trader-0.15.0.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-wiz_trader-0.15.0.dist-info/top_level.txt,sha256=lnYS_g8LlA6ryKYnvY8xIQ6K2K-xzOsd-99AWgnW6VY,11
-wiz_trader-0.15.0.dist-info/RECORD,,