1
"""
2
文件夹自动整理脚本
3
功能：自动将文件夹中的文件按后缀名分类到不同子文件夹
4
"""
5

6
import os
7
import shutil
8
from pathlib import Path
9
from typing import Dict, List
10

11
# ============================================================
12
# 第一步：配置区域（请根据你的实际情况修改下面这些值）
13
# ============================================================
14

15
# 要整理的文件夹路径（请替换成你自己的文件夹路径）
16
# 注意事项：
17
#   - Windows 用户示例：D:\\Downloads 或 D:/Downloads
18
#   - Mac/Linux 用户示例：/Users/你的用户名/Downloads
19
TARGET_FOLDER = "/tmp/test_folder"  # <-- 这里改成你要整理的文件夹路径！
20

21
# ============================================================
22
# 第二步：定义文件分类规则（后缀名 -> 文件夹名称）
23
# ============================================================
24

25
FILE_CATEGORIES: Dict[str, List[str]] = {
26
    # 图片素材：常见图片格式
27
    "【图片素材】": [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".webp", ".svg", ".ico", ".tiff"],
28

29
    # 视频素材：常见视频格式
30
    "【视频素材】": [".mp4", ".mov", ".avi", ".mkv", ".wmv", ".flv", ".webm", ".m4v"],
31

32
    # 音频素材：常见音频格式
33
    "【音频素材】": [".mp3", ".wav", ".flac", ".aac", ".ogg", ".wma", ".m4a"],
34

35
    # 文档：常见文档格式
36
    "【文档】": [".doc", ".docx", ".pdf", ".txt", ".rtf", ".odt", ".xlsx", ".xls", ".pptx", ".ppt"],
37

38
    # 压缩文件：常见压缩格式
39
    "【压缩文件】": [".zip", ".rar", ".7z", ".tar", ".gz", ".bz2"],
40

41
    # 代码文件：常见代码格式
42
    "【代码文件】": [".py", ".js", ".html", ".css", ".java", ".cpp", ".c", ".go", ".rs", ".php"],
43

44
    # 安装包：常见安装包格式
45
    "【安装包】": [".exe", ".dmg", ".pkg", ".deb", ".rpm", ".msi"],
46
}
47

48
# ============================================================
49
# 第三步：核心整理函数（一般不需要修改）
50
# ============================================================
51

52
def get_file_category(file_extension: str) -> str | None:
53
    """
54
    根据文件后缀名获取对应的分类文件夹名称
55

56
    参数:
57
        file_extension: 文件后缀名（如 ".jpg"）
58

59
    返回:
60
        分类文件夹名称，如果不在规则中则返回 None
61
    """
62
    # 统一转为小写进行匹配
63
    ext_lower = file_extension.lower()
64

65
    for category, extensions in FILE_CATEGORIES.items():
66
        if ext_lower in extensions:
67
            return category
68

69
    return None  # 未分类的文件
70

71

72
def create_category_folder(base_path: Path, category_name: str) -> Path:
73
    """
74
    创建分类文件夹（如果不存在）
75

76
    参数:
77
        base_path: 目标文件夹路径
78
        category_name: 分类文件夹名称
79

80
    返回:
81
        创建好的文件夹路径
82
    """
83
    category_path = base_path / category_name
84
    category_path.mkdir(exist_ok=True)  # exist_ok=True 表示如果已存在不报错
85
    return category_path
86

87

88
def organize_files(target_folder: str) -> Dict[str, List[str]]:
89
    """
90
    整理文件夹中的所有文件
91

92
    参数:
93
        target_folder: 要整理的文件夹路径
94

95
    返回:
96
        整理结果统计（每个分类移动了哪些文件）
97
    """
98
    target_path = Path(target_folder)
99

100
    # 检查目标文件夹是否存在
101
    if not target_path.exists():
102
        raise FileNotFoundError(f"目标文件夹不存在：{target_folder}")
103

104
    if not target_path.is_dir():
105
        raise NotADirectoryError(f"路径不是文件夹：{target_folder}")
106

107
    # 统计结果
108
    results: Dict[str, List[str]] = {
109
        "文件夹": [],      # 子文件夹
110
        "其他": [],         # 未分类的文件
111
        "已分类": [],       # 成功分类的文件（图片、视频、文档等）
112
    }
113

114
    # 首先创建所有需要的分类文件夹（确保它们存在）
115
    for category in FILE_CATEGORIES.keys():
116
        create_category_folder(target_path, category)
117
    # 创建"文件夹"和"其他"分类
118
    create_category_folder(target_path, "文件夹")
119
    create_category_folder(target_path, "其他")
120

121
    # 遍历目标文件夹中的所有文件和子文件夹
122
    for item in target_path.iterdir():
123
        # 跳过本身就是分类文件夹的项（避免把自己移动到自己的子目录）
124
        if item.name in FILE_CATEGORIES.keys() or item.name in ["文件夹", "其他"]:
125
            continue
126

127
        if item.is_dir():
128
            # -----------------------------
129
            # 处理子文件夹：移动到"文件夹"分类
130
            # -----------------------------
131
            folder_dest = target_path / "文件夹" / item.name
