以下简单示例,举例说明实际使用场景
一、背景
由于历史原因,需要批量从数据库中拉取出用例数据,用例又需对应需求,因此使用excel建了一张存有需求id信息的原表,从excel表中Load出需求id后,使用需求id到数据库查询出用例,再转存到新的excel用例表。
二、使用方法
import threading
import openpyxl
import pymysql
import os
testsql = "select testname,teststeps,'通过' from test where id in ('{}');"
#读取xlsx文件
workbook = openpyxl.load_workbook('D:\\test\\test.xlsx')
worksheet = workbook.active
def load_test(row1, row2):
#创建数据库连接
conn = pymysql.connect(
host='1.2.3.4',
port=3306,
user='root',
password='abc',
database='test',
charset='utf8mb4'
)
cursor = conn.cursor()
data = []
for i in range(row1, row2):
if worksheet.cell(row=i, column=1).value:
# 获取需求id
testid = worksheet.cell(row=i, column=1).value
data.append(testid)
for reqid in data:
testsqlnew = testsql.format(str(reqid))
cursor.execute(testsqlnew)