python批量删除文件

写程序测试时总会碰到要删除日志文件夹下的日志或者删除一些历史文件，每次都会生成，再测试的时候为了查找错误原因方便总是要在测试前删除这些文件，手动删除比较麻烦，所以写一个批量删除脚本

import os

def removeFileInFirstDir(targetDir): 
    for file in os.listdir(targetDir): 
        targetFile = os.path.join(targetDir,  file) 
        if os.path.isfile(targetFile): 
            os.remove(targetFile)

lists=['log','downloaded']

if  __name__ =="__main__": 
    for lista in lists:
        removeFileInFirstDir(lista) 

脚本与文件夹要同级