Shell/Python实现Mysql读txt文本

五郎.

#!/usr/bin/python　　
# mysql_import.py
　　

　　
#使用MySQLdb去实现
　　
import MySQLdb,os,time
　　

　　
os.system("sed -e 's/\"//g' -e 's#/#-#g' test.txt >t.txt");
　　

　　
try:
　　
    print time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(time.time()))  #记录开始时间
　　
    conn=MySQLdb.connect(host='127.0.0.1',user='root',passwd='zhang1992')
　　
    print "conn success"
　　
except:
　　
    print "conn error!"
　　
    exit()
　　
cur=conn.cursor()
　　
cur.execute('drop database if exists myimport')
　　
cur.execute('create database myimport')
　　
i=0
　　
count=0
　　
with open('t.txt','r') as ft:
　　
    for row in ft.readlines():
　　
        ID,NAME,CREATED=list(row.strip("\n").split(","))
　　
        if i==0:
　　
            cid=ID
　　
            cname=NAME
　　
            created=CREATED
　　
            cur.execute("create table if not exists myimport.import_obj(id int unsigned auto_increment primary key,%s int unsigned unique not null,%s varchar(60) not null,%s datetime not null)"%(ID,NAME,CREATED))
　　
            i=1
　　
        else:
　　
            sql="insert into myimport.import_obj(%s,%s,%s) value(%d,'%s','%s')"%(cid,cname,created,int(ID),NAME,CREATED)
　　
            cur.execute(sql)
　　

　　
os.system("rm -f t.txt")
　　
cur.close()
　　
conn.commit()
　　
conn.close()
　　

　　
print time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(time.time()))  #记录结束始时间