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

以你之姓@ 2023-10-16 09:54 10阅读 0赞

**目录**

批量造数据

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

目录结构

定时任务

--------------------

## **批量造数据** ##

*  连接Mysql的信息

1 import pymysql
     2 # 数据库连接信息
     3 # 多个库要有多个conn
     4 conn = pymysql.connect(
     5     host="主机",
     6     user="用户名",
     7     password="密码",
     8     database="库名"
     9 )
    10 conn1 = pymysql.connect(
    11     host="主机",
    12     user="用户名",
    13     password="密码",
    14     database="库名"
    15 )
    16 
    17 # 创建游标对象
    18 cursor = conn.cursor()
    19 cursor1 = conn1.cursor()
    20 
    21 # 执行对应的SQL
    22 cursor.execute
    23 # 获取执行结果
    24 Result=cursor.fetchall()

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

*  分批读取csv文件中的值

1 csv_file_path = 'csv文件目录'
     2 with open(csv_file_path, 'r',encoding='utf-8') as file:
     3     reader = csv.reader(file)
     4     next(reader)  # Skip the header row
     5 
     6     batch_size = 100  # 每批处理的数量
     7     total_items = 3100  # 总共需要处理的数量
     8 
     9     for i in range(0, total_items, batch_size):
    10         # 在每次循环中处理 batch_size 个项目
    11         # 可以在循环体内部使用 i 作为起始索引
    12       
    13         for j in range(i, min(i + batch_size, total_items)):
    14             row = next(reader)
    15             # 打印这一行的数据
    16             print(row)

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

*  随机生成统代

1 import random
     2 import string
     3 def generate_credit_code():
     4     # 生成第1位登记管理部门代码
     5     管理部门代码 = ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D']
     6     register_department = random.choice(管理部门代码)
     7     # print('管理部门代码为',register_department)
     8 
     9     # 生成2-9位组织机构代码
    10     organizations_code = []
    11     for _ in range(8):
    12         org_code = ''
    13         for _ in range(8):
    14             org_code += random.choice(string.ascii_uppercase + string.digits)
    15         organizations_code.append(org_code)
    16     organizations_code=random.choice(organizations_code)
    17     # print('组织机构代码为',organizations_code)
    18 
    19 
    20     # 生成10-17位统一社会信用代码
    21     unification_credit_code = ''
    22     for _ in range(8):
    23         unification_credit_code += random.choice(string.ascii_uppercase + string.digits)
    24     # print('统一社会信用代码为',unification_credit_code)
    25 
    26         # 组合统一社会信用代码
    27     credit_code = f"{register_department}{''.join(organizations_code)}{unification_credit_code}"
    28     return credit_code

*  随机生成注册号

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

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

### 目录结构 ###

（有些乱。。。

![178673361a76ecce6b27563fe06fc867.png][]

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

![7874be80a5fc46293b56edfd3bb3d62e.png][]

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

1 import pytest
     2 import requests
     3 import pymysql
     4 from config import readconfig
     5 readcon = readconfig.Read()
     6 
     7 
     8 @pytest.fixture(scope="session")
     9 # 这个方法是pytest封装公共方法的一个文件，文件名必须是（conftest.py)
    10 # 作用： 其他地方在使用这个方法时就不用from XX import cc 然后也不用实例化了
    11 
    12 
    13 def test_login():
    14     msg = {
    15         "username": '用户名',
    16         "password": '加密后的密码'
    17     }
    18
    19     url =readcon.get_URL("baseurl")
    20     cc = requests.post(url+"api/uxxxxxxr/login",  params=msg)
    21     getjson = cc.json()
    22 
    23     # 获取token
    24     tok = getjson['data']['token']
    25     userid = getjson['data']['userId']
    26     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())

--------------------

**以下是我收集到的比较好的学习教程资源，虽然不是什么很值钱的东西，如果你刚好需要，可以评论区，留言【777】直接拿走就好了**

![c1653dd19f35495583cef3678b57b6b0.png][]

![775632765b6744fbbecbd8fdd097d1ab.png][]

**各位想获取资料的朋友请****点赞 + 评论 + 收藏****，三连！**

**三连**之后我会在评论区挨个私信发给你们~

[178673361a76ecce6b27563fe06fc867.png]: https://img-blog.csdnimg.cn/img_convert/178673361a76ecce6b27563fe06fc867.png
[7874be80a5fc46293b56edfd3bb3d62e.png]: https://img-blog.csdnimg.cn/img_convert/7874be80a5fc46293b56edfd3bb3d62e.png
[c1653dd19f35495583cef3678b57b6b0.png]: https://img-blog.csdnimg.cn/c1653dd19f35495583cef3678b57b6b0.png
[775632765b6744fbbecbd8fdd097d1ab.png]: https://img-blog.csdnimg.cn/775632765b6744fbbecbd8fdd097d1ab.png