python读取txt文件将数据导入mysql数据库

在测试的时候,需要造模拟数据,或者是将生产环境的数据导出到测试环境中去,本文记录通过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需要在循环内,循环一次执行一次

猜你喜欢

转载自blog.csdn.net/HBT036017/article/details/104969165
今日推荐