Python3.5+PyQt5多執行緒+itchat實現微信防撤回桌面版程式碼
阿新 • • 發佈:2019-01-10
weChatThread執行緒類
之前一直不會python多執行緒,寫這個程式的時候,發現不用多執行緒會陷入無限未響應狀態。於是學了半天python多執行緒,但是在主函式裡寫的時候,發現一個問題,Ui主執行緒和工作執行緒沒有分離,使用itchat等庫的時候會堵塞主執行緒,換句話說PyQt中子執行緒不能操作GUI介面。之前寫的多執行緒仍然屬於Ui主執行緒,是其子執行緒,所以才造成未響應。
既然知道問題了,那就查資料解決問題,後來,在幾篇部落格上找到了解決辦法
然後仿照第一篇部落格,重寫了QThread類,並借鑑第三篇部落格,學會了PyQt多執行緒中的訊號/槽機制,用來傳遞引數。
#!/usr/bin/env python3 # -*- coding: utf-8 -*- __author__ = 'memgq' from PyQt5.QtCore import QThread,pyqtSignal import itchat import time,os import shutil import re from itchat.content import * class weChatWord(QThread): getMsgSignal= pyqtSignal(str) #pyqtSignal()必須寫在__init__前面,裡面可接收的引數型別挺多的,str,list,dict都支援 def __init__(self,parent=None): super(weChatWord,self).__init__(parent) self.msg_list=[] self.type_list=['Picture','Recording', 'Attachment','Video'] def clearList(self): ''' 清空快取訊息和檔案 :return:''' tm_now=time.time() len_list=len(self.msg_list) if len_list>0: for i in range(len_list): if tm_now-self.msg_list[i]['msg_time']>121: if self.msg_list[i]['msg_type'] in self.type_list: try: os.remove(".\\BackUp\\"+self.msg_list[i]['msg_content']) except Exception as e: print(e) finally: pass else:break self.msg_list=self.msg_list[i:] def run(self): ''' 重寫run()函式, :return: ''' @itchat.msg_register([TEXT, PICTURE, MAP, CARD, SHARING, RECORDING, ATTACHMENT, VIDEO, FRIENDS], isFriendChat=True, isGroupChat=True) def getMsg(msg): ''' 註冊訊息型別,並對不同型別的訊息執行不用的操作 :param msg: :return: ''' msg_dict={} # pprint.pprint(msg) msg_id = msg['MsgId'] # 訊息ID msg_time = msg['CreateTime'] msg_url=None msg_group="" if (itchat.search_friends(userName=msg['FromUserName'])): if itchat.search_friends(userName=msg['FromUserName'])['RemarkName']: msg_from = itchat.search_friends(userName=msg['FromUserName'])['RemarkName'] # 訊息傳送人備註 elif itchat.search_friends(userName=msg['FromUserName'])['NickName']: # 訊息傳送人暱稱 msg_from = itchat.search_friends(userName=msg['FromUserName'])['NickName'] # 訊息傳送人暱稱 else: msg_from = r"讀取傳送訊息好友失敗" else: msg_group = msg['User']['NickName'] msg_from = msg['ActualNickName'] msg_type = msg['Type'] if msg_type in ['Text', 'Friends','Sharing']: msg_content = msg['Text'] msg_url = msg['Url'] elif msg_type in self.type_list: msg_content=msg['FileName'] msg['Text'](msg['FileName']) shutil.move(msg_content,r'.\\BackUp\\') elif msg['Type'] == 'Card': msg_content = msg['RecommendInfo']['NickName'] + r" 的名片" elif msg['Type'] == 'Map': x, y, location = re.search("<location x=\"(.*?)\" y=\"(.*?)\".*label=\"(.*?)\".*", msg['OriContent']).group(1, 2, 3) if location is None: msg_content = r"緯度->" + x.__str__() + " 經度->" + y.__str__() else: msg_content = r"" + location msg_dict={'msg_id':msg_id,'msg_time':msg_time,'msg_from':msg_from,'msg_group':msg_group, 'msg_content':msg_content,'msg_type':msg_type,'msg_url':msg_url} self.msg_list.append(msg_dict) self.clearList() @itchat.msg_register([NOTE],isFriendChat=True, isGroupChat=True) def recall(msg): ''' 當訊息型別為通知類的時候,查詢訊息內容是否為撤回訊息,如果是,則執行撤回後的防撤回操作 :param msg: :return: ''' # pprint.pprint(msg) msg_content=msg['Content'] if re.search(r'\<replacemsg\>\<!\[CDATA\[(.*)撤回了一條訊息\]\]\>\<\/replacemsg\>',msg_content): msg_note=re.search(r'\<replacemsg\>\<!\[CDATA\[(.*)\]\]\>\<\/replacemsg\>',msg_content).group(1) old_msg_id=re.search(r'\<msgid\>([0-9]+)\</msgid\>',msg_content).group(1) for each in self.msg_list: if each['msg_id']==old_msg_id: timeArray = time.localtime() otherStyleTime = time.strftime("%Y-%m-%d %H:%M:%S,", timeArray) msg_note = msg_note + ',撤回內容為:' + each['msg_content'] if each['msg_group']!='': msg_note = "群組("+each['msg_group']+")中"+msg_note msg_note=otherStyleTime+msg_note itchat.send(msg_note,toUserName='filehelper') self.msg_list.pop(self.msg_list.index(each)) self.getMsgSignal.emit(msg_note) break #建立BuckUp資料夾 if not os.path.exists(".\\BackUp\\"): os.mkdir('.\\BackUp\\') #啟動itchat() itchat.auto_login(hotReload=True) itchat.run()
主程式類
class mainwindowapp(QMainWindow,wechatunrecall.Ui_MainWindow): def __init__(self): super().__init__() self.setupUi(self) self.createActions() self.createTrayIcon() self.pushButton.clicked.connect(self.saveLog) self.pushButton_2.clicked.connect(self.clearlog) self.pushButton_3.clicked.connect(self.houtai) self.trayIcon.activated.connect(self.iconActivated) timeArray = time.localtime() otherStyleTime = time.strftime("%Y-%m-%d %H:%M:%S", timeArray) self.setLog(otherStyleTime+",程式執行時,請用手機掃描彈出的二維碼進行登入,並確保電腦上自帶的Window照片查" "看器可用,撤回的圖片檔案等可下載附件連同執行日誌儲存在程式目錄下BackUp資料夾中。\n") self.weChatBigWord() def saveLog(self): ''' 儲存日誌 :return: ''' if not os.path.exists(".\\BackUp\\"): os.mkdir(".\\BackUp\\") timeArray = time.localtime() otherStyleTime = time.strftime("%Y-%m-%d%H%M%S", timeArray) text=self.textBrowser.toPlainText() logPath=".\\BackUp\\"+otherStyleTime+'.txt' with open(logPath,'w') as f: f.write(text) def setLog(self,msg): ''' 往執行日誌視窗寫撤回訊息的內容 :param msg: :return: ''' self.textBrowser.append(msg) def createTrayIcon(self): ''' 建立托盤圖示,可以讓程式最小化到windows托盤中執行 :return: ''' self.trayIconMenu=QMenu(self) self.trayIconMenu.addAction(self.restoreAction) self.trayIconMenu.addSeparator() self.trayIconMenu.addAction(self.quitAction) self.trayIcon=QSystemTrayIcon(self) self.trayIcon.setContextMenu(self.trayIconMenu) self.trayIcon.setIcon(QIcon('./media/images/maincion.png')) self.setWindowIcon(QIcon('./media/images/maincion.png')) self.trayIcon.show() def createActions(self): ''' 為托盤圖示新增功能 :return: ''' self.restoreAction=QAction("恢復",self,triggered=self.showNormal) self.quitAction=QAction("退出",self,triggered=QApplication.instance().quit) def iconActivated(self,reason): ''' 啟用托盤功能 :param reason: :return: ''' if reason in (QSystemTrayIcon.Trigger, QSystemTrayIcon.DoubleClick): self.showNormal() def houtai(self): self.hide() def clearlog(self): self.textBrowser.clear() def weChatBigWord(self): ''' weChatThread類例項化,並啟動執行緒 :return: ''' from weChatThread import weChatWord self.wcBWThread=weChatWord() self.wcBWThread.getMsgSignal.connect(self.setLog) self.wcBWThread.start()
程式介面
程式介面仍然由Qtdesigner設計
後記
第一次嘗試多執行緒程式設計,並且具體應用到實際專案中去,收穫良多。