132

133
            # 如果目标路径已存在，添加序号区分
134
            if folder_dest.exists():
135
                counter = 1
136
                while folder_dest.exists():
137
                    new_name = f"{item.name}_{counter}"
138
                    folder_dest = target_path / "文件夹" / new_name
139
                    counter += 1
140

141
            # 移动文件夹
142
            shutil.move(str(item), str(folder_dest))
143
            results["文件夹"].append(item.name)
144
            print(f"  📁 {item.name} -> 文件夹/")
145

146
        elif item.is_file():
147
            # -----------------------------
148
            # 处理文件：根据后缀名分类
149
            # -----------------------------
150
            file_extension = item.suffix
151

152
            if file_extension:
153
                # 有后缀名的文件，尝试匹配分类
154
                category = get_file_category(file_extension)
155

156
                if category:
157
                    # 属于已知分类（如图片、视频、文档等）
158
                    category_path = target_path / category
159
                    dest_path = category_path / item.name
160

161
                    # 如果目标路径已存在，添加序号区分
162
                    if dest_path.exists():
163
                        counter = 1
164
                        while dest_path.exists():
165
                            new_name = f"{item.stem}_{counter}{item.suffix}"
166
                            dest_path = category_path / new_name
167
                            counter += 1
168

169
                    # 移动文件
170
                    shutil.move(str(item), str(dest_path))
171
                    results["已分类"].append(item.name)
172
                    print(f"  ✅ {item.name} -> {category}/")
173
                else:
174
                    # 未匹配到任何分类，移动到"其他"
175
                    other_path = target_path / "其他"
176
                    dest_path = other_path / item.name
177

178
                    if dest_path.exists():
179
                        counter = 1
180
                        while dest_path.exists():
181
                            new_name = f"{item.stem}_{counter}{item.suffix}"
182
                            dest_path = other_path / new_name
183
                            counter += 1
184

185
                    shutil.move(str(item), str(dest_path))
186
                    results["其他"].append(item.name)
187
                    print(f"  📄 {item.name} -> 其他/")
188
            else:
189
                # 没有后缀名的文件，移动到"其他"
190
                other_path = target_path / "其他"
191
                dest_path = other_path / item.name
192

193
                if dest_path.exists():
194
                    counter = 1
195
                    while dest_path.exists():
196
                        new_name = f"{item.name}_{counter}"
197
                        dest_path = other_path / new_name
198
                        counter += 1
199

200
                shutil.move(str(item), str(dest_path))
201
                results["其他"].append(item.name)
202
                print(f"  📄 {item.name} -> 其他/")
203

204
    return results
205

206

207
def print_summary(results: Dict[str, List[str]]):
208
    """
209
    打印整理结果摘要
210
    """
211
    print("\n" + "=" * 50)
212
    print("📊 整理完成！结果统计如下：")
213
    print("=" * 50)
214

215
    total = len(results["文件夹"]) + len(results["已分类"]) + len(results["其他"])
216
    print(f"\n📈 总计整理：{total} 个项目")
217

218
    # 显示各个分类的统计
219
    if results["文件夹"]:
220
        print(f"\n📁 移入【文件夹】分类：{len(results['文件夹'])} 个")
221
        for name in results["文件夹"][:5]:  # 最多显示5个
222
            print(f"   - {name}")
223
        if len(results["文件夹"]) > 5:
224
            print(f"   ... 还有 {len(results['文件夹']) - 5} 个")
225

226
    if results["已分类"]:
227
        print(f"\n✅ 移入各类素材分类：{len(results['已分类'])} 个（图片/视频/音频/文档等）")
228
        for name in results["已分类"][:5]:
229
            print(f"   - {name}")
230
        if len(results["已分类"]) > 5:
231
            print(f"   ... 还有 {len(results['已分类']) - 5} 个")
232

233
    if results["其他"]:
234
        print(f"\n📄 移入【其他】分类：{len(results['其他'])} 个")
235
        for name in results["其他"][:5]:
236
            print(f"   - {name}")
237
        if len(results["其他"]) > 5:
238
            print(f"   ... 还有 {len(results['其他']) - 5} 个")
239

240
    print("\n" + "=" * 50)
241

242

243
# ============================================================
244
# 第四步：程序入口
245
# ============================================================
246

247
if __name__ == "__main__":
248
    print("🚀 文件夹自动整理工具")
249
    print("=" * 50)
250
    print(f"📂 目标文件夹：{TARGET_FOLDER}")
251
    print("=" * 50)
252

253
    try:
254
        # 执行整理
255
        results = organize_files(TARGET_FOLDER)
256

257
        # 打印结果
258
        print_summary(results)
259

260
    except FileNotFoundError as e:
261
        print(f"\n❌ 错误：{e}")
262
        print("\n请检查以下几点：")
263
        print("1. 目标文件夹路径是否正确？")
264
        print("2. 路径中的文件夹名称是否拼写正确？")
265

266
    except PermissionError:
267
        print("\n❌ 错误：没有访问该文件夹的权限")
268

269
    except Exception as e:
270
        print(f"\n❌ 发生未知错误：{e}")