如何用Python的Docker SDK将Django上传的Zip文件直接传入容器且不落地？

你的核心需求是不落地、不加载到内存，将Django上传的zip文件直接传入Docker容器处理，原代码存在几个关键问题导致无法正常工作：

原代码的核心问题 #

• 容器启动后会立即退出：command='mkdir /tmp/upload'执行完就结束，没有前台进程维持容器运行，后续exec_run会因容器已停止而失败。
• 变量引用错误：处理输出时用了未定义的data变量，实际应该是output.output。
• 文件写入不完整：通过stream.write写入后未正确关闭流，可能导致容器内文件损坏。
• 命令逻辑冗余：echo y | unzip可以用unzip -o参数直接覆盖，无需交互。
• 未处理压缩后的文件输出：cat $filename.zip的内容没有被正确捕获。

修复后的完整代码 #

import docker

client = docker.from_env()

# 启动容器时用前台进程维持运行，确保容器不会退出
container = client.containers.run(
    image='myimage:latest',
    command='tail -f /dev/null',  # 保持容器运行的前台命令
    remove=True,
    detach=True,
    network_mode='none',
)

try:
    # 1. 创建上传目录（直接在exec里执行，避免启动时命令导致容器退出）
    container.exec_run(cmd='mkdir -p /tmp/upload')

    # 2. 直接通过stdin将文件流传入容器，无需分块写入更高效
    # 注意：zipfile是Django上传的InMemoryUploadedFile对象
    exec_result = container.exec_run(
        cmd=f'cat > /tmp/upload/{zipfile.name}',
        stdin=zipfile,  # 直接传入文件对象，Docker SDK会处理流传输
        stream=True
    )
    # 等待命令执行完成，确保文件写入完毕
    for _ in exec_result[1]:
        pass

    # 3. 容器内处理逻辑优化
    container_exec_command = f"""
        cd /tmp/upload
        # 用-o参数自动覆盖，无需echo y
        unzip -o -P {filePassword} {zipfile.name}
        # 确保只获取一个文件，若多个则取第一个
        filename=$(find . -maxdepth 1 -type f ! -name '*.zip' | head -n1 | xargs basename)
        # 执行sha256和file命令，将结果输出到stdout
        sha256sum "$filename"
        file "$filename"
        # 压缩文件，同时将压缩后的内容输出到stdout
        zip -P pass1234 - "$filename"
    """
    # 执行命令并捕获输出，区分stdout和stderr
    exit_code, output = container.exec_run(
        cmd=['sh', '-c', container_exec_command],
        detach=False,
        stdout=True,
        stderr=True
    )

    if exit_code != 0:
        # 处理错误，打印stderr内容
        print(f"容器内命令执行失败: {output.decode('utf-8')}")
    else:
        # 拆分输出：前两行是sha256和file结果，后面是zip文件的二进制内容
        output_str = output.decode('utf-8', errors='ignore')
        # 找到zip内容的起始位置（zip文件的魔数是PK\x03\x04）
        zip_start = output_str.find('PK\x03\x04')
        if zip_start != -1:
            # 提取sha256和file结果
            result_part = output_str[:zip_start].strip().split('\n')
            sha256_hash = result_part[0].split()[0]
            file_info = result_part[1]
            # 提取zip二进制内容
            zip_content = output[zip_start:]

            print(f"sha256: {sha256_hash}")
            print(f"文件信息: {file_info}")
            # 这里可以将zip_content返回给Django响应，或者处理
        else:
            print("未找到压缩后的文件内容")

finally:
    # 停止容器，因为启动时设置了remove=True，停止后会自动删除
    container.stop()

关键优化点 #

• 容器存活保障：用tail -f /dev/null作为启动命令，确保容器持续运行，支持后续exec操作。
• 流式传输文件：直接将Django的InMemoryUploadedFile对象传给exec_run的stdin参数，Docker SDK会自动处理流式传输，完全避免加载到内存（只要Django配置了非内存上传，比如临时文件，但即使是InMemoryUploadedFile，也是分块处理的）。
• 命令逻辑优化：用unzip -o替代交互输入，用zip -将压缩内容直接输出到stdout，避免在容器内生成额外文件。
• 输出分离处理：区分命令执行的结果文本和压缩文件的二进制内容，正确提取所需信息。

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