40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
# pragma pylint: disable=W0603
|
|
""" Wallet """
|
|
import logging
|
|
from typing import Dict
|
|
from collections import namedtuple
|
|
from freqtrade.exchange import Exchange
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Wallets(object):
|
|
|
|
# wallet data structure
|
|
wallet = namedtuple(
|
|
'wallet',
|
|
['exchange', 'currency', 'free', 'used', 'total']
|
|
)
|
|
|
|
def __init__(self, exchange: Exchange) -> None:
|
|
self.exchange = exchange
|
|
self.wallets: Dict[str, self.wallet] = {}
|
|
|
|
def _update_wallets(self) -> None:
|
|
balances = self.exchange.get_balances()
|
|
|
|
for currency in balances:
|
|
info = {
|
|
'exchange': self.exchange.id,
|
|
'currency': currency,
|
|
'free': balances[currency['free']],
|
|
'used': balances[currency['used']],
|
|
'total': balances[currency['total']]
|
|
}
|
|
self.wallets[currency] = self.wallet(**info)
|
|
|
|
logger.info('Wallets synced ...')
|
|
|
|
def update(self) -> None:
|
|
self._update_wallets()
|