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
|
#!/usr/bin/env python
"""bcfg2-admin is a script that helps to administrate a Bcfg2 deployment."""
from optparse import OptionParser
import logging
import Bcfg2.Server.Core
import Bcfg2.Logger
import Bcfg2.Options
# Compatibility import
from Bcfg2.Bcfg2Py3k import StringIO
log = logging.getLogger('bcfg2-admin')
import Bcfg2.Server.Admin
def mode_import(modename):
"""Load Bcfg2.Server.Admin.<mode>."""
modname = modename.capitalize()
mod = getattr(__import__("Bcfg2.Server.Admin.%s" %
(modname)).Server.Admin, modname)
return getattr(mod, modname)
def get_modes():
"""Get all available modes, except for the base mode."""
return [x.lower() for x in Bcfg2.Server.Admin.__all__ if x != 'mode']
def create_description():
"""Create the description string from the list of modes."""
modes = get_modes()
description = StringIO()
description.write("Available modes are:\n\n")
for mode in modes:
try:
description.write((" %-15s %s\n" %
(mode, mode_import(mode).__shorthelp__)))
except ImportError:
continue
return description.getvalue()
def main():
Bcfg2.Logger.setup_logging('bcfg2-admin', to_console=True, level=40)
usage = "Usage: %prog [options] MODE [args]"
parser = OptionParser(usage=usage)
parser.set_defaults(configfile=Bcfg2.Options.CFILE.default)
parser.add_option("-C", "--configfile",
dest="configfile",
help="Path to bcfg2.conf",
metavar="FILE")
parser.disable_interspersed_args()
(options, args) = parser.parse_args()
# Provide help if requested or no args were specified
if len(args) < 1 or args[0] == 'help':
if len(args) > 1:
# Get help for a specific mode by passing it the help argument
args = [args[1], args[0]]
else:
# Print short help for all modes
parser.print_help()
print(create_description())
raise SystemExit(0)
if args[0] in get_modes():
modname = args[0].capitalize()
if len(args) == 1 or args[1] == 'help':
print(mode_import(modname).__longhelp__)
raise SystemExit(0)
try:
mode_cls = mode_import(modname)
except ImportError:
e = sys.exc_info()[1]
log.error("Failed to load admin mode %s: %s" % (modname, e))
raise SystemExit(1)
mode = mode_cls(options.configfile)
mode(args[1:])
if hasattr(mode, 'bcore'):
mode.bcore.shutdown()
else:
log.error("Unknown mode %s" % args[0])
parser.print_help()
print(create_description())
raise SystemExit(1)
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
raise SystemExit(1)
|
