mirror of
https://github.com/aljazceru/zabbix.git
synced 2025-12-17 16:24:19 +01:00
110 lines
2.7 KiB
Python
110 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
||
##################################################
|
||
# AUTHOR: Neo <netkiller@msn.com>
|
||
# WEBSITE: http://www.netkiller.cn
|
||
# Description：zabbix 通过 status 模块监控 nginx
|
||
# Note：Zabbix 3.2
|
||
# DateTime: 2016-11-22
|
||
##################################################
|
||
try:
|
||
import time, json
|
||
import os, sys, errno
|
||
import logging, logging.handlers
|
||
#import configparser
|
||
from optparse import OptionParser, OptionGroup
|
||
#from urllib import
|
||
import urllib.request
|
||
except ImportError as err:
|
||
print("Error: %s" %(err))
|
||
|
||
class Elasticsearch():
|
||
stats = {
|
||
'cluster': 'http://localhost:9200/_cluster/stats',
|
||
'nodes' : 'http://localhost:9200/_nodes/stats',
|
||
'indices': 'http://localhost:9200/_stats',
|
||
'health' : 'http://localhost:9200/_cluster/health'
|
||
}
|
||
|
||
def __init__(self):
|
||
self.ttl = 60
|
||
pass
|
||
|
||
def lock(self,name):
|
||
try:
|
||
fd = os.open(name, os.O_WRONLY | os.O_CREAT | os.O_EXCL)
|
||
os.close(fd)
|
||
return True
|
||
except OSError as e:
|
||
if e.errno == errno.EEXIST:
|
||
return False
|
||
raise
|
||
def expire(self,key, ttl):
|
||
if not os.path.exists(key):
|
||
return True
|
||
elif (time.time() - os.path.getmtime(key)) > ttl:
|
||
return True
|
||
else:
|
||
return False
|
||
#return self.lock(key+".lock")
|
||
|
||
def open(self, module):
|
||
text=""
|
||
json_obj = None
|
||
cache = '/tmp/elastizabbix-{0}.json'.format(module)
|
||
if self.expire(cache, self.ttl):
|
||
text = urllib.request.urlopen(self.stats[module]).read().decode("utf-8")
|
||
with open(cache, 'w') as f: f.write(text)
|
||
json_obj = json.loads(text)
|
||
else:
|
||
json_obj = json.load(open(cache))
|
||
|
||
return json_obj
|
||
|
||
def get(self, module, keyworld):
|
||
json_obj = self.open(module);
|
||
keys = []
|
||
for i in keyworld.split('.'):
|
||
keys.append(i)
|
||
key = '.'.join(keys)
|
||
if key in json_obj:
|
||
json_obj = json_obj.get(key)
|
||
keys = []
|
||
return json_obj
|
||
|
||
def discover(self,module):
|
||
d= {'data': []}
|
||
if module == 'nodes':
|
||
for k,v in self.get('nodes', 'nodes').items():
|
||
d['data'].append({'{#NAME}': v['name'], '{#NODE}': k})
|
||
|
||
if module == "indices":
|
||
for k,v in self.get('indices', 'indices').items():
|
||
d['data'].append({'{#NAME}': k})
|
||
return json.dumps(d)
|
||
|
||
|
||
def main(self):
|
||
parser = OptionParser(usage='usage: %prog <module> <keyword>', version="%prog 1.0.0", description='Elasticsearch for Zabbix')
|
||
(options, args) = parser.parse_args()
|
||
|
||
if not len(args) == 2:
|
||
parser.print_help()
|
||
sys.exit(1)
|
||
|
||
module = args[0]
|
||
keyword = args[1]
|
||
|
||
if module in self.stats.keys():
|
||
print(self.get(module, keyword))
|
||
elif module == "discover":
|
||
print(self.discover(keyword))
|
||
else:
|
||
parser.print_help()
|
||
|
||
if __name__ == '__main__':
|
||
try:
|
||
elastic = Elasticsearch()
|
||
elastic.main()
|
||
except KeyboardInterrupt:
|
||
print ("Crtl+C Pressed. Shutting down.")
|