def test_allows_provided_subcommands(self):
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers()
subparser = subparsers.add_parser("magic")
subparser.set_defaults(chosen="magic")
with argparse_util.allow_missing_subcommand():
args = parser.parse_args(["magic"])
self.assertEqual(args, argparse.Namespace(chosen="magic"))
def configure(self, argv=("", ), **kwargs):
"""Configures TensorBoard behavior via flags.
This method will populate the "flags" property with an argparse.Namespace
representing flag values parsed from the provided argv list, overridden by
explicit flags from remaining keyword arguments.
Args:
argv: Can be set to CLI args equivalent to sys.argv; the first arg is
taken to be the name of the path being executed.
kwargs: Additional arguments will override what was parsed from
argv. They must be passed as Python data structures, e.g.
`foo=1` rather than `foo="1"`.
Returns:
Either argv[:1] if argv was non-empty, or [''] otherwise, as a mechanism
for absl.app.run() compatibility.
Raises:
ValueError: If flag values are invalid.
"""
base_parser = argparse_flags.ArgumentParser(
prog="tensorboard",
description=(
"TensorBoard is a suite of web applications for "
"inspecting and understanding your TensorFlow runs "
"and graphs. https://github.com/tensorflow/tensorboard "),
)
subparsers = base_parser.add_subparsers(
help="TensorBoard subcommand (defaults to %r)" %
_SERVE_SUBCOMMAND_NAME)
serve_subparser = subparsers.add_parser(
_SERVE_SUBCOMMAND_NAME,
help="start local TensorBoard server (default subcommand)",
)
serve_subparser.set_defaults(
**{_SUBCOMMAND_FLAG: _SERVE_SUBCOMMAND_NAME})
if len(argv) < 2 or argv[1].startswith("-"):
# This invocation, if valid, must not use any subcommands: we
# don't permit flags before the subcommand name.
serve_parser = base_parser
else:
# This invocation, if valid, must use a subcommand: we don't take
# any positional arguments to `serve`.
serve_parser = serve_subparser
for (name, subcommand) in six.iteritems(self.subcommands):
subparser = subparsers.add_parser(
name,
help=subcommand.help(),
description=subcommand.description(),
)
subparser.set_defaults(**{_SUBCOMMAND_FLAG: name})
subcommand.define_flags(subparser)
for loader in self.plugin_loaders:
loader.define_flags(serve_parser)
arg0 = argv[0] if argv else ""
with argparse_util.allow_missing_subcommand():
flags = base_parser.parse_args(
argv[1:]) # Strip binary name from argv.
if getattr(flags, _SUBCOMMAND_FLAG, None) is None:
# Manually assign default value rather than using `set_defaults`
# on the base parser to work around Python bug #9351 on old
# versions of `argparse`: <https://bugs.python.org/issue9351>
setattr(flags, _SUBCOMMAND_FLAG, _SERVE_SUBCOMMAND_NAME)
self.cache_key = manager.cache_key(
working_directory=os.getcwd(),
arguments=argv[1:],
configure_kwargs=kwargs,
)
if arg0:
# Only expose main module Abseil flags as TensorBoard native flags.
# This is the same logic Abseil's ArgumentParser uses for determining
# which Abseil flags to include in the short helpstring.
for flag in set(absl_flags.FLAGS.get_key_flags_for_module(arg0)):
if hasattr(flags, flag.name):
raise ValueError("Conflicting Abseil flag: %s" % flag.name)
setattr(flags, flag.name, flag.value)
for k, v in kwargs.items():
if not hasattr(flags, k):
raise ValueError("Unknown TensorBoard flag: %s" % k)
setattr(flags, k, v)
if getattr(flags, _SUBCOMMAND_FLAG) == _SERVE_SUBCOMMAND_NAME:
for loader in self.plugin_loaders:
loader.fix_flags(flags)
self.flags = flags
return [arg0]
def test_still_complains_on_missing_arguments(self):
parser = argparse.ArgumentParser()
parser.add_argument("please_provide_me")
with argparse_util.allow_missing_subcommand():
with self.assertRaises(SystemExit):
parser.parse_args([])