Zabbix监控调用map.get并且下载图文件

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 17/8/28 上午9:09
# @Author  : lee
# @File    : ZabbixMaps.py
# @Software: PyCharm
# 说明: 输入机器要查询的园区 知道序号后,第二次 园区+序号出图  阿里 m6 园区命令分别是(ali,m6,yq)


import urllib.request
import http.cookiejar
import http
import urllib.parse
import urllib.request
import json
import requests
from prettytable import PrettyTable
import time
import sys

t = str(time.time()).replace('.', '')  #去掉.的时间戳 用于给图片命名
dict_maps_name = {}

class ZabbixMaps:
    def __init__(self,area,number=-1):#,number):#),graph):
        self.area = area
        self.number = int(number)
        #self.number = number
        # self.graph = graph     #
        self.header = {"Content-Type": "application/json-rpc"}

    def get_Area_url(self): # 获得token 并返回
        if self.area == 'area1':
            url_head = 'xxx.xxxx/'  # 写的网址的头,一会要接其他网址
        elif self.area == 'm6':
            url_head = 'xxxxxx/'
        elif self.area == 'yq':
            url_head = '172.18.2.99'
        self.url_little = url_head
        url = 'http://%s/zabbix/api_jsonrpc.php'% url_head
        return url
    def get_token(self): # 获得token 并返回
        data = {"jsonrpc": "2.0",
                "method": "user.login",
                "params": {
                    "user": '  ',
                    "password": '  ',
                },
                "id": 1,
                "auth": None
                }

        token = requests.post(url=self.get_Area_url(), headers=self.header, data=json.dumps(data))
        json_dict_token = json.loads(token.text)
        # print(json_dict_token['result'])
        #print(json_dict_token['result'])
        return json_dict_token['result']
    def get_mapid(self):
        counter = 0
        data = {
            "jsonrpc": "2.0",
            "method": "map.get",
            "params": {
                "output":["name","sysmapoid"], # ["hostid","name"],
            },
            "auth": self.get_token(),
            "id": 1
                        }

        # #url_m6 = 'http://xxx/zabbix/api_jsonrpc.php'
        hosts = requests.post(url=self.get_Area_url(), headers=self.header, data=json.dumps(data))
        json_dict_mapid = json.loads(hosts.text)
        #json_dict_hostid_result = json_dict_hostid['result'][0]['hostid']
        #print(json_dict_mapid['result'])
        tables = PrettyTable(['序号','监控项'])
        tables.align['序号','监控项'] = 'c'
        tables.horizontal_char = '-'
        tables.junction_char = "-"
        tables.left_padding_width = 3
        for i in json_dict_mapid['result']:
            dict_maps_name[counter]=i['sysmapid'],i['name']
            counter += 1
        # return json_dict_mapid['result']
        for i in dict_maps_name:
            tables.add_row([i,dict_maps_name[i][1]])

        if self.number == -1:
            print(tables)
            print('请输入机房位置和序号:')

        else:
            #print(dict_maps_name)
            return dict_maps_name

    def url_jpg(self):
        if self.number == -1:
            url_jpg = -1

        else:
            # print(dict_graphid_name[1])
            mapid = self.get_mapid()[self.number][0]
            #print(self.get_mapid()[self.number][0])
            url_jpg = 'http://%s/zabbix/map.php?noedit=1&sysmapid=%s' % (self.url_little,mapid)
            #url_2hours = 'http://%s/zabbix/xxx.php?graphid=%s&period=7200' % (self.get_IP_Area(),self.graph_id)
        #print(url_jpg)
        return url_jpg



    def get_jpg(self):
        #print(type(self.get_IP_Area()))
        # post数据接收和处理的页面
        posturl = 'http://%s/zabbix/index.php?request=charts.php' % self.url_little  # 从数据包中分析出,处理post请求的url
        #print(posturl)
         #Referer: http://xxx.ops.com.cn/zabbix/index.php?request=charts.php%3Fddreset%3D1


        # 设置一个cookie处理器,它负责从服务器下载cookie到本地,并且在发送请求时带上本地的cookie
        cj = http.cookiejar.LWPCookieJar()
        cookie_support = urllib.request.HTTPCookieProcessor(cj)
        opener = urllib.request.build_opener(cookie_support, urllib.request.HTTPHandler)
        urllib.request.install_opener(opener)


        # 构造header,一般header至少要包含一下两项。这两项是从抓到的包里分析得出的。
        headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64; rv:52.0) Gecko/20100101 Firefox/52.0',
                  'Referer': 'http://xxx.ops.xxx.com.cn/zabbix/index.php'}

        # 构造Post数据,他也是从抓大的包里分析得出的。
        postData = {    'name': ' ',
                        'password': '  ',
                         'autologin':1,
                        'enter':'Sign in'
                     }


        # 需要给Post数据编码

        postData = urllib.parse.urlencode(postData).encode('utf-8')
        # 通过urllib2提供的request方法来向指定Url发送我们构造的数据,并完成登录过程
        request = urllib.request.Request(posturl, postData, headers)
        response = urllib.request.urlopen(request)
        text = response.read().decode()

        #下载图片
        jpg_link = self.url_jpg()  #图片链接
        if jpg_link == -1:
            pass
        else:
            path = '/Users/lee/PycharmProjects/pythonProjects//zabbix/ZabbixMaps/ZAbbixMap.jpg' #%s_%s.jpg' % (self.graph_id,t)

            urllib.request.urlretrieve(jpg_link, path)




