python是门优雅的语言

file_list=os.listdir(path)

#断言函数，满足条件继续执行，即如果文件数量不是1直接报错
assert len(file_list)==1
file=file_list[0]
data_path=path+file

#定义要读取的列
usecols=['商家编码','CC','宝贝ID','库存']

#获取当前年份
_year=datetime.datetime.now().year

#使用正则 从文件名获取月、日
_month,_day=re.findall('(.*)TM',file)[0].split('.')

#拼接日期
date='-'.join((str(_year),_month.zfill(2),_day.zfill(2)))

#读取数据
data=pd.read_excel(data_path,usecols=usecols).reset_index().rename(columns={'index':'idx'})
data['日期']=date
w_sql(sql_name,data,db_name)
print("-"*10+date+'库存数据入库完成'+"-"*10)

#删除文件
os.remove(data_path)

date=input()
file_list=os.listdir(path)

if len(file_list)==1：
	print("文件有误，请检查")
else:
	
	#构造日期
	date='2020.'+file[0][:-10]
	date_p=time.strptime(date,"%Y.%m.%d")
	date_f=time.strftime("%Y-%m-%d",date_p)
	
	#读取数据并处理
	data_path=path+file_list[0]
	data=pd.read_excel(data_path)
	data=data[['商家编码','CC','宝贝ID','库存']]
	data.reset_index(inplace=True)
	data.rename(columns={'index':'idx'},inplace=True)
	data['日期']=date
	
	#写入数据库
	w_sql(sql_name,data,db_name)
	print("-"*10+date+'库存数据入库完成'+"-"*10)
	#删除文件
	os.remove(data_path)