一個突發奇想,當你把usb插進去的時候,你能不能自動執行usb上的程式。查了一下,發現只有windows上可以,具體的大家也可以搜尋(搜尋關鍵字usb autorun)到。但是,如果我想,例如,當一個usb插入時,在後台自動把usb裡的重要文件神不知不覺地拷貝到本地或上傳到某個伺服器,就需要特殊的軟體輔助。
於是我心想,能不能用python寫一個程序,讓它在後台運行。每當有u盤插入的時候,就自動拷貝其中重要文件。
首先我們打開電腦終端,進入/Volumes目錄,這時候插入U盤,可以發現它被掛載在了這個目錄之下,也就是說,我們只要在固定時間掃描這個目錄,當這個目錄有新資料夾出現的時候,很可能有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/下所有的資料夾當作根目錄,往下搜尋
#
好了,我們現在需要遞歸usb資料夾,找到所有的file,查看大小,如果小於,例如3M,就拷貝進home,大於就捨去。
#現在我們拿剛才的資料夾舉例子,如果想把file1.txt拷貝到folder2:
#還有許多使用工具在shutil裡面這裡就不詳述了。
os.path.getsize(檔案名稱)返回的是一個單位為byte的數值,如果用來查看檔案大小,我們需要手動寫一個函數,將其換算成容易閱讀的形式。
這裡我們只要選擇檔案大小小於3M的即可,3M = 3 * 1024kB = 3 * 1024*1024byte
結合shutil.copy2就可以把選定大小的檔案複製進我們的目標資料夾了
這裡就需要正規表示式來幫助我們了。
正規表示式內容很多,《python核心程式設計》中用了整整一章來講,所以我們也不深入了。
如下,我們讓指定檔案後綴以及指定檔案大小可以複製進我們的目標檔案:
#用更複雜的正規表示式可以更好地指定檔案類型
#這時候我在目錄下建立了一個檔案叫做newfile
總之,對每個檔案進行修改時間的篩選可以只複製那些近期,或是特定時期修改或新增的文件,這個功能在特定情況下很有用。
#其實,標題這麼起只是為了吸引大家注意,這就是一個小程序,也談不上病毒。
我更想透過這個例子,展現python對於檔案處理的強大能力,引發大家的學習熱情。
以上實作都是基於macos,linux應該一樣,windows稍加修改也可以成功。以上技巧僅供學習,切勿用於其他行為
以上是細思恐極,插上 U 碟就開始執行 Python 程式碼的詳細內容。更多資訊請關注PHP中文網其他相關文章!