第一个python脚本

工作中有些工具是用python写的脚本
有一个功能是需要遍历删除某个文件夹下的留存日期在三个月前的文件。具体代码如下：

import os
import time


def fileremove(filename, remove_time):
    aa = os.path.getmtime(filename)
    bb = remove_time * 86400
    cc = time.time()
    if int(aa) < int(cc)-int(bb):
        os.remove(filename)


def ListDir(filedir, deltime):
    if os.path.exists(filedir):
        if os.path.isfile(filedir):
            fileremove(filedir, deltime)
        else:
            for i in os.listdir(filedir):
                if os.path.isfile(filedir + '/' +i):
                    fileremove(filedir + '/' + i, deltime)
                else:
                    new_dir = filedir + '/' + i
                    ListDir(new_dir, deltime)

file_dir = "d:/pyTest/others"
if __name__ == "__main__":
    ListDir(file_dir, 3)

上面的代码的功能就是删除d:/pyTest/others文件夹下3天以前的文件。

如果想删除当前文件夹，那么改成

if __name__ == "__main__":
    file_dir = os.getcwd()
    ListDir(file_dir, 3)