data_deal模块主要用于保存数据,并在生成word报告后清楚缓存的数据,相对比较简单。
1. store_path
1.1 模块解释
获取文件所在的目录,便于其他模块直接应用此路径。
1.2 模块源码
'''
获取保存所需路劲
'''
import os
def get_path():
file_path = os.getcwd().split('Automatic_report')[0]
out_path = file_path + 'Automatic_report\\data_deal\\'
return out_path
if __name__ == '__main__':
get_path()
2. clear_cache
2.1 模块解释
清理缓存在此处的一些文件数据。
2.2 模块源码
'''
清除缓存
'''
from Automatic_report.data_deal.store_path import *
def clear(file_path):
dir_type = ['jpg', 'xlsx', 'docx']
# 读取文件夹的内的所有文件
dir_list = os.listdir(file_path)
for name in dir_list:
try:
_type = name.split('.')[1]
except:
_type = 'NO'
if _type in dir_type:
# 删除文件
os.unlink(file_path+name)
else:
pass
print('缓存清理完毕!!')
if __name__ == '__main__':
file_path = get_path()
clear(file_path)