python 读取txt文件中满足要求的行写到新文件中_python 查询满足数据条件的数据并写入到文本文件中（用于压测参数化使用）...

记录一下做压力测试过程中，需要用到参数化（获取数据库满足条件的记录并将结果写入到text中）

我这次用到二个接口参数化则用了unittest：

# -*- coding:utf-8 -*-

import unittest

import pymysql

class Content(unittest.TestCase):

def setUp(self):

self.conn = pymysql.connect(host='数据库主机', user='用户名', password='密码', db='数据库', port=端口, charset='utf8')

self.pt_cur = self.conn.cursor()

def test_a001_serach(self):

"""签到页面150"""

sql = "select x, y from 数据表 where `条件a`=0 and '条件b'=1;"

self.pt_cur.execute(sql)

result = self.pt_cur.fetchall()

# 本地要写入的文件夹比，如我本地的为（D:/activity_mobile.text）

with open("D:/activity_mobile", "w") as f:

for i, j in list(result):

a = "%s," % i

b = '"%s"' % j

f.write(a + b + "\n")

def test_a002_banner_new(self):

"""轮询500"""

sql = "select x from 数据表 where `条件A`=0;"

self.pt_cur.execute(sql)

result = self.pt_cur.fetchall()

with open("D:/signinCode", "w") as f:

for i in result:

a = "%s" % i

# print(a)

f.write(a + "\n")

# 数据库连接关闭

def tearDown(self):

self.conn.close()

if __name__ == '__main__':

unittest.main()