Как симулировать ZipFile.open в Python 2.5?
Я хочу извлечь файл из почтового индекса по определенному пути, игнорируя путь к файлу в архиве. Это очень легко в 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.5ZipFile.open метод недоступен. Я не мог найти решение для stackoverflow, ноэто сообщение на форуме было хорошее решение, которое используетZipInfo.file_offset
искать нужную точку в почтовом индексе и использовать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())
Обратите внимание, как я распаковываю и читаю поле, которое дает длину дополнительного поля, потому что вызовlen
наZipInfo.extra
Атрибут дает на 4 байта меньше, что приводит к неправильному вычислению смещения. Возможно, я что-то здесь упускаю?
Кто-нибудь может улучшить это решение для Python 2.5?
Редактировать: Я должен был сказать, очевидное решение, предложенное ChrisAdams
dest_file.write(zip_file.read(name))
потерпит неудачу сMemoryError
для любого файла разумного размера, содержащегося в почтовом индексе, потому что он пытается вылить весь файл в память за один раз. У меня большие файлы, поэтому мне нужно вывести содержимое на диск.
Кроме того, обновление Python является очевидным решением, но оно совершенно не в моих руках и практически невозможно.