摘要：今天的騷操作，分享的是一個(gè)自動(dòng)拷貝電腦盤內(nèi)容的程序。當(dāng)別人的盤插入到你的電腦上時(shí)，腳本會(huì)自動(dòng)把盤所有的內(nèi)容，全部拷貝到你的電腦上。騷源碼程序主要是由三個(gè)函數(shù)組成，。每個(gè)操作都會(huì)記錄到日志中，以便日后查看相應(yīng)的拷貝信息。

閱讀文本大概需要 6 分鐘。

Python 這門語言有非常多有趣的內(nèi)容，比如給微信好友自動(dòng)發(fā)消息、查看微信好友撤回的消息、通過微信控制電腦等等。在我公眾號(hào)上也分享過很多有趣的程序，近期我會(huì)去搜集一些騷操作的 Python 程序分享給大家，并命名「Python 騷操作」專欄，持續(xù)更新！文中一些程序可能來自別人的文章，但如果我覺得程序還有所改進(jìn)的地方，我會(huì)繼續(xù)添加一些代碼，并且我會(huì)在文末標(biāo)注原程序出處。

今天的「Python 騷操作」，分享的是一個(gè)自動(dòng)拷貝電腦 U 盤內(nèi)容的程序。當(dāng)別人的 U 盤插入到你的電腦上時(shí)，Python 腳本會(huì)自動(dòng)把 U 盤所有的內(nèi)容，全部拷貝到你的電腦上。腳本每隔 5 秒掃描當(dāng)前電腦是否有 U 盤插入，如果是已拷貝過的內(nèi)容，則不會(huì)重復(fù)拷貝，每個(gè)相應(yīng)的操作都會(huì)記錄到日志文件中。

如果恰好這個(gè) U 盤里有某些室友的「學(xué)習(xí)視頻」，比如下的圖片。

那騷氣指數(shù) 10 顆星！

注意此程序只是用來學(xué)習(xí)交流，切勿違法使用。

Python 騷思路

在 windows 系統(tǒng)中，當(dāng)有 U 盤插入到你的電腦上時(shí)，會(huì)給它分配一個(gè)磁盤名稱，比如在我的電腦上分配的是「J」。所以我會(huì)先定義一個(gè)「USB_FILE」變量，表示 U 盤所在的目錄。

我們利用 os 庫來判斷當(dāng)前系統(tǒng)是否有「USB_FILE」，如果掃描到當(dāng)前系統(tǒng)有這個(gè)目錄，說明有 U 盤插入到當(dāng)前電腦上。隨后我們通過 shutil 庫，把 U 盤里的內(nèi)容全部拷貝到自己的電腦上。在每個(gè)操作之間我添加了 input_log() 方法，用來把當(dāng)前相應(yīng)的操作記錄到日志文件中。?

Python 騷源碼

程序主要是由三個(gè)函數(shù)組成，get_dirs_size()、main()、input_log()。

get_dirs_size()

get_dirs_size() 主要是返回要拷貝 U 盤的長度，用來判斷當(dāng)前要拷貝的內(nèi)容是否已經(jīng)重復(fù)，如果重復(fù)則無需重新拷貝。

def?get_dirs_size(dir):
????size?=?0
????for?root,?dirs,?files?in?os.walk(dir):
????????size?+=?sum([getsize(join(root,?name))?for?name?in?files])
????print(size)
????return?size

main()

main() 用來執(zhí)行主邏輯代碼，即每隔 5 秒檢查當(dāng)前電腦是否有 U 盤插入。如果檢查到，則記錄當(dāng)前 U 盤的長度。在下一次檢測的時(shí)候，如果當(dāng)前的長度沒有發(fā)生變化，則無需重新拷貝。每個(gè)操作都會(huì)記錄到日志中，以便日后查看相應(yīng)的拷貝信息。

def?main():
????old_size?=?0
????new_dirsize?=?0
????while(1):
????????if?os.path.exists(USB_PATH):
????????????print("檢測到?U?盤")
????????????input_log("檢測到?U?盤")
????????????new_dirsize?=?get_dirs_size(USB_PATH)
????????????if?old_size?!=?new_dirsize:
????????????????for?filename?in?os.listdir(USB_PATH):
????????????????????print("拷貝了："?+?filename)
????????????????????input_log("拷貝了："?+?filename)
????????????????shutil.copytree(USB_PATH,?SAVE_PATH)
????????????????old_size?=?new_dirsize
????????????else:
????????????????print("U?盤沒有變化")
????????????????input_log("U?盤沒有變化")
????????else:
????????????print("暫時(shí)沒有?U?判斷")
????????print("開始休眠")
????????input_log("開始休眠")
????????sleep(5)

????????print("休眠結(jié)束，重新檢測")
????????input_log("休眠結(jié)束，重新檢測")

input_log() 利用 logging 庫，來把相應(yīng)的操作信息保存到一個(gè)日志文件中。

def?input_log(message):
????handler?=?logging.handlers.RotatingFileHandler(LOG_FILE,?maxBytes=1024?*?1024,?backupCount=5,
???????????????????????????????????????????????????encoding="utf-8")??#?實(shí)例化handler
????fmt?=?"%(asctime)s?-?%(levelname)s?-?%(message)s"

????formatter?=?logging.Formatter(fmt)??#?實(shí)例化formatter
????handler.setFormatter(formatter)??#?為handler添加formatter

????logger?=?logging.getLogger("tst")??#?獲取名為tst的logger
????logger.addHandler(handler)??#?為logger添加handler
????logger.setLevel(logging.DEBUG)

????logger.info(message)

Python 騷教程

此次的程序使用非常簡單，首先下載源碼，后臺(tái)回復(fù)「拷貝」即可獲取源碼。然后修改下 USB_PATH 和 SAVE_PATH，這兩個(gè)參數(shù)表示的是你 U 盤所屬的位置和你想要保存的目錄。

LOG_FILE?=?"test.log"
USB_PATH?=?"J:"
SAVE_PATH?=?"D:disk_copy"

原程序參考來源：

http://uuzdaisuki.com/2018/03...

本文首發(fā)于公眾號(hào)「癡?！?，后臺(tái)回復(fù)「1024」即可獲取最新編程資源。

比如這樣的：史上最全 Python 學(xué)習(xí)資料，PDF 電子書大合集