mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 22:41:20 +08:00
215 lines
6.7 KiB
Python
215 lines
6.7 KiB
Python
# -*- 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 *
|
||
from PyQt5.QtGui import QPixmap, QFont
|
||
from PyQt5.QtWidgets import *
|
||
|
||
from app import config
|
||
from app.DataBase import msg, misc
|
||
from app.Ui.Icon import Icon
|
||
from . import mainwindow
|
||
from .chat import ChatWindow
|
||
from .contact import ContactWindow
|
||
from .tool.tool_window import ToolWindow
|
||
from ..person import MePC
|
||
|
||
# 美化样式表
|
||
Stylesheet = """
|
||
QPushButton {
|
||
background-color: rgb(240,240,240);
|
||
border:none;
|
||
}
|
||
QPushButton:hover{
|
||
background-color: rgb(209,209,209);
|
||
}
|
||
/*去掉item虚线边框*/
|
||
QListWidget, QListView, QTreeWidget, QTreeView {
|
||
outline: 0px;
|
||
}
|
||
/*设置左侧选项的最小最大宽度,文字颜色和背景颜色*/
|
||
QListWidget {
|
||
min-width: 120px;
|
||
max-width: 120px;
|
||
color: black;
|
||
background: white;
|
||
border:none;
|
||
}
|
||
QListWidget::item{
|
||
height:60;
|
||
}
|
||
/*被选中时的背景颜色和左边框颜色*/
|
||
QListWidget::item:selected {
|
||
background: rgb(204, 204, 204);
|
||
border-left: 4px solid rgb(9, 187, 7);
|
||
color: black;
|
||
font-weight: bold;
|
||
}
|
||
/*鼠标悬停颜色*/
|
||
HistoryPanel::item:hover {
|
||
background: rgb(52, 52, 52);
|
||
}
|
||
"""
|
||
|
||
|
||
class MainWinController(QMainWindow, mainwindow.Ui_MainWindow):
|
||
exitSignal = pyqtSignal(bool)
|
||
okSignal = pyqtSignal(bool)
|
||
|
||
# username = ''
|
||
def __init__(self, username, parent=None):
|
||
super(MainWinController, self).__init__(parent)
|
||
self.setupUi(self)
|
||
self.setWindowIcon(Icon.MainWindow_Icon)
|
||
self.setStyleSheet(Stylesheet)
|
||
self.listWidget.clear()
|
||
self.resize(QSize(800, 600))
|
||
self.action_desc.triggered.connect(self.about)
|
||
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('./app/data/icons/loading.svg'))
|
||
|
||
def load_data(self):
|
||
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 = MePC()
|
||
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)
|
||
else:
|
||
QMessageBox.information(
|
||
self,
|
||
'温馨提示',
|
||
'点击 工具->获取信息 重启后可以显示本人头像哦'
|
||
)
|
||
|
||
def init_ui(self):
|
||
self.listWidget.setVisible(False)
|
||
self.stackedWidget.setVisible(False)
|
||
self.listWidget.currentRowChanged.connect(self.setCurrentIndex)
|
||
tool_item = QListWidgetItem(Icon.MyInfo_Icon, '工具', self.listWidget)
|
||
chat_item = QListWidgetItem(Icon.Chat_Icon, '聊天', self.listWidget)
|
||
contact_item = QListWidgetItem(Icon.Contact_Icon, '好友', self.listWidget)
|
||
myinfo_item = QListWidgetItem(Icon.MyInfo_Icon, '我的', self.listWidget)
|
||
|
||
tool_window = ToolWindow()
|
||
tool_window.get_info_signal.connect(self.set_my_info)
|
||
tool_window.load_finish_signal.connect(self.loading)
|
||
self.stackedWidget.addWidget(tool_window)
|
||
self.listWidget.setCurrentRow(0)
|
||
self.stackedWidget.setCurrentIndex(0)
|
||
self.chat_window = ChatWindow()
|
||
# chat_window = QWidget()
|
||
self.stackedWidget.addWidget(self.chat_window)
|
||
self.contact_window = ContactWindow()
|
||
self.stackedWidget.addWidget(self.contact_window)
|
||
label = QLabel('该功能暂不支持哦')
|
||
label.setFont(QFont("微软雅黑", 50))
|
||
label.setAlignment(Qt.AlignCenter)
|
||
self.stackedWidget.addWidget(label)
|
||
tool_window.load_finish_signal.connect(self.loading)
|
||
self.contact_window.load_finish_signal.connect(self.loading)
|
||
self.chat_window.load_finish_signal.connect(self.loading)
|
||
|
||
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.get_avatar_buffer(wxid)
|
||
except AttributeError:
|
||
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)
|
||
me = MePC()
|
||
me.set_avatar(img_bytes)
|
||
self.myavatar.setScaledContents(True)
|
||
self.myavatar.setPixmap(self.avatar)
|
||
|
||
def stop_loading(self, a0):
|
||
self.label.setVisible(False)
|
||
|
||
def loading(self, a0):
|
||
self.load_num += 1
|
||
# self.label.setVisible(False)
|
||
print('加载一个了')
|
||
if self.load_num == 2:
|
||
print('ok了')
|
||
self.label.clear()
|
||
self.label.hide()
|
||
self.okSignal.emit(True)
|
||
self.listWidget.setVisible(True)
|
||
self.stackedWidget.setVisible(True)
|
||
|
||
def about(self):
|
||
"""
|
||
关于
|
||
"""
|
||
QMessageBox.about(self, "关于",
|
||
f"版本：{config.version}\n"
|
||
f"QQ交流群:{config.contact}\n"
|
||
"地址：https://github.com/LC044/WeChatMsg\n"
|
||
f"新特性:\n{''.join(['*' + i for i in config.description])}"
|
||
)
|
||
|
||
def close(self) -> bool:
|
||
del self.listWidget
|
||
del self.stackedWidget
|
||
msg.close()
|
||
self.contact_window.close()
|
||
|
||
|
||
class LoadWindowThread(QThread):
|
||
windowSignal = pyqtSignal(QWidget)
|
||
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)
|