我用 AI 写了个脚本,下载文件夹从此自己变整齐

# 分类规则：扩展名 -> 目标子文件夹
CATEGORIES = {
  “图片”:   {“.jpg”, “.jpeg”, “.png”, “.gif”, “.webp”, “.bmp”, “.svg”},
  “文档”:   {“.pdf”, “.doc”, “.docx”, “.xls”, “.xlsx”, “.ppt”, “.pptx”, “.txt”, “.md”},
  “压缩包”: {“.zip”, “.rar”, “.7z”, “.tar”, “.gz”, “.tgz”, “.bz2”, “.xz”},
  “安装包”: {“.exe”, “.msi”, “.dmg”, “.pkg”, “.deb”},
  “影音”:   {“.mp4”, “.mkv”, “.mov”, “.mp3”, “.wav”, “.flac”},
}

DOWNLOADS = Path.home() / “Downloads”      # 下载文件夹
LOG = DOWNLOADS / “move_log.json”          # 移动日志

def category_of(ext):
  for name, exts in CATEGORIES.items():
    if ext.lower() in exts:
      return name
  return “其它”

def unique_path(target):
  # 目标已存在则加序号，绝不覆盖
  if not target.exists():
    return target
  stem, suffix, i = target.stem, target.suffix, 1
  while True:
    cand = target.with_name(f”{stem}({i}){suffix}”)
    if not cand.exists():
      return cand
    i += 1

def tidy(dry_run=False):
  moves = []
  for item in DOWNLOADS.iterdir():
    if item.is_dir() or item.name == “move_log.json”:
      continue                       # 跳过子文件夹和日志本身
    dest_dir = DOWNLOADS / category_of(item.suffix)
    dest = unique_path(dest_dir / item.name)
    moves.append((str(item), str(dest)))
    if dry_run:
      print(f”[预演] {item.name}  ->  {dest_dir.name}/”)
    else:
      dest_dir.mkdir(exist_ok=True)
      shutil.move(str(item), str(dest))
      print(f”已移动 {item.name}  ->  {dest_dir.name}/”)
  if not dry_run and moves:
    # 日志累积：保留未还原的记录，支持多次整理后一键还原
    existing = json.loads(LOG.read_text(encoding=“utf-8”)) if LOG.exists() else []
    LOG.write_text(json.dumps(existing + moves, ensure_ascii=False, indent=2), encoding=“utf-8”)
  print(f”\n完成，共处理 {len(moves)} 个文件” + (“（仅预演，未实际移动）” if dry_run else “”))

def undo():
  if not LOG.exists():
    print(“没有找到移动日志，无法回退。”); return
  moves = json.loads(LOG.read_text(encoding=“utf-8”))
  for src, dest in reversed(moves):
    if not Path(dest).exists():
      continue                       # 文件已不在原地（可能手动处理过），跳过
    target = unique_path(Path(src))  # 原位置已有同名则加序号，绝不覆盖
    shutil.move(dest, str(target))
    if str(target) == src:
      print(f”已还原 {Path(dest).name}”)
    else:
      print(f”已还原 {Path(dest).name}  ->  {target.name}（已加序号避免覆盖）”)
  LOG.unlink()
  print(“已全部还原。”)

if __name__ == “__main__”:
  if “–undo” in sys.argv:
    undo()
  else:
    tidy(dry_run=“–dry-run” in sys.argv)

# 第二步：确认没问题，真正执行
python tidy_downloads.py

# 后悔了？一键还原到整理前
python tidy_downloads.py –undo