mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-22 04:45:05 +00:00
34 lines
946 B
Python
34 lines
946 B
Python
from hachoir.field import CompressedField
|
|
|
|
try:
|
|
from zlib import decompressobj, MAX_WBITS
|
|
|
|
class DeflateStream:
|
|
|
|
def __init__(self, stream, wbits=None):
|
|
if wbits:
|
|
self.gzip = decompressobj(-MAX_WBITS)
|
|
else:
|
|
self.gzip = decompressobj()
|
|
|
|
def __call__(self, size, data=None):
|
|
if data is None:
|
|
data = b''
|
|
return self.gzip.decompress(self.gzip.unconsumed_tail + data, size)
|
|
|
|
class DeflateStreamWbits(DeflateStream):
|
|
|
|
def __init__(self, stream):
|
|
DeflateStream.__init__(self, stream, True)
|
|
|
|
def Deflate(field, wbits=True):
|
|
if wbits:
|
|
CompressedField(field, DeflateStreamWbits)
|
|
else:
|
|
CompressedField(field, DeflateStream)
|
|
return field
|
|
has_deflate = True
|
|
except ImportError:
|
|
def Deflate(field, wbits=True):
|
|
return field
|
|
has_deflate = False
|