def decode(obj: bytes) -> Message:
"""
Decode bytes into a 'Tac' message.
:param obj: the bytes object.
:return: the 'Tac' message.
"""
message_pb = ProtobufMessage()
tac_pb = tac_pb2.TacMessage()
message_pb.ParseFromString(obj)
message_id = message_pb.dialogue_message.message_id
dialogue_reference = (
message_pb.dialogue_message.dialogue_starter_reference,
message_pb.dialogue_message.dialogue_responder_reference,
)
target = message_pb.dialogue_message.target
tac_pb.ParseFromString(message_pb.dialogue_message.content)
performative = tac_pb.WhichOneof("performative")
performative_id = TacMessage.Performative(str(performative))
performative_content = dict() # type: Dict[str, Any]
if performative_id == TacMessage.Performative.REGISTER:
agent_name = tac_pb.register.agent_name
performative_content["agent_name"] = agent_name
elif performative_id == TacMessage.Performative.UNREGISTER:
pass
elif performative_id == TacMessage.Performative.TRANSACTION:
transaction_id = tac_pb.transaction.transaction_id
performative_content["transaction_id"] = transaction_id
ledger_id = tac_pb.transaction.ledger_id
performative_content["ledger_id"] = ledger_id
sender_address = tac_pb.transaction.sender_address
performative_content["sender_address"] = sender_address
counterparty_address = tac_pb.transaction.counterparty_address
performative_content["counterparty_address"] = counterparty_address
amount_by_currency_id = tac_pb.transaction.amount_by_currency_id
amount_by_currency_id_dict = dict(amount_by_currency_id)
performative_content[
"amount_by_currency_id"] = amount_by_currency_id_dict
fee_by_currency_id = tac_pb.transaction.fee_by_currency_id
fee_by_currency_id_dict = dict(fee_by_currency_id)
performative_content[
"fee_by_currency_id"] = fee_by_currency_id_dict
quantities_by_good_id = tac_pb.transaction.quantities_by_good_id
quantities_by_good_id_dict = dict(quantities_by_good_id)
performative_content[
"quantities_by_good_id"] = quantities_by_good_id_dict
nonce = tac_pb.transaction.nonce
performative_content["nonce"] = nonce
sender_signature = tac_pb.transaction.sender_signature
performative_content["sender_signature"] = sender_signature
counterparty_signature = tac_pb.transaction.counterparty_signature
performative_content[
"counterparty_signature"] = counterparty_signature
elif performative_id == TacMessage.Performative.CANCELLED:
pass
elif performative_id == TacMessage.Performative.GAME_DATA:
amount_by_currency_id = tac_pb.game_data.amount_by_currency_id
amount_by_currency_id_dict = dict(amount_by_currency_id)
performative_content[
"amount_by_currency_id"] = amount_by_currency_id_dict
exchange_params_by_currency_id = (
tac_pb.game_data.exchange_params_by_currency_id)
exchange_params_by_currency_id_dict = dict(
exchange_params_by_currency_id)
performative_content[
"exchange_params_by_currency_id"] = exchange_params_by_currency_id_dict
quantities_by_good_id = tac_pb.game_data.quantities_by_good_id
quantities_by_good_id_dict = dict(quantities_by_good_id)
performative_content[
"quantities_by_good_id"] = quantities_by_good_id_dict
utility_params_by_good_id = tac_pb.game_data.utility_params_by_good_id
utility_params_by_good_id_dict = dict(utility_params_by_good_id)
performative_content[
"utility_params_by_good_id"] = utility_params_by_good_id_dict
fee_by_currency_id = tac_pb.game_data.fee_by_currency_id
fee_by_currency_id_dict = dict(fee_by_currency_id)
performative_content[
"fee_by_currency_id"] = fee_by_currency_id_dict
agent_addr_to_name = tac_pb.game_data.agent_addr_to_name
agent_addr_to_name_dict = dict(agent_addr_to_name)
performative_content[
"agent_addr_to_name"] = agent_addr_to_name_dict
currency_id_to_name = tac_pb.game_data.currency_id_to_name
currency_id_to_name_dict = dict(currency_id_to_name)
performative_content[
"currency_id_to_name"] = currency_id_to_name_dict
good_id_to_name = tac_pb.game_data.good_id_to_name
good_id_to_name_dict = dict(good_id_to_name)
performative_content["good_id_to_name"] = good_id_to_name_dict
version_id = tac_pb.game_data.version_id
performative_content["version_id"] = version_id
if tac_pb.game_data.info_is_set:
info = tac_pb.game_data.info
info_dict = dict(info)
performative_content["info"] = info_dict
elif performative_id == TacMessage.Performative.TRANSACTION_CONFIRMATION:
transaction_id = tac_pb.transaction_confirmation.transaction_id
performative_content["transaction_id"] = transaction_id
amount_by_currency_id = (
tac_pb.transaction_confirmation.amount_by_currency_id)
amount_by_currency_id_dict = dict(amount_by_currency_id)
performative_content[
"amount_by_currency_id"] = amount_by_currency_id_dict
quantities_by_good_id = (
tac_pb.transaction_confirmation.quantities_by_good_id)
quantities_by_good_id_dict = dict(quantities_by_good_id)
performative_content[
"quantities_by_good_id"] = quantities_by_good_id_dict
elif performative_id == TacMessage.Performative.TAC_ERROR:
pb2_error_code = tac_pb.tac_error.error_code
error_code = ErrorCode.decode(pb2_error_code)
performative_content["error_code"] = error_code
if tac_pb.tac_error.info_is_set:
info = tac_pb.tac_error.info
info_dict = dict(info)
performative_content["info"] = info_dict
else:
raise ValueError(
"Performative not valid: {}.".format(performative_id))
return TacMessage(message_id=message_id,
dialogue_reference=dialogue_reference,
target=target,
performative=performative,
**performative_content)
def encode(msg: Message) -> bytes:
"""
Encode a 'Tac' message into bytes.
:param msg: the message object.
:return: the bytes.
"""
msg = cast(TacMessage, msg)
message_pb = ProtobufMessage()
dialogue_message_pb = DialogueMessage()
tac_msg = tac_pb2.TacMessage()
dialogue_message_pb.message_id = msg.message_id
dialogue_reference = msg.dialogue_reference
dialogue_message_pb.dialogue_starter_reference = dialogue_reference[0]
dialogue_message_pb.dialogue_responder_reference = dialogue_reference[
1]
dialogue_message_pb.target = msg.target
performative_id = msg.performative
if performative_id == TacMessage.Performative.REGISTER:
performative = tac_pb2.TacMessage.Register_Performative(
) # type: ignore
agent_name = msg.agent_name
performative.agent_name = agent_name
tac_msg.register.CopyFrom(performative)
elif performative_id == TacMessage.Performative.UNREGISTER:
performative = tac_pb2.TacMessage.Unregister_Performative(
) # type: ignore
tac_msg.unregister.CopyFrom(performative)
elif performative_id == TacMessage.Performative.TRANSACTION:
performative = tac_pb2.TacMessage.Transaction_Performative(
) # type: ignore
transaction_id = msg.transaction_id
performative.transaction_id = transaction_id
ledger_id = msg.ledger_id
performative.ledger_id = ledger_id
sender_address = msg.sender_address
performative.sender_address = sender_address
counterparty_address = msg.counterparty_address
performative.counterparty_address = counterparty_address
amount_by_currency_id = msg.amount_by_currency_id
performative.amount_by_currency_id.update(amount_by_currency_id)
fee_by_currency_id = msg.fee_by_currency_id
performative.fee_by_currency_id.update(fee_by_currency_id)
quantities_by_good_id = msg.quantities_by_good_id
performative.quantities_by_good_id.update(quantities_by_good_id)
nonce = msg.nonce
performative.nonce = nonce
sender_signature = msg.sender_signature
performative.sender_signature = sender_signature
counterparty_signature = msg.counterparty_signature
performative.counterparty_signature = counterparty_signature
tac_msg.transaction.CopyFrom(performative)
elif performative_id == TacMessage.Performative.CANCELLED:
performative = tac_pb2.TacMessage.Cancelled_Performative(
) # type: ignore
tac_msg.cancelled.CopyFrom(performative)
elif performative_id == TacMessage.Performative.GAME_DATA:
performative = tac_pb2.TacMessage.Game_Data_Performative(
) # type: ignore
amount_by_currency_id = msg.amount_by_currency_id
performative.amount_by_currency_id.update(amount_by_currency_id)
exchange_params_by_currency_id = msg.exchange_params_by_currency_id
performative.exchange_params_by_currency_id.update(
exchange_params_by_currency_id)
quantities_by_good_id = msg.quantities_by_good_id
performative.quantities_by_good_id.update(quantities_by_good_id)
utility_params_by_good_id = msg.utility_params_by_good_id
performative.utility_params_by_good_id.update(
utility_params_by_good_id)
fee_by_currency_id = msg.fee_by_currency_id
performative.fee_by_currency_id.update(fee_by_currency_id)
agent_addr_to_name = msg.agent_addr_to_name
performative.agent_addr_to_name.update(agent_addr_to_name)
currency_id_to_name = msg.currency_id_to_name
performative.currency_id_to_name.update(currency_id_to_name)
good_id_to_name = msg.good_id_to_name
performative.good_id_to_name.update(good_id_to_name)
version_id = msg.version_id
performative.version_id = version_id
if msg.is_set("info"):
performative.info_is_set = True
info = msg.info
performative.info.update(info)
tac_msg.game_data.CopyFrom(performative)
elif performative_id == TacMessage.Performative.TRANSACTION_CONFIRMATION:
performative = tac_pb2.TacMessage.Transaction_Confirmation_Performative(
) # type: ignore
transaction_id = msg.transaction_id
performative.transaction_id = transaction_id
amount_by_currency_id = msg.amount_by_currency_id
performative.amount_by_currency_id.update(amount_by_currency_id)
quantities_by_good_id = msg.quantities_by_good_id
performative.quantities_by_good_id.update(quantities_by_good_id)
tac_msg.transaction_confirmation.CopyFrom(performative)
elif performative_id == TacMessage.Performative.TAC_ERROR:
performative = tac_pb2.TacMessage.Tac_Error_Performative(
) # type: ignore
error_code = msg.error_code
ErrorCode.encode(performative.error_code, error_code)
if msg.is_set("info"):
performative.info_is_set = True
info = msg.info
performative.info.update(info)
tac_msg.tac_error.CopyFrom(performative)
else:
raise ValueError(
"Performative not valid: {}".format(performative_id))
dialogue_message_pb.content = tac_msg.SerializeToString()
message_pb.dialogue_message.CopyFrom(dialogue_message_pb)
message_bytes = message_pb.SerializeToString()
return message_bytes
def encode(self, msg: Message) -> bytes:
"""
Encode a 'Tac' message into bytes.
:param msg: the message object.
:return: the bytes.
"""
msg = cast(TacMessage, msg)
tac_msg = tac_pb2.TacMessage()
tac_msg.message_id = msg.message_id
dialogue_reference = msg.dialogue_reference
tac_msg.dialogue_starter_reference = dialogue_reference[0]
tac_msg.dialogue_responder_reference = dialogue_reference[1]
tac_msg.target = msg.target
performative_id = msg.performative
if performative_id == TacMessage.Performative.REGISTER:
performative = tac_pb2.TacMessage.Register_Performative(
) # type: ignore
agent_name = msg.agent_name
performative.agent_name = agent_name
tac_msg.register.CopyFrom(performative)
elif performative_id == TacMessage.Performative.UNREGISTER:
performative = tac_pb2.TacMessage.Unregister_Performative(
) # type: ignore
tac_msg.unregister.CopyFrom(performative)
elif performative_id == TacMessage.Performative.TRANSACTION:
performative = tac_pb2.TacMessage.Transaction_Performative(
) # type: ignore
tx_id = msg.tx_id
performative.tx_id = tx_id
tx_sender_addr = msg.tx_sender_addr
performative.tx_sender_addr = tx_sender_addr
tx_counterparty_addr = msg.tx_counterparty_addr
performative.tx_counterparty_addr = tx_counterparty_addr
amount_by_currency_id = msg.amount_by_currency_id
performative.amount_by_currency_id.update(amount_by_currency_id)
tx_sender_fee = msg.tx_sender_fee
performative.tx_sender_fee = tx_sender_fee
tx_counterparty_fee = msg.tx_counterparty_fee
performative.tx_counterparty_fee = tx_counterparty_fee
quantities_by_good_id = msg.quantities_by_good_id
performative.quantities_by_good_id.update(quantities_by_good_id)
tx_nonce = msg.tx_nonce
performative.tx_nonce = tx_nonce
tx_sender_signature = msg.tx_sender_signature
performative.tx_sender_signature = tx_sender_signature
tx_counterparty_signature = msg.tx_counterparty_signature
performative.tx_counterparty_signature = tx_counterparty_signature
tac_msg.transaction.CopyFrom(performative)
elif performative_id == TacMessage.Performative.GET_STATE_UPDATE:
performative = tac_pb2.TacMessage.Get_State_Update_Performative(
) # type: ignore
tac_msg.get_state_update.CopyFrom(performative)
elif performative_id == TacMessage.Performative.CANCELLED:
performative = tac_pb2.TacMessage.Cancelled_Performative(
) # type: ignore
tac_msg.cancelled.CopyFrom(performative)
elif performative_id == TacMessage.Performative.GAME_DATA:
performative = tac_pb2.TacMessage.Game_Data_Performative(
) # type: ignore
amount_by_currency_id = msg.amount_by_currency_id
performative.amount_by_currency_id.update(amount_by_currency_id)
exchange_params_by_currency_id = msg.exchange_params_by_currency_id
performative.exchange_params_by_currency_id.update(
exchange_params_by_currency_id)
quantities_by_good_id = msg.quantities_by_good_id
performative.quantities_by_good_id.update(quantities_by_good_id)
utility_params_by_good_id = msg.utility_params_by_good_id
performative.utility_params_by_good_id.update(
utility_params_by_good_id)
tx_fee = msg.tx_fee
performative.tx_fee = tx_fee
agent_addr_to_name = msg.agent_addr_to_name
performative.agent_addr_to_name.update(agent_addr_to_name)
currency_id_to_name = msg.currency_id_to_name
performative.currency_id_to_name.update(currency_id_to_name)
good_id_to_name = msg.good_id_to_name
performative.good_id_to_name.update(good_id_to_name)
version_id = msg.version_id
performative.version_id = version_id
if msg.is_set("info"):
performative.info_is_set = True
info = msg.info
performative.info.update(info)
tac_msg.game_data.CopyFrom(performative)
elif performative_id == TacMessage.Performative.TRANSACTION_CONFIRMATION:
performative = tac_pb2.TacMessage.Transaction_Confirmation_Performative(
) # type: ignore
tx_id = msg.tx_id
performative.tx_id = tx_id
amount_by_currency_id = msg.amount_by_currency_id
performative.amount_by_currency_id.update(amount_by_currency_id)
quantities_by_good_id = msg.quantities_by_good_id
performative.quantities_by_good_id.update(quantities_by_good_id)
tac_msg.transaction_confirmation.CopyFrom(performative)
elif performative_id == TacMessage.Performative.TAC_ERROR:
performative = tac_pb2.TacMessage.Tac_Error_Performative(
) # type: ignore
error_code = msg.error_code
ErrorCode.encode(performative.error_code, error_code)
if msg.is_set("info"):
performative.info_is_set = True
info = msg.info
performative.info.update(info)
tac_msg.tac_error.CopyFrom(performative)
else:
raise ValueError(
"Performative not valid: {}".format(performative_id))
tac_bytes = tac_msg.SerializeToString()
return tac_bytes