十行 Python 代碼寫一個USB病毒

推薦 | 編程派公眾號（ID：codingpy）

昨天在上廁所的時候突發奇想，當你把usb插進去的時候，能不能自動執行usb上的程序。查了一下，發現只有windows上可以，具體的大家也可以搜索(搜索關鍵詞usb autorun)到。但是，如果我想，比如，當一個usb插入時，在後台自動把usb里的重要文件神不知鬼不覺地拷貝到本地或者上傳到某個伺服器，就需要特殊的軟體輔助。

於是我心想，能不能用python寫一個程序，讓它在後台運行。每當有u盤插入的時候，就自動拷貝其中重要文件。

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

首先我們打開電腦終端，進入/Volumes目錄，這時候插入U盤，可以發現它被掛載在了這個目錄之下，也就是說，我們只要在固定時間掃描這個目錄，當這個目錄有新文件夾出現的時候，很可能有U盤被插入了。

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

就像標題所示，我們真的只用了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

現在我們來測試一下

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

單獨查看 dirs

單獨查看 files

好了，我們現在需要遞歸usb文件夾，找到所有的file，查看大小，如果小於，比如3M，就拷貝進home，大於就捨去。

shutil模塊

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

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

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

結合shutil.copy2就可以把選定大小的文件複製進我們的目標文件夾了

如何指定文件類型

這裡就需要正則表達式來幫助我們了。

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

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

如下，我們讓指定文件後綴以及指定文件大小可以複製進我們的目標文件：

別忘了導入 re

用更加複雜的正則表達式可以更好地指定文件類型

根據修改時間篩選文件

這時候我在目錄下創建了一個文件叫做newfile

總之，對每一個文件進行修改時間的篩選可以只複製那些近期，或者特定時期修改或者添加過的文件，這個功能在特定情況下很有用。

總結

其實，標題這麼起只是為了吸引大家注意，這就是一個小程序，也談不上病毒。我更想通過這個例子，展示python對於文件處理的強大能力，引發大家的學習熱情。以上實現都是基於macos，linux應該一樣，windows稍加修改也可以成功。

回復下方「關鍵詞」，獲取優質資源

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！