mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 14:31:18 +08:00
417 lines
16 KiB
Python
417 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
@File : contact.py
|
||
@Author : Shuaikang Zhou
|
||
@Time : 2022/12/13 15:07
|
||
@IDE : Pycharm
|
||
@Version : Python3.10
|
||
@comment : ···
|
||
"""
|
||
import datetime
|
||
|
||
from PyQt5.QtCore import *
|
||
from PyQt5.QtGui import *
|
||
from PyQt5.QtWidgets import *
|
||
|
||
from .analysis import analysis
|
||
from .contactUi import *
|
||
from .emotion import emotion
|
||
from .report import report
|
||
from .userinfo import userinfoUi
|
||
from ...DataBase import data, output
|
||
|
||
EMOTION = 1
|
||
ANALYSIS = 2
|
||
|
||
|
||
class StackedWidget():
|
||
def __init__(self):
|
||
pass
|
||
|
||
|
||
class ContactController(QWidget, Ui_Dialog):
|
||
exitSignal = pyqtSignal()
|
||
urlSignal = pyqtSignal(QUrl)
|
||
|
||
# username = ''
|
||
|
||
def __init__(self, Me, parent=None):
|
||
super(ContactController, self).__init__(parent)
|
||
self.emotionView = None
|
||
self.analysisView = None
|
||
self.chatroomFlag = None
|
||
self.ta_avatar = None
|
||
self.setupUi(self)
|
||
self.setWindowTitle('WeChat')
|
||
self.setWindowIcon(QIcon('./app/data/icon.png'))
|
||
# self.setStyleSheet('''QWidget{background-color:rgb(255, 255, 255);}''')
|
||
self.initui()
|
||
self.Me = Me
|
||
self.Thread = ChatMsg(self.Me.username, None)
|
||
self.contacts = {}
|
||
self.last_btn = None
|
||
self.chat_flag = True
|
||
self.show_flag = False
|
||
self.ta_username = None
|
||
self.last_pos = 0
|
||
self.last_msg_time = 0 # 上次信息的时间
|
||
self.last_talkerId = None
|
||
self.now_talkerId = None
|
||
self.last_analysis = None
|
||
self.now_analysis = None
|
||
self.view_emotion = {}
|
||
self.view_analysis = {}
|
||
self.showContact()
|
||
|
||
self.now_btn = self.userinfo
|
||
self.last_btn = None
|
||
|
||
def initui(self):
|
||
# 槽函数连接
|
||
self.btn_back.clicked.connect(self.back)
|
||
# self.btn_output.clicked.connect(self.output)
|
||
self.btn_analysis.clicked.connect(self.analysis)
|
||
self.btn_emotion.clicked.connect(self.emotionale_Analysis)
|
||
self.btn_report.clicked.connect(self.annual_report)
|
||
|
||
self.lay0 = QVBoxLayout()
|
||
# self.widget.setLayout(self.lay0)
|
||
# self.widget.setStyleSheet('''QWidget{background-color:rgb(255, 255, 255);}''')
|
||
# self.stackedWidget.setStyleSheet('''QWidget{background-color:rgb(240, 240, 240);}''')
|
||
self.stackedWidget.setStyleSheet('''QWidget{background-color:rgb(255, 255, 255);}''')
|
||
self.frame = QtWidgets.QFrame()
|
||
self.frame.setObjectName("frame")
|
||
self.userinfo = userinfoUi.Ui_Frame() # 联系人信息界面
|
||
self.userinfo.setupUi(self.frame)
|
||
self.userinfo.progressBar.setVisible(False)
|
||
self.stackedWidget.addWidget(self.frame)
|
||
# self.lay0.addWidget(self.frame)
|
||
menu = QMenu(self)
|
||
self.toDocxAct = QAction(QIcon('app/data/icons/word.svg'), '导出Docx', self)
|
||
self.toCSVAct = QAction(QIcon('app/data/icons/csv.svg'), '导出CSV', self)
|
||
self.toHtmlAct = QAction(QIcon('app/data/icons/html.svg'), '导出HTML', self)
|
||
self.toolButton_output.setPopupMode(QToolButton.MenuButtonPopup)
|
||
menu.addAction(self.toDocxAct)
|
||
menu.addAction(self.toCSVAct)
|
||
menu.addAction(self.toHtmlAct)
|
||
self.toolButton_output.setMenu(menu)
|
||
# self.toolButton_output.addSeparator()
|
||
self.toHtmlAct.triggered.connect(self.output)
|
||
self.toDocxAct.triggered.connect(self.output)
|
||
self.toCSVAct.triggered.connect(self.output)
|
||
|
||
def showContact(self):
|
||
"""
|
||
显示联系人
|
||
:return:
|
||
"""
|
||
print('show')
|
||
if self.show_flag:
|
||
return
|
||
self.show_flag = True
|
||
rconversations = data.get_rconversation()
|
||
max_hight = max(len(rconversations) * 80, 680)
|
||
# 设置滚动区域的高度
|
||
self.scrollAreaWidgetContents.setGeometry(
|
||
QtCore.QRect(0, 0, 300, max_hight))
|
||
|
||
for i in range(len(rconversations)):
|
||
rconversation = rconversations[i]
|
||
username = rconversation[1]
|
||
# 创建联系人按钮对象
|
||
# 将实例化对象添加到self.contacts储存起来
|
||
pushButton_2 = Contact(self.scrollAreaWidgetContents, i, rconversation)
|
||
pushButton_2.setGeometry(QtCore.QRect(0, 80 * i, 300, 80))
|
||
pushButton_2.setLayoutDirection(QtCore.Qt.LeftToRight)
|
||
pushButton_2.clicked.connect(pushButton_2.show_msg)
|
||
pushButton_2.usernameSingal.connect(self.Contact)
|
||
self.contacts[username] = pushButton_2
|
||
|
||
def Contact(self, talkerId):
|
||
"""
|
||
聊天界面 点击联系人头像时候显示聊天数据
|
||
:param talkerId:
|
||
:return:
|
||
"""
|
||
self.now_talkerId = talkerId
|
||
# self.frame.setVisible(True)
|
||
self.setViewVisible(self.now_talkerId)
|
||
# 把当前按钮设置为灰色
|
||
if self.last_talkerId and self.last_talkerId != talkerId:
|
||
print('对方账号：', self.last_talkerId)
|
||
self.contacts[self.last_talkerId].setStyleSheet(
|
||
"QPushButton {background-color: rgb(220,220,220);}"
|
||
"QPushButton:hover{background-color: rgb(208,208,208);}\n"
|
||
)
|
||
self.last_talkerId = talkerId
|
||
self.contacts[talkerId].setStyleSheet(
|
||
"QPushButton {background-color: rgb(198,198,198);}"
|
||
"QPushButton:hover{background-color: rgb(209,209,209);}\n"
|
||
)
|
||
# 设置联系人的基本信息
|
||
conRemark = self.contacts[talkerId].conRemark
|
||
nickname = self.contacts[talkerId].nickname
|
||
alias = self.contacts[talkerId].alias
|
||
|
||
self.label_remark.setText(conRemark)
|
||
self.ta_username = talkerId
|
||
if '@chatroom' in talkerId:
|
||
self.chatroomFlag = True
|
||
else:
|
||
self.chatroomFlag = False
|
||
self.ta_avatar = self.contacts[talkerId].avatar
|
||
self.userinfo.l_remark.setText(conRemark)
|
||
pixmap = QPixmap(self.ta_avatar).scaled(60, 60) # 按指定路径找到图片
|
||
self.userinfo.l_avatar.setPixmap(pixmap)
|
||
self.userinfo.l_nickname.setText(f'昵称：{nickname}')
|
||
self.userinfo.l_username.setText(f'微信号：{alias}')
|
||
self.userinfo.lineEdit.setText(conRemark)
|
||
|
||
def output(self):
|
||
"""
|
||
导出聊天记录
|
||
:return:
|
||
"""
|
||
self.setViewVisible(self.now_talkerId)
|
||
|
||
if self.sender() == self.toDocxAct:
|
||
self.outputThread = output.Output(self.Me, self.now_talkerId)
|
||
self.outputThread.progressSignal.connect(self.output_progress)
|
||
self.outputThread.rangeSignal.connect(self.set_progressBar_range)
|
||
self.outputThread.okSignal.connect(self.hide_progress_bar)
|
||
self.outputThread.start()
|
||
elif self.sender() == self.toCSVAct:
|
||
print('开始导出csv')
|
||
self.outputThread = output.Output(self.Me, self.now_talkerId, type_=output.Output.CSV)
|
||
self.outputThread.progressSignal.connect(self.output_progress)
|
||
self.outputThread.rangeSignal.connect(self.set_progressBar_range)
|
||
self.outputThread.okSignal.connect(self.hide_progress_bar)
|
||
self.outputThread.start()
|
||
print('导出csv')
|
||
elif self.sender() == self.toHtmlAct:
|
||
print('功能暂未实现')
|
||
QMessageBox.warning(self,
|
||
"别急别急",
|
||
"马上就实现该功能"
|
||
)
|
||
|
||
def hide_progress_bar(self, int):
|
||
reply = QMessageBox(self)
|
||
reply.setIcon(QMessageBox.Information)
|
||
reply.setWindowTitle('OK')
|
||
reply.setText("导出聊天记录成功")
|
||
reply.addButton("确认", QMessageBox.AcceptRole)
|
||
reply.addButton("取消", QMessageBox.RejectRole)
|
||
# reply.addButton("忽略", QMessageBox.DestructiveRole)
|
||
print(reply)
|
||
api = reply.exec_()
|
||
self.userinfo.progressBar.setVisible(False)
|
||
|
||
def set_progressBar_range(self, value):
|
||
self.userinfo.progressBar.setVisible(True)
|
||
self.userinfo.progressBar.setRange(0, value)
|
||
|
||
def output_progress(self, value):
|
||
self.userinfo.progressBar.setProperty('value', value)
|
||
|
||
def analysis(self):
|
||
"""
|
||
聊天分析
|
||
:return:
|
||
"""
|
||
self.frame.setVisible(False) # 将userinfo界面设置为不可见
|
||
# 判断talkerId是否已经分析过了
|
||
# 是：则清空其他界面，直接显示该界面
|
||
# 否：清空其他界面，创建用户界面并显示
|
||
if 'room' in self.now_talkerId:
|
||
QMessageBox.warning(
|
||
self, '警告',
|
||
'暂不支持群组'
|
||
)
|
||
return
|
||
if self.now_talkerId in self.view_analysis:
|
||
self.setViewVisible(self.now_talkerId, mod=ANALYSIS)
|
||
return True
|
||
else:
|
||
self.setViewVisible(self.now_talkerId)
|
||
self.view_analysis[self.now_talkerId] = analysis.AnalysisController(self.now_talkerId)
|
||
# self.lay0.addWidget(self.view_analysis[self.now_talkerId])
|
||
self.last_analysis = self.now_talkerId
|
||
self.stackedWidget.addWidget(self.view_analysis[self.now_talkerId])
|
||
self.setViewVisible(self.now_talkerId, mod=ANALYSIS)
|
||
|
||
def emotionale_Analysis(self):
|
||
print('情感分析', data.get_conRemark(self.now_talkerId))
|
||
self.frame.setVisible(False) # 将userinfo界面设置为不可见
|
||
# 判断talkerId是否已经分析过了
|
||
# 是：则清空其他界面，直接显示该界面
|
||
# 否：清空其他界面，创建用户界面并显示
|
||
if self.now_talkerId in self.view_emotion:
|
||
self.setViewVisible(self.now_talkerId, mod=EMOTION)
|
||
return True
|
||
else:
|
||
self.setViewVisible(self.now_talkerId)
|
||
self.view_emotion[self.now_talkerId] = emotion.EmotionController(self.ta_username)
|
||
# self.lay0.addWidget(self.view_emotion[self.now_talkerId])
|
||
self.last_analysis = self.now_talkerId
|
||
self.stackedWidget.addWidget(self.view_emotion[self.now_talkerId])
|
||
self.setViewVisible(self.now_talkerId, mod=EMOTION)
|
||
pass
|
||
|
||
def annual_report(self):
|
||
self.report = report.ReportController(123)
|
||
self.report.show()
|
||
|
||
def showUserinfo(self):
|
||
pass
|
||
# self.analysisView = analysis.AnalysisController(self.now_talkerId)
|
||
# # self.lay0 = QHBoxLayout()
|
||
# # self.widget.setLayout(self.lay0)
|
||
# self.lay0.addWidget(self.analysisView)
|
||
|
||
def setViewVisible(self, wxid: str, mod=None):
|
||
"""
|
||
设置当前可见窗口
|
||
"""
|
||
match mod:
|
||
case None:
|
||
self.stackedWidget.setCurrentWidget(self.frame)
|
||
case 1:
|
||
if not self.last_analysis:
|
||
return False
|
||
self.stackedWidget.setCurrentWidget(self.view_emotion[self.now_talkerId])
|
||
case 2:
|
||
if not self.last_analysis:
|
||
return False
|
||
self.stackedWidget.setCurrentWidget(self.view_analysis[self.now_talkerId])
|
||
|
||
def back(self):
|
||
"""
|
||
将userinfo界面设置为可见，其他界面设置为不可见
|
||
"""
|
||
self.frame.setVisible(True)
|
||
self.setViewVisible(self.now_talkerId)
|
||
|
||
|
||
class Contact(QtWidgets.QPushButton):
|
||
"""
|
||
联系人类，继承自pyqt的按钮，里面封装了联系人头像等标签
|
||
"""
|
||
usernameSingal = pyqtSignal(str)
|
||
|
||
def __init__(self, Ui, id=None, contact=None):
|
||
super(Contact, self).__init__(Ui)
|
||
self.layoutWidget = QtWidgets.QWidget(Ui)
|
||
self.layoutWidget.setObjectName("layoutWidget")
|
||
self.gridLayout1 = QtWidgets.QGridLayout(self.layoutWidget)
|
||
self.gridLayout1.setSizeConstraint(QtWidgets.QLayout.SetMaximumSize)
|
||
self.gridLayout1.setContentsMargins(10, 10, 10, 10)
|
||
self.gridLayout1.setSpacing(10)
|
||
self.gridLayout1.setObjectName("gridLayout1")
|
||
self.label_time = QtWidgets.QLabel(self.layoutWidget)
|
||
font = QtGui.QFont()
|
||
font.setPointSize(8)
|
||
self.label_time.setFont(font)
|
||
self.label_time.setLayoutDirection(QtCore.Qt.RightToLeft)
|
||
self.label_time.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignTrailing | QtCore.Qt.AlignVCenter)
|
||
self.label_time.setObjectName("label_time")
|
||
self.gridLayout1.addWidget(self.label_time, 0, 2, 1, 1)
|
||
self.label_remark = QtWidgets.QLabel(self.layoutWidget)
|
||
font = QtGui.QFont()
|
||
font.setFamily("Adobe 黑体 Std R")
|
||
font.setPointSize(10)
|
||
self.label_remark.setFont(font)
|
||
self.label_remark.setObjectName("label_remark")
|
||
self.gridLayout1.addWidget(self.label_remark, 0, 1, 1, 1)
|
||
self.label_msg = QtWidgets.QLabel(self.layoutWidget)
|
||
font = QtGui.QFont()
|
||
font.setPointSize(8)
|
||
self.label_msg.setFont(font)
|
||
self.label_msg.setObjectName("label_msg")
|
||
self.gridLayout1.addWidget(self.label_msg, 1, 1, 1, 2)
|
||
self.label_avatar = QtWidgets.QLabel(self.layoutWidget)
|
||
self.label_avatar.setMinimumSize(QtCore.QSize(60, 60))
|
||
self.label_avatar.setMaximumSize(QtCore.QSize(60, 60))
|
||
self.label_avatar.setLayoutDirection(QtCore.Qt.RightToLeft)
|
||
self.label_avatar.setAutoFillBackground(False)
|
||
self.label_avatar.setStyleSheet("background-color: #ffffff;")
|
||
self.label_avatar.setInputMethodHints(QtCore.Qt.ImhNone)
|
||
self.label_avatar.setFrameShape(QtWidgets.QFrame.NoFrame)
|
||
self.label_avatar.setFrameShadow(QtWidgets.QFrame.Plain)
|
||
self.label_avatar.setAlignment(QtCore.Qt.AlignLeading | QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter)
|
||
self.label_avatar.setObjectName("label_avatar")
|
||
self.gridLayout1.addWidget(self.label_avatar, 0, 0, 2, 1)
|
||
self.gridLayout1.setColumnStretch(0, 1)
|
||
self.gridLayout1.setColumnStretch(1, 6)
|
||
self.gridLayout1.setRowStretch(0, 5)
|
||
self.gridLayout1.setRowStretch(1, 3)
|
||
self.setLayout(self.gridLayout1)
|
||
self.setStyleSheet(
|
||
"QPushButton {background-color: rgb(220,220,220);}"
|
||
"QPushButton:hover{background-color: rgb(208,208,208);}\n"
|
||
)
|
||
self.msgCount = contact[0]
|
||
self.username = contact[1]
|
||
self.conversationTime = contact[6]
|
||
self.msgType = contact[7]
|
||
self.digest = contact[8]
|
||
hasTrunc = contact[10]
|
||
attrflag = contact[11]
|
||
if hasTrunc == 0:
|
||
if attrflag == 0:
|
||
self.digest = '[动画表情]'
|
||
elif attrflag == 67108864:
|
||
try:
|
||
remark = data.get_conRemark(contact[9])
|
||
msg = self.digest.split(':')[1].strip('\n').strip()
|
||
self.digest = f'{remark}:{msg}'
|
||
except Exception as e:
|
||
# print(self.digest)
|
||
# print(e)
|
||
pass
|
||
else:
|
||
pass
|
||
self.show_info(id)
|
||
|
||
def show_info(self, id):
|
||
|
||
self.avatar = data.get_avator(self.username)
|
||
# print(avatar)
|
||
self.conRemark = data.get_conRemark(self.username)
|
||
self.nickname, self.alias = data.get_nickname(self.username)
|
||
time = datetime.datetime.now().strftime("%m-%d %H:%M")
|
||
msg = '还没说话'
|
||
pixmap = QPixmap(self.avatar).scaled(60, 60) # 按指定路径找到图片
|
||
self.label_avatar.setPixmap(pixmap) # 在label上显示图片
|
||
self.label_remark.setText(self.conRemark)
|
||
self.label_msg.setText(self.digest)
|
||
self.label_time.setText(data.timestamp2str(self.conversationTime)[2:])
|
||
|
||
def show_msg(self):
|
||
self.usernameSingal.emit(self.username)
|
||
|
||
|
||
class ChatMsg(QThread):
|
||
"""
|
||
发送信息线程
|
||
"""
|
||
isSend_signal = pyqtSignal(tuple)
|
||
okSignal = pyqtSignal(int)
|
||
|
||
def __init__(self, my_u, ta_u, parent=None):
|
||
super().__init__(parent)
|
||
self.sec = 2 # 默认1000秒
|
||
self.my_u = my_u
|
||
self.ta_u = ta_u
|
||
self.my_avatar = data.get_avator(my_u)
|
||
self.msg_id = 0
|
||
|
||
def run(self):
|
||
self.ta_avatar = data.get_avator(self.ta_u)
|
||
messages = data.get_message(self.ta_u, self.msg_id)
|
||
# messages.reverse()
|
||
for message in messages:
|
||
self.isSend_signal.emit(message)
|
||
self.msg_id += 1
|
||
self.okSignal.emit(1)
|