if __name__ == "__main__":
    #一个参数
    # item = ZabbixMaps('ali')
    # item.url_jpg()
    # item.get_mapid()


    # 两个参数
    item = ZabbixMaps('ali',2)
    # item.get_token()
    # item.get_mapid()
    # item.url_jpg()
    # item.get_jpg()
    # print('图片下载完成!')
    if len(sys.argv) == 2:
        try:
            item = ZabbixMaps(sys.argv[1])
            item.url_jpg()
            item.get_mapid()
        except:
            print("输入有误")
    elif len(sys.argv) != 2:
        try:
            item = ZabbixMaps(sys.argv[1],sys.argv[2])
        # item.get_token()
            #item.get_mapid()
            item.url_jpg()
            item.get_jpg()
            print('图片下载完成!')
        except:
            print("选项输入有误")

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏葡萄城控件技术团队

MVC5:使用Ajax和HTML5实现文件上传功能

引言 在实际编程中,经常遇到实现文件上传并显示上传进度的功能,基于此目的,本文就为大家介绍不使用flash 或任何上传文件的插件来实现带有进度显示的文件上传功能...

715100
来自专栏娱乐心理测试

小程序完整demo

30450
来自专栏破晓之歌

Python 的 GUI 开发工具 原

Flexx 是一个纯 Python 工具包,用来创建图形化界面应用程序。其使用 Web 技术进行界面的渲染。你可以用 Flexx 来创建桌面应用,同时也可以导出...

1K20
来自专栏流浪猫的golang

go chan 两种应用模型

    协程完成: 7     协程完成: 8     协程完成: 2     协程完成: 5     协程完成: 4     协程完成: 6     协程完成...

8820
来自专栏葡萄城控件技术团队

如何在 ASP.NET MVC 中集成 AngularJS(2)

在如何在 ASP.NET MVC 中集成 AngularJS(1)中,我们介绍了 ASP.NET MVC 捆绑和压缩、应用程序版本自动刷新和工程构建等内容。 下...

273100
来自专栏码生

ReactNative 面试题

今天有一个 ReactNative 的面试。 时间紧迫,临时写了几个问题,初级面试问题。 都是关键字

1.1K30
来自专栏小白安全

XSS跨站脚本攻击的原理分析与解剖

《xss攻击手法》一开始在互联网上资料并不多(都是现成的代码,没有从基础的开始),直到刺的《白帽子讲WEB安全》和cn4rry的《XSS跨站脚本攻击剖析与防...

34450
来自专栏進无尽的文章

扒虫篇-Bug日志 Ⅱ

事情是这样的:一个风和日丽的下午,我正在 itunesConnect 中注册一个APP,基本信息都保存了,在编辑版本信息时,都弄的差不多了,可是没有保存,结果不...

9410
来自专栏移动开发之家

Flutter完整开发实战详解(三、 打包与填坑篇)

作为系列文章的第三篇,继篇章一和篇章二之后,本篇将为你着重展示:Flutter开发过程的打包流程、APP包对比、细节技巧与问题处理。本篇主要描述的Flutter...

69130
来自专栏携程技术中心

干货 | 前端常用的通信技术

作者简介 陈为平,携程市场部前端工程师,目前主要负责“携程运动”项目的大前端相关工作。 前段时间在忙开发携程运动项目和相应的微信小程序,其中和后端通信犹为频繁。...

44860

扫码关注云+社区

领取腾讯云代金券