from binance.enums import *


class ExchangeInfo:
    def __init__(self, client):
        self.client = client

        self.intervals = None
        self.symbols = None
        self.balances = None
        # Set the above values from information retrieved from exchange.
        self.set_exchange_data()

    def set_exchange_data(self):
        # Pull all balances from client while discarding assets with zero balance
        account = self.client.futures_coin_account_balance()
        self.balances = [asset for asset in account if float(asset['balance']) > 0]

        # Pull all available symbols from client
        exchange_info = self.client.get_exchange_info()
        self.symbols = exchange_info['symbols']

        # Available intervals
        self.intervals = (
            KLINE_INTERVAL_1MINUTE, KLINE_INTERVAL_3MINUTE,
            KLINE_INTERVAL_5MINUTE, KLINE_INTERVAL_15MINUTE,
            KLINE_INTERVAL_30MINUTE, KLINE_INTERVAL_1HOUR,
            KLINE_INTERVAL_2HOUR, KLINE_INTERVAL_4HOUR,
            KLINE_INTERVAL_6HOUR, KLINE_INTERVAL_8HOUR,
            KLINE_INTERVAL_12HOUR, KLINE_INTERVAL_1DAY,
            KLINE_INTERVAL_3DAY, KLINE_INTERVAL_1WEEK,
            KLINE_INTERVAL_1MONTH
        )