wiz-trader 0.25.0__py3-none-any.whl → 0.26.0__py3-none-any.whl

wiz_trader/__init__.py

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

wiz_trader/apis/__init__.py

@@ -3,4 +3,4 @@
 
 from .client import WizzerClient
 
-__all__ = ["WizzerClient"]
+__all__ = ["WizzerClient"]

wiz_trader/apis/client.py

@@ -197,7 +197,14 @@ class WizzerClient:
     "instrument.metrics": "/instruments/metrics",
     "instrument.option_chain": "/instruments/options/chain",
     "instrument.expiry_list": "/instruments/options/chain/expirylist",
-    "instrument.future_list": "/instruments/futures/list"
+    "instrument.future_list": "/instruments/futures/list",
+
+    # Classification API endpoints
+    "classification.types": "/datahub/classifications/types",
+    "classification.values": "/datahub/classifications/{type}",
+    "indices.filter": "/datahub/indices/filter",
+    "instruments.filter": "/datahub/instruments/filter",
+    "instruments.filter_with_index": "/datahub/indices/{id}/instruments"
   }
   
   def __init__(
@@ -1598,6 +1605,84 @@ class WizzerClient:
     logger.debug("Fetching futures list for: %s", identifier)
     return self._make_request("POST", endpoint, json=data)
 
+
+  def get_classification_types(self) -> list:
+    """
+    Retrieve all available classification types.
+    
+    Returns:
+      list: A list of classification types.
+    """
+    endpoint = self._routes["classification.types"]
+    logger.debug("Fetching classification types.")
+    return self._make_request("GET", endpoint)
+
+  def get_classifications(self, classification_type: str) -> list:
+    """
+    Retrieve all values for a specific classification type.
+    
+    Args:
+      classification_type (str): The type of classification to retrieve values for.
+      
+    Returns:
+      list: A list of classification values.
+    """
+    endpoint = self._routes["classification.values"].format(type=classification_type)
+    logger.debug(f"Fetching classification values for type: {classification_type}")
+    return self._make_request("GET", endpoint)
+
+  def filter_indices(self, filters: Dict[str, List[str]], match_all: bool = True) -> list:
+    """
+    Filter indices based on multiple classification criteria.
+    
+    Args:
+      filters (Dict[str, List[str]]): A dictionary of filters where keys are classification
+                                      types and values are lists of classification values.
+      match_all (bool): If True, performs an AND logic search. If False, performs an OR logic search.
+      
+    Returns:
+      list: A list of indices that match the filter criteria.
+    """
+    endpoint = self._routes["indices.filter"]
+    data = {
+      "filters": filters,
+      "matchAll": match_all
+    }
+    logger.debug(f"Filtering indices with filters: {filters} and match_all: {match_all}")
+    return self._make_request("POST", endpoint, json=data)
+
+  def filter_instruments(
+    self,
+    filters: Optional[Dict[str, List[str]]] = None,
+    match_all: bool = True,
+    index_identifier: Optional[str] = None
+  ) -> list:
+    """
+    Filter instruments based on classification criteria, with an option to filter within an index.
+    
+    Args:
+      filters (Optional[Dict[str, List[str]]]): A dictionary of filters where keys are classification
+                                      types and values are lists of classification values. Can be None.
+      match_all (bool): If True, performs an AND logic search. If False, performs an OR logic search.
+      index_identifier (Optional[str]): If provided, filters instruments within a specific index.
+      
+    Returns:
+      list: A list of instruments that match the filter criteria.
+    """
+    data = {
+        "filters": filters or {},
+        "matchAll": match_all
+    }
+
+    if index_identifier:
+        endpoint = self._routes["instruments.filter_with_index"].format(id=index_identifier)
+        logger.debug(f"Filtering instruments for index: {index_identifier} with filters: {filters} and match_all: {match_all}")
+    else:
+        endpoint = self._routes["instruments.filter"]
+        logger.debug(f"Filtering instruments with filters: {filters} and match_all: {match_all}")
+        
+    return self._make_request("POST", endpoint, json=data)
+
   def _make_request(
     self, 
     method: str, 
@@ -1640,4 +1725,4 @@ class WizzerClient:
       logger.error("API request failed: %s", e, exc_info=True)
       if hasattr(e.response, 'text'):
         logger.error("Response content: %s", e.response.text)
-      raise
+      raise

wiz_trader/quotes/client.py

@@ -7,7 +7,6 @@ from typing import Callable, List, Optional, Any, Iterator
 
 import websockets
 from websockets.exceptions import ConnectionClosed
-from websockets.protocol import State
 
 # Setup module‐level logger with a default handler if none exists.
 logger = logging.getLogger(__name__)
@@ -163,7 +162,7 @@ class QuotesClient:
     # -- Async core methods (for internal use) --
 
     async def _subscribe_async(self, instruments: List[str]) -> None:
-        if self.ws and self.ws.state == State.OPEN:
+        if self.ws and self.ws.open:
             new = set(instruments) - self.subscribed_instruments
             if new:
                 self.subscribed_instruments |= new
@@ -178,7 +177,7 @@ class QuotesClient:
             self.subscribed_instruments |= set(instruments)
 
     async def _unsubscribe_async(self, instruments: List[str]) -> None:
-        if self.ws and self.ws.state == State.OPEN:
+        if self.ws and self.ws.open:
             to_remove = set(instruments) & self.subscribed_instruments
             if to_remove:
                 self.subscribed_instruments -= to_remove

{wiz_trader-0.25.0.dist-info → wiz_trader-0.26.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wiz_trader
-Version: 0.25.0
+Version: 0.26.0
 Summary: A Python SDK for connecting to the Wizzer.
 Home-page: https://bitbucket.org/wizzer-tech/quotes_sdk.git
 Author: Pawan Wagh
@@ -59,7 +59,7 @@ Dynamic: requires-python
 WizTrader is a Python SDK for connecting to the Wizzer trading platform. It provides two main client classes:
 
 - **QuotesClient**: For real-time market data streaming via WebSockets
-- **WizzerClient**: For REST API access to trading, order management, and market data
+- **WizzerClient**: For REST API access to trading, order management, and market data, including data pipelines and classification APIs.
 
 This documentation covers all the ways to use the SDK, with examples for each endpoint and common use cases.
 
@@ -403,6 +403,73 @@ for component in components[:5]:
     print(f"{component['tradingSymbol']} - {component.get('weightage', 'N/A')}%")
 ```
 
+#### List Classification Types
+
+Retrieve all available classification types.
+
+```python
+classification_types = client.get_classification_types()
+# Response: ['basicIndustry', 'industry', 'macroEconomicSector', 'sector']
+```
+
+#### List Classification Values
+
+Retrieve all values for a specific classification type.
+
+```python
+basic_industry_values = client.get_classifications("basicIndustry")
+# Response: ['Aerospace & Defense', 'Agricultural Food & other Products', ...]
+```
+
+#### Filter Indices by Classifications
+
+Filter indices based on multiple classification criteria. The `match_all` parameter controls the filter logic:
+- `True`: AND logic (all filters must match)
+- `False`: OR logic (any filter can match)
+
+```python
+# Example: Find indices in the 'Commodities' macro-economic sector OR 'Aerospace & Defense' industry
+filters = {
+    "macroEconomicSector": ["Commodities"],
+    "industry": ["Aerospace & Defense"]
+}
+or_logic_indices = client.filter_indices(filters=filters, match_all=False)
+
+# Example: Find indices that are in BOTH 'Housing Finance Company' and 'Financial Services'
+filters = {
+    "basicIndustry": ["Housing Finance Company"],
+    "sector": ["Financial Services"]
+}
+and_logic_indices = client.filter_indices(filters=filters, match_all=True)
+```
+
+#### Filter Instruments by Classifications
+
+Filter instruments based on classification criteria.
+
+```python
+# Example: Find instruments in the 'Technology' sector
+filters = {"sector": ["Technology"]}
+tech_instruments = client.filter_instruments(filters=filters, match_all=True)
+```
+
+#### Filter Instruments by Classifications and Index
+
+Filter instruments that belong to both specific classifications and a given index.
+
+```python
+# Example: Find 'Financial Services' instruments within the 'NIFTY NEXT 50' index
+filters = {
+    "basicIndustry": ["Housing Finance Company"],
+    "sector": ["Financial Services"]
+}
+filtered_instruments = client.filter_instruments(
+    index_identifier="NSE:NIFTY NEXT 50:26054",
+    filters=filters,
+    match_all=False
+)
+```
+
 #### Get Historical OHLCV Data
 ## Overview
 

wiz_trader-0.26.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+wiz_trader/__init__.py,sha256=J8tpFwWCVMnCGEwHmeOLFzA3KvKEF3lNyScNCjZPhZ8,183
+wiz_trader/apis/__init__.py,sha256=6sUr1nzmplNdld0zryMrQSt0jHT2GhOiFYgKKVHzk8U,133
+wiz_trader/apis/client.py,sha256=VEotcYfPkjmpp2seJtTWyEdppa6q9NecX89iWxnFQf0,63154
+wiz_trader/quotes/__init__.py,sha256=RF9g9CNP6bVWlmCh_ad8krm3-EWOIuVfLp0-H9fAeEM,108
+wiz_trader/quotes/client.py,sha256=EkRxCudEq0J6ReP5nBgDzrL1XbtgWhrzpZT27xXBMBU,10891
+wiz_trader-0.26.0.dist-info/METADATA,sha256=zlzua-zw8mkNRa5SwnWQLZ-SyIYSPlowcJzTKwKFhIM,89856
+wiz_trader-0.26.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wiz_trader-0.26.0.dist-info/top_level.txt,sha256=lnYS_g8LlA6ryKYnvY8xIQ6K2K-xzOsd-99AWgnW6VY,11
+wiz_trader-0.26.0.dist-info/RECORD,,

wiz_trader-0.25.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-wiz_trader/__init__.py,sha256=0pjCUUIWdupdbX9Rk7OcwJlW287LB1OvtkOFJES4m00,182
-wiz_trader/apis/__init__.py,sha256=ItWKMOl4omiW0g2f-M7WRW3v-dss_ULd9vYnFyIIT9o,132
-wiz_trader/apis/client.py,sha256=GY1aAaV4ia1tnFnB2qaNqnv-qeUvkVlvw9xOKN54qIs,59786
-wiz_trader/quotes/__init__.py,sha256=RF9g9CNP6bVWlmCh_ad8krm3-EWOIuVfLp0-H9fAeEM,108
-wiz_trader/quotes/client.py,sha256=oPDOKqc9EChID_V0_O7ziCdcDnyM_jC7uuKfkc1GwmI,10959
-wiz_trader-0.25.0.dist-info/METADATA,sha256=SaChOGUc2nsJ7rno-ENsd5-VYI1XrKOSHHJOLfAJhM0,87770
-wiz_trader-0.25.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wiz_trader-0.25.0.dist-info/top_level.txt,sha256=lnYS_g8LlA6ryKYnvY8xIQ6K2K-xzOsd-99AWgnW6VY,11
-wiz_trader-0.25.0.dist-info/RECORD,,