您的位置:首页 > 编程语言 > Python开发

用python导入mysql

2016-04-13 17:57 597 查看
#coding: utf-8
'''
@2016/04/13:
To load raw-data.csv into db.
Data is from Citeulike.
'''
import re
import mysql.connector

def main():
# directory on iiot's server
infile = open('./raw-data.csv','r')
outfile = open('./error.txt','a+')

#database on iiot's server
cnx = mysql.connector.connect(user=' ',password=' ',
host = ' ',port =  ,
db=' ')
cursor = cnx.cursor()
insert_query = 'insert into article values (%s,%s,%s,%s,%s)'
for read in infile:
# 1       2        3             4           5
# doc.id,"title","citeulike.id","raw.title","raw.abstract"
try:
res = re.search('(\d*),"(.*)",(\d*),"(.*)","(.*)"',read)
print res.group(1)
data = (res.group(1),res.group(2),res.group(3),res.group(4),res.group(5))
cursor.execute(insert_query,data)
cnx.commit()
except Exception,e:
outfile.write(str(e)+'\n'+read+'\n')

if __name__=='__main__':
main()


把.csv文件导入mysql,.csv文件以','分隔,在每一项字符串中含有',' 不能用load data local infile .. fields terminated by ',',否则字符串被截断。

用re
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: