Python arg_splitter.ArgSplitter类代码示例

  def create(cls, env, config, known_scope_infos, args=None, bootstrap_option_values=None):
    """Create an Options instance.

    :param env: a dict of environment variables.
    :param :class:`pants.option.config.Config` config: data from a config file.
    :param known_scope_infos: ScopeInfos for all scopes that may be encountered.
    :param args: a list of cmd-line args; defaults to `sys.argv` if None is supplied.
    :param bootstrap_option_values: An optional namespace containing the values of bootstrap
           options. We can use these values when registering other options.
    """
    # We need parsers for all the intermediate scopes, so inherited option values
    # can propagate through them.
    complete_known_scope_infos = cls.complete_scopes(known_scope_infos)
    splitter = ArgSplitter(complete_known_scope_infos)
    args = sys.argv if args is None else args
    goals, scope_to_flags, target_specs, passthru, passthru_owner, unknown_scopes = splitter.split_args(args)

    option_tracker = OptionTracker()

    if bootstrap_option_values:
      target_spec_files = bootstrap_option_values.target_spec_files
      if target_spec_files:
        for spec in target_spec_files:
          with open(spec, 'r') as f:
            target_specs.extend([line for line in [line.strip() for line in f] if line])

    help_request = splitter.help_request

    parser_hierarchy = ParserHierarchy(env, config, complete_known_scope_infos, option_tracker)
    bootstrap_option_values = bootstrap_option_values
    known_scope_to_info = {s.scope: s for s in complete_known_scope_infos}
    return cls(goals, scope_to_flags, target_specs, passthru, passthru_owner, help_request,
               parser_hierarchy, bootstrap_option_values, known_scope_to_info,
               option_tracker, unknown_scopes)

  def create(cls, env, config, known_scope_infos, args=None, bootstrap_option_values=None):
    """Create an Options instance.

    :param env: a dict of environment variables.
    :param config: data from a config file (must support config.get[list](section, name, default=)).
    :param known_scope_infos: ScopeInfos for all scopes that may be encountered.
    :param args: a list of cmd-line args; defaults to `sys.argv` if None is supplied.
    :param bootstrap_option_values: An optional namespace containing the values of bootstrap
           options. We can use these values when registering other options.
    """
    # We need parsers for all the intermediate scopes, so inherited option values
    # can propagate through them.
    complete_known_scope_infos = cls.complete_scopes(known_scope_infos)
    splitter = ArgSplitter(complete_known_scope_infos)
    args = sys.argv if args is None else args
    goals, scope_to_flags, target_specs, passthru, passthru_owner = splitter.split_args(args)

    if bootstrap_option_values:
      target_spec_files = bootstrap_option_values.target_spec_files
      if target_spec_files:
        for spec in target_spec_files:
          with open(spec) as f:
            target_specs.extend(filter(None, [line.strip() for line in f]))

    help_request = splitter.help_request

    parser_hierarchy = ParserHierarchy(env, config, complete_known_scope_infos)
    values_by_scope = {}  # Arg values, parsed per-scope on demand.
    bootstrap_option_values = bootstrap_option_values
    known_scope_to_info = {s.scope: s for s in complete_known_scope_infos}
    return cls(goals, scope_to_flags, target_specs, passthru, passthru_owner, help_request,
               parser_hierarchy, values_by_scope, bootstrap_option_values, known_scope_to_info)

  def __init__(self, env, config, known_scopes, args=sys.argv, bootstrap_option_values=None):
    """Create an Options instance.

    :param env: a dict of environment variables.
    :param config: data from a config file (must support config.get[list](section, name, default=)).
    :param known_scopes: a list of all possible scopes that may be encountered.
    :param args: a list of cmd-line args.
    :param bootstrap_option_values: An optional namespace containing the values of bootstrap
           options. We can use these values when registering other options.
    """
    splitter = ArgSplitter(known_scopes)
    self._goals, self._scope_to_flags, self._target_specs, self._passthru, self._passthru_owner = \
      splitter.split_args(args)

    if bootstrap_option_values:
      target_spec_files = bootstrap_option_values.target_spec_files
      if target_spec_files:
        for spec in target_spec_files:
          with open(spec) as f:
            self._target_specs.extend(filter(None, [line.strip() for line in f]))

    self._help_request = splitter.help_request
    self._parser_hierarchy = ParserHierarchy(env, config, known_scopes, self._help_request)
    self._values_by_scope = {}  # Arg values, parsed per-scope on demand.
    self._bootstrap_option_values = bootstrap_option_values
    self._known_scopes = set(known_scopes)

 def _split(self, args_str, expected_scope_to_flags, expected_target_specs, expected_help=False):
   splitter = ArgSplitter(ArgSplitterTest._known_scopes)
   args = shlex.split(str(args_str))
   scope_to_flags, target_specs = splitter.split_args(args)
   self.assertEquals(expected_scope_to_flags, scope_to_flags)
   self.assertEquals(expected_target_specs, target_specs)
   self.assertEquals(expected_help, splitter.is_help)

 def _split(self, args_str, expected_goals, expected_scope_to_flags, expected_target_specs,
            expected_passthru=None, expected_passthru_owner=None, expected_is_help=False):
   expected_passthru = expected_passthru or []
   splitter = ArgSplitter(ArgSplitterTest._known_scopes)
   args = shlex.split(str(args_str))
   goals, scope_to_flags, target_specs, passthru, passthru_owner = splitter.split_args(args)
   self.assertEquals(expected_goals, goals)
   self.assertEquals(expected_scope_to_flags, scope_to_flags)
   self.assertEquals(expected_target_specs, target_specs)
   self.assertEquals(expected_passthru, passthru)
   self.assertEquals(expected_passthru_owner, passthru_owner)
   self.assertEquals(expected_is_help, splitter.is_help)

  def __init__(self, env, config, known_scopes, args=sys.argv, legacy_parser=None):
    """Create an Options instance.

    :param env: a dict of environment variables.
    :param config: data from a config file (must support config.get(section, name, default=)).
    :param known_scopes: a list of all possible scopes that may be encountered.
    :param args: a list of cmd-line args.
    :param legacy_parser: optional instance of optparse.OptionParser, used to register and access
           the old-style flags during migration.
    """
    splitter = ArgSplitter(known_scopes)
    self._scope_to_flags, self._target_specs, self._passthru = splitter.split_args(args)
    self._is_help = splitter.is_help
    self._parser_hierarchy = ParserHierarchy(env, config, known_scopes, legacy_parser)
    self._legacy_parser = legacy_parser  # Old-style options, used temporarily during transition.
    self._legacy_values = None  # Values parsed from old-stype options.
    self._values_by_scope = {}  # Arg values, parsed per-scope on demand.

 def _split(self, args_str, expected_goals, expected_scope_to_flags, expected_target_specs,
            expected_passthru=None, expected_passthru_owner=None,
            expected_is_help=False, expected_help_advanced=False, expected_help_all=False):
   expected_passthru = expected_passthru or []
   splitter = ArgSplitter(ArgSplitterTest._known_scopes)
   args = shlex.split(str(args_str))
   goals, scope_to_flags, target_specs, passthru, passthru_owner = splitter.split_args(args)
   self.assertEquals(expected_goals, goals)
   self.assertEquals(expected_scope_to_flags, scope_to_flags)
   self.assertEquals(expected_target_specs, target_specs)
   self.assertEquals(expected_passthru, passthru)
   self.assertEquals(expected_passthru_owner, passthru_owner)
   self.assertEquals(expected_is_help, splitter.help_request is not None)
   self.assertEquals(expected_help_advanced,
                     splitter.help_request is not None and splitter.help_request.advanced)
   self.assertEquals(expected_help_all,
                     splitter.help_request is not None and splitter.help_request.all_scopes)

 def _error_split(self, args_str):
   parser = ArgSplitter(ArgSplitterTest._known_scopes)
   args = shlex.split(str(args_str))
   with pytest.raises(ArgSplitterError):
     parser.split_args(args)

 def _split_version(self, args_str):
   splitter = ArgSplitter(ArgSplitterTest._known_scope_infos)
   args = shlex.split(args_str)
   splitter.split_args(args)
   self.assertTrue(splitter.help_request is not None and splitter.help_request.version)

 def _split_no_goal(self, args_str):
   splitter = ArgSplitter(ArgSplitterTest._known_scope_infos)
   splitter.split_args(shlex.split(args_str))
   self.assertTrue(isinstance(splitter.help_request, NoGoalHelp))

 def _split_unknown_goal(self, args_str, unknown_goals):
   splitter = ArgSplitter(ArgSplitterTest._known_scope_infos)
   splitter.split_args(shlex.split(args_str))
   self.assertTrue(isinstance(splitter.help_request, UnknownGoalHelp))
   self.assertSetEqual(set(unknown_goals), set(splitter.help_request.unknown_goals))

 def _split_version_request(self, args_str):
   splitter = ArgSplitter(ArgSplitterTest._known_scope_infos)
   splitter.split_args(shlex.split(args_str))
   self.assertTrue(isinstance(splitter.help_request, VersionHelp))