让我水一篇吧!没啥技能点,单纯的觉得今天写的代码很优雅(哈哈哈~容许我不要脸一波,不要脸是真的,优雅 也是真的,越看越喜欢这 段代码)!学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)==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)