Merge pull request #97 from InjectiveLabs/f/fix_stream_markets

feat: add market_ids as optional param in market stream

Author: achilleas-kal

pyinjective/async_client.py

@@ -353,8 +353,9 @@ async def get_spot_markets(self, **kwargs):
         )
         return await self.stubSpotExchange.Markets(req)
 
-    async def stream_spot_markets(self):
-        req = spot_exchange_rpc_pb.StreamMarketsRequest()
+    async def stream_spot_markets(self, **kwargs):
+        req = spot_exchange_rpc_pb.StreamMarketsRequest(
+            market_ids=kwargs.get("market_ids"))
         return self.stubSpotExchange.StreamMarkets(req)
 
     async def get_spot_orderbook(self, market_id: str):
@@ -441,8 +442,9 @@ async def get_derivative_markets(self, **kwargs):
         )
         return await self.stubDerivativeExchange.Markets(req)
 
-    async def stream_derivative_markets(self):
-        req = derivative_exchange_rpc_pb.StreamMarketRequest()
+    async def stream_derivative_markets(self, **kwargs):
+        req = derivative_exchange_rpc_pb.StreamMarketRequest(
+            market_ids=kwargs.get("market_ids"))
         return self.stubDerivativeExchange.StreamMarket(req)
 
     async def get_derivative_orderbook(self, market_id: str):

pyinjective/client.py

@@ -381,8 +381,9 @@ def get_spot_markets(self, **kwargs):
         )
         return self.stubSpotExchange.Markets(req)
 
-    def stream_spot_markets(self):
-        req = spot_exchange_rpc_pb.StreamMarketsRequest()
+    def stream_spot_markets(self, **kwargs):
+        req = spot_exchange_rpc_pb.StreamMarketsRequest(
+            market_ids=kwargs.get("market_ids"))
         metadata = self.get_cookie(type="exchange")
         res = self.stubSpotExchange.StreamMarkets(req, metadata=metadata)
         self.set_cookie(res,type="exchange")
@@ -484,8 +485,9 @@ def get_derivative_markets(self, **kwargs):
         )
         return self.stubDerivativeExchange.Markets(req)
 
-    def stream_derivative_markets(self):
-        req = derivative_exchange_rpc_pb.StreamMarketRequest()
+    def stream_derivative_markets(self, **kwargs):
+        req = derivative_exchange_rpc_pb.StreamMarketRequest(
+            market_ids=kwargs.get("market_ids"))
         metadata = self.get_cookie(type="exchange")
         res = self.stubDerivativeExchange.StreamMarket(req, metadata=metadata)
         self.set_cookie(res,type="exchange")