现在我们来测试一下
root存放的是当前位置,它会把./testwalk/下所有的文件夹作为根目录,往下搜索
单独查看 dirs
单独查看 files
好了,我们现在需要递归usb文件夹,找到所有的file,查看大小,如果小于,比如3M,就拷贝进home,大于就舍去。
shutil模块
现在我们拿刚才的文件夹举例子,如果想把file1.txt拷贝到folder2:
还有许多使用工具在shutil里面这里就不详述了。
os.path.getsize()判断大小
os.path.getsize(文件名)返回的是一个单位为byte的数值,如果用来查看文件大小,我们则需要手动写一个函数,将其换算成容易阅读的形式。
这里我们只要选择文件大小小于3M的即可,3M = 3 * 1024kB = 3 * 1024*1024byte
结合shutil.copy2就可以把选定大小的文件复制进我们的目标文件夹了
如何指定文件类型
这里就需要正则表达式来帮助我们了。
正则表达式内容很多,《python核心编程》中用了整整一章来讲,所以我们也不深入了。下面是官方文档,感兴趣的可以看一下。
https://docs.python.org/2/library/re.html
如下,我们让指定文件后缀以及指定文件大小可以复制进我们的目标文件:
别忘了导入 re
用更加复杂的正则表达式可以更好地指定文件类型
根据修改时间筛选文件
这时候我在目录下创建了一个文件叫做newfile
总之,对每一个文件进行修改时间的筛选可以只复制那些近期,或者特定时期修改或者添加过的文件,这个功能在特定情况下很有用。
总结
其实,标题这么起只是为了吸引大家注意,这就是一个小程序,也谈不上病毒。我更想通过这个例子,展示python对于文件处理的强大能力,引发大家的学习热情。以上实现都是基于macos,linux应该一样,windows稍加修改也可以成功。
查看更多关于十行代码使用Python写一个USB病毒的详细内容...