BXW

博哥的blog

博哥的技术博客

0%

python一键整理文件夹脚本

发表于 更新于 分类于

我以前经常把下载的东西放在桌面或者“下载”文件夹中，使用后再也没管它，久而久之文件夹就变得乱七八糟，再整理的时候非常痛苦，巴不得有一个自动化的工具帮我归类文档。

不知道你是不是有过跟我一样糟糕的体验，不过，前段时间我用Python做了一个根据后缀名整理文件的工具，先来看看效果：

整理前

整理后

准备和使用

安装python和pip pip使用
需要安装pathlib扩展

1
pip install pathlib

运行

1
python autofile.py '要整理的文件夹'
原理

这份代码的原理很简单，为了整理这些文件，我们要遍历当前指定的文件夹：如果是文件夹则跳过，如果是文件则检测后缀，分类到对应的文件夹中。文件夹名字及其对应的文件后缀如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
DIRECTORIES = {
    "图片": [".jpeg", ".jpg", ".tiff", ".gif", ".bmp", ".png", ".bpg", "svg",
               ".heif", ".psd"],
    "视频": [".avi", ".flv", ".wmv", ".mov", ".mp4", ".webm", ".vob", ".mng",
               ".qt", ".mpg", ".mpeg", ".3gp", ".mkv"],
    "文档": [".oxps", ".epub", ".pages", ".docx", ".doc", ".fdf", ".ods",
                  ".odt", ".pwi", ".xsn", ".xps", ".dotx", ".docm", ".dox",
                  ".rvg", ".rtf", ".rtfd", ".wpd", ".xls", ".xlsx", ".ppt",
                  "pptx",".csv",",pdf"],
    "压缩文件": [".a", ".ar", ".cpio", ".iso", ".tar", ".gz", ".rz", ".7z",
                 ".dmg", ".rar", ".xar", ".zip"],
    "影音": [".aac", ".aa", ".aac", ".dvf", ".m4a", ".m4b", ".m4p", ".mp3",
              ".msv", "ogg", "oga", ".raw", ".vox", ".wav", ".wma"],
    "文本": [".txt", ".in", ".out"],
    "编程": [".py",".html5", ".html", ".htm", ".xhtml",".c",".cpp",".java",".css"],
    "可执行程序": [".exe"],
}

接下来是自动整理的部分代码，先遍历指定的文件夹，识别后缀并分类到对应的文件夹中。

1
2
3
4
5
6
7
8
9
10
11
for entry in os.scandir(dirval):
    if entry.is_dir():
        # 如果是文件夹则跳过
        continue
    file_path = Path(dirval + "\\" + entry.name)
    file_format = file_path.suffix.lower()
    # 避免后缀大小写判断问题，全转为小写
    if file_format in FILE_FORMATS:
        directory_path = Path(dirval + "\\" + FILE_FORMATS[file_format])
        directory_path.mkdir(exist_ok=True)
        file_path.rename(directory_path.joinpath(entry.name))

还有出现不存在在我们字典里的后缀的情况，把它们分类到名为 其他文件 的文件夹中：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
try:
    os.mkdir(dirval + "\\" +"其他文件")
except:
    pass

for dir in os.scandir(dirval):
    try:
        if dir.is_dir():
            # 删除空文件夹
            os.rmdir(dir)
        else:
            temp = str(Path(dir)).split('\\')
            # 分割文件路径
            path = '\\'.join(temp[:-1])
            print(path + '\\其他文件\\' + str(temp[-1]))
            os.rename(str(Path(dir)), path + '\\其他文件\\' + str(temp[-1]))
    except:
        pass
自定义整理

如果你并不想按照设定好的文件整理，希望能自定义地将某些特定后缀的文件放到另一个文件夹，比如 .py 文件统一收纳到“Python脚本”文件夹中，你可以这么改 DIRECTORIES 变量：
添加自定义参数

