wiz-trader 0.21.0__tar.gz → 0.22.0__tar.gz

{wiz_trader-0.21.0/src/wiz_trader.egg-info → wiz_trader-0.22.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wiz_trader
-Version: 0.21.0
+Version: 0.22.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.21.0 → wiz_trader-0.22.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wiz_trader"
-version = "0.21.0"
+version = "0.22.0"
 description = "A Python SDK for connecting to the Wizzer."
 readme = "README.md"
 authors = [

{wiz_trader-0.21.0 → wiz_trader-0.22.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
   name='wiz_trader',
-  version='0.21.0',
+  version='0.22.0',
   description='A Python SDK for connecting to the Wizzer.',
   long_description=open('README.md').read() if open('README.md') else "",
   long_description_content_type='text/markdown',

{wiz_trader-0.21.0 → wiz_trader-0.22.0}/src/wiz_trader/__init__.py

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

{wiz_trader-0.21.0 → wiz_trader-0.22.0}/src/wiz_trader/quotes/client.py

@@ -193,6 +193,38 @@ class QuotesClient:
             asyncio.set_event_loop(loop)
         loop.create_task(self._subscribe_async(instruments))
 
+    async def _unsubscribe_async(self, instruments: List[str]) -> None:
+        """
+        Internal async unsubscription. Use `unsubscribe()` wrapper to schedule this.
+        """
+        # Only send unsubs if the socket is open
+        if self.ws and self.ws.state == State.OPEN:
+            to_remove = set(instruments) & self.subscribed_instruments
+            if to_remove:
+                for batch in self._chunk_list(list(to_remove), self.batch_size):
+                    logger.info("Unsubscribing from %d instruments", len(batch))
+                    await self.ws.send(json.dumps({
+                        "action": self.ACTION_UNSUBSCRIBE,
+                        "instruments": batch
+                    }))
+                    await asyncio.sleep(0.1)
+                # remove them from our local set
+                self.subscribed_instruments -= to_remove
+        else:
+            # if we're not connected yet, just remove from the queue
+            self.subscribed_instruments -= set(instruments)
+
+    def unsubscribe(self, instruments: List[str]) -> None:
+        """
+        Schedule an async unsubscribe so users can call this without 'await'.
+        """
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+        loop.create_task(self._unsubscribe_async(instruments))
+    
     # You could add a similar unsubscribe wrapper if needed
 
     async def close(self) -> None:

{wiz_trader-0.21.0 → wiz_trader-0.22.0/src/wiz_trader.egg-info}/PKG-INFO

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