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

前言

在最初搭建公司监控系统的时候，最头疼的是需要把同类项目组的相同图形添加到一个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.py
import urllib2
import json
import argparse
def 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_list
def 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 python
import urllib2
import sys
import json
import 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接口认证的函数，登录成功会返回一个Token
def 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()

来源：https://blog.hackroad.com/operations-engineer/linux_server/13474.html