python读取数据库并把数据写入本地文件


一,介绍

上周用jmeter做性能测试时,接口B传入的参数需要依赖接口A生成的借贷申请ID,接口A运行完需要把生成的借贷申请ID导出来到一个文件,作为参数传给接口B,刚开始的时候,手动去数据库倒,

倒了几次感觉有些麻烦,就写了一段python读取数据库并将读到数据写入到本地文件

 

 

 

二,python读取数据库代码如下:

 

# -*- coding:utf-8 -*-
import pymysql


def get_loan_number(file):
connect = pymysql.Connect(
host="0.0.0.0",
port=3306,
user="test",
passwd="12345678",
db="test",
charset='utf8'
)
print("写入中,请等待……")
cursor = connect.cursor()
sql = "select id from application where status='SUBMITTING' and contract like 'Performance-%' and " \
"loan_org_party='166490194444444444'"
cursor.execute(sql)
number = cursor.fetchall()
fp = open(file, "w")
loan_count = 0
for loanNumber in number:
loan_count += 1
fp.write(loanNumber[0] + "\n")
fp.close()
cursor.close()
connect.close()
print("写入完成,共写入%d条数据……" % loan_count)


if __name__ == "__main__":
file = r"C:\Users\test\Desktop\loanNUmber.txt"
get_loan_number(file)

运行结果如下:

 

 

三、写到本地数据如下:





四、jmeter 用CSV Data Set Config读取本地参数文件

 

 

 

 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM