1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
##############################################################################
# Copyright (c) 2015 Ericsson AB and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
""" Argument parser for yardstick command line tool
"""
import argparse
import logging
from pkg_resources import get_distribution
class CmdParser(argparse.ArgumentParser):
def __init__(self):
argparse.ArgumentParser.__init__(self)
self.output_file_default = "/tmp/yardstick.out"
self._version = "yardstick version %s " % \
get_distribution('yardstick').version
self.__add_arguments()
def __add_arguments(self):
self.add_argument("-d", "--debug",
help="increase output verbosity to debug",
action="store_true")
self.add_argument("-v", "--verbose",
help="increase output verbosity to info",
action="store_true")
self.add_argument("-V", "--version",
help="display version",
version=self._version,
action="version")
self.add_argument("--keep-deploy",
help="keep context deployed in cloud",
action="store_true")
self.add_argument("--parse-only",
help="parse the benchmark config file and exit",
action="store_true")
self.add_argument("--output-file",
help="file where output is stored, default %s" %
self.output_file_default,
default=self.output_file_default)
self.add_argument("taskfile", type=str,
help="path to taskfile", nargs=1)
def parse_args(self):
args = argparse.ArgumentParser.parse_args(self)
logger = logging.getLogger('yardstick')
logger.setLevel(logging.WARNING)
if args.verbose:
logger.setLevel(logging.INFO)
if args.debug:
logger.setLevel(logging.DEBUG)
return args
|