python写的zabbix自动添加和删除主机的脚本,原理是模拟登陆zabbix web页面中添加、删除主机的操作。如果你是添加或是删除多台主机的话可以写个脚本循环或是多线程来调用此脚本删除添加即可。

  1. #!/usr/bin/env python  
  2. # -*- coding: utf-8 -*-  
  3. #zabbix的自动发现功能有时由于网络等原因自动添加主机比较慢  
  4. #或者不能添加地址段内所有的主机,基于此,写此脚本根据单IP  
  5. #和单IP应用来添加删除主机。  
  6. #在zabbix2.0.3/4上测试通过  
  7.  
  8. import re,urllib,urllib2,cookielib  
  9. import MySQLdb  
  10.  
  11. zabbix_server = '192.168.1.2' 
  12.  
  13. class web_form:  
  14.     post_data=""#登陆提交的参数  
  15.     def __init__(self):  
  16.         '''''初始化类,并建立cookies值''' 
  17.         cj = cookielib.CookieJar()  
  18.         opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))  
  19.         opener.addheaders = [('User-agent''Mozilla/5.0 (Windows NT 6.2; WOW64; rv:19.0) Gecko/20100101 Firefox/19.0')]  
  20.         urllib2.install_opener(opener)  
  21.  
  22.     def login(self,loginurl):  
  23.         '''''模拟登陆,获取认证的后的session''' 
  24.         req = urllib2.Request(loginurl,self.post_data)  
  25.         _response = urllib2.urlopen(req)  
  26.         for i in str(_response.info()).split('\n'):  
  27.             if 'Set-Cookie' in i:  
  28.                 sid = i.split("=")[1].split(";")[0][16:]  
  29.                 break 
  30.         _d=_response.read()  
  31.         return _d,sid  
  32.  
  33.     def getpagehtml(self,pageurl):  
  34.         '''''获取目标网站任意一个页面的html代码''' 
  35.         req2=urllib2.Request(pageurl,self.post_data)  
  36.         _response2=urllib2.urlopen(req2)  
  37.         status = _response2.code  
  38.         _d2=_response2.read()  
  39.         return _d2  
  40.  
  41. def center(sql):  
  42.     try:  
  43.         center_ip = zabbix_server  
  44.         center_user = 'root' 
  45.         center_passwd = '123456789' 
  46.         conn = MySQLdb.connect(host = center_ip,user = center_user,passwd = center_passwd,connect_timeout=10)  
  47.         cursor = conn.cursor()   
  48.         cursor.execute(sql)  
  49.         alldata = cursor.fetchall()  
  50.         cursor.close()  
  51.         conn.close()  
  52.         return alldata  
  53.     except Exception,e:  
  54.         return '0' 
  55.  
  56. def operation(action,hostip,post_dirc):  
  57.     boss = web_form()  
  58.     #参递一个post参数  
  59.     url = "http://%s/zabbix" % zabbix_server  
  60.     boss.post_data = urllib.urlencode({"autologin":"1","enter":"Sign in","name":"admin","password":"zabbix"})  
  61.     login,sid = boss.login("%s/index.php" % url)#首先登陆zabbix,获取认证的sid  
  62.     if action == 'add':#添加主机  
  63.         sql = "SELECT hostid FROM zabbix.hosts WHERE host='%s'" % hostip  
  64.         hostid = center(sql)  
  65.         if len(hostid) != 0:  
  66.             print "Server %s is already in zabbix status !" % hostip  
  67.         else:  
  68.             post_form = {'form':'创建主机','form_refresh':'2','host':hostip,'interfaces[1][dns]':'','interfaces[1][interfaceid]':'1','interfaces[1][ip]':hostip,'interfaces[1][isNew]':'true','interfaces[1][port]':'10050','interfaces[1][type]':'1','interfaces[1][useip]':'1','inventory_mode':'-1','ipmi_authtype':'-1','ipmi_password':'','ipmi_privilege':'2','ipmi_username':'','macros[0][macro]':'','macros[0][value]':'','mainInterfaces[1]':'1','newgroup':'','proxy_hostid':'0','save':'存档','sid':sid,'status':'0','visiblename':hostip}  
  69.             post_form = dict(post_form,**post_dirc)  
  70.             boss.post_data=urllib.urlencode(post_form)  
  71.             add_host = boss.getpagehtml("%s/hosts.php" % url)  
  72.             sql = "SELECT hostid FROM zabbix.hosts WHERE host='%s'" % hostip  
  73.             hostid = center(sql)  
  74.             if len(hostid) != 0:  
  75.                 print "Add %s in zabbix success !" % hostip  
  76.             else:  
  77.                 print "Add %s in zabbix failure !" % hostip  
  78.     elif action == 'del':#删除主机  
  79.         sql = "SELECT hostid FROM zabbix.hosts WHERE host='%s'" % hostip  
  80.         hostid = center(sql)#从zabbix服务器端数据库取出要删除ip的hostid号  
  81.         if len(hostid) != 0:  
  82.             hostid = hostid[0][0]  
  83.             a = 'hosts[%s]' % hostid  
  84.             boss.post_data=urllib.urlencode({'form_refresh':'1','go':'delete','goButton':'确认 (1)',a:hostid,'sid':sid})  
  85.             del_host = boss.getpagehtml("%s/hosts.php" % url)  
  86.             hostid = center(sql)  
  87.             sql = "SELECT hostid FROM zabbix.hosts WHERE host='%s'" % hostip  
  88.             if len(hostid) != 0:  
  89.                 print "Delete %s from zabbix failure !" % hostip  
  90.             else:  
  91.                 print "Delete %s from zabbix success !" % hostip  
  92.         else:  
  93.             print "Server %s not in zabbix status !" % hostip  
  94.  
  95. def main(block,action,hostip):#根据IP的应用,确定需要添加到zabbix中的组和需要应用的模板  
  96.     if block == "adb" or block == "ldb" or block == "ccs":  
  97.         sql = "SELECT groupid,name FROM zabbix.groups WHERE name IN ('WD_ADB_LDB','Linux servers')" 
  98.         groupdata = center(sql)  
  99.         sql = "SELECT hostid,HOST FROM zabbix.hosts WHERE HOST IN ('linux_Server','MySQL_status','Linux_disk_io')" 
  100.         templates = center(sql)  
  101.     elif "_s" in block:  
  102.         sql = "SELECT groupid,name FROM zabbix.groups WHERE name IN ('Linux servers','WD_SLAVE')" 
  103.         groupdata = center(sql)  
  104.         sql = "SELECT hostid,HOST FROM zabbix.hosts WHERE HOST IN ('linux_Server','MySQL_status','Slave_Status','Linux_disk_io')" 
  105.         templates = center(sql)  
  106.     else:  
  107.         sql = "SELECT groupid,name FROM zabbix.groups WHERE name IN ('Linux servers','WD_GS')" 
  108.         groupdata = center(sql)  
  109.         sql = "SELECT hostid,HOST FROM zabbix.hosts WHERE HOST IN ('linux_Server','Linux_disk_io')" 
  110.         templates = center(sql)  
  111.     templatesdirc = {}  
  112.     groupdirc = {}  
  113.     for i in groupdata:  
  114.         groups = 'groups[%s]' % str(i[0])  
  115.         groupdirc[groups] = str(i[0])  
  116.     for i in templates:  
  117.         groups = 'templates[%s]' % str(i[0])  
  118.         templatesdirc[groups] = str(i[1])  
  119.     post_dirc = dict(groupdirc,**templatesdirc)  
  120.     operation(action,hostip,post_dirc)  
  121.  
  122. if __name__=="__main__":  
  123.     block = 'adb_s' 
  124.     hostip = '6.6.6.6' 
  125.     action = 'add' 
  126.     action = 'del' 
  127.     main(block,action,hostip)