#!/usr/bin/python3.7
# encoding: utf-8

import urllib.request
import json
import pydnsbl
import ssl
ssl._create_default_https_context = ssl._create_unverified_context


TOKEN_URL = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
# 企业的id，在管理端->"我的企业" 可以看到
# CORP_ID = "CORP_ID"
CORP_ID = "CORP_ID****"
# 某个自建应用的id及secret, 在管理端 -> 企业应用 -> 自建应用, 点进相应应用可以看到
APP_ID = "APP_ID*****"
CORP_SECRET = "**********"

class Wechat(object):
    "send monitor message by wechat"

    def __init__(self):
        self.CORP_ID = CORP_ID
        self.CORP_SECRET = CORP_SECRET
        self.APP_ID = APP_ID
        self.BASEURL = 'https://qyapi.weixin.qq.com/cgi-bin/'
        self.TOKEN_URL = 'gettoken?corpid={0}&corpsecret={1}'.format(
            self.CORP_ID, self.CORP_SECRET)

    # 获取认证 token
    def Get_Token(self):
        try:
            response = urllib.request.urlopen(
                '{0}{1}'.format(self.BASEURL, self.TOKEN_URL))
            access_token = json.loads(
                response.read().decode('utf-8'))['access_token']
            with open('token', 'w') as f:
                f.write(access_token)
        except KeyError:
            raise KeyError
        return access_token

    def checker(self):
        Result = []
        ip_checker = pydnsbl.DNSBLIpChecker()
        Result1 = ip_checker.check('68.128.212.240')  # Exchange Server 01
        Result2 = ip_checker.check('68.128.212.241')  # Exchange Server 02
        Result3 = ip_checker.check('68.128.212.242')  # Exchange Server 03
        Result.append(Result1)
        Result.append(Result2)
        Result.append(Result3)
        # domain_checker = pydnsbl.DNSBLDomainChecker()
        # Result4 = domain_checker.check('luxiu2.com')  # Domain Name 01
        # Result5 = domain_checker.check('videour.com')  # Domain Name 02
        # Result6 = domain_checker.check('jesdoit.com')  # Domain Name 03
        # Result.append(Result4)
        # Result.append(Result5)
        # Result.append(Result6)
        return Result

    # 本地 token
    def Local_Token(self):
        try:
            with open('token', 'r') as f:
                token = f.readline().strip()
                if token == '':
                    token = self.Get_Token()
                    return token
                else:
                    return token
        except IOError:
            token = self.Get_Token()
            return token

    # 获取报警人员名单
    def Get_User(self, dep_id=1, fchild=1):
        #token = self.Get_Token()
        token = self.Local_Token()
        send_url = '{0}user/list?access_token={1}&department_id={2}&fetch_child{3}'.format(
            self.BASEURL, token, dep_id, fchild,)
        respone = urllib.request.urlopen(url=send_url).read()
        stat = json.loads(respone)['userlist']
        user = ''
        for k in stat:
            user += '{0} '.format(k['mobile'])
        mobile = ','.join(user.split())
        with open('user.txt', 'w') as f:
            f.write(mobile)

    # 发送报警信息
    def Send_Message(self, content):
        self.content = {
            # "touser":  "User01|User02|User03",   # 成员， @all及所有人  "UserID1|UserID2|UserID3",//企业微信的唯一userid，非必输
            "touser": 'jasonhuang',
            # "toparty": '1',                          # 部门，@all 及所有部门  "PartyID1|PartyID2",//部门id，非必输，如果输入了就只给指定部门发送消息
            "msgtype": 'text',                       # 消息类型，文本，图片
            "agentid": self.APP_ID,                  # 企业应用 id
            "safe": "0",                             #
            "text": {
                    "content": content                   # 报警内容
            }
        }
        token = self.Local_Token()
        # 构建告警信息，必须是 json 格式
        msg = messages_content = json.dumps(self.content)
        send_url = '{0}message/send?access_token={1}'.format(
            self.BASEURL, token)
        respone = urllib.request.urlopen(
            url=send_url, data=msg.encode("utf-8")).read()
        stat = json.loads(respone.decode())['errcode']
        if stat == 0:
            print('Succesfully Send To Wechat')
        else:
            token = self.Get_Token()
            send_url = '{0}message/send?access_token={1}'.format(
                self.BASEURL, token)
            respone = urllib.request.urlopen(url=send_url, data=msg).read()
            return respone


if __name__ == '__main__':
    msgs = []
    msg = Wechat().checker()
    for i in msg:
        j = str(i)
        if j.find('BLACKLISTED') != -1:
            msgs.append(j)
    if msgs:
        msgsend = str(msgs)
        wechat = Wechat()
        wechat.Send_Message('邮件服务器公网IP状态告警:'+msgsend)
        print('Error'+msgsend)
    else:
        print('All Exchange Server Internet IP Status are Normal')