def CreateFlowArgs(self, flow_name=None):
"""Creates flow arguments object for a flow with a given name."""
if not self._flow_descriptors:
self._flow_descriptors = {}
result = self._context.SendRequest("ListFlowDescriptors", None)
for item in result.items:
self._flow_descriptors[item.name] = item
try:
flow_descriptor = self._flow_descriptors[flow_name]
except KeyError:
raise UnknownFlowName(flow_name)
return utils.CopyProto(utils.UnpackAny(flow_descriptor.default_args))
def CreateFlowArgs(
self,
flow_name: str,
) -> Any:
"""Creates flow arguments object for a flow with a given name."""
if not self._flow_descriptors:
self._flow_descriptors = {}
result = self._context.SendRequest("ListFlowDescriptors", None)
if not isinstance(result, flow_pb2.ApiListFlowDescriptorsResult):
raise TypeError(f"Unexpected response type: {type(result)}")
for item in result.items:
self._flow_descriptors[item.name] = item
try:
flow_descriptor = self._flow_descriptors[flow_name]
except KeyError:
raise UnknownFlowName(flow_name)
return utils.CopyProto(utils.UnpackAny(flow_descriptor.default_args))
def UnpackAny(self, proto_any): """Resolves the type and unpacks the given protobuf Any object.""" return utils.UnpackAny(proto_any)
def payload(self):
return utils.UnpackAny(self.data.payload)
def args(self): return utils.UnpackAny(self.data.args)
def payload(self) -> Union[message.Message, utils.UnknownProtobuf]:
return utils.UnpackAny(self.data.payload)
def args(self) -> Union[message.Message, utils.UnknownProtobuf]:
return utils.UnpackAny(self.data.args)
def UnpackAny(
self,
proto_any: any_pb2.Any,
) -> Union[message.Message, utils.UnknownProtobuf]:
"""Resolves the type and unpacks the given protobuf Any object."""
return utils.UnpackAny(proto_any)
