118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
|
import requests
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
class ZabbixAPI:
|
||
|
|
def __init__(self, user, password, zabbix_url):
|
||
|
|
# Zabbix URL
|
||
|
|
self.zabbix_url = zabbix_url
|
||
|
|
self.header = {"Content-Type": "application/json"}
|
||
|
|
# login information
|
||
|
|
self.user = user
|
||
|
|
self.password = password
|
||
|
|
self.auth = ""
|
||
|
|
|
||
|
|
self.templates_id = []
|
||
|
|
self.groups_id = []
|
||
|
|
|
||
|
|
def login(self):
|
||
|
|
# Parameter for zabbix API
|
||
|
|
parameter = json.dumps({
|
||
|
|
"jsonrpc": "2.0",
|
||
|
|
"method": "user.login",
|
||
|
|
"params": {
|
||
|
|
"user": self.user,
|
||
|
|
"password": self.password
|
||
|
|
},
|
||
|
|
"id": 1,
|
||
|
|
"auth": None
|
||
|
|
})
|
||
|
|
# Get response text from zabbix API by using POST
|
||
|
|
response = requests.post(self.zabbix_url, parameter, headers=self.header).text
|
||
|
|
# Get key from response text
|
||
|
|
try:
|
||
|
|
self.auth = json.loads(response)['result']
|
||
|
|
except KeyError:
|
||
|
|
print("Login failed")
|
||
|
|
print("Login Success")
|
||
|
|
print(self.auth)
|
||
|
|
|
||
|
|
def get_template_id(self, templates):
|
||
|
|
# Init variable for templates_id in list
|
||
|
|
self.templates_id = []
|
||
|
|
# Parameter for Get TemplateID in Zabbix API
|
||
|
|
parameter = json.dumps({
|
||
|
|
"jsonrpc": "2.0",
|
||
|
|
"method": "template.get",
|
||
|
|
"params": {
|
||
|
|
"output": "extend",
|
||
|
|
"filter": {
|
||
|
|
"host": templates
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"auth": self.auth,
|
||
|
|
"id": 1
|
||
|
|
})
|
||
|
|
# Get response text from zabbix API by using POST
|
||
|
|
response = requests.post(self.zabbix_url, parameter, headers=self.header).text
|
||
|
|
# Append list templates_id from response text
|
||
|
|
for i in json.loads(response)['result']:
|
||
|
|
self.templates_id.append(str(i['templateid']))
|
||
|
|
print(self.templates_id)
|
||
|
|
|
||
|
|
def get_group_id(self, groups):
|
||
|
|
# Init variable for groups_id in list
|
||
|
|
self.groups_id = []
|
||
|
|
# Parameter for Get GroupID in Zabbix API
|
||
|
|
parameter = json.dumps({
|
||
|
|
"jsonrpc": "2.0",
|
||
|
|
"method": "hostgroup.get",
|
||
|
|
"params": {
|
||
|
|
"output": "extend",
|
||
|
|
"filter": {
|
||
|
|
"name": groups
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"auth": self.auth,
|
||
|
|
"id": 1
|
||
|
|
})
|
||
|
|
# Get response text from zabbix API by using POST
|
||
|
|
response = requests.post(self.zabbix_url, parameter, headers=self.header).text
|
||
|
|
# Append list templates_id from response text
|
||
|
|
for i in json.loads(response)['result']:
|
||
|
|
self.groups_id.append(str(i['groupid']))
|
||
|
|
print(self.groups_id)
|
||
|
|
|
||
|
|
def create_host(self, hostname, interface_ip, interface_port):
|
||
|
|
# Put the group ID and template ID into dict
|
||
|
|
group = []
|
||
|
|
template = []
|
||
|
|
for ids in self.groups_id:
|
||
|
|
group.append({"groupid": str(ids)})
|
||
|
|
for ids in self.templates_id:
|
||
|
|
template.append({"templateid": str(ids)})
|
||
|
|
# Parameter for CreateHost in Zabbix API
|
||
|
|
parameter = json.dumps({
|
||
|
|
"jsonrpc": "2.0",
|
||
|
|
"method": "host.create",
|
||
|
|
"params": {
|
||
|
|
"host": interface_ip,
|
||
|
|
"name": hostname,
|
||
|
|
"interfaces": [{
|
||
|
|
"type": 1,
|
||
|
|
"main": 1,
|
||
|
|
"useip": 1,
|
||
|
|
"ip": interface_ip,
|
||
|
|
"dns": "",
|
||
|
|
"port": interface_port
|
||
|
|
}],
|
||
|
|
"groups": group,
|
||
|
|
"templates": template
|
||
|
|
},
|
||
|
|
"auth": self.auth,
|
||
|
|
"id": 1
|
||
|
|
})
|
||
|
|
# Create host with zabbix API by using POST
|
||
|
|
response = requests.post(self.zabbix_url, parameter, headers=self.header).text
|
||
|
|
print(response)
|