Add support for event balance_update.

Author: Davi0kProgramsThings

bfxapi/types/__init__.py

@@ -1,5 +1,6 @@
 from .dataclasses import (
     BalanceAvailable,
+    BalanceInfo,
     BaseMarginInfo,
     Candle,
     CurrencyConversion,

bfxapi/types/dataclasses.py

@@ -608,6 +608,12 @@ class DerivativePositionCollateralLimits(_Type):
     max_collateral: float
 
 
+@dataclass
+class BalanceInfo(_Type):
+    aum: float
+    aum_net: float
+
+
 # endregion
 
 # region Dataclass definitions for types of merchant use

bfxapi/types/serializers.py

@@ -904,4 +904,10 @@
     labels=["min_collateral", "max_collateral"],
 )
 
+BalanceInfo = generate_labeler_serializer(
+    name="BalanceInfo",
+    klass=dataclasses.BalanceInfo,
+    labels=["aum", "aum_net"],
+)
+
 # endregion

bfxapi/websocket/_event_emitter/bfx_event_emitter.py

@@ -67,6 +67,7 @@
     "base_margin_info",
     "symbol_margin_info",
     "funding_info_update",
+    "balance_update",
     "notification",
     "on-req-notification",
     "ou-req-notification",

bfxapi/websocket/_handlers/auth_events_handler.py

@@ -34,6 +34,7 @@ class AuthEventsHandler:
         "ws": "wallet_snapshot",
         "wu": "wallet_update",
         "fiu": "funding_info_update",
+        "bu": "balance_update",
     }
 
     __SERIALIZERS: Dict[Tuple[str, ...], serializers._Serializer] = {
@@ -45,6 +46,7 @@ class AuthEventsHandler:
         ("fls", "fln", "flu", "flc"): serializers.FundingLoan,
         ("ws", "wu"): serializers.Wallet,
         ("fiu",): serializers.FundingInfo,
+        ("bu",): serializers.BalanceInfo,
     }
 
     def __init__(self, event_emitter: EventEmitter) -> None: