Python批量删除特定后缀名的文件和目录

利用 Python 在日常工作中，删除指定目录已经子目录下的特定后缀的文件名。

Python Version: 2.7

# coding:utf-8
import os
import sys
import os
import shutil

#获取当前路径
def fileDir() :
    path = sys.path[ 0 ]
    print(path)
    #判断为脚本文件还是编译后文件，如果是脚本文件则返回脚本目录，否则返回编件译后的文件路径
    if os.path.isdir( path ) :
        return path
    elif os.path.isfile( path ) :
        return os.path.dirname( path )
    
#获取文件后缀名
def suffix(fileName, *suffixName) :
    array = map(fileName.endswith, suffixName)
    if True in array :
        return True
    else :
        return False
    
#删除目录下扩展名为.exe,.bak的文件
def deleteFile() :
    target_dir = fileDir()
    for root, dir_names, file_names in os.walk(target_dir):
        for file in file_names:
            target_file = os.path.join(root, file)
            if suffix(file, '.doc', '.xls'):
                os.remove(target_file)
            # 文件夹名字
            if file == 'a':
                shutil.rmtree(os.path.join(root, dir_names))

    
if __name__ == '__main__' :
    deleteFile()