def get_user(self, public_key):
"""Gets the agent associated with the public_key
Args:
public_key (str): The public key of the agent
Returns:
agent_pb2.Agent: Agent with the provided public_key
"""
address = addresser.get_user_address(public_key)
state_entries = self._context.get_state(addresses=[address],
timeout=self._timeout)
if state_entries:
container = user_pb2.UserContainer()
container.ParseFromString(state_entries[0].data)
for user in container.entries:
if user.public_key == public_key:
return user
def set_user(self, public_key, username, role, timestamp):
user_address = addresser.get_user_address(public_key)
user = user_pb2.User(public_key=public_key,
username=username,
role=role)
container = user_pb2.UserContainer()
state_entries = self._context.get_state(addresses=[user_address],
timeout=self._timeout)
if state_entries:
container.ParseFromString(state_entries[0].data)
container.entries.extend([user])
data = container.SerializeToString()
updated_state = {}
updated_state[user_address] = data
self._context.set_state(updated_state, timeout=self._timeout)