性能压测脚本造数据_cursor 辅助测试-优快云博客

本文链接：https://blog.youkuaiyun.com/u014045237/article/details/145791324

批量生成token

#coding:utf-8
import openpyxl
import requests
import random
import json
from loguru import logger
#请求的url地址
url_mobile = "https://eapp-gw-uat.lionaitech.com/user/app/mobile"
url_token = "https://eapp-gw-uat.lionaitech.com/auth/oauth/token"
#请求的消息头
headers_mobile={}
headers_mobile["Content-Type"] = "application/json"
headers_token={}
headers_token["Content-Type"] = "application/x-www-form-urlencoded"
write_workbook = openpyxl.Workbook()
sheet_write = None
#写入excel文件的路径
result_file_path = "D:/exeed_token/token.xlsx"

list_phone=["134","135","136","137","138","139","147","150","152","157","158","159","173","178","182","183","130","131","132","145","155","156","175","176","166","133","149","153","173","177","180","181","189","199"]
def get_phone_number():
    for number in range(500):
        random_element = random.choice(list_phone)
        yy = random_element
        for i in range(8):
            x = random.randint(0, 9)
            yy = yy + str(x)
        logger.info("yy:" + yy)
        message_send(yy,number)
# def read_excel():
#     #打开excel文件
#     workbook = openpyxl.load_workbook(filepath)
#
#     names = workbook.sheetnames #获取表格里面的工作薄
#     print(names)
#     for i in range(0,len(names)):
#         sheet_object = workbook[names[i]]
#         print(sheet_object.max_row)
#         for j in range(2,sheet_object.max_row+1):
#             cell_value = str(sheet_object.cell(j, 1).value)
#             print(cell_value)
#             message_send(cell_value,j)

def message_send(phone_number,number):
    mobileStr = f'{{"phone": "{phone_number}", "sixCode": true}}'
    logger.info(mobileStr)
    #获取验证码
    json_body_mobile = {
        "body":mobileStr
    }
    logger.info(str(json_body_mobile))
    responseResult = requests.post(url_mobile, headers=headers_mobile, json=json_body_mobile)
    response_data = responseResult.json()
    #logger.info(response_data)

    json_body_token = {}
    json_body_token["code"] = "123456"
    json_body_token["social"] = phone_number
    json_body_token["grant_type"] = "sms"
    json_body_token["client_secret"] = "c2Th7Jp8Kw8Ly9Px5Hl8Kn1Uu6Kq1Bf0b2Iz4Es8Vi9Za9Mv9Md5Qo8Gg7Ji1Zz2"
    json_body_token["client_id"] = "202069311861295711"
    responseResult_token = requests.post(url_token, headers=headers_token, data=json_body_token)
    response_data_token = responseResult_token.json()
    logger.info(response_data_token)
    response_data_access_token = response_data_token["body"]["access_token"]
    logger.info(response_data_access_token)
    write_excel_data(number+2,phone_number,response_data_access_token)

def write_excel_ini():
    global sheet_write
    sheet_write = write_workbook.create_sheet("Sheet1", 0)
    sheet_write.cell(row=1, column=1).value = '用户手机号'  # 用户工号
    sheet_write.cell(row=1, column=2).value = '对应的token'  # 生成的token
    write_workbook.save(result_file_path)
def write_excel_data(j,phone_number,response_data_access_token):
    sheet_write.cell(row=j, column=1).value = phone_number  # 手机号
    sheet_write.cell(row=j, column=2).value = response_data_access_token # 生成的token
    write_workbook.save(result_file_path)

if __name__ == '__main__':
    #write_excel_ini()
    #read_excel()
    write_excel_ini()
    get_phone_number()

造手机号

//定义手机号前3位
String[] FirstThreeNumber = {"134","135","136","137","138","139","147","150","152","157","158","159","173","178","182","183","130","131","132","145","155","156","175","176","166","133","149","153","173","177","180","181","189","199"};
//随机获取前三位手机号的index
int n= (int)(Math.random() * FirstThreeNumber.length);
//获取手机前三位
String yy = FirstThreeNumber[n];
//循环获取手机号后8位
for(int i = 0; i < 8; i++){
    int x = (int)(Math.random() * 9);
    yy = yy + x;
};
vars.put("phone", yy);

jmeter插入数据到csv

//导入json包
import org.json.JSONObject;
import org.json.JSONArray;
import org.apache.jmeter.samplers.SampleResult;
SampleResult rsp =ctx.getPreviousResult();       
String response_data = rsp.getResponseDataAsString();  //获取上一个接口返回数据，此处是String类型
log.info("\n" + response_data);
//将String类型的返回值构造成JSONObject对象
JSONObject jsondata=new JSONObject(response_data);  //放进json对象里
String data_test = jsondata.get("data").toString();
JSONObject jsonstring =new JSONObject(data_test);
String orderNo=jsonstring.get("orderNo").toString();  //取出orderNo对应的value值       
vars.put("orderNo",orderNo);
//log.info(cost_time);
//log.info(score);
//写入文件
FileWriter fstream=new FileWriter("F:/123/pc.csv",true);//本地存储文件路径
BufferedWriter out = new BufferedWriter(fstream);
out.write("${var}");//token
out.write(",");//换列
out.write(orderNo);//token对应的单号
out.write(System.getProperty("line.separator"));//换行
out.close();
fstream.close();

1万经销商_1万物料_1万收货地址.py

#coding:utf-8
import pymysql
import openpyxl
from loguru import logger

sheet_write = None

write_workbook = openpyxl.Workbook()

result_file_path = "D:/odin项目资料增补资源池导入数据/导入周订单模板_造数据.xlsx"
NumberCount  = 1
def db_insert():
    # 连接数据库
    conn = pymysql.connect(host="172.25.188.206", port=3306, user="odin", password="odin@123", db="chery_otd_admin")
    cursor = conn.cursor()
    global NumberCount
    for i in range(60034, 60034 + 9999):  # 注意这里加了1
        NumberCount=NumberCount+1
        c = i
        try:
            #dealer_base_info
            insert_query_dealer_base_info="""INSERT INTO chery_otd_admin.dealer_base_info
            (tenant_id, owner_id, dept_id, sales_erp, service_erp, id, dealer_code, name, abbreviation, nature, group_code, policy_group_name, product_sequence, zone_code, province_code, city_code, district_code, postal_code, status, remark, last_erp_sync_time, row_version, create_by, create_time, update_by, update_time, del_flag)
            VALUES(100001, "admin", 1, {}, {}, 349999{}, "QR2026{}", "徐州市金慧汽车销售服务有限公司60001", "徐州金慧", 3, NULL, NULL, 1, "CNHD00", "320000", "320300", NULL, NULL, 1, "订单导入测试", NULL, NULL, "admin", "2024-12-26 18:40:43", "", "0000-00-00 00:00:00", "0")""".format(c,c,c,c)

            cursor.execute(insert_query_dealer_base_info)

            print(cursor.rowcount, "记录插入成功。dealer_base_info")
            conn.commit()

            #dealer_sales
            insert_query_dealer_sales = """INSERT INTO chery_otd_admin.dealer_sales
            (tenant_id, owner_id, dept_id, id, dealer_code, store_code, sales_erp, business_license, remark, unified_social_credit_code, erp_same_flag, email, name, address, nature, `level`, longitude, latitude, category, hotline, fax, erp_dept_code, images, operation_status, opening_time, freeze_time, quit_net_time, quit_net_protect_start_time, quit_net_protect_end_time, invoice_unit_name, invoice_tax_code, invoice_address, invoice_mobile, invoice_account, invoice_bank, row_version, create_by, create_time, update_by, update_time, del_flag)
            VALUES(100001, 'admin', 1, 29166642435335{}, 'QR2026{}', NULL, '{}', NULL, NULL, NULL, NULL, NULL, '瑞合晟(天津)供应链科技有限公司', '天津市西青区大寺镇储华道28号环城国际汽配城E区17-20号', 1, '1', NULL, NULL, 1, NULL, NULL, NULL, NULL, 1, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, '', '0000-00-00 00:00:00', '', '0000-00-00 00:00:00', '0')""".format(c,c,c)

            cursor.execute(insert_query_dealer_sales)

            print(cursor.rowcount, "记录插入成功。dealer_sales")
            conn.commit()

            #dealer_delivery_address
            insert_query_dealer_delivery_address = """INSERT INTO chery_otd_admin.dealer_delivery_address
            (tenant_id, id, dealer_code, record_id, purpose_type, apply_order_type, province_code, province_name, city_code, city_name, district_code, district_name, address, details_address, priority, status, remark, row_version, create_by, create_time, update_by, update_time, del_flag, dept_id, owner_id, mobile, id_card, contact_person, dealer_erp)
            VALUES(100001, 39609652231749{}, 'QR2026{}', '61033', 1, NULL, NULL, '河北省', NULL, '保定市', NULL, NULL, '北京市朝阳区八一路1号', '河北省保定市高碑店市北京市朝阳区八一路{}', NULL, 1, NULL, NULL, 'system', '2024-04-07 15:36:17', 'system', NULL, '0', 11, 'system', '123', '123', 'AAA', '{}')""".format(c,c,c,c)


            cursor.execute(insert_query_dealer_delivery_address)

            print(cursor.rowcount, "记录插入成功。dealer_delivery_address")
            conn.commit()

            #product
            insert_query_product = """INSERT INTO chery_otd_product.product
            (id, code, name, sapcode, previous_name, brand_sequence_code, brand_code, brand_category_id, brand_define_id, brand_series_code, brand_series_category_id, brand_series_define_id, vehicle_series_code, vehicle_series_category_id, vehicle_series_define_id, vehicle_type_code, vehicle_type_category_id, vehicle_type_define_id, model_year_code, model_year_category_id, model_year_define_id, powertrain_code, powertrain_category_id, powertrain_define_id, vehicle_version_code, vehicle_version_category_id, vehicle_version_define_id, vehicle_type_alias, ka_exclusive_flag, appearance_color_code, appearance_color_name, interior_color_code, interior_color_name, seat_code, guide_price, internal_transfer_price, emission_standard, power_type_code, manual_automatic, engine_cylinder, material_type_code, can_be_ordered, status, announcement, announcement_expiration_time, remark, create_by, owner_id, update_by, create_time, update_time, del_flag, tenant_id, row_version, original_id)
            VALUES(29747027648405{}, 'T6471LRCLJ{}', '奇瑞/2025款瑞虎8PRO/2025/1.6T DCT国六/豪华版黑色内饰5座/石墨黑RDE', 'T6471LRCLJW0035', 'SQR6471T18TBE1.6T标准黑黑五座23MY-RDE新碳晶黑', '1', 'QR', 459009869833388578, 694465780427082997, 'QR1', 7843366934728390973, 2123809238089907872, 'QR8X', 806823959603689404, 7910804568757978775, 'QR75', 2736051744559315341, 6259520164680610077, 'QR8X_2025', 1874714131928690690, 8353462053204610160, 'QR8X-PR016', 1874714170562424833, 5690215616684247186, 'QR8X-VS0066', 1874714209640755202, 7532110108795027617, NULL, '0', 'JK', '星云紫', 'QR66-057', '黑黑', '7座', 18300.0000, 120010.0000, '6', '燃油', '手动', '6', '1', '1', 1, NULL, '2025-01-02 15:09:51', NULL, 'yuangong62', 'yuangong62', 'admin', '2025-01-02 14:22:24', '2025-01-16 15:30:16', '0', 100001, 3, NULL)""".format(c,c)

            cursor.execute(insert_query_product)

            print(cursor.rowcount, "记录插入成功。dealer_delivery_address")
            conn.commit()

            #dealer_sales_vehicle_type
            insert_query_dealer_sales_vehicle_type = """INSERT INTO chery_otd_product.dealer_sales_vehicle_type
            (id, dealer_code, vehicle_type_code, powertrain_code, powertrain_category_id, authorize_expire_time, `attribute`, create_by, update_by, create_time, update_time, del_flag, tenant_id, owner_id, row_version)
            VALUES(28976119609971{}, 'QR2026{}', 'QR75', 'QR8X-PR016', 1874714170562424833, NULL, '2', 'yuangong62', 'yuangong62', '2025-03-06 19:35:22', '2025-03-06 19:35:22', '0', 100001, 'yuangong62', 0)""".format(c,c)

            cursor.execute(insert_query_dealer_sales_vehicle_type)

            print(cursor.rowcount, "记录插入成功。dealer_sales_vehicle_type")
            conn.commit()

            #vehicle_order_type_info_config

            insert_query_vehicle_order_type_info_config = """INSERT INTO chery_order_deliver.vehicle_order_type_info_config
            (id, order_type, dealer_code, sales_office_code, vehicle_funds_type_status, vehicle_funds_type_code, receiving_address_id, use_rebate, create_by, owner_id, update_by, create_time, update_time, del_flag, tenant_id, row_version)
            VALUES(29649444164798{}, '1', 'QR2026{}', '2000', '1', '8201', 92, '1', 'huojinfeng8', 'huojinfeng8', 'huojinfeng8', '2024-12-06 16:06:13', '2024-12-06 16:06:13', '0', 100001, 0)""".format(c,c)

            cursor.execute(insert_query_vehicle_order_type_info_config)

            print(cursor.rowcount, "记录插入成功。vehicle_order_type_info_config")
            conn.commit()
            #同样的数据写入到excel里面
        finally:
            print("exception")
        write_excel_data(NumberCount,c)

    # 关闭连接
    cursor.close()
    conn.close()

def write_excel_ini():
    global sheet_write
    sheet_write = write_workbook.create_sheet("Sheet1", 0)
    sheet_write.cell(row=1, column=1).value = '经销商ERP号'  # 经销商ERP号
    sheet_write.cell(row=1, column=2).value = '销售办公室'  # 销售办公室
    sheet_write.cell(row=1, column=3).value = '物料号'  # 物料号
    sheet_write.cell(row=1, column=4).value = '计划种类'  # 计划种类
    sheet_write.cell(row=1, column=5).value = '数量'  # 数量
    sheet_write.cell(row=1, column=6).value = '运输方式'  #运输方式
    sheet_write.cell(row=1, column=7).value = '收货地址'  #收货地址
    sheet_write.cell(row=1, column=8).value = '是否使用返利'  # 是否使用返利
    sheet_write.cell(row=1, column=9).value = '备注'  #备注
    write_workbook.save(result_file_path)

def write_excel_data(j,c):
    print("写入到excel的调用次数")
    sheet_write.cell(row=j, column=1).value = c # 经销商ERP号
    sheet_write.cell(row=j, column=2).value = "销售部"  # 销售办公室
    sheet_write.cell(row=j, column=3).value = "T6471LRCLJ{}".format(c)  #物料号
    sheet_write.cell(row=j, column=4).value = "备库订单"  # 计划种类
    sheet_write.cell(row=j, column=5).value = "1"  # 数量
    sheet_write.cell(row=j, column=6).value = "物流运输"  # 运输方式
    sheet_write.cell(row=j, column=7).value = "河北省保定市高碑店市北京市朝阳区八一路{}".format(c)  #收货地址
    sheet_write.cell(row=j, column=8).value = "是" #是否使用返利
    sheet_write.cell(row=j, column=9).value = "大批量数据-性能压测"#备注
    write_workbook.save(result_file_path)

if __name__ == '__main__':
    #write_excel_ini()
    #read_excel()
    write_excel_ini()
    db_insert()

1万月度分经销商目标

import pymysql
import openpyxl

result_file_path = "D:/odin各种导入数据的资源/导入月度分经销商目标模板_code_sit.xlsx"

sheet_write = None
write_workbook = openpyxl.Workbook()
NumberCount = 1
def db_insert():
    # 连接数据库
    conn = pymysql.connect(host="172.25.188.206", port=3306, user="odin", password="odin@123", db="chery_otd_admin")
    cursor = conn.cursor()
    global NumberCount
    for i in range(60005, 60005 + 9999):  # 注意这里加了1
        NumberCount = NumberCount + 1
        c = i
        #-- 创建车型编码
        insert_query_product_category_define = """INSERT INTO chery_otd_product.product_category_define
        (id, code, name, ka_name, root_category_id, parent_id, status, product_category_level, `type`, create_by, owner_id, update_by, create_time, update_time, del_flag, tenant_id, row_version, original_id, source_type, brand_type)
        VALUES(72598001646806{}, 'QR{}', '瑞虎8PLUS鲲鹏e+（北方版）', '', NULL, 7910804568757978775, 1, 4, 'vehicleType', 'admin', 'admin', 'yuangong62', '2024-11-12 17:08:00.952000', '2025-01-09 15:12:23.991000', '0', 100001, 1, NULL, 2, NULL);""".format(c,c)
        cursor.execute(insert_query_product_category_define)

        print(cursor.rowcount, "记录插入成功。insert_query_product_category_define")
        conn.commit()

        #经销商经营权限,目标
        insert_query_dealer_sales_vehicle_type = """INSERT INTO chery_otd_product.dealer_sales_vehicle_type
        (id, dealer_code, vehicle_type_code, powertrain_code, powertrain_category_id, authorize_expire_time, `attribute`, create_by, update_by, create_time, update_time, del_flag, tenant_id, owner_id, row_version)
        VALUES(78976119609971{}, 'QR2026{}', 'QR{}', 'QR8X-PR016', 1874714170562424833, NULL, '2', 'yuangong62', 'yuangong62', '2025-03-06 19:35:22', '2025-03-06 19:35:22', '0', 100001, 'yuangong62', 0);""".format(c,c,c)

        cursor.execute(insert_query_dealer_sales_vehicle_type)

        print(cursor.rowcount, "记录插入成功。insert_query_dealer_sales_vehicle_type")
        conn.commit()

         #---insert_query_sys_dept---
        insert_query_sys_dept = """INSERT INTO chery_otd_admin.sys_dept
        (tenant_id, dept_id, parent_id, name, dept_type, sync_wechat_status, source_org_code, sort_order, create_by, update_by, create_time, update_time, del_flag, owner_id, dealer_code, sync_all_son_depart, sync_wechat_time, wechat_dept_id)
        VALUES(100001, 78996888440320{}, 1899688552456663042, '徐州金慧', '3', '0', 'QR2026{}', 1, 'yuangong62', ' ', '2025-03-12 13:08:10', NULL, '0', 'yuangong62', 'QR2026{}', '0', NULL, NULL);""".format(c,c,c)

        cursor.execute(insert_query_sys_dept)

        print(cursor.rowcount, "记录插入成功。insert_query_sys_dept")
        conn.commit()
        write_excel_data(NumberCount, c)

    # 关闭连接
    cursor.close()
    conn.close()

def write_excel_data(j,c):
    print("写入到excel的调用次数"+str(j))
    sheet_write.cell(row=j, column=1).value = '2025'  # 年份
    sheet_write.cell(row=j, column=2).value = '5'  # 月份
    sheet_write.cell(row=j, column=3).value = '提车目标'  # 目标分类
    sheet_write.cell(row=j, column=4).value = c  # 经销商ERP号
    sheet_write.cell(row=j, column=5).value = 'QR{}'.format(c)  # 车型编码
    sheet_write.cell(row=j, column=6).value = '车型名称'  # 车型名称
    sheet_write.cell(row=j, column=7).value = '30'  # 目标数
    write_workbook.save(result_file_path)
def write_excel_ini():
    global sheet_write
    sheet_write = write_workbook.create_sheet("Sheet1", 0)
    sheet_write.cell(row=1, column=1).value = '*年份'  # 年份
    sheet_write.cell(row=1, column=2).value = '*月份'  # 月份
    sheet_write.cell(row=1, column=3).value = '*目标分类'  # 目标分类
    sheet_write.cell(row=1, column=4).value = '*经销商ERP号'  # 经销商ERP号
    sheet_write.cell(row=1, column=5).value = '*车型编码'  # 车型编码
    sheet_write.cell(row=1, column=6).value = '车型名称'  # 车型名称
    sheet_write.cell(row=1, column=7).value = '*目标数'  #目标数
    sheet_write.cell(row=1, column=8).value = '错误信息'  # 错误信息
    write_workbook.save(result_file_path)

