def set_agent(self, public_key, email, name, timestamp):
"""Creates a new agent in state
Args:
public_key (str): The public key of the agent
email (str): The email address of the agent
name (str): The human-readable name of the agent
timestamp (int): Unix UTC timestamp of when the agent was created
"""
address = addresser.get_agent_address(public_key)
agent = agent_pb2.Agent(public_key=public_key,
email=email,
name=name,
timestamp=timestamp)
container = agent_pb2.AgentContainer()
state_entries = self._context.get_state(addresses=[address],
timeout=self._timeout)
if state_entries:
container.ParseFromString(state_entries[0].data)
container.entries.extend([agent])
data = container.SerializeToString()
updated_state = {}
updated_state[address] = data
self._context.set_state(updated_state, timeout=self._timeout)
def get_agent(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_agent_address(public_key)
state_entries = self._context.get_state(
addresses=[address], timeout=self._timeout)
if state_entries:
container = agent_pb2.AgentContainer()
container.ParseFromString(state_entries[0].data)
for agent in container.entries:
if agent.public_key == public_key:
return agent
return None