#-*- encoding:utf-8 -*-import csvimport sys,osimport pymysql  def read_csv(filename):  '''  讀取csv文件  '''  data = []  with open(filename) as f:    f_csv = csv.reader(f)    headers = next(f_csv)    #數(shù)據(jù)格式[1111,22222,1111,1111,.....]    for row in f_csv:      # Process row      field1=row[0]      data.append(row)    print headers  return data def load_data():  '''  插入數(shù)據(jù)庫(kù)  '''  filename = sys.argv[1]  try:  #獲取一個(gè)數(shù)據(jù)庫(kù)連接，注意如果是UTF-8類型的，需要制定數(shù)據(jù)庫(kù)    conn=pymysql.connect(host='192.168.1.161',user='naliworld',passwd='password!',db='search',port=3306,charset='utf8')    cur=conn.cursor()#獲取一個(gè)游標(biāo)    data=read_csv(filename)    for row in data:      # Process row      field1=row[0]      sql='''insert into search.tb_text_uid_list(appId,type,uid,creator,createTime) values({},{},{},{},{}) '''.format(3,1,field1,'/'admin/'','/'2018-08-14 13:44:09/'')      print sql      cur.execute(sql)    cur.close()#關(guān)閉游標(biāo)    conn.commit()    conn.close()#釋放數(shù)據(jù)庫(kù)資源  except Exception as e :print(e) def get_sql():  '''  插入數(shù)據(jù)庫(kù)生成插入sql  '''  sql_list = []  filename = sys.argv[1]  data=read_csv(filename)  for row in data:    # Process row    field1=row[0]    sql='''replace into search.tb_text_uid_list(appId,type,uid,creator,createTime) values({},{},{},{},{}) '''.format(3,1,field1,'/'admin/'','/'2018-08-14 13:44:09/'')    sql_list.append(sql)  file_object = open('sql.txt', 'w')  file_object.writelines([line+';/n' for line in sql_list])  file_object.close( ) if __name__ == "__main__":  get_sql()