File size: 2,433 Bytes
d94b56e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
# -*- coding: utf-8 -*-
#
# pyhwp : hwp file format parser in python
# Copyright (C) 2010-2023 mete0r <https://github.com/mete0r>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
from io import BytesIO
import codecs
import zlib
from .utils import GeneratorReader
class ZLibIncrementalDecoder(codecs.IncrementalDecoder):
def __init__(self, errors='strict', wbits=15):
assert errors == 'strict'
self.errors = errors
self.wbits = wbits
self.reset()
def decode(self, input, final=False):
c = self.decompressobj.decompress(input)
if final:
c += self.decompressobj.flush()
return c
def reset(self):
self.decompressobj = zlib.decompressobj(self.wbits)
def decompress_gen(source, bufsize=4096):
dec = ZLibIncrementalDecoder(wbits=-15)
exausted = False
while not exausted:
input = source.read(bufsize)
if len(input) < bufsize:
exausted = True
yield dec.decode(input, exausted)
def decompress_experimental(source, bufsize=4096):
''' decompress inputstream
stream: a file-like readable
returns a file-like readable
'''
return GeneratorReader(decompress_gen(source, bufsize))
def decompress(stream):
''' decompress inputstream
stream: a file-like readable
returns a file-like readable
'''
# #176 참고. #175의 임시방편을 사용한다.
compressed_maybe = stream.read()
try:
decompressed = zlib.decompress(compressed_maybe, -15) # without gzip header
except zlib.error:
return BytesIO(compressed_maybe)
else:
return BytesIO(decompressed)
|