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