finally:
self.conn.commit()
if __name__ == '__main__':
pr = Profiler()
# unit test case 1
# mysql=DB_mysql()
# sql_str='select * from city'
# df = mysql.query(sql_str)
# mysql = DB_mysql()
# tbl_nme = 'world.city'
# data ={
# 'ID':5001,
# 'NAME':'TonyCity',
# 'COUNTRYCODE':'AFG',
# 'DISTRICT':'Herat',
# 'POPULATION':190000
# }
# mysql.insert(tbl_nme,data)
# unit test case 2
M = DB_mysql()
# csv文件目录
dir = 'D:/temp\MSRA/results_fullset_20191122'
file_list = os.listdir(dir)
for i in range(len(file_list)):
file_path = os.path.join(dir, file_list[i])
if os.path.isfile(file_path):
M.read_fm_csv(file_path)
pr.stop()
