Skip to content

Commit fca13d2

Browse files
committed
Address comment
1 parent 9527b55 commit fca13d2

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

hyperliquid/info.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -598,22 +598,24 @@ def query_sub_accounts(self, user: str) -> Any:
598598
def query_user_to_multi_sig_signers(self, multi_sig_user: str) -> Any:
599599
return self.post("/info", {"type": "userToMultiSigSigners", "user": multi_sig_user})
600600

601-
def subscribe(self, subscription: Subscription, callback: Callable[[Any], None]) -> int:
601+
def _validate_subscription(self, subscription: Subscription) -> None:
602602
if (
603603
subscription["type"] == "l2Book"
604604
or subscription["type"] == "trades"
605605
or subscription["type"] == "candle"
606606
or subscription["type"] == "bbo"
607607
):
608608
subscription["coin"] = self.name_to_coin[subscription["coin"]]
609+
610+
def subscribe(self, subscription: Subscription, callback: Callable[[Any], None]) -> int:
611+
self._validate_subscription(subscription)
609612
if self.ws_manager is None:
610613
raise RuntimeError("Cannot call subscribe since skip_ws was used")
611614
else:
612615
return self.ws_manager.subscribe(subscription, callback)
613616

614617
def unsubscribe(self, subscription: Subscription, subscription_id: int) -> bool:
615-
if subscription["type"] == "l2Book" or subscription["type"] == "trades" or subscription["type"] == "candle":
616-
subscription["coin"] = self.name_to_coin[subscription["coin"]]
618+
self._validate_subscription(subscription)
617619
if self.ws_manager is None:
618620
raise RuntimeError("Cannot call unsubscribe since skip_ws was used")
619621
else:

0 commit comments

Comments
 (0)