def ListClients(query=None, context=None): """List clients conforming to a givent query.""" args = api_pb2.ApiListClientsArgs(query=query) items = context.SendIteratorRequest("ListClients", args) return utils.MapItemsIterator( lambda data: Client(data=data, context=context), items)
def ListFiles(self): args = api_pb2.ApiListFilesArgs( client_id=self.client_id, file_path=self.path) items = self._context.SendIteratorRequest("ListFiles", args) def MapDataToFile(data): return File(client_id=self.client_id, data=data, context=self._context) return utils.MapItemsIterator(MapDataToFile, items)
def ListHuntApprovals(context=None): """List all hunt approvals belonging to requesting user.""" items = context.SendIteratorRequest("ListHuntApprovals", api_pb2.ApiListHuntApprovalsArgs()) def MapHuntApproval(data): return HuntApproval(data=data, username=context.username, context=context) return utils.MapItemsIterator(MapHuntApproval, items)
def ListFlows(self, offset=0, count=0): """List flows that ran on this client.""" args = api_pb2.ApiListFlowsArgs(client_id=self.client_id, offset=offset, count=count) items = self._context.SendIteratorRequest("ListFlows", args) return utils.MapItemsIterator( lambda data: flow.Flow(data=data, context=self._context), items)
def ListApprovals(self, state=api_pb2.ApiListClientApprovalsArgs.ANY): args = api_pb2.ApiListClientApprovalsArgs(client_id=self.client_id, state=state) items = self._context.SendIteratorRequest("ListClientApprovals", args) def MapClientApproval(data): return ClientApproval(data=data, username=self._context.username, context=self._context) return utils.MapItemsIterator(MapClientApproval, items)
def ListResults(self): args = api_pb2.ApiListFlowResultsArgs( client_id=self.client_id, flow_id=self.flow_id) items = self._context.SendIteratorRequest("ListFlowResults", args) return utils.MapItemsIterator(lambda data: FlowResult(data=data), items)
def ListResults(self): args = api_pb2.ApiListHuntResultsArgs(hunt_id=self.hunt_id) items = self._context.SendIteratorRequest("ListHuntResults", args) return utils.MapItemsIterator( lambda data: HuntResult(data=data, context=self._context), items)