@@ -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