使用腳本向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)

運行結果：