morgan/python-bzz
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated to include return as an output type

Browse Source
This commit is contained in:
Morgan Kelley
2025-09-21 14:09:26 -06:00
parent 775203c6ce
commit 5bcbd881fb
1 changed files with 31 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
main.py
View File

@@ -4,10 +4,13 @@ from pathlib import Path
class BZZCompressor: class BZZCompressor:
def decompress(self, input_file_path, input_file, output_folder="out/") -> bytes: def decompress(
self, input_file_path, input_file, output_type="file", output_folder="out/"
) -> bytes:
data = bytes() data = bytes()
output_buffer = bytearray() output_buffer = bytearray()
overflow_buffer = bytearray() overflow_buffer = bytearray()
return_files = []
# read the input file # read the input file
try: try:
@@ -214,38 +217,43 @@ class BZZCompressor:
# I need to # I need to
out_data = output_buffer out_data = output_buffer
try: if output_type == "file":
with open( try:
f"{output_folder}/{input_file}_{str(file_num).zfill(3)}.file{file['type'][2:]}", with open(
"wb", f"{output_folder}/{input_file}_{str(file_num).zfill(3)}.file{file['type'][2:]}",
) as outfile: "wb",
outfile.write(out_data) ) as outfile:
# print( outfile.write(out_data)
# f"File {output_folder}/{input_file}_{str(file_num).zfill(3)}.file{file['type'][2:]} saved successfully!" # print(
# ) # f"File {output_folder}/{input_file}_{str(file_num).zfill(3)}.file{file['type'][2:]} saved successfully!"
except IOError as e: # )
raise IOError( except IOError as e:
f"Unable to write file for {input_file_path}/{input_file} on {file_num}/{len(files)}. Error: {e}" raise IOError(
) f"Unable to write file for {input_file_path}/{input_file} on {file_num}/{len(files)}. Error: {e}"
)
elif output_type == "return":
return_files.append(out_data)
index = starting_index if not skip_overflow and output_type == "file":
index = starting_index
skip_overflow = True skip_overflow = True
if len(data) > index: if len(data) > index:
for item in data[index:]: for item in data[index:]:
overflow_buffer.append(item) overflow_buffer.append(item)
for item in overflow_buffer: for item in overflow_buffer:
if item != 0x00: if item != 0x00:
skip_overflow = False skip_overflow = False
if not skip_overflow:
with open(f"{output_folder}/{input_file}.overflow.file", "wb") as outfile: with open(f"{output_folder}/{input_file}.overflow.file", "wb") as outfile:
outfile.write(overflow_buffer) outfile.write(overflow_buffer)
print( print(
f"File {output_folder}/{input_file}.overflow.file saved successfully!" f"File {output_folder}/{input_file}.overflow.file saved successfully!"
) )
elif output_type == "return":
return return_files
if __name__ == "__main__": if __name__ == "__main__":