def get_balance(
self, api: LedgerApi, message: LedgerApiMessage, dialogue: LedgerApiDialogue,
) -> LedgerApiMessage:
"""
Send the request 'get_balance'.
:param api: the API object.
:param message: the Ledger API message
:return: None
"""
balance = api.get_balance(message.address)
if balance is None:
response = self.get_error_message(
ValueError("No balance returned"), api, message, dialogue
)
else:
response = cast(
LedgerApiMessage,
dialogue.reply(
performative=LedgerApiMessage.Performative.BALANCE,
target_message=message,
balance=balance,
ledger_id=message.ledger_id,
),
)
return response
def get_balance(
self, api: LedgerApi, message: LedgerApiMessage, dialogue: LedgerApiDialogue,
) -> LedgerApiMessage:
"""
Send the request 'get_balance'.
:param api: the API object.
:param message: the Ledger API message
:return: None
"""
balance = api.get_balance(message.address)
if balance is None:
response = self.get_error_message(
ValueError("No balance returned"), api, message, dialogue
)
else:
response = LedgerApiMessage(
performative=LedgerApiMessage.Performative.BALANCE,
message_id=message.message_id + 1,
target=message.message_id,
dialogue_reference=dialogue.dialogue_label.dialogue_reference,
balance=balance,
ledger_id=message.ledger_id,
)
response.counterparty = message.counterparty
dialogue.update(response)
return response