对python插入数据库和生成插入sql的示例讲解
如下所示:
#-*-encoding:utf-8-*-
importcsv
importsys,os
importpymysql
defread_csv(filename):
'''
读取csv文件
'''
data=[]
withopen(filename)asf:
f_csv=csv.reader(f)
headers=next(f_csv)
#数据格式[1111,22222,1111,1111,.....]
forrowinf_csv:
#Processrow
field1=row[0]
data.append(row)
printheaders
returndata
defload_data():
'''
插入数据库
'''
filename=sys.argv[1]
try:
#获取一个数据库连接,注意如果是UTF-8类型的,需要制定数据库
conn=pymysql.connect(host='192.168.1.161',user='naliworld',passwd='password!',db='search',port=3306,charset='utf8')
cur=conn.cursor()#获取一个游标
data=read_csv(filename)
forrowindata:
#Processrow
field1=row[0]
sql='''insertintosearch.tb_text_uid_list(appId,type,uid,creator,createTime)values({},{},{},{},{})'''.format(3,1,field1,'\'admin\'','\'2018-08-1413:44:09\'')
printsql
cur.execute(sql)
cur.close()#关闭游标
conn.commit()
conn.close()#释放数据库资源
exceptExceptionase:print(e)
defget_sql():
'''
插入数据库生成插入sql
'''
sql_list=[]
filename=sys.argv[1]
data=read_csv(filename)
forrowindata:
#Processrow
field1=row[0]
sql='''replaceintosearch.tb_text_uid_list(appId,type,uid,creator,createTime)values({},{},{},{},{})'''.format(3,1,field1,'\'admin\'','\'2018-08-1413:44:09\'')
sql_list.append(sql)
file_object=open('sql.txt','w')
file_object.writelines([line+';\n'forlineinsql_list])
file_object.close()
if__name__=="__main__":
get_sql()
以上这篇对python插入数据库和生成插入sql的示例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。