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) + '
')
        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文件，里面就是你想要的文件了