Python3操作MySQL，查詢數據並保存到文件中

    我們在測試過程中，可能需要到數據庫中拉去一些數據，為從測試准備。比如最近在做接口性能測試的時候，就需要很多數據來支撐，所以就需要的數據庫去查詢數據，下面就是python3 查詢 mysql 並且保存到文件中，然后為接口做測試做准備。

#coding=utf-8

import pymysql as MySQLdb  #這里是python3  如果你是python2.x的話，import MySQLdb

host = xxxx
user = xxxx
passwd = xxxx
port = xxxx
db = xxxx
class SelectMySQL(object):
    def select_data(self,sql):
        result = []
        try:
            conn = MySQLdb.connect(host=host,
                               port=port,
                               user=user,
                               passwd=passwd,
                               db=db,
                               charset='utf8', )
            cur = conn.cursor()
            cur.execute(sql)
            alldata = cur.fetchall()
            # print(alldata)
            for rec in alldata:
                result.append(rec[0]) #注意，我這里只是把查詢出來的第一列數據保存到結果中了,如果是多列的話，稍微修改下就ok了
        except Exception as e:
            print('Error msg: ' + e)
        finally:
            cur.close()
            conn.close()
  
        return result

    def get_result(self, sql, filename):
        print(sql)
        results = self.select_data(sql)
        print('The amount of datas: %d' % (len(results)))
        with open(filename, 'w') as f:
            for result in results:
                f.write(str(result) + '\n')
        print('Data write is over!')
        return results

if __name__ == '__main__':
    sql = "select name from user"
    select = SelectMySQL()
    result1 = select.get_result(sql,'namemsg.txt')
    print(result1)

在你腳本的目錄下就會有namemsg.txt文件，里面就是你想要的文件了