Zabbix 快速创建screen

1. #!/usr/bin/env python
2. import urllib2
3. import json
4. import argparse
7. def authenticate(url, username, password):
8. values ={‘jsonrpc‘:‘2.0‘,
9. ‘method‘:‘user.login‘,
10. ‘params‘:{
11. ‘user‘: username,
12. ‘password‘: password
13. },
14. ‘id‘:‘0‘
15. }
17. data = json.dumps(values)
18. req = urllib2.Request(url, data,{‘Content-Type‘:‘application/json-rpc‘})
19. response = urllib2.urlopen(req, data)
20. output = json.loads(response.read())
22. try:
23. message = output[‘result‘]
24. except:
25. message = output[‘error‘][‘data‘]
26. print message
27. quit()
29. return output[‘result‘]
32. def getGraph(hostname, url, auth, graphtype, dynamic, columns):
33. if(graphtype ==0):
34. selecttype =[‘graphid‘]
35. select =‘selectGraphs‘
36. if(graphtype ==1):
37. selecttype =[‘itemid‘,‘value_type‘]
38. select =‘selectItems‘
40. values ={‘jsonrpc‘:‘2.0‘,
41. ‘method‘:‘host.get‘,
42. ‘params‘:{
43. select: selecttype,
44. ‘output‘:[‘hostid‘,‘host‘],
45. ‘searchByAny‘:1,
46. ‘filter‘:{
47. ‘host‘: hostname
48. }
49. },
50. ‘auth‘: auth,
51. ‘id‘:‘2‘
52. }
54. data = json.dumps(values)
55. req = urllib2.Request(url, data,{‘Content-Type‘:‘application/json-rpc‘})
56. response = urllib2.urlopen(req, data)
57. host_get = response.read()
59. output = json.loads(host_get)
60. # print json.dumps(output)
62. graphs =[]
63. if(graphtype ==0):
64. for i in output[‘result‘][0][‘graphs‘]:
65. graphs.append(i[‘graphid‘])
67. if(graphtype ==1):
68. for i in output[‘result‘][0][‘items‘]:
69. if int(i[‘value_type‘])in(0,3):
70. graphs.append(i[‘itemid‘])
72. graph_list =[]
73. x =0
74. y =0
76. for graph in graphs:
77. graph_list.append({
78. "resourcetype": graphtype,
79. "resourceid": graph,
80. "width":"500",
81. "height":"100",
82. "x": str(x),
83. "y": str(y),
84. "colspan":"1",
85. "rowspan":"1",
86. "elements":"0",
87. "valign":"0",
88. "halign":"0",
89. "style":"0",
90. "url":"",
91. "dynamic": str(dynamic)
92. })
93. x +=1
94. if x == columns:
95. x =0
96. y +=1
98. return graph_list
101. def screenCreate(url, auth, screen_name, graphids, columns):
102. # print graphids
103. if len(graphids)% columns ==0:
104. vsize = len(graphids)/ columns
105. else:
106. vsize =(len(graphids)/ columns)+1
108. values ={"jsonrpc":"2.0",
109. "method":"screen.create",
110. "params":[{
111. "name": screen_name,
112. "hsize": columns,
113. "vsize": vsize,
114. "screenitems":[]
115. }],
116. "auth": auth,
117. "id":2
118. }
120. for i in graphids:
121. values[‘params‘][0][‘screenitems‘].append(i)
123. data = json.dumps(values)
124. req = urllib2.Request(url, data,{‘Content-Type‘:‘application/json-rpc‘})
125. response = urllib2.urlopen(req, data)
126. host_get = response.read()
128. output = json.loads(host_get)
130. try:
131. message = output[‘result‘]
132. except:
133. message = output[‘error‘][‘data‘]
135. print json.dumps(message)
138. def main():
139. url =‘http://10.1.173.215/zabbix/api_jsonrpc.php‘
140. username ="Admin"
141. password ="zabbix"
143. parser = argparse.ArgumentParser(description=‘Create Zabbix screen from all of a host Items or Graphs.‘)
144. parser.add_argument(‘hostname‘, metavar=‘H‘, type=str,
145. help=‘Zabbix Host to create screen from‘)
146. parser.add_argument(‘screenname‘, metavar=‘N‘, type=str,
147. help=‘Screen name in Zabbix. Put quotes around it if you want spaces in the name.‘)
148. parser.add_argument(‘-c‘, dest=‘columns‘, type=int, default=3,
149. help=‘number of columns in the screen (default: 3)‘)
150. parser.add_argument(‘-d‘, dest=‘dynamic‘, action=‘store_true‘,
151. help=‘enable for dynamic screen items (default: disabled)‘)
152. parser.add_argument(‘-t‘, dest=‘screentype‘, action=‘store_true‘,
153. help=‘set to 1 if you want item simple graphs created (default: 0, regular graphs)‘)
155. args = parser.parse_args()
156. hostname = args.hostname
157. screen_name = args.screenname
158. columns = args.columns
159. dynamic =(1if args.dynamic else0)
160. screentype =(1if args.screentype else0)
162. auth = authenticate(url, username, password)
163. graphids = getGraph(hostname, url, auth, screentype, dynamic, columns)
165. print"Screen Name: "+ screen_name
166. print"Total Number of Graphs: "+ str(len(graphids))
168. screenCreate(url, auth, screen_name, graphids, columns)
170. if __name__ ==‘__main__‘:
171. main()

Usage:

1. [[email protected] ~]#./Auto_add_screens.py ‘10.1.185.114‘‘10.1.185.114‘
2. ScreenName:10.1.185.114
3. TotalNumber of Graphs:14
4. {"screenids":["26"]}

来自为知笔记(Wiz)