如何将多路径同临时目录提取逻辑封装为通用Python函数？

我代码里多次重复这样的逻辑：用tempfile.TemporaryDirectory创建临时目录，把多个路径（比如path1、path2）依次提取到这个目录，每次提取后检查返回码，失败就抛出RuntimeError，还要执行目录有效性检查和验证。但现有的extract_tmpdir函数每次调用都会生成新的临时目录，满足不了多路径共用同一个临时目录的需求，该怎么把这个逻辑封装成通用函数？

原重复代码示例 #

with tempfile.TemporaryDirectory() as tmpdir:
    retm = extract_subpath(path1, tmpdir)
    if retm["returncode"] != 0:
        raise RuntimeError(f"Failed to extract file: {retm['stderr'].strip()}")

    file = check_valid(tmpdir)

    ret = extract_subpath(path2, tmpdir)
    if ret["returncode"] != 0:
        raise RuntimeError(f"Failed to extract file: {ret['stderr'].strip()}")

    validate(tmpdir)

现有存在问题的函数 #

def extract_tmpdir(path: str) -> str:
    with tempfile.TemporaryDirectory() as tmpdir:
        ret = extract_subpath(path, tmpdir)

        if ret["returncode"] != 0:
            raise RuntimeError(f"Failed to extract file: {ret['stderr'].strip()}")

        return tmpdir

下面提供几种实用的封装方式，都能实现多路径共用同一个临时目录的需求：

方式1：上下文管理器类（推荐，结构清晰） #

把临时目录管理和提取逻辑封装成类，用with语句自动管理临时目录的创建和销毁，同时支持多次提取操作：

import tempfile
from typing import Optional

class TempDirExtractor:
    def __init__(self):
        self.tmpdir: Optional[str] = None
        self._temp_dir_instance = None

    def __enter__(self):
        # 创建临时目录并保存实例
        self._temp_dir_instance = tempfile.TemporaryDirectory()
        self.tmpdir = self._temp_dir_instance.__enter__()
        return self

    def extract(self, path: str) -> None:
        """提取指定路径到当前临时目录，失败抛出异常"""
        if not self.tmpdir:
            raise RuntimeError("请在'with'代码块内使用TempDirExtractor")
        
        ret = extract_subpath(path, self.tmpdir)
        if ret["returncode"] != 0:
            raise RuntimeError(f"提取文件失败: {ret['stderr'].strip()}")

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 自动清理临时目录
        if self._temp_dir_instance:
            self._temp_dir_instance.__exit__(exc_type, exc_val, exc_tb)

使用示例：

with TempDirExtractor() as extractor:
    # 提取第一个路径
    extractor.extract(path1)
    # 检查目录有效性
    target_file = check_valid(extractor.tmpdir)
    # 提取第二个路径
    extractor.extract(path2)
    # 执行最终验证
    validate(extractor.tmpdir)
    # 可以在这里处理提取后的文件，比如返回结果
    # return target_file

方式2：带回调的通用函数（轻量灵活） #

如果不想用类，可以写一个创建临时目录并接受回调的函数，回调里可以完成所有需要在同一个临时目录下执行的操作：

def with_shared_tempdir(operation):
    """创建临时目录，传入回调执行操作，自动清理目录"""
    with tempfile.TemporaryDirectory() as tmpdir:
        return operation(tmpdir)

# 单独封装通用的提取检查逻辑
def extract_to_dir(path: str, target_dir: str) -> None:
    ret = extract_subpath(path, target_dir)
    if ret["returncode"] != 0:
        raise RuntimeError(f"提取文件失败: {ret['stderr'].strip()}")

使用示例：

def my_task(tmpdir):
    # 执行所有需要在同一个临时目录下的操作
    extract_to_dir(path1, tmpdir)
    target_file = check_valid(tmpdir)
    extract_to_dir(path2, tmpdir)
    validate(tmpdir)
    return target_file

# 执行任务
result = with_shared_tempdir(my_task)

方式3：生成器式上下文管理器（简洁紧凑） #

用contextlib.contextmanager装饰器实现生成器式的上下文管理器，直接返回临时目录和绑定好的提取函数：

from contextlib import contextmanager

@contextmanager
def shared_tempdir_extractor():
    """生成可复用的临时目录，附带提取方法"""
    with tempfile.TemporaryDirectory() as tmpdir:
        # 定义绑定当前tmpdir的提取函数
        def extract(path):
            ret = extract_subpath(path, tmpdir)
            if ret["returncode"] != 0:
                raise RuntimeError(f"提取文件失败: {ret['stderr'].strip()}")
        
        # 返回临时目录和提取函数
        yield tmpdir, extract

使用示例：

with shared_tempdir_extractor() as (tmpdir, extract):
    extract(path1)
    target_file = check_valid(tmpdir)
    extract(path2)
    validate(tmpdir)

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