python批量添加zabbix Screens的两个脚本分享

Python  /  管理员 发布于 7年前   160

前言

在最初搭建公司监控系统的时候，最头疼的是需要把同类项目组的相同图形添加到一个Screens,由于只能一个一个的添加，非常耗时耗经历。

下面分享两个脚本来解决这个头疼的问题。

1.将单个主机的所有图形添加到一个Screens

使用方法

#更改main()函数里的url、username、password#参数一：主机名#参数二：筛选图名称python zabbix_screen_host.py 'zabbixserver' 'zabbixserver'

zabbix_screen_host.py脚本内容

#!/usr/bin/env python#zabbix_screen_host.pyimport urllib2import jsonimport argparsedef authenticate(url, username, password): values = {'jsonrpc': '2.0', 'method': 'user.login', 'params': {  'user': username,  'password': password }, 'id': '0' } data = json.dumps(values) req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'}) response = urllib2.urlopen(req, data) output = json.loads(response.read()) try: message = output['result'] except: message = output['error']['data'] print message quit() return output['result']def getGraph(hostname, url, auth, graphtype, dynamic, columns): if (graphtype == 0): selecttype = ['graphid'] select = 'selectGraphs' if (graphtype == 1): selecttype = ['itemid', 'value_type'] select = 'selectItems' values = {'jsonrpc': '2.0', 'method': 'host.get', 'params': {  select: selecttype,  'output': ['hostid', 'host'],  'searchByAny': 1,  'filter': {  'host': hostname  } }, 'auth': auth, 'id': '2' } data = json.dumps(values) req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'}) response = urllib2.urlopen(req, data) host_get = response.read() output = json.loads(host_get) # print json.dumps(output) graphs = [] if (graphtype == 0): for i in output['result'][0]['graphs']: graphs.append(i['graphid']) if (graphtype == 1): for i in output['result'][0]['items']: if int(i['value_type']) in (0, 3): graphs.append(i['itemid']) graph_list = [] x = 0 y = 0 for graph in graphs: graph_list.append({ "resourcetype": graphtype, "resourceid": graph, "width": "500", "height": "100", "x": str(x), "y": str(y), "colspan": "1", "rowspan": "1", "elements": "0", "valign": "0", "halign": "0", "style": "0", "url": "", "dynamic": str(dynamic) }) x += 1 if x == columns: x = 0 y += 1 return graph_listdef screenCreate(url, auth, screen_name, graphids, columns): # print graphids if len(graphids) % columns == 0: vsize = len(graphids) / columns else: vsize = (len(graphids) / columns) + 1 values = {"jsonrpc": "2.0", "method": "screen.create", "params": [{  "name": screen_name,  "hsize": columns,  "vsize": vsize,  "screenitems": [] }], "auth": auth, "id": 2 } for i in graphids: values['params'][0]['screenitems'].append(i) data = json.dumps(values) req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'}) response = urllib2.urlopen(req, data) host_get = response.read() output = json.loads(host_get) try: message = output['result'] except: message = output['error']['data'] print json.dumps(message) def main(): url = 'http://zabbixip/zabbix/api_jsonrpc.php' username = "***" password = "***" parser = argparse.ArgumentParser(description='Create Zabbix screen from all of a host Items or Graphs.') parser.add_argument('hostname', metavar='H', type=str,  help='Zabbix Host to create screen from') parser.add_argument('screenname', metavar='N', type=str,  help='Screen name in Zabbix. Put quotes around it if you want spaces in the name.') parser.add_argument('-c', dest='columns', type=int, default=3,  help='number of columns in the screen (default: 3)') parser.add_argument('-d', dest='dynamic', action='store_true',  help='enable for dynamic screen items (default: disabled)') parser.add_argument('-t', dest='screentype', action='store_true',  help='set to 1 if you want item simple graphs created (default: 0, regular graphs)') args = parser.parse_args() hostname = args.hostname screen_name = args.screenname columns = args.columns dynamic = (1 if args.dynamic else 0) screentype = (1 if args.screentype else 0) auth = authenticate(url, username, password) graphids = getGraph(hostname, url, auth, screentype, dynamic, columns) print "Screen Name: " + screen_name print "Total Number of Graphs: " + str(len(graphids)) screenCreate(url, auth, screen_name, graphids, columns)if __name__ == '__main__': main()

2.将同组主机的同一图形添加到一个Screens

使用方法

#更改main()函数里的url、username、password#-g ：组名称#-G：图形名称#-n ：筛选（screen）图名称#-c : 一行有多少图形python zabbix_screen_group.py -g 'zabbix' -G 'icmp-ping' -n 'zabbix-icmp-ping' -c 2

zabbix_screen_group.py脚本内容

