def modify_btc_account(
self,
account: BTCAddress,
append_or_remove: str,
add_or_sub: Callable[[FVal, FVal], FVal],
already_queried_balance: Optional[FVal] = None,
) -> None:
"""Either appends or removes a BTC acccount.
If already_queried_balance is not None then instead of querying the balance
of the account we can use the already queried one.
Call with 'append', operator.add to add the account
Call with 'remove', operator.sub to remove the account
May raise:
- RemotError if there is a problem querying blockchain.info or cryptocompare
"""
btc_usd_price = Inquirer().find_usd_price(A_BTC)
remove_with_populated_balance = (append_or_remove == 'remove'
and len(self.balances.btc) != 0)
# Query the balance of the account except for the case when it's removed
# and there is no other account in the balances
if append_or_remove == 'append' or remove_with_populated_balance:
if already_queried_balance is None:
balances = get_bitcoin_addresses_balances([account])
balance = balances[account]
else:
balance = already_queried_balance
usd_balance = balance * btc_usd_price
if append_or_remove == 'append':
self.balances.btc[account] = Balance(amount=balance,
usd_value=usd_balance)
elif append_or_remove == 'remove':
if account in self.balances.btc:
del self.balances.btc[account]
else:
raise AssertionError(
'Programmer error: Should be append or remove')
if len(self.balances.btc) == 0:
# If the last account was removed balance should be 0
self.totals[A_BTC].amount = ZERO
self.totals[A_BTC].usd_value = ZERO
else:
self.totals[A_BTC].amount = add_or_sub(self.totals[A_BTC].amount,
balance)
self.totals[A_BTC].usd_value = add_or_sub(
self.totals[A_BTC].usd_value, usd_balance)
# At the very end add/remove it from the accounts
getattr(self.accounts.btc, append_or_remove)(account)
def query_btc_balances(self) -> None:
"""Queries blockchain.info/blockcypher for the balance of all BTC accounts
May raise:
- RemotError if there is a problem querying blockchain.info or cryptocompare
"""
if len(self.accounts.btc) == 0:
return
self.balances.btc = {}
btc_usd_price = Inquirer().find_usd_price(A_BTC)
total = FVal(0)
balances = get_bitcoin_addresses_balances(self.accounts.btc)
for account, balance in balances.items():
total += balance
self.balances.btc[account] = Balance(
amount=balance,
usd_value=balance * btc_usd_price,
)
self.totals[A_BTC] = Balance(amount=total, usd_value=total * btc_usd_price)
def sync_btc_accounts_with_db(self) -> None:
"""Call this function after having deleted BTC accounts from the DB to
sync the chain manager's balances and accounts with the DB
For example this is called after removing an xpub which deletes all derived
addresses from the DB.
"""
db_btc_accounts = self.database.get_blockchain_accounts().btc
accounts_to_remove = []
for btc_account in self.accounts.btc:
if btc_account not in db_btc_accounts:
accounts_to_remove.append(btc_account)
balances_mapping = get_bitcoin_addresses_balances(accounts_to_remove)
balances = [balances_mapping.get(x, ZERO) for x in accounts_to_remove]
self.modify_blockchain_accounts(
blockchain=SupportedBlockchain.BITCOIN,
accounts=accounts_to_remove,
append_or_remove='remove',
add_or_sub=operator.sub,
already_queried_balances=balances,
)