Add and implement new IncompleteCredentialError in bfxapi.client.

This commit is contained in:
Davide Casale
2023-06-23 17:27:25 +02:00
parent 755ee767a8
commit faffb7fe82
3 changed files with 32 additions and 9 deletions

View File

@@ -1,11 +1,18 @@
from typing import List, Literal, Optional
from typing import \
TYPE_CHECKING, TypedDict, List, Literal, Optional
from bfxapi._utils.logger import ColorLogger
from bfxapi.exceptions import IncompleteCredentialError
from bfxapi.rest import BfxRestInterface
from bfxapi.websocket import BfxWebSocketClient
from bfxapi.urls import REST_HOST, WSS_HOST
if TYPE_CHECKING:
_Credentials = TypedDict("_Credentials", \
{ "api_key": str, "api_secret": str, "filters": Optional[List[str]] })
class Client:
def __init__(
self,
@@ -15,16 +22,28 @@ class Client:
rest_host: str = REST_HOST,
wss_host: str = WSS_HOST,
filters: Optional[List[str]] = None,
wss_timeout: Optional[float] = 60 * 15,
timeout: Optional[float] = 60 * 15,
log_filename: Optional[str] = None,
log_level: Literal["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"] = "INFO"
) -> None:
credentials: Optional["_Credentials"] = None
if api_key and api_secret:
credentials = \
{ "api_key": api_key, "api_secret": api_secret, "filters": filters }
elif api_key:
raise IncompleteCredentialError( \
"You must provide both an API-KEY and an API-SECRET (missing API-KEY).")
elif api_secret:
raise IncompleteCredentialError( \
"You must provide both an API-KEY and an API-SECRET (missing API-SECRET).")
self.rest = BfxRestInterface(rest_host, api_key, api_secret)
logger = ColorLogger("bfxapi", level=log_level)
if log_filename:
logger.register(filename=log_filename)
self.rest = BfxRestInterface(rest_host, api_key, api_secret)
self.wss = BfxWebSocketClient(wss_host, api_key, api_secret,
filters=filters, wss_timeout=wss_timeout, logger=logger)
self.wss = BfxWebSocketClient(wss_host, \
credentials=credentials, timeout=timeout, logger=logger)