From 30419f979c5e3958c29c00259430254ec9d7bfa2 Mon Sep 17 00:00:00 2001 From: Yavin7 Date: Sun, 21 Sep 2025 14:13:20 -0600 Subject: [PATCH] Fixed overflow stuff with new file output --- main.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index e816527..5677957 100644 --- a/main.py +++ b/main.py @@ -232,9 +232,14 @@ class BZZCompressor: 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) + return_files.append( + { + type: file["type"], + data: out_data, + } + ) - if not skip_overflow and output_type == "file": + if output_type == "file": index = starting_index skip_overflow = True @@ -247,11 +252,14 @@ class BZZCompressor: if item != 0x00: skip_overflow = False - with open(f"{output_folder}/{input_file}.overflow.file", "wb") as outfile: - outfile.write(overflow_buffer) - print( - f"File {output_folder}/{input_file}.overflow.file saved successfully!" - ) + if not skip_overflow: + with open( + f"{output_folder}/{input_file}.overflow.file", "wb" + ) as outfile: + outfile.write(overflow_buffer) + print( + f"File {output_folder}/{input_file}.overflow.file saved successfully!" + ) elif output_type == "return": return return_files