删除目录下所有gif后缀的文件

[转]    如果需要删除某目录下所有相同后缀名的文件，可用以下代码

import os
import sys
import os.path
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( file, *suffixName ) :
    array = map( file.endswith, suffixName )
    if True in array :
        return True
    else :
        return False

#删除目录下扩展名为gif的文件
def deleteFile() :
    targetDir = fileDir()
    for file in os.listdir( targetDir ) :
        targetFile = os.path.join( targetDir, file )
        if suffix( file, '.gif'):
            os.remove( targetFile )

deleteFile()