# coding:utf-8

import urllib.request

import json

import sys

import time

import re

class dingHook(object):

    def __init__(self,sys):

        self.msg = sys.argv[1:]

        self.url = "https://oapi.dingtalk.com/robot/send?access_token=30eee8d20287b0051d7c25cc6e468dcdd01XXXX"

    def request(self, url, method, data=None, head={}):

        request = urllib.request.Request(url=url, headers=head)

        request.get_method = lambda: method

        httpRes = urllib.request.urlopen(request, data)

        content = httpRes.read()

        httpRes.close()

        return content

    def post_start(self):

        data = {}

        data["msgtype"] = "text"

        data["text"] = {}

        data["text"]["content"] = " jenkins：开始执行{}自动化......".format(self.msg[0])

        data = json.dumps(data).encode(encoding='UTF8')

        head = {"Content-Type": "application/json"}

        content = self.request(self.url, "POST", data, head)

        print(content)

        return content

    def post_link(self, text, title, picurl="", messageurl=""):

        data = {}

        data["msgtype"] = "link"

        data["link"] = {}

        data["link"]["text"] = text

        data["link"]["title"] = title

        data["link"]["picUrl"] = picurl

        data["link"]["messageUrl"] = messageurl

        data = json.dumps(data).encode(encoding='UTF8')

        head = {"Content-Type": "application/json"}

        content = self.request(self.url, "POST", data, head)

        print(content)

        return content

    def post_result(self):

        i = 0

        result = "FAILURE"

        # 获取当前jenkins构建版本号

        get_buildnum = r"C:\Users\ASUS\.jenkins\jobs\{}\nextBuildNumber".format(self.msg[0])

        with open(get_buildnum, encoding='utf-8')  as fd:

             num = fd.read()

        buildNumber = str(int(num)-1)

        print(buildNumber)

        # 获取jenkins自动化执行结果

        get_buildresult = r"C:\Users\ASUS\.jenkins\jobs\{}\builds\{}\log".format(self.msg[0], buildNumber)

        # with open(get_buildresult)  as output_result:

        #      print("result:",output_result.name)

        #      build_result = output_result.read()

        #获取结果报告中的返回结果

        name_re = r'\\Report\\report\\(.*?.html)'

        with open(get_buildresult )  as fd:

             log_content = fd.read()

        html_name = re.findall(name_re, log_content)

        get_RF_result = r'C:\Users\ASUS\.jenkins\jobs\{}\htmlreports\HTML_Report\{}'.format(self.msg[0], html_name[0])

        print(get_RF_result)

        # jenkins自动化结果报告地址

        messageurl = "http://localhost:8080/job/{}/HTML_Report" .format(self.msg[0])

        # 钉钉结果报告消息图片

        picurl = "D:/work/ClTest/pased.jpg"

        picurl = "http://localhost:8080/job/{}/HTML_Report/pased.jpg".format(self.msg[0])

         # 钉钉结果报告标题

        title = "【%s】执行结果" % self.msg[0]

        print(title)

        try:

            with open(get_buildresult )  as fd:

                build_result = fd.read()

                #防止生成报告没这么快，报错时先等待5s

                if "No such file or directory" in build_result:

                    time.sleep(5)

                #判断jenkins自动化是否执行结束

                if  "Finished" in build_result:

                    with open(get_RF_result,encoding='UTF-8' )  as fd:

                            output_rf_result = fd.read()

                    result = r'<h2>Results</h2>'

                    failure_index = output_rf_result.find(result)

                    result_content = output_rf_result[failure_index:]

                    print(result_content)

                    failure_api = re.findall(r'<td[^>]*>(.*?)</td>', result_content,re.I|re.M)

                    print(failure_api)

                    failure_api = ["- " + a + '\n\n' for a in failure_api]

                    for a in failure_api:

                        if 'Failed' in a:

                             result = " {}自动化执行失败!\n".format(self.msg[0])

                             print(a)

                             content = self.post_link(text=result, title=title, messageurl=messageurl, picurl=picurl)

                             return content

                    result = "{}自动化执行成功!\n".format(self.msg[0])

                    content = self.post_link(text=result, title=title, messageurl=messageurl, picurl=picurl)

                    return content

                    # fail_case = int(case_count_list[1])

                    # suc_case = int(case_count_list[3])

                    # if fail_case == 0:

                    #     result = "SUCCESS-成功:%d-失败:%d" % (suc_case,fail_case)

                    # else:

                    #     result = "FAILURE-成功:%d-失败:%d" % (suc_case,fail_case)

        except Exception as e:

            print(e)

            time.sleep(10)

    def ding_hook(self, itype="start"):

        if itype == "start":

            self.post_start()

        elif itype == "end":

            self.post_result()

if __name__ == '__main__':

    dh = dingHook(sys)

    print(dh.ding_hook(sys.argv[2]))