使用脚本向Tapd批量导入测试用例

问题：目前公司管理项目工具是Tapd，我们测试用例管理也在上面，Tapd每次只支持一个sheet导入，假如一个excel中存在多个sheet，导入时还是很花费时间，所以写了一个脚本使用接口来导入，当然本质还是一个一个的导入，但是至少解放了双手！

设计思路如下：

 

 

 代码如下：共两个python文件

1.creatCase.py

import openpyxl
from openpyxl import Workbook, workbook

# 读取数据，将原始文件的sheet拆分成一个一个的电子表格文件
def copy_file(sorce_file):
    wb = openpyxl.load_workbook(sorce_file)
    sheets = wb.sheetnames
    print(sheets)
    if not sheets:
        print('文件为空')
    else:
        for sheetname in sheets.__iter__():
            if sheetname == '目录':
                print('存在目录表单，跳过')
            else:
                print('复制' + format(sheetname))
                copy_sheet(sorce_file, sheetname)


def copy_sheet(sorce_file_path, sheet_name):
    # 源文件
    wb1 = openpyxl.load_workbook(sorce_file_path)
    new_file_path = 'testcase_list\\'+sheet_name + '.xlsx'
    sheet1 = wb1[sheet_name]
    wb = Workbook()
    ws = wb.active
    max_row = sheet1.max_row  # 最大行数
    max_column = sheet1.max_column  # 最大列数

    for m in range(1, max_row + 1):
        for n in range(97, 97 + max_column):  # chr(97)='a'

            n = chr(n)  # ASCII字符
            i = '%s%d' % (n, m)  # 单元格编号
            cell1 = sheet1[i].value
            ws[i].value = cell1

    wb.save(new_file_path)  # 保存数据
    wb1.close()  # 关闭excel
    wb.close()


if __name__ == '__main__':
    sorce_file = 'C:/xxx/xxx/xxx/.../xx.xlsx'
    copy_file(sorce_file)

2.接口操作

uploadCase.py

import requests
import openpyxl
import creatCase

user_agent = r'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_3) AppleWebKit/537.36' \
             r' (KHTML, like Gecko) Chrome/61.0.3163.79 Safari/537.36'
headers = {'User-Agent': user_agent, 'Connection': 'keep-alive'}

# 登录tapd
def login_fun(login_url,name,pwd,project_id):
    values = {'data[Login][ref]': 'http://www.tapd.cn/'+project_id+'/sparrow/tcase/tcase_list',
              'data[Login][encrypt_key]': 'bORItxocKUHkRTlyhYmDMLZGQGtBvBc/GWZKPTdll6s=',
              'data[Login][encrypt_iv]': '4J11QPi39mXTJtJ0HEqnCA==',
              'data[Login][site]': 'TAPD',
              'data[Login][via]': 'encrypt_password',
              'data[Login][email]': name,
              'data[Login][password]': pwd,
              'data[Login][login]': 'login',
              'dsc_token': '16vSWxWRr85KB5Rn'}

    s= requests.Session()
    content = s.post(url=login_url, headers=headers, data=values)
    return s

# tapd中新增测试用例父目录
def creat_parafold(login_url,name,pwd,project_id,paraname):
    creat_parafold_url = 'https://www.tapd.cn/'+project_id+'/sparrow/tcase_categories/quick_add_category'
    data1 = {'data[TcaseCategory][parent_id]':'0','data[TcaseCategory][name]':paraname}
    s=login_fun(login_url,name,pwd,project_id)
    creat_parare =s.post(url=creat_parafold_url, headers=headers, data=data1)
    paraid_value = creat_parare.json()['id']
    return paraid_value

# 将excel中的所有sheetname以测试用例子目录形式创建到tapd测试用例父目录下，并在对应目录下上传测试用例
def creatFold_importCase(login_url,name,pwd,project_id,casepath,paraname):
    creatCase.copy_file(casepath)
    wb = openpyxl.load_workbook(casepath)
    sheets = wb.sheetnames

    creat_fold_url = 'https://www.tapd.cn/'+project_id+'/sparrow/tcase_categories/quick_add_category'
    key1 = "data[TcaseCategory][parent_id]"
    key2 = 'data[TcaseCategory][name]'
    s = login_fun(login_url, name, pwd, project_id)
    if paraname:
        paraid = creat_parafold(login_url,name,pwd,project_id,paraname)
        print(paraid)
    else:
        paraid = 0
        print(paraid)

    # dict_id = {}
    for value2 in sheets:
        value1= {key1: paraid, key2: value2}
        creat_re = s.post(url=creat_fold_url,headers=headers,data=value1)
        id_value=creat_re.json()['id']
        print(id_value)
        uploadfile_url = 'https://www.tapd.cn/'+project_id+'/imports/import_tcase'
        filepath = 'testcase_list/'+value2+'.xlsx'
        file = {
            "content": open(filepath, "rb"),
            "Content-Type": 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
            "Content-Disposition": "form-data",
            "filename": value2+'.xlsx'}

        case_data = {'data[tcase][extra_params]': '',
                     'data[tcase][import_type]': 'add',
                     'data[tcase][selected_category_id]': id_value,
                     'data[tcase][Apply]': '下一步',
                     'dsc_token': "16vSWxWRr85KB5Rn"}
        upload_res = s.post(url=uploadfile_url, headers=headers, files=file, data=case_data, allow_redirects=False)
        print(upload_res)
        sure_url=upload_res.headers['Location']
        print(sure_url)
        sure_data = {'data[tcase][selected_category_id]': id_value,
                     'data[tcase][Apply]': '确认导入',
                     'dsc_token': '16vSWxWRr85KB5Rn'}
        sure_res = s.post(url=sure_url, headers=headers, data=sure_data)
        print(sure_res)


if __name__ == '__main__':
    # 进入项目的测试用例页面，点击退出后，再登录时的请求url
    login_url='xxxx'
    # 登录账号
    mobilephone='xxx'
    # 登录密码，可在登录是请求参数中查看，urlencode形式
    pwd='xxx'
    # 导入项目id
    project_id='xxx'
    # 测试用例所在目录
    casepath='C:/xxx/xxx/xxx测试用例.xlsx'
    # paraname代表创建测试目录的根目录，如果传空，则新增的测试子目录默认创建在【所有的】目录下，如果传一个字符串，则后续测试子目录创建在该目录下
    paraname="xxx项目V1.0版本"
    creatFold_importCase(login_url,mobilephone,pwd,project_id,casepath,paraname)

运行结果：