在测试的时候,需要造模拟数据,或者是将生产环境的数据导出到测试环境中去,本文记录通过python读取txt文件,并将数据导入mysql数据库
1、示例代码:
import pymysql
# 打开数据库连接
conn = pymysql.connect(host="localhost", port=3306, user="root", passwd="123456", db="test1")
# 使用cursor()方法获取操作游标
cursor = conn.cursor()
#通过open()方法以只读的方式打开文件,编码格式为UTF-8
file = open("d:/test/test1.txt", 'r', encoding='UTF-8')
#通过readlines()方法读取文件的每一行赋值给lines
lines = file.readlines()
#如果lines为真,执行循环的内容
if lines:
for line in lines: #lines是一个列表,列表中的每隔元素就是txt文件中的一行数据
print(line) #这一步是为了验证是否能如期获取到列表
line = line.strip('\n').split(',') #strip()去掉字符串头尾的符号,通过","将每行数据拆分
a = line[0] #我的txt文件中每行三个元素,所以将三个元素分别赋值给变量abc
b = line[1]
c = line[2]
sql = "insert into student values(%s,%s,%s)" #数据库数据插入语句
param = (a, b, c) #param参数是要输入的数据
cursor.execute(sql, param) #cursor.execute(sql,param)方法执行插入语句
conn.commit() #提交
file.close() #关闭所有的连接
cursor.close()
conn.close()
2、需要注意的问题
1、txt文件中的“,”(逗号)间隔符一定要是英文格式的,我在这里除了问题,排查了半天
2、sql语句中的占位符,一定要是%s,不能加引号
3、文件的路径,一般可以用“/”,使用“\”的时候需要用’’\’’,因为python中“\”是转义符
4、cursor.execute需要在循环内,循环一次执行一次