def encode(msg: Message) -> bytes:
"""
Encode a 'LedgerApi' message into bytes.
:param msg: the message object.
:return: the bytes.
"""
msg = cast(LedgerApiMessage, msg)
ledger_api_msg = ledger_api_pb2.LedgerApiMessage()
ledger_api_msg.message_id = msg.message_id
dialogue_reference = msg.dialogue_reference
ledger_api_msg.dialogue_starter_reference = dialogue_reference[0]
ledger_api_msg.dialogue_responder_reference = dialogue_reference[1]
ledger_api_msg.target = msg.target
performative_id = msg.performative
if performative_id == LedgerApiMessage.Performative.GET_BALANCE:
performative = ledger_api_pb2.LedgerApiMessage.Get_Balance_Performative(
) # type: ignore
ledger_id = msg.ledger_id
performative.ledger_id = ledger_id
address = msg.address
performative.address = address
ledger_api_msg.get_balance.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.GET_RAW_TRANSACTION:
performative = ledger_api_pb2.LedgerApiMessage.Get_Raw_Transaction_Performative(
) # type: ignore
terms = msg.terms
Terms.encode(performative.terms, terms)
ledger_api_msg.get_raw_transaction.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.SEND_SIGNED_TRANSACTION:
performative = ledger_api_pb2.LedgerApiMessage.Send_Signed_Transaction_Performative(
) # type: ignore
signed_transaction = msg.signed_transaction
SignedTransaction.encode(performative.signed_transaction,
signed_transaction)
ledger_api_msg.send_signed_transaction.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.GET_TRANSACTION_RECEIPT:
performative = ledger_api_pb2.LedgerApiMessage.Get_Transaction_Receipt_Performative(
) # type: ignore
transaction_digest = msg.transaction_digest
TransactionDigest.encode(performative.transaction_digest,
transaction_digest)
ledger_api_msg.get_transaction_receipt.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.BALANCE:
performative = ledger_api_pb2.LedgerApiMessage.Balance_Performative(
) # type: ignore
ledger_id = msg.ledger_id
performative.ledger_id = ledger_id
balance = msg.balance
performative.balance = balance
ledger_api_msg.balance.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.RAW_TRANSACTION:
performative = ledger_api_pb2.LedgerApiMessage.Raw_Transaction_Performative(
) # type: ignore
raw_transaction = msg.raw_transaction
RawTransaction.encode(performative.raw_transaction,
raw_transaction)
ledger_api_msg.raw_transaction.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.TRANSACTION_DIGEST:
performative = ledger_api_pb2.LedgerApiMessage.Transaction_Digest_Performative(
) # type: ignore
transaction_digest = msg.transaction_digest
TransactionDigest.encode(performative.transaction_digest,
transaction_digest)
ledger_api_msg.transaction_digest.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.TRANSACTION_RECEIPT:
performative = ledger_api_pb2.LedgerApiMessage.Transaction_Receipt_Performative(
) # type: ignore
transaction_receipt = msg.transaction_receipt
TransactionReceipt.encode(performative.transaction_receipt,
transaction_receipt)
ledger_api_msg.transaction_receipt.CopyFrom(performative)
elif performative_id == LedgerApiMessage.Performative.ERROR:
performative = ledger_api_pb2.LedgerApiMessage.Error_Performative(
) # type: ignore
code = msg.code
performative.code = code
if msg.is_set("message"):
performative.message_is_set = True
message = msg.message
performative.message = message
if msg.is_set("data"):
performative.data_is_set = True
data = msg.data
performative.data = data
ledger_api_msg.error.CopyFrom(performative)
else:
raise ValueError(
"Performative not valid: {}".format(performative_id))
ledger_api_bytes = ledger_api_msg.SerializeToString()
return ledger_api_bytes
def decode(obj: bytes) -> Message:
"""
Decode bytes into a 'LedgerApi' message.
:param obj: the bytes object.
:return: the 'LedgerApi' message.
"""
ledger_api_pb = ledger_api_pb2.LedgerApiMessage()
ledger_api_pb.ParseFromString(obj)
message_id = ledger_api_pb.message_id
dialogue_reference = (
ledger_api_pb.dialogue_starter_reference,
ledger_api_pb.dialogue_responder_reference,
)
target = ledger_api_pb.target
performative = ledger_api_pb.WhichOneof("performative")
performative_id = LedgerApiMessage.Performative(str(performative))
performative_content = dict() # type: Dict[str, Any]
if performative_id == LedgerApiMessage.Performative.GET_BALANCE:
ledger_id = ledger_api_pb.get_balance.ledger_id
performative_content["ledger_id"] = ledger_id
address = ledger_api_pb.get_balance.address
performative_content["address"] = address
elif performative_id == LedgerApiMessage.Performative.GET_RAW_TRANSACTION:
pb2_terms = ledger_api_pb.get_raw_transaction.terms
terms = Terms.decode(pb2_terms)
performative_content["terms"] = terms
elif performative_id == LedgerApiMessage.Performative.SEND_SIGNED_TRANSACTION:
pb2_signed_transaction = (
ledger_api_pb.send_signed_transaction.signed_transaction)
signed_transaction = SignedTransaction.decode(
pb2_signed_transaction)
performative_content["signed_transaction"] = signed_transaction
elif performative_id == LedgerApiMessage.Performative.GET_TRANSACTION_RECEIPT:
pb2_transaction_digest = (
ledger_api_pb.get_transaction_receipt.transaction_digest)
transaction_digest = TransactionDigest.decode(
pb2_transaction_digest)
performative_content["transaction_digest"] = transaction_digest
elif performative_id == LedgerApiMessage.Performative.BALANCE:
ledger_id = ledger_api_pb.balance.ledger_id
performative_content["ledger_id"] = ledger_id
balance = ledger_api_pb.balance.balance
performative_content["balance"] = balance
elif performative_id == LedgerApiMessage.Performative.RAW_TRANSACTION:
pb2_raw_transaction = ledger_api_pb.raw_transaction.raw_transaction
raw_transaction = RawTransaction.decode(pb2_raw_transaction)
performative_content["raw_transaction"] = raw_transaction
elif performative_id == LedgerApiMessage.Performative.TRANSACTION_DIGEST:
pb2_transaction_digest = ledger_api_pb.transaction_digest.transaction_digest
transaction_digest = TransactionDigest.decode(
pb2_transaction_digest)
performative_content["transaction_digest"] = transaction_digest
elif performative_id == LedgerApiMessage.Performative.TRANSACTION_RECEIPT:
pb2_transaction_receipt = (
ledger_api_pb.transaction_receipt.transaction_receipt)
transaction_receipt = TransactionReceipt.decode(
pb2_transaction_receipt)
performative_content["transaction_receipt"] = transaction_receipt
elif performative_id == LedgerApiMessage.Performative.ERROR:
code = ledger_api_pb.error.code
performative_content["code"] = code
if ledger_api_pb.error.message_is_set:
message = ledger_api_pb.error.message
performative_content["message"] = message
if ledger_api_pb.error.data_is_set:
data = ledger_api_pb.error.data
performative_content["data"] = data
else:
raise ValueError(
"Performative not valid: {}.".format(performative_id))
return LedgerApiMessage(message_id=message_id,
dialogue_reference=dialogue_reference,
target=target,
performative=performative,
**performative_content)