Macintosh-Tools/python-rsrcfork
1
0
Fork 0
mirror of https://github.com/dgelessus/python-rsrcfork.git synced 2024-06-25 14:29:35 +00:00
Code Issues Projects Releases Wiki Activity
53e73be980
python-rsrcfork/rsrcfork/compress/__init__.py

52 lines
1.8 KiB
Python
Raw Normal View History

Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
from . import dcmp0
from . import dcmp1
from . import dcmp2
Move compressed header info constants/classes to .compress.common This allows the constants/classes to be accessed from the individual decompressor submodules.
2019-09-23 21:14:06 +00:00
from .common import DecompressError, CompressedApplicationHeaderInfo, CompressedHeaderInfo, CompressedSystemHeaderInfo
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
__all__ = [
"DecompressError",
"decompress",
]
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
def _decompress_application(data: bytes, header_info: CompressedApplicationHeaderInfo, *, debug: bool=False) -> bytes:
if header_info.dcmp_id == 0:
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
decompress_func = dcmp0.decompress
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
elif header_info.dcmp_id == 1:
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
decompress_func = dcmp1.decompress
else:
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
raise DecompressError(f"Unsupported 'dcmp' ID: {header_info.dcmp_id}, expected 0 or 1")
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
Pass complete header info to individual decompressors
2019-09-23 21:19:20 +00:00
return decompress_func(header_info, data, debug=debug)
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
def _decompress_system(data: bytes, header_info: CompressedSystemHeaderInfo, *, debug: bool=False) -> bytes:
if header_info.dcmp_id == 2:
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
decompress_func = dcmp2.decompress
else:
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
raise DecompressError(f"Unsupported 'dcmp' ID: {header_info.dcmp_id}, expected 2")
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
Pass complete header info to individual decompressors
2019-09-23 21:19:20 +00:00
return decompress_func(header_info, data, debug=debug)
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
def decompress(data: bytes, *, debug: bool=False) -> bytes:
"""Decompress the given compressed resource data."""
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
header_info = CompressedHeaderInfo.parse(data)
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
if debug:
print(f"Compressed resource data header: {header_info}")
if isinstance(header_info, CompressedApplicationHeaderInfo):
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
decompress_func = _decompress_application
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
elif isinstance(header_info, CompressedSystemHeaderInfo):
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
decompress_func = _decompress_system
else:
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
raise DecompressError(f"Unsupported compression type: 0x{header_info.compression_type:>04x}")
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
Refactor parsing of compressed resource headers In preparation for #3, the compressed resource data headers are parsed and stored as proper objects. For now these objects are only used internally by the decompression code, but in the future they can be exposed.
2019-09-23 21:10:55 +00:00
decompressed = decompress_func(data[header_info.header_length:], header_info, debug=debug)
if len(decompressed) != header_info.decompressed_length:
raise DecompressError(f"Actual length of decompressed data ({len(decompressed)}) does not match length stored in resource ({header_info.decompressed_length})")
Refactor the rsrcfork.compress module into separate submodules Each decompressor now has its own module, since they share almost no code between each other.
2019-08-22 19:19:10 +00:00
return decompressed
Reference in New Issue Copy Permalink