if __name__ == '__main__':
    #write_excel_ini()
    #read_excel()
    write_excel_ini()
    db_insert()

1万月度分大区目标

import pymysql
import openpyxl

result_file_path = "D:/odin各种导入数据的资源/导入月度分大区目标_code_sit.xlsx"
sheet_write = None
write_workbook = openpyxl.Workbook()
NumberCount = 1

def db_insert():
    # 连接数据库
    conn = pymysql.connect(host="172.25.188.206", port=3306, user="odin", password="odin@123", db="chery_otd_admin")
    cursor = conn.cursor()
    global NumberCount
    for i in range(60004, 60004 + 10000):  # 注意这里加了1
        NumberCount = NumberCount + 1
        c = i
        #新增大区
        insert_query_sys_zone = """INSERT INTO chery_otd_admin.sys_zone
        (tenant_id, id, zone_name, zone_code, product_sequence, zone_status, remark, create_by, create_time, update_by, update_time, del_flag)
        VALUES(100001, 39000061276779{}, '性能大区{}', 'CNDB00{}', '1', 1, '大批量数据-性能压测', 'admin', '2024-11-25 13:00:01', 'yuangong62', '2025-01-03 13:42:52', '0');""".format(c,c,c)

        cursor.execute(insert_query_sys_zone)
        print(cursor.rowcount, "记录插入成功。insert_query_sys_zone")
        conn.commit()

        #新增大区组织
        insert_query_sys_dept = """INSERT INTO chery_otd_admin.sys_dept
        (tenant_id, dept_id, parent_id, name, dept_type, sync_wechat_status, source_org_code, sort_order, create_by, update_by, create_time, update_time, del_flag, owner_id, dealer_code, sync_all_son_depart, sync_wechat_time, wechat_dept_id)
        VALUES(100001, 39000591359923{}, 1899688552456663042, '性能大区{}', '2', '0', 'CNDB00{}', 1, 'yuangong62', ' ', '2025-03-13 13:39:34', NULL, '0', 'yuangong62', NULL, '0', NULL, NULL);""".format(c,c,c)

        cursor.execute(insert_query_sys_dept)
        print(cursor.rowcount, "记录插入成功。insert_query_sys_dept")
        conn.commit()
        write_excel_data(NumberCount, c)

    # 关闭连接
    cursor.close()
    conn.close()

def write_excel_data(j,c):
    print("写入到excel的调用次数" + str(j))
    sheet_write.cell(row=j, column=1).value = '2025'  # 年份
    sheet_write.cell(row=j, column=2).value = '3'  # 月份
    sheet_write.cell(row=j, column=3).value = 'C网-性能大区{}'.format(c)  #大区名称
    sheet_write.cell(row=j, column=4).value = '提车目标'  # 目标分类
    sheet_write.cell(row=j, column=5).value = ''  # 车型编码
    sheet_write.cell(row=j, column=6).value = ''  # 车型名称
    sheet_write.cell(row=j, column=7).value = '30' #目标数
    write_workbook.save(result_file_path)

def write_excel_ini():
    global sheet_write
    sheet_write = write_workbook.create_sheet("Sheet1", 0)
    sheet_write.cell(row=1, column=1).value = '*年份'  # 年份
    sheet_write.cell(row=1, column=2).value = '*月份'  # 月份
    sheet_write.cell(row=1, column=3).value = '*大区名称'  # 大区名称
    sheet_write.cell(row=1, column=4).value = '*目标分类'  # 目标分类
    sheet_write.cell(row=1, column=5).value = '车型编码'  # 车型编码
    sheet_write.cell(row=1, column=6).value = '车型名称'  # 车型名称
    sheet_write.cell(row=1, column=7).value = '*目标数'  #目标数
    write_workbook.save(result_file_path)

if __name__ == '__main__':
    write_excel_ini()
    db_insert()