69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
#!/usr/bin/env python2.7
|
|
|
|
import collectd
|
|
import json
|
|
from six.moves import xmlrpc_client as xmlrpclib
|
|
|
|
pybmurl = ""
|
|
api = ""
|
|
|
|
|
|
def init_callback():
|
|
global api
|
|
api = xmlrpclib.ServerProxy(pybmurl)
|
|
collectd.info('pybitmessagestatus.py init done')
|
|
|
|
|
|
def config_callback(ObjConfiguration):
|
|
global pybmurl
|
|
apiUsername = ""
|
|
apiPassword = ""
|
|
apiInterface = "127.0.0.1"
|
|
apiPort = 8445
|
|
for node in ObjConfiguration.children:
|
|
key = node.key.lower()
|
|
if key.lower() == "apiusername" and node.values:
|
|
apiUsername = node.values[0]
|
|
elif key.lower() == "apipassword" and node.values:
|
|
apiPassword = node.values[0]
|
|
elif key.lower() == "apiinterface" and node.values:
|
|
apiInterface = node.values[0]
|
|
elif key.lower() == "apiport" and node.values:
|
|
apiPort = node.values[0]
|
|
pybmurl = "http://{}:{}@{}:{}/".format(apiUsername, apiPassword, apiInterface, str(int(apiPort)))
|
|
collectd.info('pybitmessagestatus.py config done')
|
|
|
|
|
|
def read_callback():
|
|
try:
|
|
clientStatus = json.loads(api.clientStatus())
|
|
except (ValueError, TypeError):
|
|
collectd.info("Exception loading or parsing JSON")
|
|
return
|
|
except: # noqa:E722
|
|
collectd.info("Exception loading or parsing JSON")
|
|
return
|
|
|
|
for i in ["networkConnections", "numberOfPubkeysProcessed",
|
|
"numberOfMessagesProcessed", "numberOfBroadcastsProcessed"]:
|
|
metric = collectd.Values()
|
|
metric.plugin = "pybitmessagestatus"
|
|
if i[0:6] == "number":
|
|
metric.type = 'counter'
|
|
else:
|
|
metric.type = 'gauge'
|
|
metric.type_instance = i.lower()
|
|
try:
|
|
metric.values = [clientStatus[i]]
|
|
except (TypeError, KeyError):
|
|
collectd.info("Value for %s missing" % (i))
|
|
metric.dispatch()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
else:
|
|
collectd.register_init(init_callback)
|
|
collectd.register_config(config_callback)
|
|
collectd.register_read(read_callback)
|