#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 17/8/28 上午9:09
# @Author  : lee
# @File    : ZabbixMaps.py
# @Software: PyCharm
# 說明: 輸入機器要查詢的園區(qū) 知道序號后磷瘤，第二次 園區(qū)+序號出圖  阿里 m6 園區(qū)命令分別是（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/'  # 寫的網(wǎng)址的頭搜变，一會要接其他網(wǎng)址
        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(['序號','監(jiān)控項'])
        tables.align['序號','監(jiān)控項'] = '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數(shù)據(jù)接收和處理的頁面
        posturl = 'http://%s/zabbix/index.php?request=charts.php' % self.url_little  # 從數(shù)據(jù)包中分析出采缚，處理post請求的url
        #print(posturl)
         #Referer: http://xxx.ops.com.cn/zabbix/index.php?request=charts.php%3Fddreset%3D1


        # 設置一個cookie處理器，它負責從服務器下載cookie到本地挠他，并且在發(fā)送請求時帶上本地的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數(shù)據(jù)贸呢，他也是從抓大的包里分析得出的。
        postData = {    'name': ' ',
                        'password': '  ',
                         'autologin':1,
                        'enter':'Sign in'
                     }


        # 需要給Post數(shù)據(jù)編碼

        postData = urllib.parse.urlencode(postData).encode('utf-8')
        # 通過urllib2提供的request方法來向指定Url發(fā)送我們構造的數(shù)據(jù)拢军，并完成登錄過程
        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__":
    #一個參數(shù)
    # item = ZabbixMaps('ali')
    # item.url_jpg()
    # item.get_mapid()


    # 兩個參數(shù)
    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("選項輸入有誤")