p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Simplify cursor to agents

This commit is contained in:
Mike Salvatore 2022-09-30 10:16:17 -04:00
parent 411b027e92
commit 3b6e4f5313
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/monkey_island/cc/repository/mongo_agent_repository.py
View File

@ -62,7 +62,7 @@ class MongoAgentRepository(IAgentRepository):
def get_running_agents(self) -> Sequence[Agent]: def get_running_agents(self) -> Sequence[Agent]:
try: try:
cursor = self._agents_collection.find({"stop_time": None}, {MONGO_OBJECT_ID_KEY: False}) cursor = self._agents_collection.find({"stop_time": None}, {MONGO_OBJECT_ID_KEY: False})
return list(map(lambda a: Agent(**a), cursor)) return [Agent(**a) for a in cursor]
except Exception as err: except Exception as err:
raise RetrievalError(f"Error retrieving running agents: {err}") raise RetrievalError(f"Error retrieving running agents: {err}")