import hashlib import tkinter as tk from tkinter import filedialog, messagebox

def scan_file(file_path): # 定义已知恶意文件的哈希值 known_malware_hashes = ['hash1', 'hash2', 'hash3']

# 计算文件的MD5哈希值
md5_hash = hashlib.md5()
with open(file_path, 'rb') as file:
    for chunk in iter(lambda: file.read(4096), b''):
        md5_hash.update(chunk)
file_hash = md5_hash.hexdigest()

# 比较文件的哈希值与已知恶意文件的哈希值
if file_hash in known_malware_hashes:
    print(f'{file_path} 是一个恶意文件')
else:
    print(f'{file_path} 是一个安全文件')

def scan_directory(directory_path): import os

for root, dirs, files in os.walk(directory_path):
    for file in files:
        file_path = os.path.join(root, file)
        scan_file(file_path)

def scan(): # 创建新窗口 scan_window = tk.Toplevel(window) scan_window.title("扫描病毒") scan_window.geometry("400x300")

# 添加返回按钮
return_icon = tk.PhotoImage(file="return_icon.png")
return_button = tk.Button(scan_window, image=return_icon, command=scan_window.destroy, bg=scan_window.cget('bg'), highlightthickness=0, bd=0)
return_button.place(relx=1, y=20, anchor="ne")

# 添加快速扫描按钮
quick_scan_button = tk.Button(scan_window, text="快速扫描", command=quick_scan)
quick_scan_button.place(relx=0.5, rely=0.2, anchor="center")

# 添加自定义扫描按钮
custom_scan_button = tk.Button(scan_window, text="自定义扫描", command=custom_scan)
custom_scan_button.place(relx=0.5, rely=0.4, anchor="center")

def quick_scan():
    # 创建假进度条
    progress_bar = tk.ttk.Progressbar(scan_window, mode='indeterminate')
    progress_bar.place(relx=0.5, rely=0.6, anchor="center")
    progress_bar.start()

    # 快速扫描操作的代码
    # ...

    # 扫描完成后停止进度条
    progress_bar.stop()
    progress_bar.destroy()

def custom_scan():
    # 创建假进度条
    progress_bar = tk.ttk.Progressbar(scan_window, mode='indeterminate')
    progress_bar.place(relx=0.5, rely=0.6, anchor="center")
    progress_bar.start()

    # 弹出文件选择对话框
    file_path = filedialog.askopenfilename()

    # 自定义扫描操作的代码
    # ...

    # 扫描完成后停止进度条
    progress_bar.stop()
    progress_bar.destroy()

创建快速扫描按钮

quick_scan_icon = tk.PhotoImage(file="quick_scan_icon.png") quick_scan_button = tk.Button(window, image=quick_scan_icon, command=scan, bg=window.cget('bg'), highlightthickness=0, bd=0) quick_scan_button.place(x=50, y=230)

创建自定义扫描按钮

custom_scan_icon = tk.PhotoImage(file="custom_scan_icon.png") custom_scan_button = tk.Button(window, image=custom_scan_icon, command=scan, bg=window.cget('bg'), highlightthickness=0, bd=0) custom_scan_button.place(x=150, y=230)

扫描指定目录下的所有文件

scan_directory('/path/to/directory'