10行Python代码写1个USB病毒！把室友的小电影都黑掉

昨天在上厕所的时候突发奇想&＃xff0c;当你把usb插进去的时候&＃xff0c;能不能自动执行usb上的程序。查了一下&＃xff0c;发现只有windows上可以&＃xff0c;具体的大家也可以搜索(搜索关键词usb autorun)到。但是&＃xff0c;如果我想&＃xff0c;比如&＃xff0c;当一个usb插入时&＃xff0c;在后台自动把usb里的重要文件神不知鬼不觉地拷贝到本地或者上传到某个服务器&＃xff0c;就需要特殊的软件辅助。

于是我心想&＃xff0c;能不能用python写一个程序&＃xff0c;让它在后台运行。每当有u盘插入的时候&＃xff0c;就自动拷贝其中重要文件。

如何判断U盘的插入与否?

Python学习交流群 631441315 每天分享Python知识、资料&＃xff01;

首先我们打开电脑终端&＃xff0c;进入/Volumes目录&＃xff0c;这时候插入U盘&＃xff0c;可以发现它被挂载在了这个目录之下&＃xff0c;也就是说&＃xff0c;我们只要在固定时间扫描这个目录&＃xff0c;当这个目录有新文件夹出现的时候&＃xff0c;很可能有U盘被插入了。

我的设计是这样的&＃xff0c;用time.sleep(3)函数&＃xff0c;让程序保持运行状态&＃xff0c;并且每隔三秒查看一下/Volumes/目录&＃xff0c;如果多出来文件夹&＃xff0c;就将其拷贝到另外的文件夹。

就像标题所示&＃xff0c;我们真的只用了10行(其实是11行&＃xff0c;凑个整&＃xff1a;)完成了这个“病毒”。我们可以发现usb中的目录&＃xff0c;在插入半分钟后全部躺在了home目录下了。

如何选择性的复制文件?

刚刚我们写了一个很简易的脚本测试了一下这个想法的可行性&＃xff0c;但是还是有问题。刚才之所以能把U盘中所有文件很快复制进去&＃xff0c;是因为U盘中只有两三个文件&＃xff0c;大小不超过15M。如果目标U盘中有很多电影&＃xff0c;音乐&＃xff0c;这些我们并不需要的文件&＃xff0c;我们的程序就应该能跳过它们&＃xff0c;仅仅选择一些重要的比如.docx比如.ppt文件&＃xff0c;或者仅仅复制最近修改过的那些文件&＃xff0c;或者排除所有大小大于5M的文件。我们可以用python做到吗?当然!

os.walk 递归文件夹中所有文件

http://www.runoob.com/python/os-walk.html

这里我放了一个别人的教程。大家可以大概了解一下&＃xff0c;总之我大概理解是这么个东西。

还是举个例子吧。

我在某目录下创建了testwalk文件夹&＃xff0c;里面有file123.txt三个文件&＃xff0c;folder123三个文件夹&＃xff0c;其中folder1中有文件file4.txt以及folder4

现在我们来测试一下

root存放的是当前位置&＃xff0c;它会把./testwalk/下所有的文件夹作为根目录&＃xff0c;往下搜索

单独查看 dirs

单独查看 files

好了&＃xff0c;我们现在需要递归usb文件夹&＃xff0c;找到所有的file&＃xff0c;查看大小&＃xff0c;如果小于&＃xff0c;比如3M&＃xff0c;就拷贝进home&＃xff0c;大于就舍去。

shutil模块

现在我们拿刚才的文件夹举例子&＃xff0c;如果想把file1.txt拷贝到folder2&＃xff1a;

还有许多使用工具在shutil里面这里就不详述了。

os.path.getsize()判断大小

os.path.getsize(文件名)返回的是一个单位为byte的数值&＃xff0c;如果用来查看文件大小&＃xff0c;我们则需要手动写一个函数&＃xff0c;将其换算成容易阅读的形式。

这里我们只要选择文件大小小于3M的即可&＃xff0c;3M &＃61; 3 * 1024kB &＃61; 3 * 1024*1024byte

结合shutil.copy2就可以把选定大小的文件复制进我们的目标文件夹了

如何指定文件类型

这里就需要正则表达式来帮助我们了。

正则表达式内容很多&＃xff0c;《python核心编程》中用了整整一章来讲&＃xff0c;所以我们也不深入了。下面是官方文档&＃xff0c;感兴趣的可以看一下。

https://docs.python.org/2/library/re.html

如下&＃xff0c;我们让指定文件后缀以及指定文件大小可以复制进我们的目标文件&＃xff1a;

别忘了导入 re

用更加复杂的正则表达式可以更好地指定文件类型

根据修改时间筛选文件

这时候我在目录下创建了一个文件叫做newfile

总之&＃xff0c;对每一个文件进行修改时间的筛选可以只复制那些近期&＃xff0c;或者特定时期修改或者添加过的文件&＃xff0c;这个功能在特定情况下很有用。

总结

其实&＃xff0c;标题这么起只是为了吸引大家注意&＃xff0c;这就是一个小程序&＃xff0c;也谈不上病毒。我更想通过这个例子&＃xff0c;展示python对于文件处理的强大能力&＃xff0c;引发大家的学习热情。以上实现都是基于macos&＃xff0c;linux应该一样&＃xff0c;windows稍加修改也可以成功。