zabbix auto add and delete screen

1. #!/usr/bin/env python
2. #-*-coding:utf-8-*-
3. # Author : SammyTan
4. # Email : 729979966@qq.com
5. # Last modified : 2016-04-25 16:24
6. # Filename : zabbix_screens_automation.py
7. # Description : Auto add delete and update zabbix screens
10. import os
11. import urllib2
12. import json
13. import argparse
15. url='http://10.1.180.166/zabbix/api_jsonrpc.php'
16. username='Admin'
17. password='zabbix'
19. def authenticate(url, username, password):
20. values ={'jsonrpc':'2.0',
21. 'method':'user.login',
22. 'params':{
23. 'user': username,
24. 'password': password
25. },
26. 'id':'0'
27. }
29. data = json.dumps(values)
30. req = urllib2.Request(url, data,{'Content-Type':'application/json-rpc'})
31. response = urllib2.urlopen(req, data)
32. output = json.loads(response.read())
34. try:
35. message = output['result']
36. except:
37. message = output['error']['data']
38. print message
39. quit()
40. return output['result']
43. def getHost():
44. values ={
45. "jsonrpc":"2.0",
46. "method":"host.get",
47. "params":{
48. "output":[
49. "host"
50. ]
51. },
52. "id":1,
53. "auth":"2e92f0d30d878442533da77c15ae7cdd"
54. }
56. data = json.dumps(values)
57. req = urllib2.Request(url, data,{'Content-Type':'application/json-rpc'})
58. response = urllib2.urlopen(req, data)
59. output = json.loads(response.read())
60. return output
63. def getGraph(hostid):
64. values ={
65. "jsonrpc":"2.0",
66. "method":"graph.get",
67. "params":{
68. "output":[
69. "graphid"
70. ],
71. "hostids": hostid
72. },
73. "auth":"2e92f0d30d878442533da77c15ae7cdd",
74. "id":1
75. }
77. data = json.dumps(values)
78. req = urllib2.Request(url, data,{'Content-Type':'application/json-rpc'})
79. response = urllib2.urlopen(req, data)
80. host_get = response.read()
81. output = json.loads(host_get)
82. #print json.dumps(output)
83. return output
85. def getScreentimeid(screenid):
86. values ={
87. "jsonrpc":"2.0",
88. "method":"screen.get",
89. "params":{
90. "output":"extend",
91. "selectScreenItems":[
92. "screenitemid"
93. ],
94. "screenids": screenid
95. },
96. "auth":"2e92f0d30d878442533da77c15ae7cdd",
97. "id":1
98. }
99. data = json.dumps(values)
100. req = urllib2.Request(url, data,{'Content-Type':'application/json-rpc'})
101. response = urllib2.urlopen(req, data)
102. host_get = response.read()
103. output = json.loads(host_get)
104. #print json.dumps(output)
105. return output
107. def getScreen():
108. values ={
109. "jsonrpc":"2.0",
110. "method":"screen.get",
111. "params":{
112. "output":[
113. "name",
114. "screenid"
115. ]
116. },
117. "auth":"2e92f0d30d878442533da77c15ae7cdd",
118. "id":1
119. }
120. data = json.dumps(values)
121. req = urllib2.Request(url, data,{'Content-Type':'application/json-rpc'})
122. response = urllib2.urlopen(req, data)
123. host_get = response.read()
124. output = json.loads(host_get)
125. return output
127. def screenDelete(screenid):
128. values ={
129. "jsonrpc":"2.0",
130. "method":"screen.delete",
131. "params":[
132. screenid
133. ],
134. "auth":"2e92f0d30d878442533da77c15ae7cdd",
135. "id":1
136. }
137. data = json.dumps(values)
138. req = urllib2.Request(url, data,{'Content-Type':'application/json-rpc'})
139. response = urllib2.urlopen(req, data)
140. host_get = response.read()
141. output = json.loads(host_get)
142. return output
145. #os.system('python /root/Scripts_Tmemplate/Auto_add_screens.py -c 2 $each_host $each_host')
147. def main():
148. host_output=getHost()
149. host_list =[]
150. # add host to host list
151. for host in host_output['result']:
152. #print host
153. host_list.append(host['host'])
155. screenname_list =[]
156. screen_output=getScreen()
157. # add screen name to saceen list
158. for name in screen_output['result']:
159. screenname_list.append(name['name'])
161. #Check host , if this host not have screen ,the self create screen .
162. for host in host_list:
163. if host notin screenname_list:# if this host not in screen list , the creend the host screen
164. os.system('python /root/Scripts_Tmemplate/Auto_add_screens.py -c 2 %s %s'%(host,host))  
165. #### 该脚本在下面
166. print'Create %s host screen'%(host)
168. #Delete invalid screen
169. for name in screenname_list:
170. for sid in screen_output['result']:
171. if name in sid['name']:
172. screenid = sid['screenid']
173. if name notin host_list:
174. delete_result = screenDelete(screenid)
175. print'Delete %s host screen'%(name)
177. #Check host graphs number, if graphs number grant than exsit screens graphs , the delete it after recreated.
178. for screenid in screen_output['result']:
179. #print screenid
180. screenGraphs = getScreentimeid(screenid['screenid'])
181. for sid in screenGraphs['result']:
182. for host in host_output['result']:
183. if host['host']in screenid['name']:
184. hostid = host['hostid']
185. hostname = host['host']
186. hostGraphs = getGraph(hostid)
187. #print host['host'],len(hostGraphs['result']),len(sid['screenitems'])
188. if len(hostGraphs['result'])> len(sid['screenitems']):
189. delete_result = screenDelete(host['hostid'])
190. os.system('python /root/Scripts_Tmemplate/Auto_add_screens.py -c 2 %s %s'%(hostname,hostname))
191. print'Update %s host screen'%(hostname)
192. else:
193. print'All screen already sync .'
195. if __name__ =='__main__':
196. main()

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":"200",
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.180.166/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=2,
149. help='number of columns in the screen (default: 2)')
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()

 

来自为知笔记(Wiz)