forked from autotest/autotest
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathautoserv_parser.py
152 lines (138 loc) · 8.28 KB
/
autoserv_parser.py
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
import os, sys, optparse
from autotest.client.shared import host_protections, utils
class base_autoserv_parser(object):
"""Custom command-line options parser for autoserv.
We can't use the general getopt methods here, as there will be unknown
extra arguments that we pass down into the control file instead.
Thus we process the arguments by hand, for which we are duly repentant.
Making a single function here just makes it harder to read. Suck it up.
"""
def __init__(self):
self.args = sys.argv[1:]
self.parser = optparse.OptionParser(usage="%prog [options] [control-file]")
self.setup_options()
# parse an empty list of arguments in order to set self.options
# to default values so that codepaths that assume they are always
# reached from an autoserv process (when they actually are not)
# will still work
self.options, self.args = self.parser.parse_args(args=[])
def setup_options(self):
self.parser.add_option("-m", action="store", type="string",
dest="machines",
help="list of machines")
self.parser.add_option("-M", action="store", type="string",
dest="machines_file",
help="list of machines from file")
self.parser.add_option("-c", action="store_true",
dest="client", default=False,
help="control file is client side")
self.parser.add_option("-s", action="store_true",
dest="server", default=False,
help="control file is server side")
self.parser.add_option("-r", action="store", type="string",
dest="results", default=None,
help="specify results directory")
self.parser.add_option("-l", action="store", type="string",
dest="label", default='',
help="label for the job")
self.parser.add_option("-G", action="store", type="string",
dest="group_name", default='',
help="The host_group_name to store in keyvals")
self.parser.add_option("-u", action="store", type="string",
dest="user",
default=os.environ.get('USER'),
help="username for the job")
self.parser.add_option("-P", action="store", type="string",
dest="parse_job",
default='',
help="Parse the results of the job using this "
"execution tag. Accessable in control "
"files as job.tag.")
self.parser.add_option("--execution-tag", action="store", type="string",
dest="execution_tag", default='',
help="Accessable in control files as job.tag; "
"Defaults to the value passed to -P.")
self.parser.add_option("-i", action="store_true",
dest="install_before", default=False,
help="reinstall machines before running the job")
self.parser.add_option("-I", action="store_true",
dest="install_after", default=False,
help="reinstall machines after running the job")
self.parser.add_option("-v", action="store_true",
dest="verify", default=False,
help="verify the machines only")
self.parser.add_option("-R", action="store_true",
dest="repair", default=False,
help="repair the machines")
self.parser.add_option("-C", "--cleanup", action="store_true",
default=False,
help="cleanup all machines after the job")
self.parser.add_option("-n", action="store_true",
dest="no_tee", default=False,
help="no teeing the status to stdout/err")
self.parser.add_option("-N", action="store_true",
dest="no_logging", default=False,
help="no logging")
self.parser.add_option('--verbose', action='store_true',
help='Include DEBUG messages in console output')
self.parser.add_option('--no_console_prefix', action='store_true',
help='Disable the logging prefix on console '
'output')
self.parser.add_option("-p", "--write-pidfile", action="store_true",
dest="write_pidfile", default=False,
help="write pidfile (pidfile name is determined "
"by --pidfile-label")
self.parser.add_option("--pidfile-label", action="store",
default="autoserv",
help="Determines filename to use as pidfile (if "
"-p is specified). Pidfile will be "
".<label>_execute. Default to autoserv.")
self.parser.add_option("--use-existing-results", action="store_true",
help="Indicates that autoserv is working with "
"an existing results directory")
self.parser.add_option("-a", "--args", dest='args',
help="additional args to pass to control file")
protection_levels = [host_protections.Protection.get_attr_name(s)
for i, s in host_protections.choices]
self.parser.add_option("--host-protection", action="store",
type="choice", dest="host_protection",
default=host_protections.default,
choices=protection_levels,
help="level of host protection during repair")
self.parser.add_option("--ssh-user", action="store",
type="string", dest="ssh_user",
default="root",
help=("specify the user for ssh"
"connections"))
self.parser.add_option("--ssh-port", action="store",
type="int", dest="ssh_port",
default=22,
help=("specify the port to use for "
"ssh connections"))
self.parser.add_option("--ssh-pass", action="store",
type="string", dest="ssh_pass",
default="",
help=("specify the password to use "
"for ssh connections"))
self.parser.add_option("--install-in-tmpdir", action="store_true",
dest="install_in_tmpdir", default=False,
help=("by default install autotest clients in "
"a temporary directory"))
self.parser.add_option("--collect-crashinfo", action="store_true",
dest="collect_crashinfo", default=False,
help="just run crashinfo collection")
self.parser.add_option("--control-filename", action="store",
type="string", default=None,
help=("filename to use for the server control "
"file in the results directory"))
def parse_args(self):
self.options, self.args = self.parser.parse_args()
if self.options.args:
self.args += self.options.args.split()
site_autoserv_parser = utils.import_site_class(
__file__, "autotest.server.site_autoserv_parser",
"site_autoserv_parser", base_autoserv_parser)
class autoserv_parser(site_autoserv_parser):
pass
# create the one and only one instance of autoserv_parser
autoserv_parser = autoserv_parser()