接口自动化简单的数据驱动框架ATP（基于excel）

import os
# print(os.path.dirname(os.path.abspath(__file__)))#当前文件的父路径：D:\python\yao_code\day10\testatp\conf
#为了以后方便用到其他目录下的数据，所以要取其他目录的绝对路径。因为其他目录也都在testatp下，所以先获取到testatp
#先用abspath，获取到的绝对路径中，路径符都是\，如果不用abspath，直接获取dirname,路径符是/

BASE_PATH=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#当前文件的父级conf的父级testatp
LOG_PATH=os.path.join(BASE_PATH,‘logs‘,‘atp.log‘)
LOG_LEVEL=‘debug‘
#logs文件夹的绝对路径。用base_path与logs拼接方法得到。因为join方法可以自动识别系统路径分隔符。不论在linux还是windows下都可以用系统自己的路径符拼接

CASE_PATH=os.path.join(BASE_PATH,‘cases‘)#原理同logs绝对路径拼接方法,用例存放的路径

#邮箱的相关参数
MAIL_HOST=‘smtp.qq.com‘
MAIL_USER=‘111111@qq.com‘#邮箱
MAIL_PASSWORD=‘fdfsfasfasfd‘#邮箱的授权码，注意，不是密码
TO=‘23323233@qq.com‘#收件人邮箱，多个的话用[‘‘,‘‘]

#这里可以用nnlog模块，该模块已经对logging类进行过封装了
import nnlog
from conf import setting
atp_log=nnlog.Logger(setting.LOG_PATH,level=setting.LOG_LEVEL)#日志的保存路径和日志级别都在setting.py配置文件中配置好的，直接饮用

import yagmail
from conf import setting
from lib.log import atp_log
def sendmail(title,content,attrs=None):
    try:
        m=yagmail.SMTP(host=setting.MAIL_HOST,user=setting.MAIL_USER,password=setting.MAIL_PASSWORD,smtp_ssl=True)#邮箱信息也在setting.py中配置
        m.send(to=setting.TO,
               subject=title,
               contents=content,
               attachments=attrs)
    except Exception as e:
        atp_log.error(‘邮件发送失败‘)

import xlrd,requests
from xlutils import copy
from lib.log import atp_log
class OpCase():
    def get_cases(self,filepath):
        cases=[]#保存用例
        print(‘filepath:%s‘%filepath)
        if filepath.endswith(‘.xls‘) or filepath.endswith(‘xlsx‘):
            try:
                book=xlrd.open_workbook(filepath)#打开文件
                sheet=book.sheet_by_index(0)#获取第一个sheet表
                nrows=sheet.nrows#获取表格行数(带标题，所以是比用例条数多一）
                for i in range(1,nrows):#从第2行循环读用例
                    row_data=sheet.row_values(i)#获取每行的数据，返回一个list
                    cases.append(row_data[4:8])#获取4,5,6,7列的数据放入cases:url,method,data,check
                atp_log.info(‘共有%s条用例‘%len(cases))
                self.filepath=filepath#这样，下面的函数就也可以用filepath
                # print(cases)
            except Exception as e:
                atp_log.error(‘文件打开失败:%s‘%e)
        else:
            atp_log.error(‘用例格式不合法，需要xls或xlsx:%s‘%filepath)
        return cases
    def my_request(self,url,data,method):
        method=method.upper()#统一请求方法为大写

        data=self.data_to_dic(data)
        print(‘data处理成字典后-------------%s‘%data)
        try:

            if method==‘GET‘:
                res=requests.get(url,params=data).text
            elif method==‘POST‘:
                res=requests.post(url,data).text
                # print(res)
            else:
                atp_log.warning(‘该请求方式暂不支持‘)
                res=‘该请求方式暂不支持‘
        except Exception as e:
            atp_log.error(‘接口不通url:%s\n 错误信息：%s‘%e)
            res=‘接口不通‘

        return res

    def data_to_dic(self,data):#data格式是a=1,b=2
        print(data)
        data=data.split(‘,‘)#[a=1,b=2]
        res={}#用来保存转换后的请求数据
        for d in data:#d的值：a=1
            k,v=d.split(‘=‘)
            res[k]=v
        print(‘res------------------%s‘%res)
        return res


    def check(self,res,check):
        # print(‘------------‘+res)
        # print(check)
        res=res.replace(‘": ‘,‘=‘).replace(‘": ‘,‘=‘)
        for c in check.split(‘,‘):
            if c not in res:
                atp_log.error(‘失败，预期结果：%s,实际结果：%s‘%(c,res))
                return ‘失败‘
        return ‘成功‘

    def write_to_excel(self,cases_res):#cases_res格式[[接口返回数据，成功]，[接口返回数据，失败]]
        book=xlrd.open_workbook(self.filepath)
        newbook=copy.copy(book)
        sheet=newbook.get_sheet(0)
        row=1
        for case_res in cases_res:
            sheet.write(row,8,case_res[0])#第8列是实际结果
            sheet.write(row,9,case_res[1])#第9列事测试结果
            row+=1
        newbook.save(self.filepath.replace(‘xlsx‘,‘xls‘))

# #如果不在python运行，用命令方式运行，那么首先运行的是这个start文件。这时候文件目录都还不在环境变量中，所以要先把当前文件包加入环境变量

import os,sys
BASE_PATH=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取到testatp
sys.path.insert(0,BASE_PATH)#将testatp加入环境变量

from lib.common import OpCase
from conf import setting
from lib.sendmail import sendmail
class CaseRun():
    def run(self):
        op=OpCase()
        res_list=[]#用来存放实际结果和测试结果。写入excel

        #遍历获取用例文件
        for t in os.listdir(setting.CASE_PATH):
            abs_path=os.path.join(setting.CASE_PATH,t)#获取到用例文件的绝对路径
            case_list=op.get_cases(abs_path)#获取到所有用例，是个二维数组
            # print(case_list)
            fail_count=0
            pass_count=0
            for case in case_list:  #循环每条用例
                url,method,data,check=case#获取到用例对应的数据
                res = op.my_request(url,data,method)  # 请求数据,返回实际结果
                status=op.check(res,check)
                print(‘status............%s,res...............%s‘%(status,res))
                res_list.append([res,status])
                if status==‘失败‘:
                    fail_count+=1
                else:
                    pass_count+=1
            # print(‘res_list:%s‘%res_list)
            op.write_to_excel(res_list)
            msg=‘本次一共运行了%s条用例，失败%s条，成功%s条‘%(len(res_list),fail_count,fail_count)

            sendmail(‘测试结果‘,msg,abs_path)



# print(os.listdir(setting.CASE_PATH))
# print(BASE_PATH)
CaseRun().run()