From 972b7fe374153525b77b2d2aafcbe37bad420ef1 Mon Sep 17 00:00:00 2001 From: yukonet <58343698+yukonet@users.noreply.github.com> Date: Wed, 4 Dec 2024 14:35:02 +0000 Subject: [PATCH] fix fee-deposit-account view --- app/api/views.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/app/api/views.py b/app/api/views.py index d40489b..a47129a 100644 --- a/app/api/views.py +++ b/app/api/views.py @@ -151,9 +151,16 @@ def dump(): @api.post('/fee-deposit-account') def get_fee_deposit_account(): - token_instance = Token(g.symbol) - return {'account': token_instance.get_fee_deposit_account(), - 'balance': token_instance.get_fee_deposit_account_balance()} + if g.symbol == 'ETH': + coin_instance = Coin(g.symbol) + return {'account': coin_instance.get_fee_deposit_account(), + 'balance': coin_instance.get_fee_deposit_coin_balance()} + elif g.symbol in config['TOKENS'][config["CURRENT_ETH_NETWORK"]].keys(): + token_instance = Token(g.symbol) + return {'account': token_instance.get_fee_deposit_account(), + 'balance': token_instance.get_fee_deposit_account_balance()} + else: + raise Exception(f'Symbol {g.symbol} cannot be processed') @api.post('/get_all_addresses') def get_all_addresses():