Python3操作MySQL，查询数据并保存到⽂件中
我们在测试过程中，可能需要到数据库中拉去⼀些数据，为从测试准备。⽐如最近在做接⼝性能测试的时候，就需要很多数据来⽀撑，所以就需要的数据库去查询数据，下⾯就是python3 查询 mysql 并且保存到⽂件中，然后为接⼝做测试做准备。
1#coding=utf-8
mysql存储文档
2
3import pymysql as MySQLdb  #这⾥是python3  如果你是python2.x的话，import MySQLdb
4
5 host = xxxx
6 user = xxxx
7 passwd = xxxx
8 port = xxxx
9 db = xxxx
10class SelectMySQL(object):
11def select_data(self,sql):
12        result = []
13try:
14            conn = t(host=host,
15                                port=port,
16                                user=user,
17                                passwd=passwd,
18                                db=db,
19                                charset='utf8', )
20            cur = conn.cursor()
21            ute(sql)
22            alldata = cur.fetchall()
23# print(alldata)
24for rec in alldata:
25                result.append(rec[0]) #注意，我这⾥只是把查询出来的第⼀列数据保存到结果中了，如果是多列的话，稍微修改下就ok了
26except Exception as e:
27print('Error msg: ' + e)
28finally:
29            cur.close()
30            conn.close()
31
32return result
33
34def get_result(self, sql, filename):
35print(sql)
36        results = self.select_data(sql)
37print('The amount of datas: %d' % (len(results)))
38        with open(filename, 'w') as f:
39for result in results:
40                f.write(str(result) + '\n')
41print('Data write is over!')
42return results
43
44if__name__ == '__main__':
45    sql = "select name from user"
46    select = SelectMySQL()
47    result1 = _result(sql,'')
48print(result1)
在你脚本的⽬录下就会有⽂件，⾥⾯就是你想要的⽂件了

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。