拳打电脑管家，脚踢金山毒霸，3分钟用Python撸一个杀毒软件

此文章已录入专栏 《Python入门到精通》

2021最新版Python小白教程，针对0基础小白和基础薄弱的伙伴学习

事情要从一次重保行动说起，由于活动还没结束，这里就先不透漏了。我抱着亲手做的康师傅红烧牛肉泡面，还没开始吃，就听见后面有人喊我名字，说是有个用户的电脑卡死了，要我过去看看。

我问她能不能等我吃完这桶泡面再去，我已经3个小时没有吃过饭了，现在饿得没有力气走到用户那里。

她说那是主任的电脑，你自己看着办吧。

我向来不畏惧权贵，但九年义务教育告诉我，别人有困难的时候一定要去帮一把……

我到现场一看，电脑上就只有一个空空桌面，连个鼠标都找不到，看来是电脑太卡了，软件图标都卡没了。

我问用户之前有没有做过什么奇怪的事情

他说没有，电脑一开机就是这个样子

我好像明白了什么，拔掉电源重启，电脑开机以后，果然跟我想的一样，360、电脑管家、金山毒霸这三兄弟整齐的站在右下角，桌面上满满当当的全家桶，这不卡才怪呢。刚好电脑上没有什么重要文件，懒得一个一个删了，招呼来联想的同志给他重新做个系统。

回去的路上我跟联想的同志吐槽：“这群流氓软件真是不讲武德，就知道欺负这种电脑小白。”

联想的同志白了我一眼：“看把你能的，有本事自己写一个”

我向来不喜欢斤斤计较，写杀毒软件只是出于爱好。

首先，杀毒这个动作，本质上就是删除病毒文件，这还不简单，os模块就能干

import os
os.remove(r'E://python/病毒.txt')

病毒软件肯定不是只有一种，因此我们需要准备一个病毒库，把文件扔到病毒库里匹配，匹配上了就把病毒文件删除

import os

病毒库 = ['病毒.txt']

path = r'E://python/病毒.txt'

if path in 病毒库:
    os.remove(path)
    print('已经删除病毒文件', file)

当然，病毒查杀肯定不能只查一个文件，需要查询所有的文件，而目录下所有的文件，可以写个函数，递归查询

def getAllFile(path):
    # 获取目录中的所有文件并遍历
    files = os.listdir(path)
    for file in files:
        # 拼接路径，如果是目录则递归获取所有文件
        new_path = path + r'/' + file
        if os.path.isdir(new_path):
            getAllFile(new_path)

既然是自己写的杀毒软件，那查杀什么地方当然要自己说了算，毕竟做网安这一行，很多工具本身就是病毒。写个input，自定义查杀目录

path = input('请输入需要查杀的目录：')
getAllFile(path)

完整代码如下：

import os

病毒库 = ['病毒.txt']

def getAllFile(path):
    # 获取目录中的所有文件并遍历
    files = os.listdir(path)
    for file in files:
        # 拼接路径，如果是目录则递归获取所有文件
        new_path = path + r'/' + file
        if os.path.isdir(new_path):
            getAllFile(new_path)
        else:
            # 如果是文件则查询病毒库，匹配成功则删除
            if file in 病毒库:
                os.remove(new_path)
                print('已经删除病毒文件', file)

path = input('请输入需要查杀的目录：')
getAllFile(path)

先自己写一个病毒

运行程序，输入需要查杀的目录

好了，病毒已经被沙雕了

小小病毒，可笑可笑。

正当我伸着懒腰想着怎么加大查杀力度时，感觉手好像碰到了一个硬邦邦的东西，我突然意识到不对劲……

原来是我两个小时前泡的那碗泡面，坨了！