mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 22:41:20 +08:00
318 lines
12 KiB
Python
318 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
@File : contact.py
|
||
@Author : Shuaikang Zhou
|
||
@Time : 2022/12/13 15:07
|
||
@IDE : Pycharm
|
||
@Version : Python3.10
|
||
@comment : ···
|
||
"""
|
||
from typing import Dict
|
||
|
||
from PyQt5.QtCore import *
|
||
from PyQt5.QtGui import *
|
||
from PyQt5.QtWidgets import *
|
||
|
||
import app.Ui.MyComponents.Contact as MyLabel
|
||
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/icons/logo.svg'))
|
||
# self.setStyleSheet('''QWidget{background-color:rgb(255, 255, 255);}''')
|
||
self.initui()
|
||
self.Me = Me
|
||
self.Thread = ChatMsg(self.Me.username, None)
|
||
self.contacts: Dict[str, MyLabel.ContactUi] = {}
|
||
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 = MyLabel.ContactUi(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].contact.conRemark
|
||
nickname = self.contacts[talkerId].contact.nickname
|
||
alias = self.contacts[talkerId].contact.alias
|
||
|
||
self.label_remark.setText(conRemark)
|
||
self.ta_username = talkerId
|
||
if '@chatroom' in talkerId:
|
||
self.chatroomFlag = True
|
||
else:
|
||
self.chatroomFlag = False
|
||
self.userinfo.l_remark.setText(conRemark)
|
||
pixmap = self.contacts[talkerId].contact.avatar
|
||
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(f"导出聊天记录成功\n在.\\data\\目录下")
|
||
reply.addButton("确认", QMessageBox.AcceptRole)
|
||
reply.addButton("取消", QMessageBox.RejectRole)
|
||
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 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)
|