用 WorkBuddy 管项目：任务拆分 + 代码生成实战

我要做一个 Python 小工具，把"下载文件夹"按文件类型自动分类整理。
请把它拆成最小任务清单，每个任务标清：输入是什么、输出是什么。

flowchart TD
    G["项目目标: 整理乱糟糟的下载文件夹"] --> T1["任务1 读取目录文件清单<br />输入: 目录路径 / 输出: 文件清单"]
    G --> T2["任务2 按扩展名分类<br />输入: 文件+扩展名 / 输出: 文件→分类"]
    G --> T3["任务3 创建分类子目录<br />输入: 分类名 / 输出: 图片/文档/视频...目录"]
    G --> T4["任务4 安全移动文件<br />输入: 文件+目标目录 / 输出: 移动且不覆盖同名"]
    G --> T5["任务5 生成整理报告<br />输入: 移动结果 / 输出: 每类几个+文件名清单"]
    T1 -.先行.-> T2
    T2 -.驱动.-> T3
    T3 -.接收.-> T4
    T4 -.汇总.-> T5

用 Python 标准库写一个脚本 organizer.py：
1. 扫描目标目录下的文件（不处理子目录）；
2. 按扩展名分到 图片/文档/视频/压缩/其他 子目录；
3. 分类目录不存在就创建；
4. 同名文件不覆盖，自动加序号；
5. 最后打印每类整理了几个、分别是哪些文件。
只使用标准库（pathlib/shutil/os），不要第三方依赖，加必要注释。

"""
下载文件夹整理工具（WorkBuddy 代码生成实战示例）
功能：按文件类型把目标目录里的文件分到 图片/文档/视频/压缩/其他 子目录，
     同名冲突自动改名，最后打印整理报告。
用法：python organizer.py <要整理的目录>
依赖：仅 Python 标准库（pathlib / shutil / os），无需安装任何第三方包
"""
import sys
import shutil
from pathlib import Path
from collections import defaultdict

# 扩展名 -> 分类 的映射表（新增类型只需在这里加一行）
CATEGORY_MAP = {
    "图片": {".jpg", ".jpeg", ".png", ".gif", ".webp", ".bmp", ".svg"},
    "文档": {".pdf", ".doc", ".docx", ".txt", ".md", ".xlsx", ".pptx", ".csv"},
    "视频": {".mp4", ".mov", ".avi", ".mkv", ".webm"},
    "压缩": {".zip", ".rar", ".7z", ".tar", ".gz"},
}
DEFAULT_CATEGORY = "其他"


def classify(ext: str) -> str:
    """根据扩展名返回分类名；不在映射表里的归到「其他」。"""
    ext = ext.lower()
    for cat, exts in CATEGORY_MAP.items():
        if ext in exts:
            return cat
    return DEFAULT_CATEGORY


def organize(target: Path) -> dict:
    """扫描 target 下的文件，按类型移动到对应子目录，返回 {分类: [文件名,...]}。"""
    moved = defaultdict(list)
    for f in target.iterdir():
        if not f.is_file():
            continue  # 跳过子目录，只整理「文件」
        cat = classify(f.suffix)
        dest_dir = target / cat
        dest_dir.mkdir(exist_ok=True)  # 分类目录不存在就建
        dest = dest_dir / f.name
        # 同名冲突：追加序号，绝不覆盖原文件
        if dest.exists():
            stem, suffix = f.stem, f.suffix
            i = 1
            while dest.exists():
                dest = dest_dir / f"{stem}_{i}{suffix}"
                i += 1
        shutil.move(str(f), str(dest))
        moved[cat].append(f.name)
    return moved


def main():
    if len(sys.argv) < 2:
        print("用法: python organizer.py <要整理的目录>")
        sys.exit(1)
    target = Path(sys.argv[1])
    if not target.is_dir():
        print(f"目录不存在: {target}")
        sys.exit(1)
    result = organize(target)
    print(f"\n整理完成：{target}")
    if not result:
        print("（目录里没有可移动的文件）")
        return
    for cat, files in result.items():
        print(f"\n【{cat}】{len(files)} 个")
        for name in files:
            print(f"  - {name}")


if __name__ == "__main__":
    main()

python organizer.py "C:/Users/你的名/Downloads"

整理完成：C:\...\mess

【压缩】1 个
  - archive.zip
【视频】1 个
  - clip.mp4
【文档】3 个
  - data.xlsx
  - note.pdf
  - readme.txt
【图片】2 个
  - photo1.png
  - scan.jpg

flowchart LR
    R["模糊需求<br />如:整理下载文件夹"] --> S["让 WorkBuddy 拆任务树"]
    S --> P["逐项派活·拿代码<br />带约束:语言/库/异常处理"]
    P --> V["每段先单独跑通"]
    V -->|全过| M["拼装成完整项目"]
    M --> D["交付 + 复盘归档"]
    V -.|报错| B["回灌错误信息给 WorkBuddy"]
    B --> P

flowchart TD
    Q["拿到一个小任务"] --> W["把任务+约束写给 WorkBuddy"]
    W --> G["它生成代码"]
    G --> Y{"你本地运行?"}
    Y -->|通过| OK["标记完成·并入项目"]
    Y -->|报错| F["把报错信息原样回灌"]
    F --> W
    OK --> Q2["下一个任务"]
    Q2 --> W