Source code for jina.parsers.flow

"""Argparser module for Flow"""
from jina.parsers.base import set_base_parser
from jina.parsers.helper import KVAppendAction, add_arg_group
from jina.parsers.orchestrate.base import mixin_essential_parser

[docs]def mixin_flow_features_parser(parser): """Add the arguments for the Flow features to the parser :param parser: the parser configure """ from jina.enums import FlowInspectType gp = add_arg_group(parser, title='Flow Feature') gp.add_argument( '--uses', type=str, help='The YAML path represents a flow. It can be either a local file path or a URL.', ) gp.add_argument( '--env', action=KVAppendAction, metavar='KEY: VALUE', nargs='*', help='The map of environment variables that are available inside runtime', ) gp.add_argument( '--inspect', type=FlowInspectType.from_string, choices=list(FlowInspectType), default=FlowInspectType.COLLECT, help=''' The strategy on those inspect deployments in the flow. If `REMOVE` is given then all inspect deployments are removed when building the flow. ''', )
[docs]def set_flow_parser(parser=None): """Set the parser for the flow :param parser: an (optional) initial parser to build upon :return: the parser """ if not parser: parser = set_base_parser() mixin_essential_parser(parser) mixin_flow_features_parser(parser) return parser