posir 发表于 2019-1-20 10:16:38

Zabbix 调用API 批量添加主机 (读取Excel)

  # !/usr/bin/env python
  # coding:utf-8
  import json
  import urllib2
  from urllib2 import URLError
  import xlrd
  #http://blog.mreald.com/178
  import sys
  class zabbixtools:
  def __init__(self):
  self.url = "http://zb3.2.com/api_jsonrpc.php"
  self.header = {"Content-Type": "application/json"}
  self.authID = self.user_login()
  def user_login(self):
  data = json.dumps(
  {
  "jsonrpc": "2.0",
  "method": "user.login",
  "params": {
  "user": "user",
  "password": "password"
  },
  "id": 0
  })
  request = urllib2.Request(self.url,data)
  for key in self.header:
  request.add_header(key,self.header)
  try:
  result = urllib2.urlopen(request)
  except URLError as e:
  print "Auth Failed, Please Check Your Name And Password:",e.code
  else:
  response = json.loads(result.read())
  result.close()
  authID = response['result']
  return authID
  def get_data(self,data,hostip=""):
  request = urllib2.Request(self.url,data)
  for key in self.header:
  request.add_header(key,self.header)
  try:
  result = urllib2.urlopen(request)
  except URLError as e:
  if hasattr(e, 'reason'):
  print 'We failed to reach a server.'
  print 'Reason: ', e.reason
  elif hasattr(e, 'code'):
  print 'The server could not fulfill the request.'
  print 'Error code: ', e.code
  return 0
  else:
  response = json.loads(result.read())
  result.close()
  return response
  def host_get(self,hostip):
  # hostip = raw_input("\033[1;35;40m%s\033[0m" % 'Enter Your Check Host:Host_ip :')
  data = json.dumps(
  {
  "jsonrpc": "2.0",
  "method": "host.get",
  "params": {
  "output": ["hostid", "name", "status", "host"],
  "filter": {"host": }
  },
  "auth": self.authID,
  "id": 1
  })
  res = self.get_data(data)['result']
  if (res != 0) and (len(res) != 0):
  # for host in res:
  print type(res)
  host = res
  return host['name']
  else:
  return ""
  def hostgroup_get(self,hostgroupName):
  data = json.dumps(
  {
  "jsonrpc": "2.0",
  "method": "hostgroup.get",
  "params": {
  "output": "extend",
  "filter": {"name": }
  },
  "auth": self.authID,
  "id": 1
  })
  res = self.get_data(data)['result']
  if (res != 0) and (len(res) != 0):
  # for host in res:
  print type(res)
  host = res
  return host['groupid']
  else:
  return ""
  def template_get(self, templateName):
  data = json.dumps({
  "jsonrpc": "2.0",
  "method": "template.get",
  "params": {
  "output": "extend",
  "filter": {
  "host": [
  templateName,
  ]
  }
  },
  "auth": self.authID,
  "id": 1,
  })
  res = self.get_data(data)['result']
  if (res != 0) and (len(res) != 0):
  # for host in res:
  print type(res)
  host = res
  return host['templateid']
  else:
  return ""
  def host_create(self, hostName, visibleName, hostIp, dnsName, proxyName, hostgroupName, templateName1,
  templateName2):
  data = json.dumps({
  "jsonrpc": "2.0",
  "method": "host.create",
  "params": {
  "host": hostName,
  "name": visibleName,
  # "proxy_hostid": self.proxy_get(proxyName),
  "interfaces": [
  {
  "type": 1,
  "main": 1,
  "useip": 1,
  "ip": hostIp,
  "dns": dnsName,
  "port": "10050"
  }
  ],
  "groups": [
  {
  "groupid": self.hostgroup_get(hostgroupName)
  }
  ],
  "templates": [
  {
  "templateid": self.template_get(templateName1)
  }
  ],
  },
  "auth": self.authID,
  "id": 1
  })
  res = self.get_data(data)['result']
  if (res != 0) and (len(res) != 0):
  # for host in res:
  print type(res)
  # print res['hostids']
  return res['hostids']
  else:
  return ""
  if __name__ == "__main__":
  test = zabbixtools()
  workbook = xlrd.open_workbook('add_zabbix_hosts.xlsx')
  for row in xrange(workbook.sheets().nrows):
  hostname = workbook.sheets().cell(row, 0).value
  visible = workbook.sheets().cell(row, 1).value
  hostip = workbook.sheets().cell(row, 2).value
  dnsname = workbook.sheets().cell(row, 3).value
  proxy = workbook.sheets().cell(row, 4).value
  hostgroup = workbook.sheets().cell(row, 5).value
  hosttemp = workbook.sheets().cell(row, 6).value
  hosttemp1 = workbook.sheets().cell(row, 7).value
  hostgroup = hostgroup.strip()
  hostnameGet = test.host_get(hostname)
  if hostnameGet.strip() == '':
  test.host_create(hostname, visible, hostip, dnsname, proxy, hostgroup, hosttemp, hosttemp1)
  else:
  print "%s have exist! Cannot recreate !\n" % hostnameGet

页: [1]
查看完整版本: Zabbix 调用API 批量添加主机 (读取Excel)