用 Python 遠程控制你的電腦
一、前言
很多時候我們有這種需求,因為程序運行比較耗時,但是我們沒有足夠的時間等待。這個時候我們就可以用 Python 做一個遠程控制電腦的小工具,實現遠程控制電腦。當然,我們能做的操作十分有限,我們可以通過郵件和電腦進行通訊、然后讓電腦以截圖的方式為我們顯示程序當前運行的狀態。然后我們通過這個狀態進行判斷,用郵件發送一些預設好的指令。如果程序意外中斷了,我們可以進行讓它重新運行,如果程序按照我們的預期執行了,我們就可以遠程關機。
二、使用 yagmail 進行郵件發送
使用 yagmail 發郵件非常簡單,我們首先需要安裝兩個模塊:pip install yagmailpip install keyring
2.1、發送一個簡單的郵件
我們用 yagmail 來發一封簡單的郵件試試:import yagmailimport keyringsender = '發件郵箱'password = '發件人密碼'receiver = '收件郵箱'# 注冊yagmail.register(sender, password)# 用 SMTP 服務器發郵件yag = yagmail.SMTP(user=sender, host='smtp.163.com')# 設置郵件的內容contents = ['Do not go gentle into that good night!']# 發送郵件yag.send(receiver, '郵件主題', contents=contents)
這里密碼可以使用真實密碼,也可以使用授權碼。授權碼的獲得需要到各個郵件的網頁獲取,這里以網易郵箱為例:
我們進入網頁版郵箱,依次點擊 1、2,進入如下頁面:
點擊開啟,然后按照指引獲取授權碼即可。
在連接郵箱服務器的時候,我們設置的一個 host,這個同樣需要在郵箱網頁中獲取。其中 163 的郵箱地址為 smtp.163.com,QQ 郵箱服務器地址為 smtp.qq.com,其它郵箱的可以自己查找一下。
郵箱內容是一個列表,我們現在是放入了一句話。
在調用 send 方法發送郵件時,我們需要傳入幾個參數。第一個是收件人,第二個是郵件的主題,第三個就是郵件的內容了。郵件內容我們傳入開始的內容列表。大家可以使用自己的郵箱測試一下。
2.2、發送附件
yagmail 非常智能,我們不需要像原始的 email 模塊一樣,先將附件上傳到郵箱服務器,然后再發給收件人。yagmail 會自動識別內容中的資源文件,然后自動上傳:import yagmailimport keyringsender = '發件郵箱'password = '發件人密碼'receiver = '收件郵箱'# 注冊yagmail.register(sender, password)# 用 SMTP 服務器發郵件yag = yagmail.SMTP(user=sender, host='smtp.163.com')# 設置郵件的內容contents = ['im.jpg']# 發送郵件yag.send(receiver, '郵件主題', contents=contents)
其中 im.jpg 是我本地的一張圖片,在發送的時候 yagmail 會自動識別,然后發送出去。我們沒做什么修改,就完成了附件的發送。
我們也可以直接發送帶有圖片的郵件,而不是以附件的形式發送:import yagmailimport keyringsender = '發件郵箱'password = '發件人密碼'receiver = '收件郵箱'# 注冊yagmail.register(sender, password)yag = yagmail.SMTP(user=sender, host='smtp.163.com')contents = ['這是一張圖片',yagmail.inline('im.jpg')]yag.send(receiver, '郵件主題', contents=contents)
我們只要用 yagmail.inline 包含圖片路徑即可。不過需要注意,發送的內容不能單是一張圖片。
三、郵件讀取
郵件的讀取需要使用到 imbox 模塊。我們需要先使用 keyring 模塊讀取密碼,keyring 可以讀取到在 yagmail 中注冊了的郵箱密碼:import yagmailimport keyringfrom imbox import Imboxsender = '你的郵箱'# 獲取郵箱密碼password = keyring.get_password(sender)# 讀取郵箱郵件with Imbox('imap.163.com', sender, password, ssl=True) as imbox:# 讀取所有未讀郵件all_msg = imbox.messages(unread=True)
Imbox 對象就是我們的郵箱對象,里面包含了所有郵箱信息。我們調用 messages 方法就可以獲取所有郵件,如果我們將 unread 設置為 True,則表示讀取未讀郵件。獲取了所有郵件,我們就可以查看更多信息了:import yagmailimport keyringfrom imbox import Imboxsender = '你的郵箱'# 獲取郵箱密碼password = keyring.get_password(sender)# 讀取郵箱郵件with Imbox('imap.163.com', sender, password, ssl=True) as imbox:# 讀取所有未讀郵件all_msg = imbox.messages(unread=True)# 遍歷郵件for uid, message in all_msg:# 輸出郵件的主題print(message.subject)# 標記已讀imbox.mark_seen(uid)# 獲取郵件的文本內容msg = message.body['plain']
我們要實現控制電腦就需要一直讀取,我們可以使用 schedule 模塊設置定時任務:import timeimport scheduledef func():print('test')schedule.every(2).seconds.do(func)while True:schedule.run_pending()time.sleep(1)
這樣的話,可以每兩秒執行一次 func 函數。
四、開發電腦控制工具
我們定義一個 ComputerController 類,用于控制電腦。該類的最基本方法就發郵件和收郵件:import timeimport yagmailimport keyringimport schedulefrom imbox import Imboxclass ComputerController():TIME_SPACE = 10SMTP_163 = 'smtp.163.com'SMTP_QQ = 'smtp.qq.com'def __init__(self, user, receiver):self.user = userself.receiver = receiver# 注冊yagmail.register(self.user, '你的密碼')def send_main(self, msg):'''發郵件'''yag = yagmail.SMTP(user=self.user, host=self.SMTP_163)contents = [msg]yag.send(self.receiver, '來自電腦的郵件', contents=contents)def receive_mail(self):'''收郵件'''# 讀取密碼password = keyring.get_password('yagmail', self.user)with Imbox('imap.163.com', 'sockwz@163.com', password, ssl=True) as imbox:all_msg = imbox.messages(unread=True)for uid, message in all_msg:if '110' == message.subject:# 標記已讀imbox.mark_seen(uid)msg = message.body['plain']print(msg)def run_controller(self):# 定時讀取郵件schedule.every(self.TIME_SPACE).seconds.do(controller.receive_mail)while True:schedule.run_pending()time.sleep(self.TIME_SPACE)
我們只需要創建這個類就可以了:if __name__ == '__main__':controller = ComputerController('發件郵箱', '收件郵箱')controller.run_controller()
我們需要傳入收件郵箱和發件郵箱,另外還需要在 init 方法中,修改密碼即可。
五、實現一些預設功能
下面我們實現一些小功能。
5.1、截圖from PIL import ImageGrabdef grab_windows(self):im = ImageGrab.grab()im.save('status.jpg')
使用 Pillow 模塊需要先安裝:pip install pillow
5.2、關機import osdef shut_down(self):'''關機'''os.system('shutdown /s /t 0')
這個是通用的兩個功能。另外我們可以使用 pynput 模塊控制鍵盤,進行一些操作。這個需要根據自己電腦的情況操作,比如我使用 PyCharm 運行程序可以使用下列代碼運行程序:def run_program(self):'''運行程序'''with self.kb.pressed(Key.shift):self.kb.press(Key.f10)self.kb.release(Key.f10)
大家可以自己定制一些操作。整個程序完整代碼如下:import osimport timeimport yagmailimport keyringimport schedulefrom imbox import Imboxfrom PIL import ImageGrabfrom pynput.keyboard import *class ComputerController():TIME_SPACE = 10SMTP_163 = 'smtp.163.com'SMTP_QQ = 'smtp.qq.com'kb = Controller()def __init__(self, user, receiver):self.user = userself.receiver = receiver# 注冊yagmail.register(self.user, '授權碼或密碼')def send_main(self, msg):'''發郵件'''yag = yagmail.SMTP(user=self.user, host=self.SMTP_163)contents = [msg]yag.send(self.receiver, '來自電腦的郵件', contents=contents)def receive_mail(self):'''收郵件'''# 讀取密碼password = keyring.get_password('yagmail', self.user)with Imbox('imap.163.com', 'sockwz@163.com', password, ssl=True) as imbox:all_msg = imbox.messages(unread=True)for uid, message in all_msg:if '110' == message.subject:# 標記已讀imbox.mark_seen(uid)msg = message.body['plain']if msg[0].__contains__('截圖'):# 截圖self.grab_windows()# 發送截圖self.send_main('status.jpg')if msg[0].__contains__('重啟'):self.restart_server()time.sleep(3)self.grab_windows()self.send_main('status.jpg')if msg[0].__contains__('關機'):self.shut_down()def grab_windows(self):im = ImageGrab.grab()im.save('status.jpg')def restart_server(self):time.sleep(3)os.system('shutdown')time.sleep(3)os.system('startup')def run_program(self):'''運行程序'''with self.kb.pressed(Key.shift):self.kb.press(Key.f10)self.kb.release(Key.f10)def shut_down(self):'''關機'''os.system('shutdown /s /t 0')def run_controller(self):schedule.every(self.TIME_SPACE).seconds.do(controller.receive_mail)while True:schedule.run_pending()time.sleep(self.TIME_SPACE)if __name__ == '__main__':controller = ComputerController('發件郵箱', '收件郵箱')controller.run_controller()
我們在讀取郵件時只讀取主題為 110 的郵件,這樣可以避免干擾郵件。所以我們在發送郵件時需要將主題設置為 110,而郵件內容則是一些關鍵詞。比如截圖就是截圖操作,關機就是關機操作。這個可以自己定義。