Python3.5+PyQt5多线程+itchat实现微信防撤回桌面版代码(二)

来源:互联网 发布:ae软件怎么下载 编辑:程序博客网 时间:2024/04/30 09:36

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,pyqtSignalimport itchatimport time,osimport shutilimport refrom 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设计

后记

第一次尝试多线程编程,并且具体应用到实际项目中去,收获良多。
最近打算学点Django,有没有推荐教程?