Allow passing bytestrings as replacements to the localunzip module as well

This commit is contained in:
Kovid Goyal 2016-06-16 09:39:31 +05:30
parent bab29014b1
commit e214e3c86c

View File

@ -283,7 +283,10 @@ class LocalZipFile(object):
if header.filename in names: if header.filename in names:
zi = ZipInfo(header.filename) zi = ZipInfo(header.filename)
zi.compress_type = header.compression_method zi.compress_type = header.compression_method
ztemp.writestr(zi, replacements[header.filename].read()) r = replacements[header.filename]
if not isinstance(r, bytes):
r = r.read()
ztemp.writestr(zi, r)
found.add(header.filename) found.add(header.filename)
else: else:
ztemp.writestr(header.filename, self.read(header.filename, ztemp.writestr(header.filename, self.read(header.filename,