1 import requests
2 import json
3 import sys
4
5
6
7 def get_token():
8 data = {
9 "jsonrpc": "2.0",
10 "method": "user.login",
11 "params": {
12 "user": username,
13 "password": password
14 },
15 "id": 0
16 }
17 r = requests.get(zaurl, headers=header, data=json.dumps(data))
18 auth = json.loads(r.text)
19 return auth["result"]
20
21
22 #指定模板，获取主机信息
23 def template_hostget(token):
24 data = {
25 "jsonrpc": "2.0",
26 "method": "template.get",
27 "params": {
28 "output": ['host', 'templateid'],
29 "templateids": ['10001'],
30 "selectHosts": [ # 返回链接到模板的主机
31 "hostid",
32 "host",
33 ],
34 },
35 "auth": token,
36 "id": 1
37 }
38
39 request = requests.post(zaurl, data=json.dumps(data), headers=header)
40 dict = json.loads(request.content)
41 return (dict['result'][0]['hosts'])
42
43 #获取连接的模板信息
44 def hostget(token, hostid):
45 data = {
46 "jsonrpc": "2.0",
47 "method": "host.get",
48 "params": {
49 "output": ["hostid","host"],
50 "selectParentTemplates": [
51 "templateid",
52 "name"
53 ],
54 "hostids": hostid
55 },
56 "auth": token,
57 "id": 1
58 }
59
60 request = requests.post(zaurl, data=json.dumps(data), headers=header)
61 dict = json.loads(request.content)
62 # print (dict)
63 print (dict['result'][0]['hostid'],dict['result'][0]['host'],dict['result'][0]['parentTemplates'])
64
65
66 def massupdate_templates_clear_hosts(token, hostid):
67 data = {
68 "jsonrpc": "2.0",
69 "method": "host.update",
70 "params": {
71 "hostid": hostid,
72 "templates_clear": ['10001']
73 },
74 "auth": token,
75 "id": 1
76 }
77 request = requests.post(zaurl, data=json.dumps(data), headers=header)
78 print(request.content)
79
80 def add_template(token, templateid, hostid):
81 data = {
82 "jsonrpc": "2.0",
83 "method": "template.massadd",
84 "params": {
85 "templates": [
86 {
87 "templateid": templateid
88 },
89 ],
90 "hosts": [
91 {
92 "hostid": hostid
93 }
94 ]
95 },
96 "auth": token,
97 "id": 2
98 }
99 request = requests.post(zaurl, data=json.dumps(data), headers=header)
100 print(request.content)
101
102 if __name__ == "__main__":
103 #修改输入参数
104 hostfile = "主机IP.txt"
105 zaurl = "http://x.x.x.x/zabbix/api_jsonrpc.php"
106 header = {"Content-Type": "application/json"}
107 username = "xx"
108 password = "xx"
109 token = get_token()
110 #获取主机和模板
111 # hostinfo = template_hostget(token)
112 # for i in hostinfo:
113 # hostget(token, i['hostid'])
114
115 #因监控机器数量太多，一次性全部修改会对zabbix数据库造成比较大的压力，所以分批修改
116 hostids = [获取到主机在zabbix里面的hostid]
117 for id in hostids:
118 # hostget(token,'15097')
119 massupdate_templates_clear_hosts(token, id)
120 add_template(token, "10146", id)
121
122
