clean_ids.py

import os

def extract_vulnerable_ids_from_filenames(folder_path):
    vulnerable_ids = []
    for root, _, files in os.walk(folder_path):
        for filename in files:
            file_path = os.path.join(root, filename)
            if os.path.isfile(file_path):
                try:
                    vulnerable_id = int(filename.split(" - ")[0])
                    vulnerable_ids.append(vulnerable_id)
                except (ValueError, IndexError):
                    pass
    return vulnerable_ids

def save_extracted_vulnerable_ids(vulnerable_ids, output_file_path):
    with open(output_file_path, "w", encoding="utf-8") as file:
        for id in vulnerable_ids:
            file.write(f"Vulnerable ID: {id}\n")

def main():
    # Specify the folder path where the remaining files are located
    remaining_files_folder_path = "C:/Users/aymen/Desktop/foundings"

    # Specify the output file path
    output_file_path = "C:/Users/aymen/Desktop/foundings/extracted_vulnerable_ids.txt"

    # Extract vulnerable IDs from filenames
    extracted_vulnerable_ids = extract_vulnerable_ids_from_filenames(remaining_files_folder_path)

    # Save extracted vulnerable IDs to the specified file
    save_extracted_vulnerable_ids(extracted_vulnerable_ids, output_file_path)

if __name__ == "__main__":
    main()