技术文章 > 头条 > 正文

用python写一个usb自动拷贝程序!

 Ly

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

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

1bc21e3fbf380ecf53d81a4f77920e5.png

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

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

d2cadd2a65d2aa6e0f4bda36abbbda2.png

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

如何选择性的复制文件?

刚刚我们写了一个很简易的脚本测试了一下这个想法的可行性,但是还是有问题。

刚才之所以能把U盘中所有文件很快复制进去,是因为U盘中只有两三个文件,大小不超过15M。

如果目标U盘中有很多电影,音乐,这些我们并不需要的文件,我们的程序就应该能跳过它们,仅仅选择一些重要的比如.docx比如.ppt文件,或者仅仅复制最近修改过的那些文件,或者排除所有大小大于5M的文件。我们可以用python做到吗?当然!

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

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

fdd9866608df6eb04f0f43ff824ec72.png

现在我们来测试一下

f690e33f01df13445955068c9d34c79.png

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

693512ef3a9cecb3b31afca3723ffdd.png

单独看dirs

0cea5f0f4032c3bfdf7571d047cc329.png

单独查看files

0bdb04734a1af5f0c6232a8dca393e8.png

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

shutil模块

c6e90b012968927391488c44305fb70.png

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

f18e9347cc35fe76340403c4261ed66.png

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

os.path.gersize()判断大小

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

b135c4bc528c7de6c08ac59cf158f83.png

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

fafd3861673945c581138ff4766d9c2.png

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

如何指定文件类型

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

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

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

如下,我们让指定文件后缀以及指定文件大小可以复制进我们的目标文件:

别忘了导入re

b267cb9841412931fef6e7fce1a894a.png

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

根据修改时间筛选文件

f9db4b4ef4d6cc5eef86635989c529e.png

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

590bc84b44e1973349de156496eb8b3.png

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

更多Python知识,请关注Python视频教程!!

免费视频教程
本文转载于:InfoQ,如有侵犯,请联系a@php.cn删除
相关文章
 Python的文字特效,炫酷了!
 Python中的唯美星空代码详解!
 Python中用代码实现旋转地球!
 Python实现生成随机验证码附代码!
 用Python实现让微信地球转起来附代码!
相关视频章节
 什么是python,为什么必须学习python
 python的练习入门——练习一
 如何在Python中循环遍历列表
 在python如何使用列表
 在python如何使用字符串
视频教程分类