googlecloud 0.0.2 → 0.0.4
Sign up to get free protection for your applications and to get access to all the features.
- data.tar.gz.sig +0 -0
- data/CHANGELOG +4 -0
- data/LICENSE +674 -0
- data/Manifest +111 -0
- data/README.md +4 -3
- data/bin/gcutil +53 -0
- data/googlecloud.gemspec +4 -3
- data/packages/gcutil-1.7.1/CHANGELOG +197 -0
- data/packages/gcutil-1.7.1/LICENSE +202 -0
- data/packages/gcutil-1.7.1/VERSION +1 -0
- data/packages/gcutil-1.7.1/gcutil +53 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/LICENSE +23 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/__init__.py +1 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/discovery.py +743 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/errors.py +123 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/ext/__init__.py +0 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/http.py +1443 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/mimeparse.py +172 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/model.py +385 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/apiclient/schema.py +303 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/__init__.py +1 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/anyjson.py +32 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/appengine.py +528 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/client.py +1139 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/clientsecrets.py +105 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/crypt.py +244 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/django_orm.py +124 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/file.py +107 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/locked_file.py +343 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/multistore_file.py +379 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/oauth2client/tools.py +174 -0
- data/packages/gcutil-1.7.1/lib/google_api_python_client/uritemplate/__init__.py +147 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/LICENSE +202 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/__init__.py +3 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/__init__.py +3 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/app.py +356 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/appcommands.py +783 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/basetest.py +1260 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/datelib.py +421 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/debug.py +60 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/file_util.py +181 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/resources.py +67 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/run_script_module.py +217 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/setup_command.py +159 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/shellutil.py +49 -0
- data/packages/gcutil-1.7.1/lib/google_apputils/google/apputils/stopwatch.py +204 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/__init__.py +0 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/auth_helper.py +140 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/auth_helper_test.py +149 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/auto_auth.py +130 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/auto_auth_test.py +75 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/basic_cmds.py +128 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/basic_cmds_test.py +111 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/command_base.py +1808 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/command_base_test.py +1651 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/compute/v1beta13.json +2851 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/compute/v1beta14.json +3361 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/disk_cmds.py +342 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/disk_cmds_test.py +474 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/firewall_cmds.py +344 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/firewall_cmds_test.py +231 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/flags_cache.py +274 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/gcutil +89 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/gcutil_logging.py +69 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/image_cmds.py +262 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/image_cmds_test.py +172 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/instance_cmds.py +1506 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/instance_cmds_test.py +1904 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/kernel_cmds.py +91 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/kernel_cmds_test.py +56 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/machine_type_cmds.py +106 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/machine_type_cmds_test.py +59 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/metadata.py +96 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/metadata_lib.py +357 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/metadata_test.py +84 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/mock_api.py +420 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/mock_metadata.py +58 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/move_cmds.py +824 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/move_cmds_test.py +307 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/network_cmds.py +178 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/network_cmds_test.py +133 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/operation_cmds.py +181 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/operation_cmds_test.py +196 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/path_initializer.py +38 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/project_cmds.py +173 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/project_cmds_test.py +111 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/scopes.py +61 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/scopes_test.py +50 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/snapshot_cmds.py +276 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/snapshot_cmds_test.py +260 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/ssh_keys.py +266 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/ssh_keys_test.py +128 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/table_formatter.py +563 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/thread_pool.py +188 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/thread_pool_test.py +88 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/utils.py +208 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/utils_test.py +193 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/version.py +17 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/version_checker.py +246 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/version_checker_test.py +271 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/zone_cmds.py +151 -0
- data/packages/gcutil-1.7.1/lib/google_compute_engine/gcutil/zone_cmds_test.py +60 -0
- data/packages/gcutil-1.7.1/lib/httplib2/LICENSE +21 -0
- data/packages/gcutil-1.7.1/lib/httplib2/httplib2/__init__.py +1630 -0
- data/packages/gcutil-1.7.1/lib/httplib2/httplib2/cacerts.txt +714 -0
- data/packages/gcutil-1.7.1/lib/httplib2/httplib2/iri2uri.py +110 -0
- data/packages/gcutil-1.7.1/lib/httplib2/httplib2/socks.py +438 -0
- data/packages/gcutil-1.7.1/lib/iso8601/LICENSE +20 -0
- data/packages/gcutil-1.7.1/lib/iso8601/iso8601/__init__.py +1 -0
- data/packages/gcutil-1.7.1/lib/iso8601/iso8601/iso8601.py +102 -0
- data/packages/gcutil-1.7.1/lib/iso8601/iso8601/test_iso8601.py +111 -0
- data/packages/gcutil-1.7.1/lib/python_gflags/AUTHORS +2 -0
- data/packages/gcutil-1.7.1/lib/python_gflags/LICENSE +28 -0
- data/packages/gcutil-1.7.1/lib/python_gflags/gflags.py +2862 -0
- data/packages/gcutil-1.7.1/lib/python_gflags/gflags2man.py +544 -0
- data/packages/gcutil-1.7.1/lib/python_gflags/gflags_validators.py +187 -0
- metadata +118 -5
- metadata.gz.sig +0 -0
@@ -0,0 +1,356 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# Copyright 2003 Google Inc. All Rights Reserved.
|
3
|
+
#
|
4
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
5
|
+
# you may not use this file except in compliance with the License.
|
6
|
+
# You may obtain a copy of the License at
|
7
|
+
#
|
8
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
9
|
+
#
|
10
|
+
# Unless required by applicable law or agreed to in writing, software
|
11
|
+
# distributed under the License is distributed on an "AS-IS" BASIS,
|
12
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
13
|
+
# See the License for the specific language governing permissions and
|
14
|
+
# limitations under the License.
|
15
|
+
|
16
|
+
"""Generic entry point for Google applications.
|
17
|
+
|
18
|
+
To use this module, simply define a 'main' function with a single
|
19
|
+
'argv' argument and add the following to the end of your source file:
|
20
|
+
|
21
|
+
if __name__ == '__main__':
|
22
|
+
app.run()
|
23
|
+
|
24
|
+
TODO(user): Remove silly main-detection logic, and force all clients
|
25
|
+
of this module to check __name__ explicitly. Fix all current clients
|
26
|
+
that don't check __name__.
|
27
|
+
"""
|
28
|
+
import errno
|
29
|
+
import os
|
30
|
+
import pdb
|
31
|
+
import socket
|
32
|
+
import stat
|
33
|
+
import struct
|
34
|
+
import sys
|
35
|
+
import time
|
36
|
+
import traceback
|
37
|
+
import gflags as flags
|
38
|
+
FLAGS = flags.FLAGS
|
39
|
+
|
40
|
+
flags.DEFINE_boolean('run_with_pdb', 0, 'Set to true for PDB debug mode')
|
41
|
+
flags.DEFINE_boolean('run_with_profiling', 0,
|
42
|
+
'Set to true for profiling the script. '
|
43
|
+
'Execution will be slower, and the output format might '
|
44
|
+
'change over time.')
|
45
|
+
flags.DEFINE_boolean('use_cprofile_for_profiling', True,
|
46
|
+
'Use cProfile instead of the profile module for '
|
47
|
+
'profiling. This has no effect unless '
|
48
|
+
'--run_with_profiling is set.')
|
49
|
+
|
50
|
+
# If main() exits via an abnormal exception, call into these
|
51
|
+
# handlers before exiting.
|
52
|
+
|
53
|
+
EXCEPTION_HANDLERS = []
|
54
|
+
help_text_wrap = False # Whether to enable TextWrap in help output
|
55
|
+
|
56
|
+
|
57
|
+
class Error(Exception):
|
58
|
+
pass
|
59
|
+
|
60
|
+
|
61
|
+
class UsageError(Error):
|
62
|
+
"""The arguments supplied by the user are invalid.
|
63
|
+
|
64
|
+
Raise this when the arguments supplied are invalid from the point of
|
65
|
+
view of the application. For example when two mutually exclusive
|
66
|
+
flags have been supplied or when there are not enough non-flag
|
67
|
+
arguments. It is distinct from flags.FlagsError which covers the lower
|
68
|
+
level of parsing and validating individual flags.
|
69
|
+
"""
|
70
|
+
|
71
|
+
def __init__(self, message, exitcode=1):
|
72
|
+
Error.__init__(self, message)
|
73
|
+
self.exitcode = exitcode
|
74
|
+
|
75
|
+
|
76
|
+
class HelpFlag(flags.BooleanFlag):
|
77
|
+
"""Special boolean flag that displays usage and raises SystemExit."""
|
78
|
+
|
79
|
+
def __init__(self):
|
80
|
+
flags.BooleanFlag.__init__(self, 'help', 0, 'show this help',
|
81
|
+
short_name='?', allow_override=1)
|
82
|
+
|
83
|
+
def Parse(self, arg):
|
84
|
+
if arg:
|
85
|
+
usage(writeto_stdout=1)
|
86
|
+
sys.exit(1)
|
87
|
+
|
88
|
+
|
89
|
+
class HelpXMLFlag(flags.BooleanFlag):
|
90
|
+
"""Similar to HelpFlag, but generates output in XML format."""
|
91
|
+
|
92
|
+
def __init__(self):
|
93
|
+
flags.BooleanFlag.__init__(self, 'helpxml', False,
|
94
|
+
'like --help, but generates XML output',
|
95
|
+
allow_override=1)
|
96
|
+
|
97
|
+
def Parse(self, arg):
|
98
|
+
if arg:
|
99
|
+
flags.FLAGS.WriteHelpInXMLFormat(sys.stdout)
|
100
|
+
sys.exit(1)
|
101
|
+
|
102
|
+
|
103
|
+
class HelpshortFlag(flags.BooleanFlag):
|
104
|
+
"""Special bool flag that calls usage(shorthelp=1) and raises SystemExit."""
|
105
|
+
|
106
|
+
def __init__(self):
|
107
|
+
flags.BooleanFlag.__init__(self, 'helpshort', 0,
|
108
|
+
'show usage only for this module',
|
109
|
+
allow_override=1)
|
110
|
+
|
111
|
+
def Parse(self, arg):
|
112
|
+
if arg:
|
113
|
+
usage(shorthelp=1, writeto_stdout=1)
|
114
|
+
sys.exit(1)
|
115
|
+
|
116
|
+
|
117
|
+
class BuildDataFlag(flags.BooleanFlag):
|
118
|
+
|
119
|
+
"""Boolean flag that writes build data to stdout and exits."""
|
120
|
+
|
121
|
+
def __init__(self):
|
122
|
+
flags.BooleanFlag.__init__(self, 'show_build_data', 0,
|
123
|
+
'show build data and exit')
|
124
|
+
|
125
|
+
def Parse(self, arg):
|
126
|
+
if arg:
|
127
|
+
sys.stdout.write(build_data.BuildData())
|
128
|
+
sys.exit(0)
|
129
|
+
|
130
|
+
|
131
|
+
def parse_flags_with_usage(args):
|
132
|
+
"""Try parsing the flags, printing usage and exiting if unparseable."""
|
133
|
+
try:
|
134
|
+
argv = FLAGS(args)
|
135
|
+
return argv
|
136
|
+
except flags.FlagsError, error:
|
137
|
+
sys.stderr.write('FATAL Flags parsing error: %s\n' % error)
|
138
|
+
sys.stderr.write('Pass --help or --helpshort to see help on flags.\n')
|
139
|
+
sys.exit(1)
|
140
|
+
|
141
|
+
|
142
|
+
_define_help_flags_called = False
|
143
|
+
|
144
|
+
|
145
|
+
def DefineHelpFlags():
|
146
|
+
"""Register help flags. Idempotent."""
|
147
|
+
# Use a global to ensure idempotence.
|
148
|
+
# pylint: disable-msg=W0603
|
149
|
+
global _define_help_flags_called
|
150
|
+
|
151
|
+
if not _define_help_flags_called:
|
152
|
+
flags.DEFINE_flag(HelpFlag())
|
153
|
+
flags.DEFINE_flag(HelpXMLFlag())
|
154
|
+
flags.DEFINE_flag(HelpshortFlag())
|
155
|
+
flags.DEFINE_flag(BuildDataFlag())
|
156
|
+
_define_help_flags_called = True
|
157
|
+
|
158
|
+
|
159
|
+
def RegisterAndParseFlagsWithUsage():
|
160
|
+
"""Register help flags, parse arguments and show usage if appropriate.
|
161
|
+
|
162
|
+
Returns:
|
163
|
+
remaining arguments after flags parsing
|
164
|
+
"""
|
165
|
+
DefineHelpFlags()
|
166
|
+
|
167
|
+
argv = parse_flags_with_usage(sys.argv)
|
168
|
+
return argv
|
169
|
+
|
170
|
+
|
171
|
+
def really_start(main=None):
|
172
|
+
"""Initializes flag values, and calls main with non-flag arguments.
|
173
|
+
|
174
|
+
Only non-flag arguments are passed to main(). The return value of main() is
|
175
|
+
used as the exit status.
|
176
|
+
|
177
|
+
Args:
|
178
|
+
main: Main function to run with the list of non-flag arguments, or None
|
179
|
+
so that sys.modules['__main__'].main is to be used.
|
180
|
+
"""
|
181
|
+
argv = RegisterAndParseFlagsWithUsage()
|
182
|
+
|
183
|
+
if main is None:
|
184
|
+
main = sys.modules['__main__'].main
|
185
|
+
|
186
|
+
try:
|
187
|
+
if FLAGS.run_with_pdb:
|
188
|
+
sys.exit(pdb.runcall(main, argv))
|
189
|
+
else:
|
190
|
+
if FLAGS.run_with_profiling:
|
191
|
+
# Avoid import overhead since most apps (including performance-sensitive
|
192
|
+
# ones) won't be run with profiling.
|
193
|
+
import atexit
|
194
|
+
if FLAGS.use_cprofile_for_profiling:
|
195
|
+
import cProfile as profile
|
196
|
+
else:
|
197
|
+
import profile
|
198
|
+
profiler = profile.Profile()
|
199
|
+
atexit.register(profiler.print_stats)
|
200
|
+
retval = profiler.runcall(main, argv)
|
201
|
+
sys.exit(retval)
|
202
|
+
else:
|
203
|
+
sys.exit(main(argv))
|
204
|
+
except UsageError, error:
|
205
|
+
usage(shorthelp=1, detailed_error=error, exitcode=error.exitcode)
|
206
|
+
|
207
|
+
|
208
|
+
def run():
|
209
|
+
"""Begin executing the program.
|
210
|
+
|
211
|
+
- Parses command line flags with the flag module.
|
212
|
+
- If there are any errors, print usage().
|
213
|
+
- Calls main() with the remaining arguments.
|
214
|
+
- If main() raises a UsageError, print usage and the error message.
|
215
|
+
"""
|
216
|
+
return _actual_start()
|
217
|
+
|
218
|
+
|
219
|
+
def _actual_start():
|
220
|
+
"""Another layer in the starting stack."""
|
221
|
+
# Get raw traceback
|
222
|
+
tb = None
|
223
|
+
try:
|
224
|
+
raise ZeroDivisionError('')
|
225
|
+
except ZeroDivisionError:
|
226
|
+
tb = sys.exc_info()[2]
|
227
|
+
assert tb
|
228
|
+
|
229
|
+
# Look at previous stack frame's previous stack frame (previous
|
230
|
+
# frame is run() or start(); the frame before that should be the
|
231
|
+
# frame of the original caller, which should be __main__ or appcommands
|
232
|
+
prev_prev_frame = tb.tb_frame.f_back.f_back
|
233
|
+
if not prev_prev_frame:
|
234
|
+
return
|
235
|
+
prev_prev_name = prev_prev_frame.f_globals.get('__name__', None)
|
236
|
+
if (prev_prev_name != '__main__'
|
237
|
+
and not prev_prev_name.endswith('.appcommands')):
|
238
|
+
return
|
239
|
+
# just in case there's non-trivial stuff happening in __main__
|
240
|
+
del tb
|
241
|
+
sys.exc_clear()
|
242
|
+
|
243
|
+
try:
|
244
|
+
really_start()
|
245
|
+
except SystemExit, e:
|
246
|
+
raise
|
247
|
+
except Exception, e:
|
248
|
+
# Call any installed exception handlers which may, for example,
|
249
|
+
# log to a file or send email.
|
250
|
+
for handler in EXCEPTION_HANDLERS:
|
251
|
+
try:
|
252
|
+
if handler.Wants(e):
|
253
|
+
handler.Handle(e)
|
254
|
+
except:
|
255
|
+
# We don't want to stop for exceptions in the exception handlers but
|
256
|
+
# we shouldn't hide them either.
|
257
|
+
sys.stderr.write(traceback.format_exc())
|
258
|
+
raise
|
259
|
+
# All handlers have had their chance, now die as we would have normally.
|
260
|
+
raise
|
261
|
+
|
262
|
+
|
263
|
+
def usage(shorthelp=0, writeto_stdout=0, detailed_error=None, exitcode=None):
|
264
|
+
"""Write __main__'s docstring to stderr with some help text.
|
265
|
+
|
266
|
+
Args:
|
267
|
+
shorthelp: print only flags from this module, rather than all flags.
|
268
|
+
writeto_stdout: write help message to stdout, rather than to stderr.
|
269
|
+
detailed_error: additional detail about why usage info was presented.
|
270
|
+
exitcode: if set, exit with this status code after writing help.
|
271
|
+
"""
|
272
|
+
if writeto_stdout:
|
273
|
+
stdfile = sys.stdout
|
274
|
+
else:
|
275
|
+
stdfile = sys.stderr
|
276
|
+
|
277
|
+
doc = sys.modules['__main__'].__doc__
|
278
|
+
if not doc:
|
279
|
+
doc = '\nUSAGE: %s [flags]\n' % sys.argv[0]
|
280
|
+
doc = flags.TextWrap(doc, indent=' ', firstline_indent='')
|
281
|
+
else:
|
282
|
+
# Replace all '%s' with sys.argv[0], and all '%%' with '%'.
|
283
|
+
num_specifiers = doc.count('%') - 2 * doc.count('%%')
|
284
|
+
try:
|
285
|
+
doc %= (sys.argv[0],) * num_specifiers
|
286
|
+
except (OverflowError, TypeError, ValueError):
|
287
|
+
# Just display the docstring as-is.
|
288
|
+
pass
|
289
|
+
if help_text_wrap:
|
290
|
+
doc = flags.TextWrap(flags.DocToHelp(doc))
|
291
|
+
if shorthelp:
|
292
|
+
flag_str = FLAGS.MainModuleHelp()
|
293
|
+
else:
|
294
|
+
flag_str = str(FLAGS)
|
295
|
+
try:
|
296
|
+
stdfile.write(doc)
|
297
|
+
if flag_str:
|
298
|
+
stdfile.write('\nflags:\n')
|
299
|
+
stdfile.write(flag_str)
|
300
|
+
stdfile.write('\n')
|
301
|
+
if detailed_error is not None:
|
302
|
+
stdfile.write('\n%s\n' % detailed_error)
|
303
|
+
except IOError, e:
|
304
|
+
# We avoid printing a huge backtrace if we get EPIPE, because
|
305
|
+
# "foo.par --help | less" is a frequent use case.
|
306
|
+
if e.errno != errno.EPIPE:
|
307
|
+
raise
|
308
|
+
if exitcode is not None:
|
309
|
+
sys.exit(exitcode)
|
310
|
+
|
311
|
+
|
312
|
+
class ExceptionHandler(object):
|
313
|
+
"""Base exception handler from which other may inherit."""
|
314
|
+
|
315
|
+
def Wants(self, unused_exc):
|
316
|
+
"""Check if this exception handler want to handle this exception.
|
317
|
+
|
318
|
+
Args:
|
319
|
+
unused_exc: Exception, the current exception
|
320
|
+
|
321
|
+
Returns:
|
322
|
+
boolean
|
323
|
+
|
324
|
+
This base handler wants to handle all exceptions, override this
|
325
|
+
method if you want to be more selective.
|
326
|
+
"""
|
327
|
+
return True
|
328
|
+
|
329
|
+
def Handle(self, exc):
|
330
|
+
"""Do something with the current exception.
|
331
|
+
|
332
|
+
Args:
|
333
|
+
exc: Exception, the current exception
|
334
|
+
|
335
|
+
This method must be overridden.
|
336
|
+
"""
|
337
|
+
raise NotImplementedError()
|
338
|
+
|
339
|
+
|
340
|
+
def InstallExceptionHandler(handler):
|
341
|
+
"""Install an exception handler.
|
342
|
+
|
343
|
+
Args:
|
344
|
+
handler: an object conforming to the interface defined in ExceptionHandler
|
345
|
+
|
346
|
+
Raises:
|
347
|
+
TypeError: handler was not of the correct type
|
348
|
+
|
349
|
+
All installed exception handlers will be called if main() exits via
|
350
|
+
an abnormal exception, i.e. not one of SystemExit, KeyboardInterrupt,
|
351
|
+
FlagsError or UsageError.
|
352
|
+
"""
|
353
|
+
if not isinstance(handler, ExceptionHandler):
|
354
|
+
raise TypeError('handler of type %s does not inherit from ExceptionHandler'
|
355
|
+
% type(handler))
|
356
|
+
EXCEPTION_HANDLERS.append(handler)
|
@@ -0,0 +1,783 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# Copyright 2007 Google Inc. All Rights Reserved.
|
3
|
+
#
|
4
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
5
|
+
# you may not use this file except in compliance with the License.
|
6
|
+
# You may obtain a copy of the License at
|
7
|
+
#
|
8
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
9
|
+
#
|
10
|
+
# Unless required by applicable law or agreed to in writing, software
|
11
|
+
# distributed under the License is distributed on an "AS-IS" BASIS,
|
12
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
13
|
+
# See the License for the specific language governing permissions and
|
14
|
+
# limitations under the License.
|
15
|
+
|
16
|
+
"""This module is the base for programs that provide multiple commands.
|
17
|
+
|
18
|
+
This provides command line tools that have a few shared global flags,
|
19
|
+
followed by a command name, followed by command specific flags,
|
20
|
+
then by arguments. That is:
|
21
|
+
tool [--global_flags] command [--command_flags] [args]
|
22
|
+
|
23
|
+
The module is built on top of app.py and 'overrides' a bit of it. However
|
24
|
+
the interface is mostly the same. The main difference is that your main
|
25
|
+
is supposed to register commands and return without further execution
|
26
|
+
of the commands; pre checking is of course welcome! Also your
|
27
|
+
global initialization should call appcommands.Run() rather than app.run().
|
28
|
+
|
29
|
+
To register commands use AddCmd() or AddCmdFunc(). AddCmd() is used
|
30
|
+
for commands that derive from class Cmd and the AddCmdFunc() is used
|
31
|
+
to wrap simple functions.
|
32
|
+
|
33
|
+
This module itself registers the command 'help' that allows users
|
34
|
+
to retrieve help for all or specific commands.
|
35
|
+
|
36
|
+
Example:
|
37
|
+
|
38
|
+
<code>
|
39
|
+
from mx import DateTime
|
40
|
+
|
41
|
+
|
42
|
+
class CmdDate(appcommands.Cmd):
|
43
|
+
\"\"\"This docstring contains the help for the date command.\"\"\"
|
44
|
+
|
45
|
+
def Run(self, argv):
|
46
|
+
print DateTime.now()
|
47
|
+
|
48
|
+
|
49
|
+
def main(argv):
|
50
|
+
appcommands.AddCmd('date', CmdDate, command_aliases=['data_now'])
|
51
|
+
|
52
|
+
|
53
|
+
if __name__ == '__main__':
|
54
|
+
appcommands.Run()
|
55
|
+
</code>
|
56
|
+
|
57
|
+
In the above example the name of the registered command on the command line is
|
58
|
+
'date'. Thus, to get the date you would execute:
|
59
|
+
tool date
|
60
|
+
The above example also added the command alias 'data_now' which allows to
|
61
|
+
replace 'tool date' with 'tool data_now'.
|
62
|
+
|
63
|
+
To get a list of available commands run:
|
64
|
+
tool help
|
65
|
+
For help with a specific command, you would execute:
|
66
|
+
tool help date
|
67
|
+
For help on flags run one of the following:
|
68
|
+
tool --help
|
69
|
+
Note that 'tool --help' gives you information on global flags, just like for
|
70
|
+
applications that do not use appcommand. Likewise 'tool --helpshort' and the
|
71
|
+
other help-flags from app.py are also available.
|
72
|
+
|
73
|
+
The above example also demonstrates that you only have to call
|
74
|
+
appcommands.Run()
|
75
|
+
and register your commands in main() to initialize your program with appcommands
|
76
|
+
(and app).
|
77
|
+
|
78
|
+
Handling of flags:
|
79
|
+
Flags can be registered just as with any other google tool using flags.py.
|
80
|
+
In addition you can also provide command specific flags. To do so simply add
|
81
|
+
flags registering code into the __init__ function of your Cmd classes passing
|
82
|
+
parameter flag_values to any flags registering calls. These flags will get
|
83
|
+
copied to the global flag list, so that once the command is detected they
|
84
|
+
behave just like any other flag. That means these flags won't be available
|
85
|
+
for other commands. Note that it is possible to register flags with more
|
86
|
+
than one command.
|
87
|
+
|
88
|
+
Getting help:
|
89
|
+
This module activates formatting and wrapping to help output. That is
|
90
|
+
the main difference to help created from app.py. So just as with app.py,
|
91
|
+
appcommands.py will create help from the main modules main __doc__.
|
92
|
+
But it adds the new 'help' command that allows you to get a list of
|
93
|
+
all available commands. Each command's help will be followed by the
|
94
|
+
registered command specific flags along with their defaults and help.
|
95
|
+
After help for all commands there will also be a list of all registered
|
96
|
+
global flags with their defaults and help.
|
97
|
+
|
98
|
+
The text for the command's help can best be supplied by overwriting the
|
99
|
+
__doc__ property of the Cmd classes for commands registered with AddCmd() or
|
100
|
+
the __doc__ property of command functions registered AddCmdFunc().
|
101
|
+
|
102
|
+
Inner working:
|
103
|
+
This module interacts with app.py by replacing its inner start dispatcher.
|
104
|
+
The replacement version basically does the same, registering help flags,
|
105
|
+
checking whether help flags were present, and calling the main module's main
|
106
|
+
function. However unlike app.py, this module epxpects main() to only register
|
107
|
+
commands and then to return. After having all commands registered
|
108
|
+
appcommands.py will then parse the remaining arguments for any registered
|
109
|
+
command. If one is found it will get executed. Otherwise a short usage info
|
110
|
+
will be displayed.
|
111
|
+
|
112
|
+
Each provided command must be an instance of Cmd. If commands get registered
|
113
|
+
from global functions using AddCmdFunc() then the helper class _FunctionalCmd
|
114
|
+
will be used in the registering process.
|
115
|
+
"""
|
116
|
+
|
117
|
+
|
118
|
+
|
119
|
+
import os
|
120
|
+
import pdb
|
121
|
+
import sys
|
122
|
+
import traceback
|
123
|
+
|
124
|
+
from google.apputils import app
|
125
|
+
import gflags as flags
|
126
|
+
|
127
|
+
FLAGS = flags.FLAGS
|
128
|
+
|
129
|
+
|
130
|
+
# module exceptions:
|
131
|
+
class AppCommandsError(Exception):
|
132
|
+
"""The base class for all flags errors."""
|
133
|
+
pass
|
134
|
+
|
135
|
+
|
136
|
+
_cmd_argv = None # remaining arguments with index 0 = sys.argv[0]
|
137
|
+
_cmd_list = {} # list of commands index by name (_Cmd instances)
|
138
|
+
_cmd_alias_list = {} # list of command_names index by command_alias
|
139
|
+
|
140
|
+
|
141
|
+
def GetAppBasename():
|
142
|
+
"""Returns the friendly basename of this application."""
|
143
|
+
return os.path.basename(sys.argv[0])
|
144
|
+
|
145
|
+
|
146
|
+
def ShortHelpAndExit(message=None):
|
147
|
+
"""Display optional message, followed by a note on how to get help, then exit.
|
148
|
+
|
149
|
+
Args:
|
150
|
+
message: optional message to display
|
151
|
+
"""
|
152
|
+
sys.stdout.flush()
|
153
|
+
if message is not None:
|
154
|
+
sys.stderr.write('%s\n' % message)
|
155
|
+
sys.stderr.write("Run '%s help' to get help\n" % GetAppBasename())
|
156
|
+
sys.exit(1)
|
157
|
+
|
158
|
+
|
159
|
+
def GetCommandList():
|
160
|
+
"""Return list of registered commands."""
|
161
|
+
# pylint: disable-msg=W0602
|
162
|
+
global _cmd_list
|
163
|
+
return _cmd_list
|
164
|
+
|
165
|
+
|
166
|
+
def GetCommandAliasList():
|
167
|
+
"""Return list of registered command aliases."""
|
168
|
+
# pylint: disable-msg=W0602
|
169
|
+
global _cmd_alias_list
|
170
|
+
return _cmd_alias_list
|
171
|
+
|
172
|
+
|
173
|
+
def GetFullCommandList():
|
174
|
+
"""Return list of registered commands, including aliases."""
|
175
|
+
all_cmds = dict(GetCommandList())
|
176
|
+
for cmd_alias, cmd_name in GetCommandAliasList().iteritems():
|
177
|
+
all_cmds[cmd_alias] = all_cmds.get(cmd_name)
|
178
|
+
return all_cmds
|
179
|
+
|
180
|
+
|
181
|
+
def GetCommandByName(name):
|
182
|
+
"""Get the command or None if name is not a registered command.
|
183
|
+
|
184
|
+
Args:
|
185
|
+
name: name of command to look for
|
186
|
+
|
187
|
+
Returns:
|
188
|
+
Cmd instance holding the command or None
|
189
|
+
"""
|
190
|
+
return GetCommandList().get(GetCommandAliasList().get(name))
|
191
|
+
|
192
|
+
|
193
|
+
def GetCommandArgv():
|
194
|
+
"""Return list of remaining args."""
|
195
|
+
return _cmd_argv
|
196
|
+
|
197
|
+
|
198
|
+
def GetMaxCommandLength():
|
199
|
+
"""Returns the length of the longest registered command."""
|
200
|
+
return max([len(cmd_name) for cmd_name in GetCommandList()])
|
201
|
+
|
202
|
+
|
203
|
+
class Cmd(object):
|
204
|
+
"""Abstract class describing and implementing a command.
|
205
|
+
|
206
|
+
When creating code for a command, at least you have to derive this class
|
207
|
+
and override method Run(). The other methods of this class might be
|
208
|
+
overridden as well. Check their documentation for details. If the command
|
209
|
+
needs any specific flags, use __init__ for registration.
|
210
|
+
"""
|
211
|
+
|
212
|
+
def __init__(self, name, flag_values, command_aliases=None):
|
213
|
+
"""Initialize and check whether self is actually a Cmd instance.
|
214
|
+
|
215
|
+
This can be used to register command specific flags. If you do so
|
216
|
+
remember that you have to provide the 'flag_values=flag_values'
|
217
|
+
parameter to any flags.DEFINE_*() call.
|
218
|
+
|
219
|
+
Args:
|
220
|
+
name: Name of the command
|
221
|
+
flag_values: FlagValues() instance that needs to be passed as
|
222
|
+
flag_values parameter to any flags registering call.
|
223
|
+
command_aliases: A list of command aliases that the command can be run as.
|
224
|
+
Raises:
|
225
|
+
AppCommandsError: if self is Cmd (Cmd is abstract)
|
226
|
+
"""
|
227
|
+
self._command_name = name
|
228
|
+
self._command_aliases = command_aliases
|
229
|
+
self._command_flags = flag_values
|
230
|
+
self._all_commands_help = None
|
231
|
+
if type(self) is Cmd:
|
232
|
+
raise AppCommandsError('Cmd is abstract and cannot be instantiated')
|
233
|
+
|
234
|
+
def Run(self, argv):
|
235
|
+
"""Execute the command. Must be provided by the implementing class.
|
236
|
+
|
237
|
+
Args:
|
238
|
+
argv: Remaining command line arguments after parsing flags and command
|
239
|
+
(that is a copy of sys.argv at the time of the function call with
|
240
|
+
all parsed flags removed).
|
241
|
+
|
242
|
+
Returns:
|
243
|
+
0 for success, anything else for failure (must return with integer).
|
244
|
+
Alternatively you may return None (or not use a return statement at all).
|
245
|
+
|
246
|
+
Raises:
|
247
|
+
AppCommandsError: Always as in must be overwritten
|
248
|
+
"""
|
249
|
+
raise AppCommandsError('%s.%s.Run() is not implemented' % (
|
250
|
+
type(self).__module__, type(self).__name__))
|
251
|
+
|
252
|
+
def CommandRun(self, argv):
|
253
|
+
"""Execute the command with given arguments.
|
254
|
+
|
255
|
+
First register and parse additional flags. Then run the command.
|
256
|
+
|
257
|
+
Returns:
|
258
|
+
Command return value.
|
259
|
+
|
260
|
+
Args:
|
261
|
+
argv: Remaining command line arguments after parsing command and flags
|
262
|
+
(that is a copy of sys.argv at the time of the function call with
|
263
|
+
all parsed flags removed).
|
264
|
+
"""
|
265
|
+
# Register flags global when run normally
|
266
|
+
FLAGS.AppendFlagValues(self._command_flags)
|
267
|
+
# Prepare flags parsing, to redirect help, to show help for command
|
268
|
+
orig_app_usage = app.usage
|
269
|
+
|
270
|
+
def ReplacementAppUsage(shorthelp=0, writeto_stdout=1, detailed_error=None,
|
271
|
+
exitcode=None):
|
272
|
+
AppcommandsUsage(shorthelp, writeto_stdout, detailed_error, exitcode=1,
|
273
|
+
show_cmd=self._command_name, show_global_flags=True)
|
274
|
+
app.usage = ReplacementAppUsage
|
275
|
+
# Parse flags and restore app.usage afterwards
|
276
|
+
try:
|
277
|
+
try:
|
278
|
+
argv = ParseFlagsWithUsage(argv)
|
279
|
+
# Run command
|
280
|
+
if FLAGS.run_with_pdb:
|
281
|
+
ret = pdb.runcall(self.Run, argv)
|
282
|
+
else:
|
283
|
+
ret = self.Run(argv)
|
284
|
+
if ret is None:
|
285
|
+
ret = 0
|
286
|
+
else:
|
287
|
+
assert isinstance(ret, int)
|
288
|
+
return ret
|
289
|
+
except app.UsageError, error:
|
290
|
+
app.usage(shorthelp=1, detailed_error=error, exitcode=error.exitcode)
|
291
|
+
finally:
|
292
|
+
# Restore app.usage and remove this command's flags from the global flags.
|
293
|
+
app.usage = orig_app_usage
|
294
|
+
for flag_name in self._command_flags.FlagDict():
|
295
|
+
delattr(FLAGS, flag_name)
|
296
|
+
|
297
|
+
def CommandGetHelp(self, unused_argv, cmd_names=None):
|
298
|
+
"""Get help string for command.
|
299
|
+
|
300
|
+
Args:
|
301
|
+
unused_argv: Remaining command line flags and arguments after parsing
|
302
|
+
command (that is a copy of sys.argv at the time of the
|
303
|
+
function call with all parsed flags removed); unused in this
|
304
|
+
default implementation, but may be used in subclasses.
|
305
|
+
cmd_names: Complete list of commands for which help is being shown at
|
306
|
+
the same time. This is used to determine whether to return
|
307
|
+
_all_commands_help, or the command's docstring.
|
308
|
+
(_all_commands_help is used, if not None, when help is being
|
309
|
+
shown for more than one command, otherwise the command's
|
310
|
+
docstring is used.)
|
311
|
+
|
312
|
+
Returns:
|
313
|
+
Help string, one of the following (by order):
|
314
|
+
- Result of the registered 'help' function (if any)
|
315
|
+
- Doc string of the Cmd class (if any)
|
316
|
+
- Default fallback string
|
317
|
+
"""
|
318
|
+
if (type(cmd_names) is list and len(cmd_names) > 1 and
|
319
|
+
self._all_commands_help is not None):
|
320
|
+
return flags.DocToHelp(self._all_commands_help)
|
321
|
+
elif self.__doc__:
|
322
|
+
return flags.DocToHelp(self.__doc__)
|
323
|
+
else:
|
324
|
+
return 'No help available'
|
325
|
+
|
326
|
+
def CommandGetAliases(self):
|
327
|
+
"""Get aliases for command.
|
328
|
+
|
329
|
+
Returns:
|
330
|
+
aliases: list of aliases for the command.
|
331
|
+
"""
|
332
|
+
return self._command_aliases
|
333
|
+
|
334
|
+
|
335
|
+
class _FunctionalCmd(Cmd):
|
336
|
+
"""Class to wrap functions as CMD instances.
|
337
|
+
|
338
|
+
Args:
|
339
|
+
cmd_func: command function
|
340
|
+
"""
|
341
|
+
|
342
|
+
def __init__(self, name, flag_values, cmd_func, all_commands_help=None,
|
343
|
+
**kargs):
|
344
|
+
"""Create a functional command.
|
345
|
+
|
346
|
+
Args:
|
347
|
+
name: Name of command
|
348
|
+
flag_values: FlagValues() instance that needs to be passed as flag_values
|
349
|
+
parameter to any flags registering call.
|
350
|
+
cmd_func: Function to call when command is to be executed.
|
351
|
+
"""
|
352
|
+
Cmd.__init__(self, name, flag_values, **kargs)
|
353
|
+
self._all_commands_help = all_commands_help
|
354
|
+
self._cmd_func = cmd_func
|
355
|
+
|
356
|
+
def CommandGetHelp(self, unused_argv, cmd_names=None):
|
357
|
+
"""Get help for command.
|
358
|
+
|
359
|
+
Args:
|
360
|
+
unused_argv: Remaining command line flags and arguments after parsing
|
361
|
+
command (that is a copy of sys.argv at the time of the
|
362
|
+
function call with all parsed flags removed); unused in this
|
363
|
+
implementation.
|
364
|
+
cmd_names: By default, if help is being shown for more than one command,
|
365
|
+
and this command defines _all_commands_help, then
|
366
|
+
_all_commands_help will be displayed instead of the class
|
367
|
+
doc. cmd_names is used to determine the number of commands
|
368
|
+
being displayed and if only a single command is display then
|
369
|
+
the class doc is returned.
|
370
|
+
|
371
|
+
Returns:
|
372
|
+
__doc__ property for command function or a message stating there is no
|
373
|
+
help.
|
374
|
+
"""
|
375
|
+
if (type(cmd_names) is list and len(cmd_names) > 1 and
|
376
|
+
self._all_commands_help is not None):
|
377
|
+
return flags.DocToHelp(self._all_commands_help)
|
378
|
+
if self._cmd_func.__doc__ is not None:
|
379
|
+
return flags.DocToHelp(self._cmd_func.__doc__)
|
380
|
+
else:
|
381
|
+
return 'No help available'
|
382
|
+
|
383
|
+
def Run(self, argv):
|
384
|
+
"""Execute the command with given arguments.
|
385
|
+
|
386
|
+
Args:
|
387
|
+
argv: Remaining command line flags and arguments after parsing command
|
388
|
+
(that is a copy of sys.argv at the time of the function call with
|
389
|
+
all parsed flags removed).
|
390
|
+
|
391
|
+
Returns:
|
392
|
+
Command return value.
|
393
|
+
"""
|
394
|
+
return self._cmd_func(argv)
|
395
|
+
|
396
|
+
|
397
|
+
def _AddCmdInstance(command_name, cmd, command_aliases=None):
|
398
|
+
"""Add a command from a Cmd instance.
|
399
|
+
|
400
|
+
Args:
|
401
|
+
command_name: name of the command which will be used in argument parsing
|
402
|
+
cmd: Cmd instance to register
|
403
|
+
command_aliases: A list of command aliases that the command can be run as.
|
404
|
+
|
405
|
+
Raises:
|
406
|
+
AppCommandsError: is command is already registered OR cmd is not a subclass
|
407
|
+
of Cmd
|
408
|
+
AppCommandsError: if name is already registered OR name is not a string OR
|
409
|
+
name is too short OR name does not start with a letter OR
|
410
|
+
name contains any non alphanumeric characters besides
|
411
|
+
'_'.
|
412
|
+
"""
|
413
|
+
# Update global command list.
|
414
|
+
# pylint: disable-msg=W0602
|
415
|
+
global _cmd_list
|
416
|
+
global _cmd_alias_list
|
417
|
+
if not issubclass(cmd.__class__, Cmd):
|
418
|
+
raise AppCommandsError('Command must be an instance of commands.Cmd')
|
419
|
+
|
420
|
+
for name in [command_name] + (command_aliases or []):
|
421
|
+
_CheckCmdName(name)
|
422
|
+
_cmd_alias_list[name] = command_name
|
423
|
+
|
424
|
+
_cmd_list[command_name] = cmd
|
425
|
+
|
426
|
+
|
427
|
+
def _CheckCmdName(name_or_alias):
|
428
|
+
"""Only allow strings for command names and aliases (reject unicode as well).
|
429
|
+
|
430
|
+
Args:
|
431
|
+
name_or_alias: properly formatted string name or alias.
|
432
|
+
|
433
|
+
Raises:
|
434
|
+
AppCommandsError: is command is already registered OR cmd is not a subclass
|
435
|
+
of Cmd
|
436
|
+
AppCommandsError: if name is already registered OR name is not a string OR
|
437
|
+
name is too short OR name does not start with a letter OR
|
438
|
+
name contains any non alphanumeric characters besides
|
439
|
+
'_'.
|
440
|
+
"""
|
441
|
+
if name_or_alias in GetCommandAliasList():
|
442
|
+
raise AppCommandsError("Command or Alias '%s' already defined" %
|
443
|
+
name_or_alias)
|
444
|
+
if not isinstance(name_or_alias, str) or len(name_or_alias) <= 1:
|
445
|
+
raise AppCommandsError("Command '%s' not a string or too short"
|
446
|
+
% str(name_or_alias))
|
447
|
+
if not name_or_alias[0].isalpha():
|
448
|
+
raise AppCommandsError("Command '%s' does not start with a letter"
|
449
|
+
% name_or_alias)
|
450
|
+
if [c for c in name_or_alias if not (c.isalnum() or c == '_')]:
|
451
|
+
raise AppCommandsError("Command '%s' contains non alphanumeric characters"
|
452
|
+
% name_or_alias)
|
453
|
+
|
454
|
+
|
455
|
+
def AddCmd(command_name, cmd_factory, **kargs):
|
456
|
+
"""Add a command from a Cmd subclass or factory.
|
457
|
+
|
458
|
+
Args:
|
459
|
+
command_name: name of the command which will be used in argument parsing
|
460
|
+
cmd_factory: A callable whose arguments match those of Cmd.__init__ and
|
461
|
+
returns a Cmd. In the simplest case this is just a subclass
|
462
|
+
of Cmd.
|
463
|
+
command_aliases: A list of command aliases that the command can be run as.
|
464
|
+
|
465
|
+
Raises:
|
466
|
+
AppCommandsError: if calling cmd_factory does not return an instance of Cmd.
|
467
|
+
"""
|
468
|
+
cmd = cmd_factory(command_name, flags.FlagValues(), **kargs)
|
469
|
+
|
470
|
+
if not isinstance(cmd, Cmd):
|
471
|
+
raise AppCommandsError('Command must be an instance of commands.Cmd')
|
472
|
+
|
473
|
+
_AddCmdInstance(command_name, cmd, **kargs)
|
474
|
+
|
475
|
+
|
476
|
+
def AddCmdFunc(command_name, cmd_func, command_aliases=None,
|
477
|
+
all_commands_help=None):
|
478
|
+
"""Add a new command to the list of registered commands.
|
479
|
+
|
480
|
+
Args:
|
481
|
+
command_name: name of the command which will be used in argument
|
482
|
+
parsing
|
483
|
+
cmd_func: command function, this function received the remaining
|
484
|
+
arguments as its only parameter. It is supposed to do the
|
485
|
+
command work and then return with the command result that
|
486
|
+
is being used as the shell exit code.
|
487
|
+
command_aliases: A list of command aliases that the command can be run as.
|
488
|
+
all_commands_help: Help message to be displayed in place of func.__doc__
|
489
|
+
when all commands are displayed.
|
490
|
+
"""
|
491
|
+
_AddCmdInstance(command_name,
|
492
|
+
_FunctionalCmd(command_name, flags.FlagValues(), cmd_func,
|
493
|
+
command_aliases=command_aliases,
|
494
|
+
all_commands_help=all_commands_help),
|
495
|
+
command_aliases)
|
496
|
+
|
497
|
+
|
498
|
+
class _CmdHelp(Cmd):
|
499
|
+
"""Standard help command.
|
500
|
+
|
501
|
+
Allows to provide help for all or specific commands.
|
502
|
+
"""
|
503
|
+
|
504
|
+
def Run(self, argv):
|
505
|
+
"""Execute help command.
|
506
|
+
|
507
|
+
If an argument is given and that argument is a registered command
|
508
|
+
name, then help specific to that command is being displayed.
|
509
|
+
If the command is unknown then a fatal error will be displayed. If
|
510
|
+
no argument is present then help for all commands will be presented.
|
511
|
+
|
512
|
+
If a specific command help is being generated, the list of commands is
|
513
|
+
temporarily replaced with one containing only that command. Thus the call
|
514
|
+
to usage() will only show help for that command. Otherwise call usage()
|
515
|
+
will show help for all registered commands as it sees all commands.
|
516
|
+
|
517
|
+
Args:
|
518
|
+
argv: Remaining command line flags and arguments after parsing command
|
519
|
+
(that is a copy of sys.argv at the time of the function call with
|
520
|
+
all parsed flags removed).
|
521
|
+
So argv[0] is the program and argv[1] will be the first argument to
|
522
|
+
the call. For instance 'tool.py help command' will result in argv
|
523
|
+
containing ('tool.py', 'command'). In this case the list of
|
524
|
+
commands is searched for 'command'.
|
525
|
+
|
526
|
+
Returns:
|
527
|
+
1 for failure
|
528
|
+
"""
|
529
|
+
if len(argv) > 1 and argv[1] in GetFullCommandList():
|
530
|
+
show_cmd = argv[1]
|
531
|
+
else:
|
532
|
+
show_cmd = None
|
533
|
+
AppcommandsUsage(shorthelp=0, writeto_stdout=1, detailed_error=None,
|
534
|
+
exitcode=1, show_cmd=show_cmd, show_global_flags=False)
|
535
|
+
|
536
|
+
def CommandGetHelp(self, unused_argv, cmd_names=None):
|
537
|
+
"""Returns: Help for command."""
|
538
|
+
cmd_help = ('Help for all or selected command:\n'
|
539
|
+
'\t%(prog)s help [<command>]\n\n'
|
540
|
+
'To retrieve help with global flags:\n'
|
541
|
+
'\t%(prog)s --help\n\n'
|
542
|
+
'To retrieve help with flags only from the main module:\n'
|
543
|
+
'\t%(prog)s --helpshort [<command>]\n\n'
|
544
|
+
% {'prog': GetAppBasename()})
|
545
|
+
return flags.DocToHelp(cmd_help)
|
546
|
+
|
547
|
+
|
548
|
+
def GetSynopsis():
|
549
|
+
"""Get synopsis for program.
|
550
|
+
|
551
|
+
Returns:
|
552
|
+
Synopsis including program basename.
|
553
|
+
"""
|
554
|
+
return '%s [--global_flags] <command> [--command_flags] [args]' % (
|
555
|
+
GetAppBasename())
|
556
|
+
|
557
|
+
|
558
|
+
def _UsageFooter(detailed_error, cmd_names):
|
559
|
+
"""Output a footer at the end of usage or help output.
|
560
|
+
|
561
|
+
Args:
|
562
|
+
detailed_error: additional detail about why usage info was presented.
|
563
|
+
cmd_names: list of command names for which help was shown or None.
|
564
|
+
Returns:
|
565
|
+
Generated footer that contains 'Run..' messages if appropriate.
|
566
|
+
"""
|
567
|
+
footer = []
|
568
|
+
if not cmd_names or len(cmd_names) == 1:
|
569
|
+
footer.append("Run '%s help' to see the list of available commands."
|
570
|
+
% GetAppBasename())
|
571
|
+
if not cmd_names or len(cmd_names) == len(GetCommandList()):
|
572
|
+
footer.append("Run '%s help <command>' to get help for <command>."
|
573
|
+
% GetAppBasename())
|
574
|
+
if detailed_error is not None:
|
575
|
+
if footer:
|
576
|
+
footer.append('')
|
577
|
+
footer.append('%s' % detailed_error)
|
578
|
+
return '\n'.join(footer)
|
579
|
+
|
580
|
+
|
581
|
+
def AppcommandsUsage(shorthelp=0, writeto_stdout=0, detailed_error=None,
|
582
|
+
exitcode=None, show_cmd=None, show_global_flags=False):
|
583
|
+
"""Output usage or help information.
|
584
|
+
|
585
|
+
Extracts the __doc__ string from the __main__ module and writes it to
|
586
|
+
stderr. If that string contains a '%s' then that is replaced by the command
|
587
|
+
pathname. Otherwise a default usage string is being generated.
|
588
|
+
|
589
|
+
The output varies depending on the following:
|
590
|
+
- FLAGS.help
|
591
|
+
- FLAGS.helpshort
|
592
|
+
- show_cmd
|
593
|
+
- show_global_flags
|
594
|
+
|
595
|
+
Args:
|
596
|
+
shorthelp: print only command and main module flags, rather than all.
|
597
|
+
writeto_stdout: write help message to stdout, rather than to stderr.
|
598
|
+
detailed_error: additional details about why usage info was presented.
|
599
|
+
exitcode: if set, exit with this status code after writing help.
|
600
|
+
show_cmd: show help for this command only (name of command).
|
601
|
+
show_global_flags: show help for global flags.
|
602
|
+
"""
|
603
|
+
if writeto_stdout:
|
604
|
+
stdfile = sys.stdout
|
605
|
+
else:
|
606
|
+
stdfile = sys.stderr
|
607
|
+
|
608
|
+
prefix = ''.rjust(GetMaxCommandLength() + 2)
|
609
|
+
# Deal with header, containing general tool documentation
|
610
|
+
doc = sys.modules['__main__'].__doc__
|
611
|
+
if doc:
|
612
|
+
help_msg = flags.DocToHelp(doc.replace('%s', sys.argv[0]))
|
613
|
+
stdfile.write(flags.TextWrap(help_msg, flags.GetHelpWidth()))
|
614
|
+
stdfile.write('\n\n\n')
|
615
|
+
if not doc or doc.find('%s') == -1:
|
616
|
+
synopsis = 'USAGE: ' + GetSynopsis()
|
617
|
+
stdfile.write(flags.TextWrap(synopsis, flags.GetHelpWidth(), ' ',
|
618
|
+
''))
|
619
|
+
stdfile.write('\n\n\n')
|
620
|
+
# Special case just 'help' registered, that means run as 'tool --help'.
|
621
|
+
if len(GetCommandList()) == 1:
|
622
|
+
cmd_names = []
|
623
|
+
else:
|
624
|
+
# Show list of commands
|
625
|
+
if show_cmd is None or show_cmd == 'help':
|
626
|
+
cmd_names = GetCommandList().keys()
|
627
|
+
cmd_names.sort()
|
628
|
+
stdfile.write('Any of the following commands:\n')
|
629
|
+
doc = ', '.join(cmd_names)
|
630
|
+
stdfile.write(flags.TextWrap(doc, flags.GetHelpWidth(), ' '))
|
631
|
+
stdfile.write('\n\n\n')
|
632
|
+
# Prepare list of commands to show help for
|
633
|
+
if show_cmd is not None:
|
634
|
+
cmd_names = [show_cmd] # show only one command
|
635
|
+
elif FLAGS.help or FLAGS.helpshort or shorthelp:
|
636
|
+
cmd_names = []
|
637
|
+
else:
|
638
|
+
cmd_names = GetCommandList().keys() # show all commands
|
639
|
+
cmd_names.sort()
|
640
|
+
# Show the command help (none, one specific, or all)
|
641
|
+
for name in cmd_names:
|
642
|
+
command = GetCommandByName(name)
|
643
|
+
cmd_help = command.CommandGetHelp(GetCommandArgv(), cmd_names=cmd_names)
|
644
|
+
cmd_help = cmd_help.strip()
|
645
|
+
all_names = ', '.join([name] + (command.CommandGetAliases() or []))
|
646
|
+
if len(all_names) + 1 >= len(prefix) or not cmd_help:
|
647
|
+
# If command/alias list would reach over help block-indent
|
648
|
+
# start the help block on a new line.
|
649
|
+
stdfile.write(flags.TextWrap(all_names, flags.GetHelpWidth()))
|
650
|
+
stdfile.write('\n')
|
651
|
+
prefix1 = prefix
|
652
|
+
else:
|
653
|
+
prefix1 = all_names.ljust(GetMaxCommandLength() + 2)
|
654
|
+
if cmd_help:
|
655
|
+
stdfile.write(flags.TextWrap(cmd_help, flags.GetHelpWidth(), prefix,
|
656
|
+
prefix1))
|
657
|
+
stdfile.write('\n\n')
|
658
|
+
else:
|
659
|
+
stdfile.write('\n')
|
660
|
+
# When showing help for exactly one command we show its flags
|
661
|
+
if len(cmd_names) == 1:
|
662
|
+
# Need to register flags for command prior to be able to use them.
|
663
|
+
# We do not register them globally so that they do not reappear.
|
664
|
+
# pylint: disable-msg=W0212
|
665
|
+
cmd_flags = command._command_flags
|
666
|
+
if cmd_flags.RegisteredFlags():
|
667
|
+
stdfile.write('%sFlags for %s:\n' % (prefix, name))
|
668
|
+
stdfile.write(cmd_flags.GetHelp(prefix+' '))
|
669
|
+
stdfile.write('\n\n')
|
670
|
+
stdfile.write('\n')
|
671
|
+
# Now show global flags as asked for
|
672
|
+
if show_global_flags:
|
673
|
+
stdfile.write('Global flags:\n')
|
674
|
+
if shorthelp:
|
675
|
+
stdfile.write(FLAGS.MainModuleHelp())
|
676
|
+
else:
|
677
|
+
stdfile.write(FLAGS.GetHelp())
|
678
|
+
stdfile.write('\n')
|
679
|
+
else:
|
680
|
+
stdfile.write("Run '%s --help' to get help for global flags."
|
681
|
+
% GetAppBasename())
|
682
|
+
stdfile.write('\n%s\n' % _UsageFooter(detailed_error, cmd_names))
|
683
|
+
if exitcode is not None:
|
684
|
+
sys.exit(exitcode)
|
685
|
+
|
686
|
+
|
687
|
+
def ParseFlagsWithUsage(argv):
|
688
|
+
"""Parse the flags, exiting (after printing usage) if they are unparseable.
|
689
|
+
|
690
|
+
Args:
|
691
|
+
argv: command line arguments
|
692
|
+
|
693
|
+
Returns:
|
694
|
+
remaining command line arguments after parsing flags
|
695
|
+
"""
|
696
|
+
# Update the global commands.
|
697
|
+
# pylint: disable-msg=W0603
|
698
|
+
global _cmd_argv
|
699
|
+
try:
|
700
|
+
_cmd_argv = FLAGS(argv)
|
701
|
+
return _cmd_argv
|
702
|
+
except flags.FlagsError, error:
|
703
|
+
ShortHelpAndExit('FATAL Flags parsing error: %s' % error)
|
704
|
+
|
705
|
+
|
706
|
+
def GetCommand(command_required):
|
707
|
+
"""Get the command or return None (or issue an error) if there is none.
|
708
|
+
|
709
|
+
Args:
|
710
|
+
command_required: whether to issue an error if no command is present
|
711
|
+
|
712
|
+
Returns:
|
713
|
+
command or None, if command_required is True then return value is a valid
|
714
|
+
command or the program will exit. The program also exits if a command was
|
715
|
+
specified but that command does not exist.
|
716
|
+
"""
|
717
|
+
# Update the global commands.
|
718
|
+
# pylint: disable-msg=W0603
|
719
|
+
global _cmd_argv
|
720
|
+
_cmd_argv = ParseFlagsWithUsage(_cmd_argv)
|
721
|
+
if len(_cmd_argv) < 2:
|
722
|
+
if command_required:
|
723
|
+
ShortHelpAndExit('FATAL Command expected but none given')
|
724
|
+
return None
|
725
|
+
command = GetCommandByName(_cmd_argv[1])
|
726
|
+
if command is None:
|
727
|
+
ShortHelpAndExit("FATAL Command '%s' unknown" % _cmd_argv[1])
|
728
|
+
del _cmd_argv[1]
|
729
|
+
return command
|
730
|
+
|
731
|
+
|
732
|
+
def _CommandsStart():
|
733
|
+
"""Main initialization.
|
734
|
+
|
735
|
+
This initializes flag values, and calls __main__.main(). Only non-flag
|
736
|
+
arguments are passed to main(). The return value of main() is used as the
|
737
|
+
exit status.
|
738
|
+
|
739
|
+
"""
|
740
|
+
app.RegisterAndParseFlagsWithUsage()
|
741
|
+
# The following is supposed to return after registering additional commands
|
742
|
+
try:
|
743
|
+
sys.modules['__main__'].main(GetCommandArgv())
|
744
|
+
# If sys.exit was called, return with error code.
|
745
|
+
except SystemExit, e:
|
746
|
+
sys.exit(e.code)
|
747
|
+
except Exception, error:
|
748
|
+
traceback.print_exc() # Print a backtrace to stderr.
|
749
|
+
ShortHelpAndExit('\nFATAL error in main: %s' % error)
|
750
|
+
|
751
|
+
if len(GetCommandArgv()) > 1:
|
752
|
+
command = GetCommand(command_required=True)
|
753
|
+
else:
|
754
|
+
command = GetCommandByName('help')
|
755
|
+
sys.exit(command.CommandRun(GetCommandArgv()))
|
756
|
+
|
757
|
+
|
758
|
+
def Run():
|
759
|
+
"""This must be called from __main__ modules main, instead of app.run().
|
760
|
+
|
761
|
+
app.run will base its actions on its stacktrace.
|
762
|
+
|
763
|
+
Returns:
|
764
|
+
app.run()
|
765
|
+
"""
|
766
|
+
app.parse_flags_with_usage = ParseFlagsWithUsage
|
767
|
+
app.really_start = _CommandsStart
|
768
|
+
app.usage = _ReplacementAppUsage
|
769
|
+
return app.run()
|
770
|
+
|
771
|
+
|
772
|
+
# Always register 'help' command
|
773
|
+
AddCmd('help', _CmdHelp)
|
774
|
+
|
775
|
+
|
776
|
+
def _ReplacementAppUsage(shorthelp=0, writeto_stdout=0, detailed_error=None,
|
777
|
+
exitcode=None):
|
778
|
+
AppcommandsUsage(shorthelp, writeto_stdout, detailed_error, exitcode=exitcode,
|
779
|
+
show_cmd=None, show_global_flags=True)
|
780
|
+
|
781
|
+
|
782
|
+
if __name__ == '__main__':
|
783
|
+
Run()
|