python+requests+excel+unittest+ddt 接口自动化数据驱动并生成 html 报告 (优化版)

import jsonimport requestsfrom common.operation_json import OperetionJson
class OperationHeader:
    def __init__(self, response):        # self.response = json.loads(response)        self.response = response
    def get_response_token(self):        '''        获取登录返回的token        '''        token = {"data":{"token":self.response['data']['token']}}        #token = {"token": self.response['data']['token']}        return token
    def write_token(self):        op_json = OperetionJson()        op_json.write_data(self.get_response_token())
    def get_response_msg(self):        reponse_msg = {"msg":self.response['msg']}        #print("reponse_msg:", reponse_msg)        return reponse_msg

#coding:utf-8import jsonclass OperetionJson:
    def __init__(self,file_path=None):        if file_path  == None:            self.file_path = '../case/cookie.json'        else:            self.file_path = file_path        self.data = self.read_data()
    #读取json文件    def read_data(self):        with open(self.file_path, 'r', encoding='utf-8') as fp:            data1 = fp.read()            if len(data1) > 0:                data = json.loads(data1)            else:                data = {}            return data
    #根据关键字获取数据    def get_data(self,id):        print(type(self.data))        return self.data[id]
    #写json    def write_data(self,data):        with open('../case/token.json','w') as fp:            fp.truncate()  # 先清空之前的数据，再写入，这样每次登录的token都是不一样的            fp.write(json.dumps(data))

 # 请求头部headers    try:        headers = eval(testdata["headers"])        if testdata["token"] == "yes":            op_json = OperetionJson("../case/token.json")            token = op_json.get_data('data')            headers = dict(headers, **token)        print("请求头部：", headers)        log.info("请求头部：", headers)    except:        headers = None

    def get_response_cookie(self):        cookie1 = requests.utils.dict_from_cookiejar(self.response.cookies)        cookie = {"data":{"gfsessionid":cookie1["gfsessionid"]}}        # {"data": {"token": self.response['data']['token']}}        print("cookie:", cookie)        return cookie    def write_cookie(self):        op = OperetionJson()        op.write_mydata(self.get_response_cookie())

    def write_mydata(self,data):        with open('../case/cookie.json','w') as fp:            fp.truncate()  # 先清空之前的数据，再写入，这样每次登录的token都是不一样的            fp.write(json.dumps(data))

    try:        headers = eval(testdata["headers"])        if testdata["cookie"] == "yes":            op_json = OperetionJson("../case/cookie.json")            token1 = op_json.get_data('data')            headers = dict(headers, **token1)        print("请求头部：", headers)        log.info("请求头部：", headers)    except:        headers = None

#coding=utf-8from email.mime.text import MIMETextimport timeimport smtplibimport getpassfrom email.mime.multipart import MIMEMultipartfrom email.mime.text import MIMETextfrom email.mime.base import MIMEBasefrom email import encodersimport emailimport os
def sendmain(file_path,mail_to = '[email protected]'):    mail_from = '[email protected]'    f = open(file_path,'rb')    mail_body=f.read()    f.close()        #msg = email.MIMEMultipart.MIMEMultipart()    msg = MIMEMultipart()
    # 构造MIMEBase对象做为文件附件内容并附加到根容器      contype = 'application/octet-stream'      maintype, subtype = contype.split('/', 1)      ## 读入文件内容并格式化      data = open(file_path, 'rb')    #file_msg = email.MIMEBase.MIMEBase(maintype, subtype)    file_msg = MIMEBase(maintype, subtype)    file_msg.set_payload(data.read( ))      data.close( )      #email.Encoders.encode_base64(file_msg)    encoders.encode_base64(file_msg)    ## 设置附件头      basename = os.path.basename(file_path)      file_msg.add_header('Content-Disposition',                          'attachment', filename = basename)      msg.attach(file_msg)      print(u'msg 附件添加成功')        msg1 = MIMEText(mail_body,_subtype='html',_charset='utf-8')    msg.attach(msg1)        if isinstance(mail_to,str):        msg['To'] = mail_to    else:         msg['To'] = ','.join(mail_to)    msg['From'] = mail_from    msg['Subject'] = u'xxxxxxxxx接口自动化测试' # 邮件标题    msg['date']=time.strftime('%Y-%m-%d-%H_%M_%S')    print(msg['date'])
    smtp = smtplib.SMTP()    smtp.connect('smtp.126.com')    smtp.login('[email protected]','aaaaaaaaaa') # 这里的密码是邮件第三方客户端认证密码    smtp.sendmail(mail_from, mail_to, msg.as_string())    smtp.quit()    print('email has send out !')'''if __name__=='__main__':    sendmain('../report/2017-08-18-10_18_57_result.html')'''

sendmain(htmlreport, mail_to=['[email protected]', '[email protected]', '[email protected]']) #多个收件人的话，直接在列表中，用,号隔开即可

def mk_test_name(name, value, index=0):    """    Generate a new name for a test case.
    It will take the original test name and append an ordinal index and a    string representation of the value, and convert the result into a valid    python identifier by replacing extraneous characters with ``_``.
    We avoid doing str(value) if dealing with non-trivial values.    The problem is possible different names with different runs, e.g.    different order of dictionary keys (see PYTHONHASHSEED) or dealing    with mock objects.    Trivial scalar values are passed as is.
    A "trivial" value is a plain scalar, or a tuple or list consisting    only of trivial values.    """
    # Add zeros before index to keep order
    index = "{0:0{1}}".format(index + 1, index_len, )    if not is_trivial(value) and type(value) is not dict: # 增加的地方，增加value的字典判断
        return "{0}_{1}_{2}".format(name, index, value.name) # 修改的地方，增加返回的值    if type(value) is dict: # 增加的地方        try: # 增加的地方            value = value["name"] + "_" + value["function"] # 增加的地方，name和function必须是execl用例中整正存在的表头，这里我是把两个表头合并了（name是我表格中接口的名称，function是表格中接口的功能描述）        except: # 增加的地方            return "{0}_{1}".format(name.index) # 增加的地方    try:        value = str(value)    except UnicodeEncodeError:        # fallback for python2        value = value.encode('ascii', 'backslashreplace')    test_name = "{0}_{1}_{2}".format(name, index, value)  # 修改的地方    return re.sub(r'\W|^(?=\d)', '_', test_name)

 DEFAULT_TESTER = getpass.getuser()

# LOG日志记录    logging.basicConfig(level=logging.DEBUG,                        format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',                        datefmt='%a, %d %b %Y %H:%M:%S',                        filename=log_path + '/' + now + r"result.log",                        filemode='w')    logger = logging.getLogger()    logger.info(all_case)

log = logging.getLogger()log.info("请求头部：", headers)