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
|
"""Nicer config class."""
import sys, os, ConfigParser, socket
__all__ = ['Config']
class Config(object):
"""Bit improved ConfigParser.
Additional features:
- Remembers section.
- Acceps defaults in get() functions.
- List value support.
"""
def __init__(self, main_section, filename, sane_config = 1, user_defs = {}):
"""Initialize Config and read from file.
@param sane_config: chooses between ConfigParser/SafeConfigParser.
"""
defs = {
'job_name': main_section,
'service_name': main_section,
'host_name': socket.gethostname(),
}
defs.update(user_defs)
self.main_section = main_section
self.filename = filename
self.sane_config = sane_config
if sane_config:
self.cf = ConfigParser.SafeConfigParser(defs)
else:
self.cf = ConfigParser.ConfigParser(defs)
if filename is None:
self.cf.add_section(main_section)
return
if not os.path.isfile(filename):
raise Exception('Config file not found: '+filename)
self.cf.read(filename)
if not self.cf.has_section(main_section):
raise Exception("Wrong config file, no section '%s'"%main_section)
def reload(self):
"""Re-reads config file."""
if self.filename:
self.cf.read(self.filename)
def get(self, key, default=None):
"""Reads string value, if not set then default."""
try:
return self.cf.get(self.main_section, key)
except ConfigParser.NoOptionError, det:
if default == None:
raise Exception("Config value not set: " + key)
return default
def getint(self, key, default=None):
"""Reads int value, if not set then default."""
try:
return self.cf.getint(self.main_section, key)
except ConfigParser.NoOptionError, det:
if default == None:
raise Exception("Config value not set: " + key)
return default
def getboolean(self, key, default=None):
"""Reads boolean value, if not set then default."""
try:
return self.cf.getboolean(self.main_section, key)
except ConfigParser.NoOptionError, det:
if default == None:
raise Exception("Config value not set: " + key)
return default
def getfloat(self, key, default=None):
"""Reads float value, if not set then default."""
try:
return self.cf.getfloat(self.main_section, key)
except ConfigParser.NoOptionError, det:
if default == None:
raise Exception("Config value not set: " + key)
return default
def getlist(self, key, default=None):
"""Reads comma-separated list from key."""
try:
s = self.cf.get(self.main_section, key).strip()
res = []
if not s:
return res
for v in s.split(","):
res.append(v.strip())
return res
except ConfigParser.NoOptionError, det:
if default == None:
raise Exception("Config value not set: " + key)
return default
def getfile(self, key, default=None):
"""Reads filename from config.
In addition to reading string value, expands ~ to user directory.
"""
fn = self.get(key, default)
if fn == "" or fn == "-":
return fn
# simulate that the cwd is script location
#path = os.path.dirname(sys.argv[0])
# seems bad idea, cwd should be cwd
fn = os.path.expanduser(fn)
return fn
def get_wildcard(self, key, values=[], default=None):
"""Reads a wildcard property from conf and returns its string value, if not set then default."""
orig_key = key
keys = [key]
for wild in values:
key = key.replace('*', wild, 1)
keys.append(key)
keys.reverse()
for key in keys:
try:
return self.cf.get(self.main_section, key)
except ConfigParser.NoOptionError, det:
pass
if default == None:
raise Exception("Config value not set: " + orig_key)
return default
def sections(self):
"""Returns list of sections in config file, excluding DEFAULT."""
return self.cf.sections()
def clone(self, main_section):
"""Return new Config() instance with new main section on same config file."""
return Config(main_section, self.filename, self.sane_config)
|
