十行Python代码就能写出一个U盘“病毒”

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

当一个usb插入时,在后台自动把usb里的重要文件神不知鬼不觉地拷贝到本地或者上传到某个服务器。

用Python写一个程序让它在后台运行,每当有U盘插入的时候,就自动拷贝其中重要的文件!

那么如何才能判断U盘的插入呢?

Python学习交流群:548377875,群内每天分享干货,包括最新的python企业案例学习资料和零基础入门教程,欢迎各位小伙伴入群学习交流

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

如何选择性的复制文件?还是举个例子吧。

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

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

现在我们来测试一下

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

单独查看 dirs

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

单独查看 files

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

shutil模块

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

现在我们拿刚才的文件夹举例子,如果想把file1.txt拷贝到folder2:

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

os.path.getsize()判断大小

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

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

这里我们只要选择文件大小小于3M的即可,3M = 3 1024kB = 3 1024*1024byte

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

如何指定文件类型

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

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

根据修改时间筛选文件

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

U盘刚插入电脑,用Python十行代码就把里面的资料“盗取”了

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

总结

其实,标题这么起只是为了吸引大家注意,这就是一个小程序,也谈不上病毒。我更想通过这个例子,展示python对于文件处理的强大能力,引发大家的学习热情。

猜你喜欢

转载自blog.csdn.net/qq_40925239/article/details/81535564