updated extract_folder

This commit is contained in:
Alexander Karpov 2023-12-04 10:30:25 +03:00
parent c294261a04
commit d7a5e8787d

View File

@ -31,10 +31,16 @@ def structure_directory_content(input_dir, output_file=None, extensions=None):
file.endswith(f".{ext}") for ext in extensions file.endswith(f".{ext}") for ext in extensions
): ):
file_path = os.path.join(root, file) file_path = os.path.join(root, file)
outfile.write(f"# {os.path.relpath(file_path, input_dir)}\n") try:
with open(file_path, "r") as infile: with open(file_path, "r") as infile:
outfile.write(infile.read()) data = infile.read()
outfile.write("\n\n") outfile.write(
f"# {os.path.relpath(file_path, input_dir)}\n"
)
outfile.write(data)
outfile.write("\n\n")
except UnicodeDecodeError:
continue
if __name__ == "__main__": if __name__ == "__main__":