#!/usr/bin/env pythonimport urllib2import sysimport jsonimport argparse #定义通过HTTP方式访问API地址的函数，后面每次请求API的各个方法都会调用这个函数def requestJson(url,values):  data = json.dumps(values) req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'}) response = urllib2.urlopen(req, data) output = json.loads(response.read())# print output try: message = output['result'] except: message = output['error']['data'] print message quit()  return output['result'] #API接口认证的函数，登录成功会返回一个Tokendef authenticate(url, username, password): values = {'jsonrpc': '2.0',  'method': 'user.login',  'params': {   'user': username,   'password': password  },  'id': '0'  } idvalue = requestJson(url,values) return idvalue #定义更加主机分组名称获取各个hostid的函数def getHosts(groupname,url,auth): host_list = [] values = {'jsonrpc': '2.0',  'method': 'hostgroup.get',  'params': {   'output': 'extend',   'filter': {   'name': groupname   },    'selectHosts' : ['hostid','host'],  },  'auth': auth,  'id': '2'  } output = requestJson(url,values) for host in output[0]['hosts']: host_list.append(host['hostid']) return host_list #定义获取graphid的函数def getGraphs(host_list,name_list, url, auth, columns, graphtype=0 ,dynamic=0): if (graphtype == 0): selecttype = ['graphid'] select = 'selectGraphs' if (graphtype == 1): selecttype = ['itemid', 'value_type'] select = 'selectItems' values=({'jsonrpc' : '2.0',  'method' : 'graph.get',  'params' : {   'output' : ['graphid','name'],   select : [selecttype,'name'],   'hostids' : host_list,   'sortfield' : 'name',   'filter' : {    'name' : name_list,     },   },  'auth' : auth,  'id' : 3  }) output = requestJson(url,values) bb = sorted(output,key = lambda x:x['graphid']) graphs = [] if (graphtype == 0): for i in bb:  print i  graphs.append(i['graphid']) if (graphtype == 1): for i in bb:  if int(i['value_type']) in (0, 3):  graphs.append(i['itemid'])  graph_list = [] x = 0 y = 0 for graph in graphs: print "x is " + str(x) print "y is " + str(y) graph_list.append({  "resourcetype": graphtype,  "resourceid": graph,  "width": "500",  "height": "100",  "x": str(x),  "y": str(y),  "colspan": "1",  "rowspan": "1",  "elements": "0",  "valign": "0",  "halign": "0",  "style": "0",  "url": "",  "dynamic": str(dynamic) }) x += 1# print type(x)# print type(columns) if x == int(columns):  x = 0  y += 1# print graph_list return graph_list #定义创建screen的函数def screenCreate(url, auth, screen_name, graphids, columns): columns = int(columns) if len(graphids) % columns == 0: vsize = len(graphids) / columns else: vsize = (len(graphids) / columns) + 1 #先使用screen.get判断给定的screen name是否存在 values0 = {  "jsonrpc" : "2.0",  "method" : "screen.get",  "params" : {   "output" : "extend",   "filter" : {   "name" : screen_name,    }    },  "auth" : auth,  "id" : 2  } values = {  "jsonrpc": "2.0",  "method": "screen.create",  "params": {   "name": screen_name,   "hsize": columns,   "vsize": vsize,   "screenitems": []  },  "auth": auth,  "id": 2  } output0 = requestJson(url,values0) print output0 #如果给定的screen name不存在则直接创建screen  if output0 == []: print "The Given Screen Name Not Exists" print "Creating Screen %s" %screen_name for i in graphids:  values['params']['screenitems'].append(i) output = requestJson(url,values) else:  #如果给定的screen name已经存在，直接创建screen是不行的，#要么先使用screen.delete把原来的screen删除掉，然后再创建，#要么直接使用screen.update更新原来那个screen，#使用screen.delete会产生新的screenid,#使用screen.update比较合理一点。 print "The Given Screen Name Already Exists" update_screenid=output0[0]["screenid"] print update_screenid print "Updating Screen Name %s Screen ID %s" %(screen_name,update_screenid) values1 = {  "jsonrpc" : "2.0",  "method" : "screen.update",  "params" : {   "screenid" : update_screenid,   "screenitems": []    },  "auth" : auth,  "id" : 2   } output1 = requestJson(url,values1) print output1 print "Updating Screen Name %s" %screen_name for i in graphids:  values1['params']['screenitems'].append(i) output = requestJson(url,values1) def main(): url = 'http://zabbixip/zabbix/api_jsonrpc.php' username = '****' password = '****' auth = authenticate(url, username, password) host_list = getHosts(groupname,url,auth) print host_list graph_ids = getGraphs(host_list,graphname, url, auth, columns) screenCreate(url, auth, screenname, graph_ids, columns)if __name__ == '__main__': parser = argparse.ArgumentParser(description='Create Zabbix screen from all of a host Items or Graphs.') parser.add_argument('-G', dest='graphname', nargs='+',metavar=('grah name'),   help='Zabbix Host Graph to create screen from') parser.add_argument('-H', dest='hostname', nargs='+',metavar=('10.19.111.145'),   help='Zabbix Host to create screen from') parser.add_argument('-g', dest='groupname', nargs='+',metavar=('linux server'),   help='Zabbix Group to create screen from') parser.add_argument('-n', dest='screenname', type=str,   help='Screen name in Zabbix. Put quotes around it if you want spaces in the name.') parser.add_argument('-c', dest='columns', type=int,   help='number of columns in the screen') args = parser.parse_args() print args hostname = args.hostname groupname = args.groupname screenname = args.screenname columns = args.columns graphname = args.graphname if columns is None: columns = len(graphname)# print columns main()

总结

以上就是这篇文章的全部内容了，希望本文的内容的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流。