标签:
#!/usr/bin/env pythonimport 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 =0for 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 +=1if x == columns: x =0 y +=1return graph_listdef screenCreate(url, auth, screen_name, graphids, columns):# print graphidsif len(graphids)% columns ==0: vsize = len(graphids)/ columnselse: 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://10.1.173.215/zabbix/api_jsonrpc.php‘ username ="Admin" password ="zabbix" 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 =(1if args.dynamic else0) screentype =(1if args.screentype else0) auth = authenticate(url, username, password) graphids = getGraph(hostname, url, auth, screentype, dynamic, columns)print"Screen Name: "+ screen_nameprint"Total Number of Graphs: "+ str(len(graphids)) screenCreate(url, auth, screen_name, graphids, columns)if __name__ ==‘__main__‘: main()[root@zabbix ~]#./Auto_add_screens.py ‘10.1.185.114‘‘10.1.185.114‘ScreenName:10.1.185.114TotalNumber of Graphs:14{"screenids":["26"]}标签:
原文地址:http://www.cnblogs.com/gyming/p/5781358.html