10 行 Python 代碼寫 1 個 USB 病毒
來源?| 知乎-DeepWwaver
昨天在上廁所的時候突發(fā)奇想，當(dāng)你把usb插進(jìn)去的時候，能不能自動執(zhí)行usb上的程序。查了一下，發(fā)現(xiàn)只有windows上可以，具體的大家也可以搜索(搜索關(guān)鍵詞usb autorun)到。但是，如果我想，比如，當(dāng)一個usb插入時，在后臺自動把usb里的重要文件神不知鬼不覺地拷貝到本地或者上傳到某個服務(wù)器，就需要特殊的軟件輔助。
于是我心想，能不能用python寫一個程序，讓它在后臺運行。每當(dāng)有u盤插入的時候，就自動拷貝其中重要文件。
如何判斷U盤的插入與否?
首先我們打開電腦終端，進(jìn)入/Volumes目錄，這時候插入U盤，可以發(fā)現(xiàn)它被掛載在了這個目錄之下，也就是說，我們只要在固定時間掃描這個目錄，當(dāng)這個目錄有新文件夾出現(xiàn)的時候，很可能有U盤被插入了。
我的設(shè)計是這樣的，用time.sleep(3)函數(shù)，讓程序保持運行狀態(tài)，并且每隔三秒查看一下/Volumes/目錄，如果多出來文件夾，就將其拷貝到另外的文件夾。
就像標(biāo)題所示，我們真的只用了10行(其實是11行，湊個整：)完成了這個“病毒”。我們可以發(fā)現(xiàn)usb中的目錄，在插入半分鐘后全部躺在了home目錄下了。
如何選擇性的復(fù)制文件?
剛剛我們寫了一個很簡易的腳本測試了一下這個想法的可行性，但是還是有問題。剛才之所以能把U盤中所有文件很快復(fù)制進(jìn)去，是因為U盤中只有兩三個文件，大小不超過15M。如果目標(biāo)U盤中有很多電影，音樂，這些我們并不需要的文件，我們的程序就應(yīng)該能跳過它們，僅僅選擇一些重要的比如.docx比如.ppt文件，或者僅僅復(fù)制最近修改過的那些文件，或者排除所有大小大于5M的文件。我們可以用python做到嗎?當(dāng)然!
os.walk 遞歸文件夾中所有文件
http://www.runoob.com/python/os-walk.html
這里我放了一個別人的教程。大家可以大概了解一下，總之我大概理解是這么個東西。
還是舉個例子吧。
我在某目錄下創(chuàng)建了testwalk文件夾，里面有file123.txt三個文件，folder123三個文件夾，其中folder1中有文件file4.txt以及folder4
現(xiàn)在我們來測試一下
root存放的是當(dāng)前位置，它會把./testwalk/下所有的文件夾作為根目錄，往下搜索
單獨查看 dirs
單獨查看 files
好了，我們現(xiàn)在需要遞歸usb文件夾，找到所有的file，查看大小，如果小于，比如3M，就拷貝進(jìn)home，大于就舍去。
shutil模塊
現(xiàn)在我們拿剛才的文件夾舉例子，如果想把file1.txt拷貝到folder2：
還有許多使用工具在shutil里面這里就不詳述了。
os.path.getsize()判斷大小
os.path.getsize(文件名)返回的是一個單位為byte的數(shù)值，如果用來查看文件大小，我們則需要手動寫一個函數(shù)，將其換算成容易閱讀的形式。
這里我們只要選擇文件大小小于3M的即可，3M = 3 * 1024kB = 3 * 1024*1024byte
結(jié)合shutil.copy2就可以把選定大小的文件復(fù)制進(jìn)我們的目標(biāo)文件夾了
如何指定文件類型
這里就需要正則表達(dá)式來幫助我們了。
正則表達(dá)式內(nèi)容很多，《python核心編程》中用了整整一章來講，所以我們也不深入了。下面是官方文檔，感興趣的可以看一下。
https://docs.python.org/2/library/re.html
如下，我們讓指定文件后綴以及指定文件大小可以復(fù)制進(jìn)我們的目標(biāo)文件：
別忘了導(dǎo)入 re
用更加復(fù)雜的正則表達(dá)式可以更好地指定文件類型
根據(jù)修改時間篩選文件
這時候我在目錄下創(chuàng)建了一個文件叫做newfile
總之，對每一個文件進(jìn)行修改時間的篩選可以只復(fù)制那些近期，或者特定時期修改或者添加過的文件，這個功能在特定情況下很有用。
總結(jié)
其實，標(biāo)題這么起只是為了吸引大家注意，這就是一個小程序，也談不上病毒。我更想通過這個例子，展示python對于文件處理的強大能力，引發(fā)大家的學(xué)習(xí)熱情。以上實現(xiàn)都是基于macos，linux應(yīng)該一樣，windows稍加修改也可以成功。
讓我知道你“在看”