1
2
3
4
5
6
DIRECTORIES = {
    # ......
    "编程": [".html5", ".html", ".htm", ".xhtml",".c",".cpp",".java",".css"],
    "Python脚本": [".py"],
    # ......
}

你可以按照这个方法，自定义收纳那些你需要整理收纳的文件格式。

如果你只想对 DIRECTORIES 变量里那些后缀的文件进行整理，不属于这些后缀的文件则不进行整理，不需要放到“其他文件”目录下，此时要怎么做？

很简单，你只需要删除以下部分代码就可以完成这个功能：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# ......
    try:
        os.mkdir(dirval + "\\" +"其他文件")
    except:
        pass

    for dir in os.scandir(dirval):
        try:
            if dir.is_dir():
                # 删除空文件夹
                os.rmdir(dir)
            else:
                temp = str(Path(dir)).split('\\')
                # 分割文件路径
                path = '\\'.join(temp[:-1])
                print(path + '\\其他文件\\' + str(temp[-1]))
                os.rename(str(Path(dir)), path + '\\其他文件\\' + str(temp[-1]))
        except:
            pass
# ......
完整代码
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import sys
import os
from pathlib import Path
 
DIRECTORIES = {
    "图片": [".jpeg", ".jpg", ".tiff", ".gif", ".bmp", ".png", ".bpg", "svg",
               ".heif", ".psd"],
    "视频": [".avi", ".flv", ".wmv", ".mov", ".mp4", ".webm", ".vob", ".mng",
               ".qt", ".mpg", ".mpeg", ".3gp", ".mkv"],
    "文档": [".oxps", ".epub", ".pages", ".docx", ".doc", ".fdf", ".ods",
                  ".odt", ".pwi", ".xsn", ".xps", ".dotx", ".docm", ".dox",
                  ".rvg", ".rtf", ".rtfd", ".wpd", ".xls", ".xlsx", ".ppt",
                  "pptx",".csv",",pdf"],
    "压缩文件": [".a", ".ar", ".cpio", ".iso", ".tar", ".gz", ".rz", ".7z",
                 ".dmg", ".rar", ".xar", ".zip"],
    "影音": [".aac", ".aa", ".aac", ".dvf", ".m4a", ".m4b", ".m4p", ".mp3",
              ".msv", "ogg", "oga", ".raw", ".vox", ".wav", ".wma"],
    "文本": [".txt", ".in", ".out"],
    "编程": [".py",".html5", ".html", ".htm", ".xhtml",".c",".cpp",".java",".css"],
    "可执行程序": [".exe"],
}
 
FILE_FORMATS = {file_format: directory
                for directory, file_formats in DIRECTORIES.items()
                for file_format in file_formats}
 
def auto_organize(dirval):
    # 整理对应的文件夹下的文件
    for entry in os.scandir(dirval):
        if entry.is_dir():
            # 如果是文件夹则跳过
            continue
        file_path = Path(dirval + "\\" + entry.name)
        file_format = file_path.suffix.lower()
        # 避免后缀大小写判断问题，全转为小写
        if file_format in FILE_FORMATS:
            directory_path = Path(dirval + "\\" + FILE_FORMATS[file_format])
            directory_path.mkdir(exist_ok=True)
            file_path.rename(directory_path.joinpath(entry.name))
 
    try:
        os.mkdir(dirval + "\\" +"其他文件")
    except:
        pass
 
    for dir in os.scandir(dirval):
        try:
            if dir.is_dir():
                # 删除空文件夹
                os.rmdir(dir)
            else:
                temp = str(Path(dir)).split('\\')
                # 分割文件路径
                path = '\\'.join(temp[:-1])
                print(path + '\\其他文件\\' + str(temp[-1]))
                os.rename(str(Path(dir)), path + '\\其他文件\\' + str(temp[-1]))
if __name__ == "__main__":
    auto_organize(sys.argv[1])