# -*- coding: utf-8 -*- """ @File : mainview.py @Author : Shuaikang Zhou @Time : 2022/12/13 15:07 @IDE : Pycharm @Version : Python3.10 @comment : 主窗口 """ import json import os.path from PyQt5.QtCore import pyqtSignal, QUrl, Qt, QThread, QSize from PyQt5.QtGui import QPixmap, QFont, QDesktopServices, QIcon from PyQt5.QtWidgets import QMainWindow, QLabel, QListWidgetItem, QMessageBox from app.DataBase import misc_db, micro_msg_db, close_db from app.ui.Icon import Icon from . import mainwindow from app.ui.menu.about_dialog import AboutDialog from .chat import ChatWindow from .contact import ContactWindow from .menu.export import ExportDialog from .tool.tool_window import ToolWindow from ..DataBase.output_pc import Output from ..components.QCursorGif import QCursorGif from ..person import Me # 美化样式表 Stylesheet = """ QWidget{ background: rgb(238,244,249); } /*去掉item虚线边框*/ QListWidget, QListView, QTreeWidget, QTreeView { outline: 0px; } QMenu::item:selected { color: black; background: rgb(230, 235, 240); } /*设置左侧选项的最小最大宽度,文字颜色和背景颜色*/ QListWidget { min-width: 120px; max-width: 380px; color: black; border:none; } QListWidget::item{ min-width: 80px; max-width: 380px; min-height: 60px; max-height: 60px; } QListWidget::item:hover { background: rgb(230, 235, 240); } /*被选中时的背景颜色和左边框颜色*/ QListWidget::item:selected { background: rgb(230, 235, 240); border-left: 2px solid rgb(62, 62, 62); color: black; font-weight: bold; } /*鼠标悬停颜色*/ HistoryPanel::item:hover { background: rgb(52, 52, 52); } QCheckBox::indicator { background: rgb(255, 255, 255); Width:20px; Height:20px; border-radius: 10px } QCheckBox::indicator:unchecked{ Width:20px; Height:20px; image: url(:/icons/icons/unselect.svg); } QCheckBox::indicator:checked{ Width:20px; Height:20px; image: url(:/icons/icons/select.svg); } """ class MainWinController(QMainWindow, mainwindow.Ui_MainWindow, QCursorGif): exitSignal = pyqtSignal(bool) okSignal = pyqtSignal(bool) # username = '' def __init__(self, username, parent=None): super(MainWinController, self).__init__(parent) self.outputThread0 = None self.outputThread = None self.setupUi(self) # self.setWindowIcon(Icon.MainWindow_Icon) pixmap = QPixmap(Icon.logo_ico_path) icon = QIcon(pixmap) self.setWindowIcon(icon) self.setStyleSheet(Stylesheet) self.listWidget.clear() self.resize(QSize(800, 600)) self.action_desc.triggered.connect(self.about) self.load_flag = False self.load_data() self.load_num = 0 self.label = QLabel(self) self.label.setGeometry((self.width() - 300) // 2, (self.height() - 100) // 2, 300, 100) self.label.setPixmap(QPixmap(':/icons/icons/loading.svg')) self.menu_output.setIcon(Icon.Output) self.action_output_CSV.setIcon(Icon.ToCSV) self.action_output_CSV.triggered.connect(self.output) self.action_output_contacts.setIcon(Icon.Output) self.action_output_contacts.triggered.connect(self.output) self.action_batch_export.setIcon(Icon.Output) self.action_batch_export.triggered.connect(self.output) self.action_desc.setIcon(Icon.Help_Icon) def load_data(self, flag=True): if os.path.exists('./app/data/info.json'): with open('./app/data/info.json', 'r', encoding='utf-8') as f: dic = json.loads(f.read()) wxid = dic.get('wxid') if wxid: me = Me() me.wxid = dic.get('wxid') me.name = dic.get('name') me.mobile = dic.get('mobile') me.wx_dir = dic.get('wx_dir') self.set_my_info(wxid) self.load_flag = True else: QMessageBox.information( self, '温馨提示', '点击 工具->获取信息 重启后可以显示本人头像哦' ) def init_ui(self): # 设置忙碌光标图片数组 self.initCursor([':/icons/icons/Cursors/%d.png' % i for i in range(8)]) self.setCursorTimeout(100) self.startBusy() self.about_view = AboutDialog(main_window=self, parent=self) def setCurrentIndex(self, row): self.stackedWidget.setCurrentIndex(row) if row == 2: self.stackedWidget.currentWidget().show_contacts() if row == 1: self.stackedWidget.currentWidget().show_chats() def setWindow(self, window): try: window.load_finish_signal.connect(self.loading) except: pass self.stackedWidget.addWidget(window) def set_my_info(self, wxid): self.avatar = QPixmap() try: img_bytes = misc_db.get_avatar_buffer(wxid) except: close_db() QMessageBox.critical(self, "数据库错误", "请重启微信后重试") import shutil shutil.rmtree('./app/Database/Msg') return if not img_bytes: return if img_bytes[:4] == b'\x89PNG': self.avatar.loadFromData(img_bytes, format='PNG') else: self.avatar.loadFromData(img_bytes, format='jfif') self.avatar.scaled(60, 60) contact_info_list = micro_msg_db.get_contact_by_username(wxid) me = Me() me.set_avatar(img_bytes) me.smallHeadImgUrl = contact_info_list[7] self.myavatar.setScaledContents(True) self.myavatar.setPixmap(self.avatar) def stop_loading(self, a0): self.label.setVisible(False) self.stopBusy() def loading(self, a0): self.load_num += 1 if self.load_num == 1: self.label.clear() self.label.hide() self.okSignal.emit(True) self.listWidget.setVisible(True) self.stackedWidget.setVisible(True) self.stopBusy() if self.load_flag: self.listWidget.setCurrentRow(1) self.stackedWidget.setCurrentIndex(1) else: self.listWidget.setCurrentRow(0) self.stackedWidget.setCurrentIndex(0) def output(self): # self.startBusy() if self.sender() == self.action_output_CSV: self.outputThread = Output(None, type_=Output.CSV_ALL) self.outputThread.startSignal.connect(lambda x: self.startBusy()) self.outputThread.okSignal.connect( lambda x: self.message('聊天记录导出成功')) self.outputThread.start() elif self.sender() == self.action_output_contacts: self.outputThread = Output(None, type_=Output.CONTACT_CSV) self.outputThread.startSignal.connect(lambda x: self.startBusy()) self.outputThread.okSignal.connect( lambda x: self.message('联系人导出成功')) self.outputThread.start() elif self.sender() == self.action_batch_export: dialog = ExportDialog(None, title='批量导出聊天记录', parent=self) result = dialog.exec_() # 使用exec_()获取用户的操作结果 def message(self, msg): self.stopBusy() QMessageBox.about(self, "提醒", msg) def about(self): """ 关于 """ # QMessageBox.about(self, "关于",config.about) # about_view = AboutDialog(self) self.about_view.show() def decrypt_success(self): QMessageBox.about(self, "解密成功", "请重新启动") self.close() def closeEvent(self, event): reply = QMessageBox.question(self, '确认退出', '确定要退出吗？', QMessageBox.Yes | QMessageBox.No, QMessageBox.No) if reply == QMessageBox.Yes: close_db() event.accept() else: event.ignore() def close(self) -> bool: close_db() super().close() self.contact_window.close() self.exitSignal.emit(True) class LoadWindowThread(QThread): okSignal = pyqtSignal(bool) def __init__(self): super().__init__() self.num = 0 def loading(self): self.num += 1 print('加载一个了') if self.num == 2: self.okSignal.emit(True) def run(self): self.chat_window = ChatWindow() self.contact_window = ContactWindow() self.contact_window.load_finish_signal.connect(self.loading) self.chat_window.load_finish_signal.connect(self.loading) print('加载完成') self.okSignal.emit(True)