Ejemplo n.º 1
0
Archivo: flow.py Proyecto: rahmiy/grr
  def Get(self):
    """Fetch flow's data and return proper Flow object."""

    args = flow_pb2.ApiGetFlowArgs(
        client_id=self.client_id, flow_id=self.flow_id)
    data = self._context.SendRequest("GetFlow", args)
    return Flow(data=data, context=self._context)
Ejemplo n.º 2
0
    def Get(self) -> "Flow":
        """Fetch flow's data and return proper Flow object."""

        args = flow_pb2.ApiGetFlowArgs(client_id=self.client_id,
                                       flow_id=self.flow_id)
        data = self._context.SendRequest("GetFlow", args)

        if not isinstance(data, flow_pb2.ApiFlow):
            raise TypeError(f"Unexpected response type: {type(data)}")

        return Flow(data=data, context=self._context)