def get_output(self, args, mock_stdout):
self.accessor.drop_all_metrics()
for metric in self.metrics:
self.accessor.create_metric(self.make_metric(
metric, self.metadata))
cmd = command_stats.CommandStats()
parser = argparse.ArgumentParser(add_help=False)
bg_utils.add_argparse_arguments(parser)
cmd.add_arguments(parser)
opts = parser.parse_args(args)
cmd.run(self.accessor, opts)
return mock_stdout.getvalue()
command_write,
)
COMMANDS = [
command_clean.CommandClean(),
command_copy.CommandCopy(),
command_daemon.CommandDaemon(),
command_delete.CommandDelete(),
command_du.CommandDu(),
command_graphite_web.CommandGraphiteWeb(),
command_info.CommandInfo(),
command_list.CommandList(),
command_read.CommandRead(),
command_repair.CommandRepair(),
command_shell.CommandShell(),
command_stats.CommandStats(),
command_syncdb.CommandSyncdb(),
command_test.CommandTest(),
command_write.CommandWrite(),
]
def _parse_opts(args):
"""Parse command line arguments."""
parser = argparse.ArgumentParser(
description="BigGraphite command line utility.")
bg_utils.add_argparse_arguments(parser)
subparsers = parser.add_subparsers(help="commands")
for command in COMMANDS:
subparser = subparsers.add_parser(command.NAME,
add_help=False) # accept -h for du