Python调用zabbix api 接口

hncys

#!/usr/bin/env python　　
# -*- coding: utf-8 -*-
　　
import json
　　
import urllib2
　　
import sys
　　

　　
class zabbixtools:
　　
        def __init__(self):
　　
            self.url = "http://172.16.4.10/zabbix/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": "luojl",
　　
                             "password": "luojl"
　　
                                  },
　　
                        "id": 0
　　
                   }
　　
                              )
　　

　　
            request = urllib2.Request(self.url,data)
　　
            for key in self.header:
　　
                request.add_header(key,self.header[key])
　　
            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[key])
　　
            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 get_hostlist(self):
　　
            data = json.dumps(
　　
                    {
　　
                        "jsonrpc":"2.0",
　　
                        "method":"host.get",
　　
                        "params":{
　　
                             "output":["hostid","name","status","host"],
　　
                             "filter":{"host":""}
　　
                                                                },
　　
                        "auth":self.authID,
　　
                        "id":2
　　
                    }
　　
                            )
　　
            request = urllib2.Request(self.url,data)
　　
            for key in self.header:
　　
                request.add_header(key,self.header[key])
　　

　　
            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
　　
            else:
　　
                response = json.loads(result.read())
　　
                result.close()
　　
                print "Number Of Hosts: ", len(response['result'])
　　
                for host in response['result']:
　　
                    print "Host ID:",host['hostid'],"Host Name:",host['name']
　　

　　
        def host_get(self,hostip):
　　
            data = json.dumps(
　　
                   {
　　
                        "jsonrpc": "2.0",
　　
                        "method": "host.get",
　　
                        "params": {
　　
                                                          "output": ["hostid","name","status","host"],
　　
                             "filter": {"host": [hostip]}
　　
                                  },
　　
                        "auth": self.authID,
　　
                        "id": 1
　　
                   }
　　
                              )
　　
            res = self.get_data(data)['result']
　　
            if (res !=0) and (len(res) != 0):
　　
                host = res[0]
　　
                if host['status'] == '1':
　　
                    print "\t","\033[1;31;40m%s\033[0m" % "Host_IP:","\033[1;31;40m%s\033[0m" % host['host'].ljust(15),'\t',"\033[1;31;40m%s\033[0m" % "Host_Name:","\033[1;31;40m%s\033[0m" % host['name'].encode('UTF-8'),'\t',"\033[1;31;40m%s\033[0m" % u'未在监控状态'.encode('UTF-8')
　　
                    return host['hostid']
　　
                elif host['status'] == '0':
　　
                    print "\t","\033[1;32;40m%s\033[0m" % "Host_IP:","\033[1;32;40m%s\033[0m" % host['host'].ljust(15),'\t',"\033[1;32;40m%s\033[0m" % "Host_Name:","\033[1;32;40m%s\033[0m" % host['name'].encode('UTF-8'),'\t',"\033[1;32;40m%s\033[0m" % u'在监控状态'.encode('UTF-8')
　　

　　
                    return host['hostid']
　　
                print
　　
            else:
　　
                print '\t',"\033[1;31;40m%s\033[0m" % "Get Host Error or cannot find this host,please check !"
　　
                return 0
　　

　　
        def abc(self):
　　
            hostip = raw_input("\033[1;35;40m%s\033[0m" % 'Enter Your Check Host:Host_ip :')
　　
            a = self.host_get(hostip)
　　
            print a
　　
        def hostgroup_get(self):
　　
            data = json.dumps(
　　
                   {
　　
                                         "jsonrpc": "2.0",
　　
                    "method": "hostgroup.get",
　　
                    "params": {
　　
                         "output": "extend"
　　
                              },
　　
                    "auth": self.authID,
　　
                    "id": 3,
　　
                    }
　　
                             )
　　
            res = self.get_data(data)
　　
            if 'result' in res.keys():
　　
                res = res['result']
　　
                if (res !=0) or (len(res) !=0):
　　
                    print "\033[1;32;40m%s\033[0m" % "Number Of Group: ", "\033[1;31;40m%d\033[0m" % len(res)
　　
                    for host in res:
　　
                        print "\t","HostGroup_id:",host['groupid'],"\t","HostGroup_Name:",host['name'].encode('UTF-8')
　　
                    print
　　
            else:
　　
                print "Get HostGroup Error,please check !"
　　

　　

　　
        def template_get(self):
　　
            data = json.dumps(
　　
                    {
　　
                     "jsonrpc": "2.0",
　　
                     "method": "template.get",
　　
                     "params": {
　　
                           "output": "extend",
　　
                               },
　　
                     "auth": self.authID,
　　
                     "id": 4,
　　
                                          }
　　
                             )
　　
            res = self.get_data(data)
　　
            if 'result' in res.keys():
　　
                res = res['result']
　　
                if (res !=0) or (len(res) !=0):
　　
                    print "\033[1;32;40m%s\033[0m" % "Number Of Template: ", "\033[1;31;40m%d\033[0m" % len(res)
　　
                    for host in res:
　　
                        print "\t","Template_id:",host['templateid'],"\t","Template_Name:",host['name'].encode('UTF-8')
　　
                    print
　　
            else:
　　
                print "Get Template Error,please check !"
　　

　　

　　

　　
if __name__ == "__main__":
　　
     test = zabbixtools()
　　
     argv = sys.argv[1].strip()
　　
     if argv == 'hostlist':
　　
        test.get_hostlist()
　　
     elif argv == 'grouplist':
　　
        test.hostgroup_get()
　　
     elif argv == 'templist':
　　
        test.template_get()
　　
     else:
　　
        test.abc()