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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
# Copyright (C) 2008-2011 Dejan Muhamedagic <dmuhamedagic@suse.de>
# Copyright (C) 2013 Kristoffer Gronlund <kgronlund@suse.com>
# See COPYING for license information.
import re
import inspect
from . import utils
from . import log
logger = log.setup_logger(__name__)
logger_utils = log.LoggerUtils(logger)
def _get_attr_cmd(attr_ext_commands, subcmd):
try:
attr_cmd = attr_ext_commands[subcmd]
if attr_cmd:
return attr_cmd
except KeyError as msg:
raise ValueError(msg)
raise ValueError("Bad attr_cmd " + repr(attr_ext_commands))
def _dispatch_attr_cmd(cmd, attr_cmd, rsc, subcmd, attr, value):
def sanity_check(arg):
if not utils.is_name_sane(arg):
raise ValueError("Expected valid name, got '%s'" % (arg))
if subcmd in ['set', 'set_p']:
if value is None:
raise ValueError("Missing value argument to set")
sanity_check(rsc)
sanity_check(attr)
sanity_check(value)
return utils.ext_cmd(attr_cmd % (rsc, attr, value)) == 0
elif subcmd in ('delete', 'show') or \
(cmd == "secret" and subcmd in ('stash', 'unstash', 'check')):
if value is not None:
raise ValueError("Too many arguments to %s" % (subcmd))
sanity_check(rsc)
sanity_check(attr)
return utils.ext_cmd(attr_cmd % (rsc, attr)) == 0
raise ValueError("Unknown command " + repr(subcmd))
def manage_attr(cmd, attr_ext_commands, rsc, subcmd, attr, value):
'''
TODO: describe.
'''
try:
attr_cmd = _get_attr_cmd(attr_ext_commands, subcmd)
if re.search("\w+=\w+", attr):
attr, value = attr.split('=')
return _dispatch_attr_cmd(cmd, attr_cmd, rsc, subcmd, attr, value)
except ValueError as msg:
cmdline = [rsc, subcmd, attr]
if value is not None:
cmdline.append(value)
logger_utils.bad_usage(cmd, ' '.join(cmdline), msg)
return False
def ptestlike(simfun, def_verb, cmd, args):
verbosity = def_verb # default verbosity
nograph = False
scores = False
utilization = False
actions = False
for p in args:
if p == "nograph":
nograph = True
elif p == "scores":
scores = True
elif p == "utilization":
utilization = True
elif p == "actions":
actions = True
elif re.match("^vv*$", p):
verbosity = p
else:
logger_utils.bad_usage(cmd, ' '.join(args))
return False
return simfun(nograph, scores, utilization, actions, verbosity)
def graph_args(args):
'''
Common parameters for two graph commands:
configure graph [<gtype> [<file> [<img_format>]]]
history graph <pe> [<gtype> [<file> [<img_format>]]]
'''
def tryarg(n, orelse):
try:
return args[n]
except IndexError:
return orelse
except TypeError:
return orelse
from .crm_gv import gv_types
gtype, outf, ftype = None, None, None
gtype = tryarg(0, "dot")
if gtype not in gv_types:
logger.error("graph type %s is not supported", gtype)
return False, gtype, outf, ftype
outf = tryarg(1, None)
if outf is not None and not utils.is_path_sane(outf):
return False, gtype, outf, ftype
ftype = tryarg(2, gtype)
return True, gtype, outf, ftype
def pretty_arguments(f, nskip=0):
'''
Returns a prettified representation
of the command arguments
'''
specs = inspect.getfullargspec(f)
named_args = []
if specs.defaults is None:
named_args += specs.args
else:
named_args += specs.args[:-len(specs.defaults)]
named_args += [("[%s]" % a) for a in specs.args[-len(specs.defaults):]]
if specs.varargs:
named_args += ['[%s ...]' % (specs.varargs)]
if nskip:
named_args = named_args[nskip:]
return ' '.join(named_args)
def validate_arguments(f, args, nskip=0):
'''
Compares the declared arguments of f to
the given arguments in args, and raises
ValueError if the arguments don't match.
nskip: When reporting an error, skip these
many initial arguments when counting.
For example, pass 1 to not count self on a
method.
Note: Does not support keyword arguments.
'''
specs = inspect.getfullargspec(f)
min_args = len(specs.args)
if specs.defaults is not None:
min_args -= len(specs.defaults)
max_args = len(specs.args)
if specs.varargs:
max_args = -1
def mknamed():
return pretty_arguments(f, nskip=nskip)
if min_args == max_args and len(args) != min_args:
raise ValueError("Expected (%s), takes exactly %d arguments (%d given)" %
(mknamed(), min_args-nskip, len(args)-nskip))
elif len(args) < min_args:
raise ValueError("Expected (%s), takes at least %d arguments (%d given)" %
(mknamed(), min_args-nskip, len(args)-nskip))
if max_args >= 0 and len(args) > max_args:
raise ValueError("Expected (%s), takes at most %d arguments (%d given)" %
(mknamed(), max_args-nskip, len(args)-nskip))
|
