python制作u盘病毒_十行代码 — 用 Python 写一个 USB “病毒”

"/Users/home/usb_copy")

# shutil.copytree 把目录下所有东西一股脑复制进/Users/home/usb_copy,

# 放进了自己的home目录下

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

如何选择性的复制文件？

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

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

http://www.runoob.com/python/os-walk.html这里我放了一个别人的教程。大家可以大概了解一下，总之我大概理解是这么个东西。

还是举个例子吧。

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

➜ testwalk touch file1.txt file2.txt file3.txt

➜ testwalk mkdir folder1 folder2 folder3

➜ testwalk cd folder1

➜ folder1 touch file4.txt && mkdir folder4

➜ folder1 cd ..

# 上面创建了这些文件以及文件夹，你也可以在图形界面上创建

# tree 是个很好玩的命令，可以直观地显示文件路径

➜ testwalk tree ./

testwalk/

├── file1.txt

├── file2.txt

├── file3.txt

├── folder1

│ ├── file4.txt

│ └── folder4

├── folder2

└── folder3

4 directories, 4 files

现在我们来测试一下

import os

for root, dirs, files in os.walk("./testwalk/"):

for name in files:

print(os.path