# python -c "import examples.websocket.authenticated.wallets"

import os

from bfxapi import Client
from bfxapi.enums import Error
from bfxapi.websocket.types import List, Wallet

bfx = Client(
    API_KEY=os.getenv("BFX_API_KEY"),
    API_SECRET=os.getenv("BFX_API_SECRET"),
    filter=["wallet"]
)

@bfx.wss.on("wss-error")
def on_wss_error(code: Error, msg: str):
    print(code, msg)

@bfx.wss.on("wallet_snapshot")
def on_wallet_snapshot(wallets: List[Wallet]):
    for wallet in wallets:
        print(f"Wallet: {wallet.wallet_type} | {wallet.currency}")
        print(f"Available balance: {wallet.available_balance}")
        print(f"Wallet trade details: {wallet.trade_details}")

@bfx.wss.on("wallet_update")
def on_wallet_update(wallet: Wallet):
    print(f"Wallet update: {wallet}")

bfx.wss.run()