aljaz/bitfinex-api-py
1
0
Fork 0
mirror of https://github.com/aljazceru/bitfinex-api-py.git synced 2025-12-19 06:44:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Apply some refactoring to sub-package bfxapi.websocket.

Browse Source
This commit is contained in:
Davide Casale
2023-10-16 05:59:24 +02:00
parent 133db74a72
commit ddce83be0c
5 changed files with 102 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
bfxapi/websocket/_client/bfx_websocket_bucket.py
View File

@@ -100,7 +100,7 @@ class BfxWebSocketBucket(Connection):
await self._websocket.send(json.dumps( \
{ "event": "conf", "flags": sum(flags) }))
@Connection.require_websocket_connection
@Connection._require_websocket_connection
async def subscribe(self,
channel: str,
sub_id: Optional[str] = None,
@@ -115,7 +115,7 @@ class BfxWebSocketBucket(Connection):
await self._websocket.send(message = \
json.dumps(subscription))
@Connection.require_websocket_connection
@Connection._require_websocket_connection
async def unsubscribe(self, sub_id: str) -> None:
for chan_id, subscription in list(self.__subscriptions.items()):
if subscription["sub_id"] == sub_id:
@@ -128,7 +128,7 @@ class BfxWebSocketBucket(Connection):
await self._websocket.send(message = \
json.dumps(unsubscription))
@Connection.require_websocket_connection
@Connection._require_websocket_connection
async def resubscribe(self, sub_id: str) -> None:
for subscription in self.__subscriptions.values():
if subscription["sub_id"] == sub_id:
@@ -136,7 +136,7 @@ class BfxWebSocketBucket(Connection):
await self.subscribe(**subscription)
@Connection.require_websocket_connection
@Connection._require_websocket_connection
async def close(self, code: int = 1000, reason: str = str()) -> None:
await self._websocket.close(code, reason)