Add config parser module to make specifying hosts easy
This commit is contained in:
parent
9ed5a2ad13
commit
669cbe6a57
13
m2g.conf.example
Normal file
13
m2g.conf.example
Normal file
@ -0,0 +1,13 @@
|
||||
[host1.domain1.local]
|
||||
address: host1.domain1.local
|
||||
|
||||
[mta.domain1.local]
|
||||
graphitename: smtp
|
||||
|
||||
[www.domain1.local]
|
||||
address: server1.domain1.local
|
||||
|
||||
[mta2.domain1.local]
|
||||
graphitename: smtp2
|
||||
|
||||
[jabber.domain1.local]
|
53
m2g_configparser.py
Executable file
53
m2g_configparser.py
Executable file
@ -0,0 +1,53 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import ConfigParser
|
||||
import sys
|
||||
|
||||
class M2GConfigParser():
|
||||
def __init__(self, config_path):
|
||||
self._config_path = config_path
|
||||
self._config_dict = {} # The entire file parsed into a dictionary.
|
||||
self._graphite_config = {} # The data we'll use for Munin queries.
|
||||
self._parsed_config = ConfigParser.SafeConfigParser()
|
||||
self._parsed_config.read(self._config_path)
|
||||
|
||||
def parse_config(self):
|
||||
"""Parse out all the config's data."""
|
||||
self.set_config_dict()
|
||||
self.set_host_list()
|
||||
|
||||
def set_config_dict(self):
|
||||
"""Populate the object's dictionary from the config file."""
|
||||
sections = self._parsed_config.sections()
|
||||
for current_section in sections:
|
||||
options = self._parsed_config.options(current_section)
|
||||
self._config_dict[current_section] = {}
|
||||
for current_option in options:
|
||||
self._config_dict[current_section][current_option] = self._parsed_config.get(current_section, current_option)
|
||||
|
||||
def get_config_dict(self):
|
||||
return self._config_dict
|
||||
|
||||
def set_host_list(self):
|
||||
"""From a parsed config, list the computed Graphite host list."""
|
||||
for section in self._config_dict:
|
||||
self._graphite_config[section] = {}
|
||||
# Set Hostname to query first
|
||||
if "address" in self._config_dict[section]:
|
||||
self._graphite_config[section]["address"] = self._config_dict[section]["address"]
|
||||
else:
|
||||
self._graphite_config[section]["address"] = section
|
||||
|
||||
# Set graphitename to display in Graphite
|
||||
if "graphitename" in self._config_dict[section]:
|
||||
self._graphite_config[section]["graphitename"] = self._config_dict[section]["graphitename"]
|
||||
else:
|
||||
self._graphite_config[section]["graphitename"] = section.split(".")[0]
|
||||
|
||||
def get_graphite_config(self):
|
||||
return self._graphite_config
|
||||
|
||||
if __name__ == "__main__":
|
||||
myconfig = M2GConfigParser(sys.argv[1])
|
||||
myconfig.parse_config()
|
||||
print "Graphite Config: %r" % myconfig.get_graphite_config()
|
Loading…
x
Reference in New Issue
Block a user