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
|
'''Option parsing library for utilities'''
__revision__ = '$Revision$'
import getopt, os, sys, ConfigParser
class OptionFailure(Exception):
pass
class Option(object):
cfpath = '/etc/bcfg2.conf'
__cfp = False
def getCFP(self):
if not self.__cfp:
self.__cfp = ConfigParser.ConfigParser()
self.__cfp.readfp(open(self.cfpath))
cfp = property(getCFP)
def getValue(self):
if self.cook:
return self.cook(self._value)
else:
return self._value
value = property(getValue)
def __init__(self, desc, default, cmd=False, odesc=False,
env=False, cf=False, cook=False):
self.desc = desc
self.default = default
self.cmd = cmd
if cmd and (cmd[0] != '-' or len(cmd) != 2):
raise OptionFailure("Poorly formed command %s" % cmd)
self.odesg = odesc
self.env = env
self.cf = cf
self.cook = cook
def buildHelpMessage(self):
msg = ''
if self.cmd:
msg = self.cmd.ljust(3)
if self.odesc:
msg += ':%-24s' % (self.odesc)
msg += "%s\n" % self.desc
return msg
def buildGetopt(self):
gstr = ''
if self.cmd:
gstr = self.cmd[1]
if self.odesc:
gstr += ':'
return gstr
def parse(self, opts, rawopts):
if self.cmd and opts:
# processing getopted data
optinfo = [opt[1] for opt in opts if opt[0] == option[0]]
if optinfo:
self._value = optinfo
return
if self.cmd and self.cmd in rawopts:
self._value = rawopts[rawopts.index(self.cmd) + 1]
return
# no command line option found
if self.env and self.env in os.environ:
self._value = os.environ[self.env]
return
if self.cf:
try:
if self.cf in locations:
self._value = self.cfp.get(*locations[self.cf])
else:
self._value = self.cfp.get(*self.cf)
except:
pass
self._value = self.default
class OptionSet(dict):
def buildGetopt(self):
return ''.join([opt.buildGetopt() for opt in self.values()])
def buildHelpMessage(self):
return ''.join([opt.buildHelpMessage() for opt in self.values()])
def helpExit(self, msg='', code=1):
if msg:
print msg
print "Usage:"
print self.buildHelpMessage()
raise SystemExit(code)
def parse(self, argv, do_getopt=True):
'''Parse options'''
ret = {}
if do_getopt:
try:
opts, args = getopt.getopt(argv, self.buildHelpGetopt(), [])
except getopt.GetoptError, err:
self.helpExit(err)
if '-h' in argv:
self.helpExit(err)
for key in self.keys():
option = self[key]
if do_getopt:
option.parse(opts, [])
else:
option.parse([], argv)
if hasattr(option, '_value'):
val = option.value
self[key] = val
class OptionParser(OptionSet):
'''OptionParser bootstraps option parsing, getting the value of the config file'''
def __init__(self, args):
self.Bootstrap = OptionSet(['configfile', Option('config file path',
'/etc/bcfg2.conf',
cmd='-C')])
self.Bootstrap.parse(sys.argv[1:], do_getopt=False)
if self.Bootstrap['configfile'] != '/etc/bcfg2.conf':
Option.cfpath = self.Bootstrap['configfile']
Option.__cfp = False
OptionSet.__init__(self, args)
|
