logo头像
Snippet 博客主题

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

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

Python Version: 2.7

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# 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()

评论系统未开启,无法评论!