Python脚本批量造数据、跑定时任务协助测试

批量造数据

• 连接Mysql的信息

import pymysql
# 数据库连接信息
# 多个库要有多个conn
conn = pymysql.connect(
    host="主机",
    user="用户名",
    password="密码",
    database="库名"
)
conn1 = pymysql.connect(
    host="主机",
    user="用户名",
    password="密码",
    database="库名"
)

# 创建游标对象
cursor = conn.cursor()
cursor1 = conn1.cursor()

# 执行对应的SQL
cursor.execute
# 获取执行结果
Result=cursor.fetchall()

　　场景一：基于已有的csv文件，分批次读取csv文件中的字段值作为变量填充到执行的SQL语句

• 分批读取csv文件中的值

csv_file_path = 'csv文件目录'
with open(csv_file_path, 'r',encoding='utf-8') as file:
    reader = csv.reader(file)
    next(reader)  # Skip the header row

    batch_size = 100  # 每批处理的数量
    total_items = 3100  # 总共需要处理的数量

    for i in range(0, total_items, batch_size):
        # 在每次循环中处理 batch_size 个项目
        # 可以在循环体内部使用 i 作为起始索引
      
        for j in range(i, min(i + batch_size, total_items)):
            row = next(reader)
            # 打印这一行的数据
            print(row)
　　　　

 　　场景二：随机生成特殊字段的值，作为变量填充到Insert语句中

• 随机生成统代

import random
import string
def generate_credit_code():
    # 生成第1位登记管理部门代码
    管理部门代码 = ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D']
    register_department = random.choice(管理部门代码)
    # print('管理部门代码为',register_department)

    # 生成2-9位组织机构代码
    organizations_code = []
    for _ in range(8):
        org_code = ''
        for _ in range(8):
            org_code += random.choice(string.ascii_uppercase + string.digits)
        organizations_code.append(org_code)
    organizations_code=random.choice(organizations_code)
    # print('组织机构代码为',organizations_code)


    # 生成10-17位统一社会信用代码
    unification_credit_code = ''
    for _ in range(8):
        unification_credit_code += random.choice(string.ascii_uppercase + string.digits)
    # print('统一社会信用代码为',unification_credit_code)

        # 组合统一社会信用代码
    credit_code = f"{register_department}{''.join(organizations_code)}{unification_credit_code}"
    return credit_code

•  随机生成注册号

mport random

#这个注册号是由15个随机数字组成的，使用random.choice方法从0-9中随机选择数字。这个方法会被调用15次，每次都会生成一个随机数字，然后通过字符串的join方法将这15个数字拼接在一起。
def generate_reg_code():
    # 15位注册号，以0开头
    reg_code  =  ''.join(random.choice('0123456789') for i in range(15))
    return reg_code

 

结合python+pytest+fixture 实现定时任务接口调用

目录结构

（有些乱。。。

 

-- config.ini     存放的是系统固定的url之类的

 

-- conftest.py  一般用于放登录接口，用户返回token，利用fixture被其他接口使用

import pytest
import requests
import pymysql
from config import readconfig
readcon = readconfig.Read()


@pytest.fixture(scope="session")
# 这个方法是pytest封装公共方法的一个文件，文件名必须是（conftest.py)
# 作用： 其他地方在使用这个方法时就不用from XX import cc 然后也不用实例化了


def test_login():
    msg = {
        "username": '用户名',
        "password": '加密后的密码'
    }

    url =readcon.get_URL("baseurl")
    cc = requests.post(url+"api/uxxxxxxr/login",  params=msg)
    getjson = cc.json()

    # 获取token
    tok = getjson['data']['token']
    userid = getjson['data']['userId']
    return tok, userid

 

定时任务

import pytest
import requests
from config import readconfig

read = readconfig.Read()
class TestCase1:
     global url, tim  # 全局变量，便于其他地方调用
     url = read.get_URL("baseurl")  
     tim = read.get_URL("timeout")
     
     def test_case1(self, test_login):
         head = {'Content-Type': 'application/json', 'Authorization': test_login[0]}  # test_login[0]为token
         NewtestCreditCodeList = []
         SelectNewtestGs = "select 字段1,字段2,字段3 from 数据表 order by id desc  limit 100"
         cursor.execute(SelectNewtestGs)
         SelectNewtestResult = cursor.fetchall()
 　　　　　　
         for tuple in SelectNewtestResult:
             NewtestCreditCodeList.append(tuple[2]) 
 
         NewtestGsCreditCodeListResult = ', '.join('"' + i + '"' for i in NewtestCreditCodeList)
         print('结果为', NewtestGsCreditCodeListResult)
         r = requests.get(url + 'api/exxxxxh/txxx/xxxxx?入参='+NewtestGsEidListResult,  headers=head)
         print(r.json())