基于分类整理文件时当前工作目录重复子目录生成问题排查

问题描述 #

我正在开发一个基于文件扩展名、类别和子类别整理文件的脚本，目标是将文件移动到目标目录内指定的子目录，遵循明确的类别与子类别结构。但遇到了棘手问题：脚本意外地在当前工作目录（cwd）内生成了其自身的重复子目录，而非在指定目标目录中构建预期的子目录结构。

最小可复现示例（MRE） #

import os
import shutil
from typing import List

def get_extension(file_name: str) -> str:
    return "." + file_name.split(".")[-1].lower()

TEXT_ = (".txt", ".rtf", ".md")
TARGET_DIR_ = "TestDIR"

file_extensions = {
    TEXT_: "Text",
}

def establish_current_dir() -> List[str]:
    try:
        os.chdir(TARGET_DIR_)
        current_dir = os.getcwd()
        files = os.listdir(current_dir)
        return files
    
    except FileNotFoundError:
        print(f"{TARGET_DIR_} directory not found.")
        return []
    
def organize(files: List[str]) -> None:
  for file in files:
      extension = get_extension(file)
      for extensions, category in file_extensions.items():
          category_dir = category
          if extension in extensions:
              if extension in [".md"]:
                  subdir_name = "markdown"
          else:
            subdir_name = "foobar"
                  
      subcategory_dir = os.path.join(category_dir, subdir_name)

      if not os.path.exists(os.path.join(TARGET_DIR_, subcategory_dir)):
          os.makedirs(subcategory_dir)

      shutil.move(file, os.path.join(TARGET_DIR_, subcategory_dir, os.path.basename(file)))

files = establish_current_dir()
organize(files)

我原本以为这段代码能正确创建目录：

if not os.path.exists(os.path.join(TARGET_DIR_, subcategory_dir)):
    os.makedirs(os.path.join(TARGET_DIR_, subcategory_dir))

但实际结果是生成了TestDIR/TestDIR/Text这样的重复目录结构，文件虽然能按类别排序，但都跑到了嵌套的子目录里，而不是预期的TestDIR/Text/markdown这类结构。我试过改成os.makedirs(subcategory_dir)，但问题依旧。

补充说明 #

我在establish_current_dir函数里用os.chdir()切换到了TARGET_DIR_，再获取该目录下的文件列表。

问题根源 #

1. 工作目录切换导致路径混乱：调用os.chdir(TARGET_DIR_)后，脚本当前工作目录变为TestDIR，后续所有相对路径都基于此目录。此时再用os.path.join(TARGET_DIR_, subcategory_dir)，相当于在TestDIR下拼接TestDIR/xxx，自然生成嵌套重复目录。
2. 循环逻辑漏洞：organize函数遍历file_extensions时，每次都会覆盖category_dir和subdir_name，若文件扩展名不在定义的类别中，最终会使用最后一个类别的category_dir和foobar子目录，逻辑不严谨。
3. 目录操作路径不匹配：判断目录是否存在时用的是os.path.join(TARGET_DIR_, subcategory_dir)，但创建目录时直接用subcategory_dir，路径不一致加上工作目录已切换，导致目录创建位置错误。

修正后的代码 #

import os
import shutil
from typing import List

def get_extension(file_name: str) -> str:
    # 处理无扩展名的文件
    if "." not in file_name:
        return ""
    return "." + file_name.split(".")[-1].lower()

TEXT_ = (".txt", ".rtf", ".md")
TARGET_DIR_ = "TestDIR"

file_extensions = {
    TEXT_: "Text",
}

def get_target_files() -> List[str]:
    target_path = os.path.abspath(TARGET_DIR_)
    try:
        # 不切换工作目录，直接获取目标目录下的文件
        files = os.listdir(target_path)
        # 返回带完整路径的文件名，避免路径混淆
        return [os.path.join(target_path, f) for f in files if os.path.isfile(os.path.join(target_path, f))]
    except FileNotFoundError:
        print(f"{TARGET_DIR_} directory not found.")
        return []
    
def organize(files: List[str]) -> None:
    target_root = os.path.abspath(TARGET_DIR_)
    for file_path in files:
        file_name = os.path.basename(file_path)
        extension = get_extension(file_name)
        
        # 初始化默认类别和子目录
        category_dir = "Other"
        subdir_name = "unsorted"
        
        # 匹配扩展名对应的类别
        for extensions, category in file_extensions.items():
            if extension in extensions:
                category_dir = category
                if extension == ".md":
                    subdir_name = "markdown"
                else:
                    # 其他文本文件用默认子目录
                    subdir_name = "general"
                break  # 找到匹配后跳出循环
        
        # 构建完整的目标子目录路径
        subcategory_dir = os.path.join(target_root, category_dir, subdir_name)
        
        # 创建目录（exist_ok=True避免重复判断）
        os.makedirs(subcategory_dir, exist_ok=True)
        
        # 移动文件到目标路径
        target_file_path = os.path.join(subcategory_dir, file_name)
        shutil.move(file_path, target_file_path)

files = get_target_files()
organize(files)

关键修改点 #

• 避免切换工作目录：改用os.path.abspath获取目标目录绝对路径，直接操作绝对路径，彻底规避相对路径带来的混乱。
• 修复循环逻辑：找到匹配扩展名后立即break，防止被后续类别覆盖；同时设置默认类别和子目录，处理未匹配的文件。
• 简化目录创建：使用os.makedirs(..., exist_ok=True)替代手动判断目录是否存在，代码更简洁可靠。
• 使用完整文件路径：获取文件时直接返回绝对路径，移动时避免路径歧义。

内容的提问来源于stack exchange，提问作者i_hate_F_sharp