def delete(agent_id):
try:
AgentState.get(agent_id).delete()
except Exception:
print("Agent might not exist")
def get_all():
return AgentState.scan()
def exists():
return AgentState.exists()
def query(agent_id):
try:
return AgentState.get(agent_id)
except Exception:
return None
def init(agent_id):
if query(agent_id) is None:
agent_state = AgentState(agent_id)
agent_state.save()
return agent_state
def delete_table():
AgentState.delete_table()
def create_table():
AgentState.create_table(read_capacity_units=10, write_capacity_units=10)