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()