如何在Python 2.5中模拟ZipFile.open？

Question

我想将文件从 zip 中提取到特定路径，忽略存档中的文件路径。这在 Python 2.6 中非常容易（我的文档字符串比代码长）

import shutil
import zipfile

def extract_from_zip(name, dest_path, zip_file):
    """Similar to zipfile.ZipFile.extract but extracts the file given by name
    from the zip_file (instance of zipfile.ZipFile) to the given dest_path
    *ignoring* the filename path given in the archive completely
    instead of preserving it as extract does.
    """
    dest_file = open(dest_path, 'wb')
    archived_file = zip_file.open(name)
    shutil.copyfileobj(archived_file, dest_file)


 extract_from_zip('path/to/file.dat', 'output.txt', zipfile.ZipFile('test.zip', 'r'))

但是在 Python 2.5 中， ZipFile.open 方法不可用。我在 stackoverflow 上找不到解决方案，但是这个论坛帖子有一个很好的解决方案，利用< code>ZipInfo.file_offset 寻找 zip 中的正确位置并使用 zlib.decompressobj 从那里解压字节。不幸的是 ZipInfo.file_offset 在 Python 2.5 中被删除了！

因此，考虑到 Python 2.5 中我们拥有的只是 ZipInfo。 header_offset，我想我只需要解析并跳过标头结构即可自己获取文件偏移量。使用维基百科作为参考（我知道）我想出了这么多更长而且不是很优雅的解决方案。

import zipfile
import zlib

def extract_from_zip(name, dest_path, zip_file):
    """Python 2.5 version :("""
    dest_file = open(dest_path, 'wb')
    info = zip_file.getinfo(name)
    if info.compress_type == zipfile.ZIP_STORED:
        decoder = None
    elif info.compress_type == zipfile.ZIP_DEFLATED:
        decoder = zlib.decompressobj(-zlib.MAX_WBITS)
    else:
        raise zipfile.BadZipFile("Unrecognized compression method")

    # Seek over the fixed size fields to the "file name length" field in
    # the file header (26 bytes). Unpack this and the "extra field length"
    # field ourselves as info.extra doesn't seem to be the correct length.
    zip_file.fp.seek(info.header_offset + 26)
    file_name_len, extra_len = struct.unpack("<HH", zip_file.fp.read(4))
    zip_file.fp.seek(info.header_offset + 30 + file_name_len + extra_len)

    bytes_to_read = info.compress_size

    while True:
        buff = zip_file.fp.read(min(bytes_to_read, 102400))
        if not buff:
            break
        bytes_to_read -= len(buff)
        if decoder:
            buff = decoder.decompress(buff)
        dest_file.write(buff)

    if decoder:
        dest_file.write(decoder.decompress('Z'))
        dest_file.write(decoder.flush())

请注意我如何解压并读取给出额外字段长度的字段，因为在 ZipInfo.extra 属性上调用 len 会减少 4 个字节，从而导致偏移量为会被错误地计算。也许我在这里遗漏了一些东西？

有人可以针对 Python 2.5 改进这个解决方案吗？

编辑：我应该说，

dest_file.write(zip_file.read(name))

对于 zip 中包含的任何大小合理的文件，ChrisAdams 建议的明显解决方案将会失败，并出现 MemoryError ，因为它试图将整个文件放入一口气记忆。我有大文件，所以我需要将内容流式传输到磁盘。

另外，升级 Python 是一种显而易见的解决方案，但它完全超出了我的控制范围，而且基本上是不可能的。

Answer 1

还没有测试过这一点，但我在 Python 2.4 中使用了非常相似的东西

import zipfile

def extract_from_zip(name, dest_path, zip_file):
    dest_file = open(dest_path, 'wb')
    dest_file.write(zip_file.read(name))
    dest_file.close()

extract_from_zip('path/to/file/in/archive.dat', 
        'output.txt', 
        zipfile.ZipFile('test.zip', 'r'))

Answer 2

我知道我回答这个问题有点晚了，但我也遇到了完全相同的问题。

我使用的解决方案是复制 python 2.6.6 版本的 zipfile 并放入一个文件夹（我称之为 python_fix）并导入它：

python_fix/zipfile.py

然后在代码中：

import python_fix.zipfile as zipfile

从那里我可以使用 2.6.6 版本的 zipfile python 2.5.1解释器（2.7.X版本在这个版本的“with”上失败”）

希望这可以帮助其他使用古老技术的人。

Answer 3

考虑到我的限制，我的问题似乎给出了答案：自己解析 ZipFile 结构，并在找到字节后使用 zlib.decompressobj 来解压缩字节。

如果您没有（/遭受）我的限制，您可以在这里找到更好的答案：

1. 如果可以，只需将 Python 2.5 升级到 2.6（或更高版本！），正如 Daenyth 的评论中所建议的那样。
2. 如果 zip 中只有小文件，可以 100% 加载到内存中，请使用 ChrisAdams 的回答
3. 如果您可以引入对外部实用程序的依赖，请对 /usr/bin/unzip 或类似的系统调用进行适当的系统调用，如下所示Vlad 的回答中建议