from PyQt5.QtCore import QThread, pyqtSignal from PyQt5.QtWidgets import QWidget, QVBoxLayout, QLabel, QHBoxLayout from app.DataBase import msg from app.components.bubble_message import BubbleMessage, ChatWidget, Notice from app.person import MePC class ChatInfo(QWidget): def __init__(self, contact, parent=None): super().__init__(parent) self.last_timestamp = 0 self.last_str_time = '' self.last_pos = 0 self.contact = contact self.init_ui() self.show_chats() def init_ui(self): self.label_reamrk = QLabel(self.contact.remark) self.hBoxLayout = QHBoxLayout() self.hBoxLayout.addWidget(self.label_reamrk) self.vBoxLayout = QVBoxLayout() self.vBoxLayout.setSpacing(0) self.vBoxLayout.addLayout(self.hBoxLayout) self.chat_window = ChatWidget() self.chat_window.scrollArea.verticalScrollBar().valueChanged.connect(self.verticalScrollBar) self.vBoxLayout.addWidget(self.chat_window) self.setLayout(self.vBoxLayout) def show_chats(self): self.show_chat_thread = ShowChatThread(self.contact) self.show_chat_thread.showSingal.connect(self.add_message) self.show_chat_thread.finishSingal.connect(self.show_finish) self.show_chat_thread.start() def show_finish(self, ok): self.setScrollBarPos() def verticalScrollBar(self, pos): """ 滚动条到0之后自动更新聊天记录 :param pos: :return: """ # print(pos) if pos > 0: return # 记录当前滚动条最大值 self.last_pos = self.chat_window.verticalScrollBar().maximum() self.update_history_messages() def update_history_messages(self): self.show_chat_thread.start() def setScrollBarPos(self): """ 将滚动条位置设置为上次看到的地方 :param pos: :return: """ self.chat_window.update() self.chat_window.show() pos = self.chat_window.verticalScrollBar().maximum() - self.last_pos self.chat_window.set_scroll_bar_value(pos) def is_5_min(self, timestamp): if abs(timestamp - self.last_timestamp) > 300: self.last_timestamp = timestamp return True return False def add_message(self, message): try: type_ = message[2] str_content = message[7] str_time = message[8] # print(type_, type(type_)) is_send = message[4] avatar = MePC().avatar if is_send else self.contact.avatar timestamp = message[5] if type_ == 1: if self.is_5_min(timestamp): time_message = Notice(self.last_str_time) self.last_str_time = str_time self.chat_window.add_message_item(time_message, 0) bubble_message = BubbleMessage( str_content, avatar, type_, is_send ) # print(str_content) # self.scroolAreaLayout.addWidget(bubble_message) self.chat_window.add_message_item(bubble_message, 0) except: print(message) class ShowChatThread(QThread): showSingal = pyqtSignal(tuple) finishSingal = pyqtSignal(int) msg_id = 0 # heightSingal = pyqtSignal(int) def __init__(self, contact): super().__init__() self.last_message_id = 9999999 self.wxid = contact.wxid def run(self) -> None: messages = msg.get_message_by_num(self.wxid, self.last_message_id) if messages: self.last_message_id = messages[-1][0] for message in messages: self.showSingal.emit(message) self.msg_id += 1 self.finishSingal.emit(1)