Python3.5+PyQt5多執行緒+itchat實現微信防撤回桌面版程式碼(二)
阿新 • • 發佈:2018-12-27
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設計
後記
第一次嘗試多執行緒程式設計,並且具體應用到實際專案中去,收穫良多。
最近打算學點Django,有沒有推薦教程?