day12_框架一tools.py代码

import xlrd,requests,time,os
from xlutils import copy
from conf.settings import DATA_PATH
def readCase(case_path):
    case_list = [] # 存放所有的测试用例，给后面运行的时候使用
    book = xlrd.open_workbook(case_path)
    sheet = book.sheet_by_index(0)
    for line in range(1,sheet.nrows):
        case = [] # 存放每条用例的小case
        line_case = sheet.row_values(line) # row_values是excel里面每一行的所有数据
        project = line_case[0] # 项目名称
        model = line_case[1] # 请求模块
        case_id = line_case[2] # case_id
        detail = line_case[3] # 用例描述
        url = line_case[4] # 请求url
        method = line_case[5] # 请求方式
        req_data = line_case[6] # 请求数据
        hope = line_case[7] # 预期结果
        tester = line_case[11] # 测试人员
        case = [project,model,case_id,detail,url,method,req_data,hope,tester]
        case_list.append(case)
    return case_list

def strToDict(data): # 把传入的字符串data转成字典，my_request函数里的data是一个字典形式
    dic = {}
    #username=ssj,passwd=123456
    dic_list = data.split(',') # ['username=ssj','passwd=123456']
    if not dic_list[0]: # 如果dic_list[0]不为真，返回一个空字典
        return dic
    for k in dic_list:
        k_list = k.split('=') # ['username','ssj']['passwd','123456']
        dic_k = k_list[0] # 'username','passwd'
        dic_v = k_list[1] # 'ssj','123456'
        dic[dic_k] = dic_v # 生成字典格式：{'username':'ssj','passwd','123456'}
    return dic

def my_request(method,url,data): # 用这个函数来请求接口数据，返回实际结果
    new_data = strToDict(data)
    try:
        if method.upper() == 'GET':
            r = requests.get(url,new_data) # new_data必须是字典格式
        else:
            r = requests.post(url,new_data) # new_data必须是字典格式
    except Exception as e:
        return '出错了，错误结果是%s' % e
    return r.text # r.text返回的是一个字符串，就是下面的response是一个字符串

def check_res(response,hope): # response是实际结果，hope是预期结果
    new_response = response.replace('": "','=').replace('": ','=') # 将实际结果生成的json串的字符串替换成=
    for check in hope.split(','): # 预期结果是error_code=0,userId=423这种格式，生成['error_code=0','userId=423']
        if check not in new_response: # 循环error_code=0和userId=423,如果不在new_response里，返回False
            return '失败'
        return '通过'

# 反写excel
def write_excel(src_case_path,res_list): # res_list是一个二维数组，每个小list里面包含请求报文、返回报文和测试结果
    src_book = xlrd.open_workbook(src_case_path) # 打开一个excel
    new_book = copy.copy(src_book) # 复制一份
    sheet = new_book.get_sheet(0) # 打开第0个sheet页
    line = 1
    for res in res_list: # res是每一行list
        req = res.get('request') # 请求报文，因为main.py里面是字典，不能通过下标取值，这种方式可以获得对应的值
        response = res.get('response') # 返回报文，因为main.py里面是字典，不能通过下标取值，这种方式可以获得对应的值
        status = res.get('status') # 测试结果，因为main.py里面是字典，不能通过下标取值，这种方式可以获得对应的值
        # req = res[0]
        # response = res[1]
        # status = res[2]
        sheet.write(line,8,req)
        sheet.write(line,9,response)
        sheet.write(line,10,status)
        line += 1
    file_name = time.strftime('%Y%m%d%H%M%S')+os.path.basename(src_case_path) # 执行完excel后的文件名
    abs_path = os.path.join(DATA_PATH,file_name).replace('xlsx','xls')
    new_book.save(abs_path)

if __name__ == '__main__':
    # res = readCase(r'D:\20171216\测试用例1.xlsx')
    # print(res)
    # res = strToDict('username=sunsj123,passwd=aA123456')
    # print(res)
    # d = {'username': 'sunsj123', 'passwd': 'aA123456'}
    # r = my_request('get','http://lanxia.fukc.com','a = 1')
    # print(r)
    #     res = '''
    #     {
    #   "error_code": 0,
    #   "login_info": {
    #     "login_time": "20171231220218",
    #     "sign": "be8f92646e93a5e8c4ce215335b7e693",
    #     "userId": 423
    #   }
    # }
    # '''
    #     r = check_res(res,'error_code=0,userId=423')
    #     print(r)
        res_list = [
        [
         'http://api.nnzhp.cn/api/user/login?username=sunsj123&passwd=aA123456',
         '{"error_code":0,"msg":"成功"}',
         '通过'
        ]
    ]
        excel_path = r'D:\20171216\测试用例1.xlsx'
        write_excel(excel_path,res_list)