Update ineptepub.py

Handle uncompressed elements (if any) in the zip file.
pull/1458/head
Apprentice Harper 3 years ago
parent bf6170e613
commit f6a568bcc1

@ -353,11 +353,15 @@ class Decryptor(object):
def decompress(self, bytes):
dc = zlib.decompressobj(-15)
bytes = dc.decompress(bytes)
try:
decompressed_bytes = dc.decompress(bytes)
ex = dc.decompress(b'Z') + dc.flush()
if ex:
bytes = bytes + ex
decompressed_bytes = decompressed_bytes + ex
except:
# possibly not compressed by zip - just return bytes
return bytes
return decompressed_bytes
def decrypt(self, path, data):
if path.encode('utf-8') in self._encrypted:

Loading…
Cancel
Save