36 lines
998 B
Python
36 lines
998 B
Python
|
from .data import Depo, DepoSingle
|
||
|
|
from decimal import Decimal
|
||
|
|
import fin_defs
|
||
|
|
import datetime
|
||
|
|
import krakenex
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class KrakenDepoFetcher:
|
||
|
|
|
||
|
|
def __init__(self, kraken_key: str, kraken_secret: str):
|
||
|
|
self.client = krakenex.API(
|
||
|
|
kraken_key,
|
||
|
|
kraken_secret,
|
||
|
|
)
|
||
|
|
|
||
|
|
def get_depo(self) -> Depo:
|
||
|
|
now = datetime.datetime.now(tz=datetime.UTC)
|
||
|
|
|
||
|
|
result = self.client.query_private('Balance')
|
||
|
|
|
||
|
|
assets: dict[fin_defs.Asset, Decimal] = {}
|
||
|
|
for account, balance_str in result['result'].items():
|
||
|
|
asset = fin_defs.WELL_KNOWN_SYMBOLS[account]
|
||
|
|
balance = Decimal(balance_str)
|
||
|
|
if balance != 0:
|
||
|
|
assets[asset] = assets.get(asset, Decimal(0)) + balance
|
||
|
|
del account, balance_str, asset, balance
|
||
|
|
|
||
|
|
return DepoSingle(
|
||
|
|
name = 'Kraken',
|
||
|
|
assets = assets,
|
||
|
|
updated_time = now,
|
||
|
|
)
|