1. 程式人生 > >Python3.5+PyQt5多線程+itchat實現微信防撤回桌面版代碼

Python3.5+PyQt5多線程+itchat實現微信防撤回桌面版代碼

logs rep not cio backup 界面 sel store for

weChatThread線程類

之前一直不會python多線程,寫這個程序的時候,發現不用多線程會陷入無限未響應狀態。於是學了半天python多線程,但是在主函數裏寫的時候,發現一個問題,Ui主線程和工作線程沒有分離,使用itchat等庫的時候會堵塞主線程,換句話說PyQt中子線程不能操作GUI界面。之前寫的多線程仍然屬於Ui主線程,是其子線程,所以才造成未響應。
既然知道問題了,那就查資料解決問題,後來,在幾篇博客上找到了解決辦法

  • PyQt 分離UI主線程與工作線程
  • python pyqt4 PyQT實現了使用QThread後臺處理數據
  • 多線程中的信號/槽

然後仿照第一篇博客,重寫了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設計

後記

第一次嘗試多線程編程,並且具體應用到實際項目中去,收獲良多。

Python3.5+PyQt5多線程+itchat實現微信防撤回桌面版代碼