mirror of
https://github.com/lcdr/utils.git
synced 2024-08-30 17:32:16 +00:00
32 lines
859 B
Python
32 lines
859 B
Python
import argparse
|
|
import os.path
|
|
import zlib
|
|
|
|
def decompress(data):
|
|
assert data[:5] == b"sd0\x01\xff"
|
|
pos = 5
|
|
out = b""
|
|
while pos < len(data):
|
|
length = int.from_bytes(data[pos:pos+4], "little")
|
|
pos += 4
|
|
out += zlib.decompress(data[pos:pos+length])
|
|
pos += length
|
|
return out
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("in_path")
|
|
parser.add_argument("--out_path", help="If not provided, output file is in the script directory")
|
|
args = parser.parse_args()
|
|
if args.out_path is None:
|
|
filename, ext = os.path.splitext(os.path.basename(args.in_path))
|
|
args.out_path = filename+"_decompressed"+ext
|
|
|
|
with open(args.in_path, "rb") as file:
|
|
data = file.read()
|
|
|
|
with open(args.out_path, "wb") as file:
|
|
file.write(decompress(data))
|
|
|
|
print("Decompressed file:", args.out_path)
|