import pefile
import os
from tqdm import tqdm
def is_pe(file_path):
    try:
        pe = pefile.PE(file_path)
        return True
    except pefile.PEFormatError:
        return False

def delete_non_pe_files(folder_path):
    for file_name in tqdm(os.listdir(folder_path)):
        file_path = os.path.join(folder_path, file_name)
        if os.path.isfile(file_path):
            if not is_pe(file_path):
                print(f"Deleting non-PE file: {file_path}")
                os.remove(file_path)
        else:
            print(f"Skipping non-file item: {file_path}")
if __name__ == '__main__':
    delete_non_pe_files(r"D:\detect_exp_d\data\benign_pre")