Script for sleaning spdx from flat .sol files, except the first one

clean_flat_files.py

import os

flattened_dir = "./flattened"

for filename in os.listdir(flattened_dir):
    file_path = os.path.join(flattened_dir, filename)

    if os.path.isfile(file_path) and filename.endswith("_flattened.sol"):
        with open(file_path, "r") as f:
            lines = f.readlines()

        first_spdx_line = next((line for line in lines if "SPDX-License-Identifier" in line), None)

        updated_lines = [first_spdx_line] + [line for line in lines if "SPDX-License-Identifier" not in line]

        with open(file_path, "w") as f:
            f.writelines(updated_lines)