您的位置:首页 > 编程语言 > Python开发

Python自定义钉钉机器人发送自动化结果报告

2018-01-20 18:10 597 查看

环境python3.5+jenkins

# 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]))


  运行命令:python urllib-test.py 工程名 end

问题:TypeError: POST data should be bytes or an iterable of bytes. It cannot be of type str.

data = json.dumps(data)这行后加个.encode(encoding='UTF8')


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: