def Handle(self, args, token=None):
iop_cls = instant_output_plugin.InstantOutputPlugin
plugin_cls = iop_cls.GetPluginClassByPluginName(args.plugin_name)
flow_urn = args.flow_id.ResolveClientFlowURN(args.client_id, token=token)
output_collection = flow.GRRFlow.TypedResultCollectionForFID(flow_urn)
plugin = plugin_cls(source_urn=flow_urn, token=token)
content_generator = instant_output_plugin.ApplyPluginToMultiTypeCollection(
plugin, output_collection, source_urn=args.client_id.ToClientURN())
return api_call_handler_base.ApiBinaryStream(
plugin.output_file_name, content_generator=content_generator)
def ProcessPlugin(self, source_urn=None):
return list(
instant_output_plugin.ApplyPluginToMultiTypeCollection(
self.plugin, self.collection, source_urn=source_urn))