python是门优雅的语言

让我水一篇吧!没啥技能点,单纯的觉得今天写的代码很优雅(哈哈哈~容许我不要脸一波,不要脸是真的,优雅 也是真的,越看越喜欢这 段代码)!学python以来第一次体验到大佬们说的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)==1print("文件有误,请检查")
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)

猜你喜欢

转载自blog.csdn.net/qq_35866846/article/details/106693774