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