将前文生成的txt文件内容读取后存入mysql数据库
python3操作数据库用到pymysql库，因为文件内是有换行操作的，这里只需要逐行读取并存入即可，主要就是与数据库的连接以及数据插入操作，记得最后要关闭文件和数据库连接
import pymysql
def store_mysql():
conn = pymysql.connect(db='test', user='root', passwd='299521', host='localhost')
cursor = conn.cursor()
# create a table
cursor.execute("drop table if exists verify_info")
sql = """create table verify_info (
id int not null auto_increment primary key,
verify_code varchar(20))"""
cursor.execute(sql)
f = open('str.txt', 'r')
for line in f.readlines():
sql = "INSERT INTO verify_info(verify_code) VALUES ('%s')" % line
cursor.execute(sql)
try:
conn.commit()
except:
conn.rollback()
f.close()
print("error happend when inserting data")
f.close()
conn.close()
if __name__ == '__main__':
store_mysql()
- 更多代码详情参考我的Github
