【python】清理脚本运行中产生的中间文件函数装饰器

最新推荐文章于 2021-03-06 09:00:00 发布

尼德兰的喵

最新推荐文章于 2021-03-06 09:00:00 发布

阅读量306

点赞数

分类专栏：芯片前端脚本文章标签： python

本文链接：https://blog.csdn.net/moon9999/article/details/113827730

版权

装饰器文件管理函数执行中间文件整洁目录

关键词由CSDN通过智能技术生成

芯片前端脚本专栏收录该内容

63 篇文章 81 订阅

订阅专栏

在脚本中调用其他工具的过程中可能会产生很多的中间文件，这个时候作为一个假装有洁癖的人是必须要把他们清理干净的，因此之前做了这个函数装饰器：

def wrap_rm_add_file(f):
    def rm_add_file(*args, **kwargs):
        before_file = os.listdir(os.getcwd())
        result = f(*args, **kwargs)
        for file in os.listdir(os.getcwd()):
            if file not in before_file:
                os.system("rm -rf %s" % file)
        return result
    return rm_add_file

这样在函数执行后，装饰器会把多出来的文件全部删除，保持了目录的干净整洁。