Fix other mypy errors and warnings.

This commit is contained in:
Davide Casale
2022-12-16 16:03:28 +01:00
parent 2595b8a760
commit 0e4cbd40a6
2 changed files with 5 additions and 3 deletions

View File

@@ -1,5 +1,7 @@
from .websocket import BfxWebsocketClient
from typing import Optional
from enum import Enum
class Constants(str, Enum):
@@ -10,7 +12,7 @@ class Constants(str, Enum):
PUB_WSS_HOST = "wss://api-pub.bitfinex.com/ws/2"
class Client(object):
def __init__(self, WSS_HOST: str = Constants.WSS_HOST, API_KEY: str = None, API_SECRET: str = None, log_level: str = "WARNING"):
def __init__(self, WSS_HOST: str = Constants.WSS_HOST, API_KEY: Optional[str] = None, API_SECRET: Optional[str] = None, log_level: str = "WARNING"):
self.wss = BfxWebsocketClient(
host=WSS_HOST,
API_KEY=API_KEY,

View File

@@ -84,7 +84,7 @@ class _RestPublicEndpoints(_Requests):
def t_tickers(self, pairs: Union[List[str], Literal["ALL"]]) -> List[TradingPairTicker]:
if isinstance(pairs, str) and pairs == "ALL":
return [ subdata for subdata in self.tickers([ "ALL" ]) if subdata["SYMBOL"].startswith("t") ]
return [ cast(TradingPairTicker, subdata) for subdata in self.tickers([ "ALL" ]) if cast(str, subdata["SYMBOL"]).startswith("t") ]
data = self.tickers([ "t" + pair for pair in pairs ])
@@ -92,7 +92,7 @@ class _RestPublicEndpoints(_Requests):
def f_tickers(self, currencies: Union[List[str], Literal["ALL"]]) -> List[FundingCurrencyTicker]:
if isinstance(currencies, str) and currencies == "ALL":
return [ subdata for subdata in self.tickers([ "ALL" ]) if subdata["SYMBOL"].startswith("f") ]
return [ cast(FundingCurrencyTicker, subdata) for subdata in self.tickers([ "ALL" ]) if cast(str, subdata["SYMBOL"]).startswith("f") ]
data = self.tickers([ "f" + currency for currency in currencies ])