编程语言
首页 > 编程语言> > 拳打电脑管家,脚踢金山毒霸,3分钟用Python撸一个杀毒软件

拳打电脑管家,脚踢金山毒霸,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)

先自己写一个病毒

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

好了,病毒已经被沙雕了

小小病毒,可笑可笑。

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

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

标签:金山毒霸,Python,查杀,new,file,杀毒软件,path,os,病毒
来源: https://blog.csdn.net/slw213106/article/details/122322699