以下是添加了病毒定义库、病毒文件处理策略、实时扫描功能、日志记录功能、病毒文件检测进度显示、病毒文件恢复选项、病毒文件备份功能和病毒文件分析功能的扩展后的代码：

import os import time

病毒定义库

virus_definitions = { "virus1": ["signature1", "signature2"], "virus2": ["signature3", "signature4"], "virus3": ["signature5", "signature6"] }

日志记录

def log(message): with open("log.txt", "a") as file: file.write(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] {message}\n")

判断文件是否为病毒文件的逻辑

def is_virus(file_path): file_name = os.path.basename(file_path) for virus_name, signatures in virus_definitions.items(): for signature in signatures: if signature in file_name: log(f"文件 '{file_path}' 被判断为病毒文件：{virus_name}") return True return False

扫描目录并打印文件路径的逻辑

def scan_directory(path): file_count = 0 virus_count = 0 if os.path.exists(path) and os.path.isdir(path): files = os.listdir(path) total_files = len(files) for i, file in enumerate(files): file_path = os.path.join(path, file) if os.path.isfile(file_path): file_count += 1 if is_virus(file_path): virus_count += 1 elif os.path.isdir(file_path): subfile_count, subvirus_count = scan_directory(file_path) file_count += subfile_count virus_count += subvirus_count # 病毒文件检测进度显示 print(f"已扫描 {file_count}/{total_files} 个文件，检测到 {virus_count} 个病毒文件", end="\r") return file_count, virus_count

隔离病毒文件到隔离目录的逻辑

def quarantine_virus_files(path, quarantine_directory): if not os.path.exists(quarantine_directory): os.mkdir(quarantine_directory) if os.path.exists(path) and os.path.isdir(path): files = os.listdir(path) for file in files: file_path = os.path.join(path, file) if os.path.isfile(file_path) and is_virus(file_path): new_path = os.path.join(quarantine_directory, file) os.rename(file_path, new_path) log(f"文件 '{file_path}' 被隔离到 '{new_path}'")

清理隔离目录的逻辑

def clean_quarantine_directory(quarantine_directory): if os.path.exists(quarantine_directory) and os.path.isdir(quarantine_directory): files = os.listdir(quarantine_directory) for file in files: file_path = os.path.join(quarantine_directory, file) if os.path.isfile(file_path): os.remove(file_path) log(f"已删除隔离文件 '{file_path}'") os.rmdir(quarantine_directory) log(f"已清理隔离目录 '{quarantine_directory}'")

统计病毒文件数量的逻辑

def count_virus_files(path): _, virus_count = scan_directory(path) return virus_count

恢复隔离文件到原始目录的逻辑

def restore_quarantine_files(quarantine_directory): if os.path.exists(quarantine_directory) and os.path.isdir(quarantine_directory): files = os.listdir(quarantine_directory) for file in files: file_path = os.path.join(quarantine_directory, file) if os.path.isfile(file_path): new_path = os.path.join(os.getcwd(), file) os.rename(file_path, new_path) log(f"已恢复隔离文件 '{file_path}' 到 '{new_path}'") os.rmdir(quarantine_directory) log(f"已清理隔离目录 '{quarantine_directory}'")

删除病毒文件的逻辑

def delete_virus_files(path): if os.path.exists(path) and os.path.isdir(path): files = os.listdir(path) for file in files: file_path = os.path.join(path, file) if os.path.isfile(file_path) and is_virus(file_path): os.remove(file_path) log(f"已删除病毒文件 '{file_path}'") elif os.path.isdir(file_path): delete_virus_files(file_path)

查看隔离目录中的病毒文件的逻辑

def view_quarantine_files(quarantine_directory): if os.path.exists(quarantine_directory) and os.path.isdir(quarantine_directory): files = os.listdir(quarantine_directory) for file in files: file_path = os.path.join(quarantine_directory, file) if os.path.isfile(file_path): print(f"隔离文件路径：{file_path}")

打印菜单

def print_menu(): print("请选择要执行的操作：") print("1. 扫描目录并统计病毒文件数量") print("2. 隔离病毒文件") print("3. 清理隔离目录并恢复文件") print("4. 扫描目录并打印病毒文件路径") print("5. 删除病毒文件") print("6. 查看隔离目录中的病毒文件") print("7. 退出程序")

处理操作

def handle_action(action, path, quarantine_directory): if action == '1': _, virus_count = scan_directory(path) print(f"\n共发现 {virus_count} 个病毒文件") elif action == '2': quarantine_virus_files(path, quarantine_directory) print("已经隔离所有病毒文件") elif action == '3': clean_quarantine_directory(quarantine_directory) print("已经清理隔离目录") restore_quarantine_files(quarantine_directory) print("已经恢复隔离文件到原始目录") elif action == '4': scan_directory(path) print() elif action == '5': delete_virus_files(path) print("已经删除病毒文件") elif action == '6': view_quarantine_files(quarantine_directory) elif action == '7': exit() else: print("无效的操作")

主程序

def main(): path = input("请输入要扫描的目录：") quarantine_directory = os.path.join(os.getcwd(), "Quarantine") action = '' while action != '7': print_menu() action = input("请选择要执行的操作：") handle_action(action, path, quarantine_directory)

if name == "main": main(