細思恐極，插上U盤就開始執行Python程式碼

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

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

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

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

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

就像標題所示，我們真的只用了10行(其實是11行，湊個整：)完成了這個「病毒」。我們可以發現usb中的目錄，在插入半分鐘後全部躺在了home目錄下了。

如何選擇性的複製檔案?

剛剛我們寫了一個很簡易的腳本測試了一下這個想法的可行性，但是還是有問題。剛才之所以能把U盤中所有文件很快複製進去，是因為U盤中只有兩三個文件，大小不超過15M。如果目標U盤中有很多電影，音樂，這些我們並不需要的文件，我們的程式就應該能跳過它們，僅僅選擇一些重要的比如.docx比如.ppt文件，或者僅僅複製最近修改過的那些文件，或排除所有大小大於5M的文件。我們可以用python做到嗎?當然！

還是舉個例子吧。

我在某目錄下創建了testwalk資料夾，裡面有file123.txt三個文件，folder123三個資料夾，其中folder1中有文件file4.txt以及folder4

現在我們來測試一下。

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核心程式設計》中用了整整一章來講，所以我們也不深入了。

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

別忘了匯入re

#用更複雜的正規表示式可以更好地指定檔案類型

#根據修改時間篩選檔案

#這時候我在目錄下建立了一個檔案叫做newfile

#總之，每個檔案修改時間的篩選可以只複製那些近期，或特定時期修改或新增的文件，這個功能在特定情況下很有用。

總結

其實，這就是一個小程序，也談不上是什麼高深的惡意程式碼。我更想透過這個例子，展現python對於文件處理的強大能力，引發大家的學習熱情。以上實作都是基於macos，linux應該一樣，windows稍加修改也可以成功。以上技巧僅供學習，切勿用於其他行為。

以上是細思恐極，插上U盤就開始執行Python程式碼的詳細內容。更多資訊請關注PHP中文網其他相關文章！