昨天在上廁所的時(shí)候突發(fā)奇想，當(dāng)你把usb插進(jìn)去的時(shí)候，能不能自動(dòng)執(zhí)行usb上的程序。查了一下，發(fā)現(xiàn)只有windows上可以，具體的大家也可以搜索(搜索關(guān)鍵詞usb autorun)到。但是，如果我想，比如，當(dāng)一個(gè)usb插入時(shí)，在后臺(tái)自動(dòng)把usb里的重要文件神不知鬼不覺(jué)地拷貝到本地或者上傳到某個(gè)服務(wù)器，就需要特殊的軟件輔助。

于是我心想，能不能用python寫一個(gè)程序，讓它在后臺(tái)運(yùn)行。每當(dāng)有u盤插入的時(shí)候，就自動(dòng)拷貝其中重要文件。

如何判斷U盤的插入與否?

首先我們打開(kāi)電腦終端，進(jìn)入/Volumes目錄，這時(shí)候插入U(xiǎn)盤，可以發(fā)現(xiàn)它被掛載在了這個(gè)目錄之下，也就是說(shuō)，我們只要在固定時(shí)間掃描這個(gè)目錄，當(dāng)這個(gè)目錄有新文件夾出現(xiàn)的時(shí)候，很可能有U盤被插入了。

我的設(shè)計(jì)是這樣的，用time.sleep(3)函數(shù)，讓程序保持運(yùn)行狀態(tài)，并且每隔三秒查看一下/Volumes/目錄，如果多出來(lái)文件夾，就將其拷貝到另外的文件夾。

就像標(biāo)題所示，我們真的只用了10行(其實(shí)是11行，湊個(gè)整：)完成了這個(gè)“病毒”。我們可以發(fā)現(xiàn)usb中的目錄，在插入半分鐘后全部躺在了home目錄下了。

如何選擇性的復(fù)制文件?

剛剛我們寫了一個(gè)很簡(jiǎn)易的腳本測(cè)試了一下這個(gè)想法的可行性，但是還是有問(wèn)題。剛才之所以能把U盤中所有文件很快復(fù)制進(jìn)去，是因?yàn)閁盤中只有兩三個(gè)文件，大小不超過(guò)15M。如果目標(biāo)U盤中有很多電影，音樂(lè)，這些我們并不需要的文件，我們的程序就應(yīng)該能跳過(guò)它們，僅僅選擇一些重要的比如.docx比如.ppt文件，或者僅僅復(fù)制最近修改過(guò)的那些文件，或者排除所有大小大于5M的文件。我們可以用python做到嗎?當(dāng)然!

os.walk 遞歸文件夾中所有文件

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

這里我放了一個(gè)別人的教程。大家可以大概了解一下，總之我大概理解是這么個(gè)東西。

還是舉個(gè)例子吧。

我在某目錄下創(chuàng)建了testwalk文件夾，里面有file123.txt三個(gè)文件，folder123三個(gè)文件夾，其中folder1中有文件file4.txt以及folder4

現(xiàn)在我們來(lái)測(cè)試一下

root存放的是當(dāng)前位置，它會(huì)把./testwalk/下所有的文件夾作為根目錄，往下搜索

單獨(dú)查看 dirs

單獨(dú)查看 files

好了，我們現(xiàn)在需要遞歸usb文件夾，找到所有的file，查看大小，如果小于，比如3M，就拷貝進(jìn)home，大于就舍去。

shutil模塊

現(xiàn)在我們拿剛才的文件夾舉例子，如果想把file1.txt拷貝到folder2：

還有許多使用工具在shutil里面這里就不詳述了。

os.path.getsize()判斷大小

os.path.getsize(文件名)返回的是一個(gè)單位為byte的數(shù)值，如果用來(lái)查看文件大小，我們則需要手動(dòng)寫一個(gè)函數(shù)，將其換算成容易閱讀的形式。

這里我們只要選擇文件大小小于3M的即可，3M = 3 * 1024kB = 3 * 1024*1024byte

結(jié)合shutil.copy2就可以把選定大小的文件復(fù)制進(jìn)我們的目標(biāo)文件夾了

如何指定文件類型

這里就需要正則表達(dá)式來(lái)幫助我們了。

正則表達(dá)式內(nèi)容很多，《python核心編程》中用了整整一章來(lái)講，所以我們也不深入了。下面是官方文檔，感興趣的可以看一下。

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

如下，我們讓指定文件后綴以及指定文件大小可以復(fù)制進(jìn)我們的目標(biāo)文件：

別忘了導(dǎo)入 re

用更加復(fù)雜的正則表達(dá)式可以更好地指定文件類型

根據(jù)修改時(shí)間篩選文件

這時(shí)候我在目錄下創(chuàng)建了一個(gè)文件叫做newfile

總之，對(duì)每一個(gè)文件進(jìn)行修改時(shí)間的篩選可以只復(fù)制那些近期，或者特定時(shí)期修改或者添加過(guò)的文件，這個(gè)功能在特定情況下很有用。

總結(jié)

其實(shí)，標(biāo)題這么起只是為了吸引大家注意，這就是一個(gè)小程序，也談不上病毒。我更想通過(guò)這個(gè)例子，展示python對(duì)于文件處理的強(qiáng)大能力，引發(fā)大家的學(xué)習(xí)熱情。以上實(shí)現(xiàn)都是基于macos，linux應(yīng)該一樣，windows稍加修改也可以成功。

-End-

最近有一些小伙伴，讓我?guī)兔φ乙恍┟嬖囶}資料，于是我翻遍了收藏的 5T 資料后，匯總整理出來(lái)，可以說(shuō)是程序員面試必備！所有資料都整理到網(wǎng)盤了，歡迎下載！

審核編輯 ：李倩