Fixed file name references

This commit is contained in:
2025-09-16 20:49:57 -06:00
parent f1c346a634
commit eb0327f753

21
main.py
View File

@@ -11,8 +11,8 @@ class BZZCompressor:
# read the input file # read the input file
try: try:
with open(f"{input_file_path}/{input_file}", "rb") as input_file: with open(f"{input_file_path}/{input_file}", "rb") as infile:
data.fromfile(input_file) data.fromfile(infile)
data = data.tobytes() data = data.tobytes()
except IOError: except IOError:
print("Could not open input file...") print("Could not open input file...")
@@ -163,20 +163,9 @@ class BZZCompressor:
out_data = output_buffer out_data = output_buffer
try: try:
if "bin_extract" in input_file_path[0:11]: with open(f"{output_folder}/{input_file}.file", "wb") as outfile:
output_file_path = input_file_path[12:]
else:
output_file_path = input_file_path
if "/" in output_file_path:
output_file_name = output_file_path.split("/")[-1] + ".file"
else:
output_file_name = output_file_path + ".file"
# TODO: Create file path, if it doesn't exist
with open(f"{output_folder}/{output_file_name}", "wb") as outfile:
outfile.write(out_data) outfile.write(out_data)
print(f"File {output_file_name} saved successfully!") print(f"File {output_folder}/{input_file}.file saved successfully!")
except IOError as e: except IOError as e:
print( print(
f"Unable to write file for {input_file_path}/{input_file}. Error: {e}" f"Unable to write file for {input_file_path}/{input_file}. Error: {e}"
@@ -195,7 +184,7 @@ if __name__ == "__main__":
output_folder_path.mkdir(parents=True, exist_ok=True) output_folder_path.mkdir(parents=True, exist_ok=True)
try: try:
compressor.decompress(dirpath, file, output_folder_path) compressor.decompress(dirpath, file, str(output_folder_path))
except Exception as e: except Exception as e:
print( print(
f"Error while decompressing {output_folder_path}/{file}. Error: {e}" f"Error while decompressing {output_folder_path}/{file}. Error: {e}"