mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-14 05:21:41 +08:00
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
import ctypes
|
||
import sys
|
||
import time
|
||
import traceback
|
||
|
||
widget = None
|
||
|
||
|
||
def excepthook(exc_type, exc_value, traceback_):
|
||
# 将异常信息转为字符串
|
||
|
||
# 在这里处理全局异常
|
||
|
||
error_message = ''.join(traceback.format_exception(exc_type, exc_value, traceback_))
|
||
msg = f"Exception Type: {exc_type.__name__}\nException Value: {exc_value}\ndetails: {error_message}"
|
||
logger.error(f'程序发生了错误:\n\n{msg}')
|
||
QMessageBox.critical(None, "Unhandled Exception", msg, QMessageBox.Ok)
|
||
|
||
# 调用原始的 excepthook，以便程序正常退出
|
||
sys.__excepthook__(exc_type, exc_value, traceback_)
|
||
|
||
|
||
# 设置 excepthook
|
||
sys.excepthook = excepthook
|
||
from PyQt5.QtGui import QFont
|
||
from PyQt5.QtWidgets import *
|
||
from PyQt5.QtCore import Qt
|
||
|
||
from app.DataBase import close_db
|
||
from app.log import logger
|
||
from app.ui import mainview
|
||
from app.ui.tool.pc_decrypt import pc_decrypt
|
||
from app.config import version
|
||
|
||
ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID("WeChatReport")
|
||
QApplication.setAttribute(Qt.AA_EnableHighDpiScaling, True)
|
||
QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps, True)
|
||
|
||
|
||
class ViewController(QWidget):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.viewMainWindow = None
|
||
self.viewDecrypt = None
|
||
|
||
def loadPCDecryptView(self):
|
||
"""
|
||
登录界面
|
||
:return:
|
||
"""
|
||
self.viewDecrypt = pc_decrypt.DecryptControl()
|
||
self.viewDecrypt.DecryptSignal.connect(self.show_success)
|
||
self.viewDecrypt.show()
|
||
|
||
def loadMainWinView(self, username=None):
|
||
"""
|
||
聊天界面
|
||
:param username: 账号
|
||
:return:
|
||
"""
|
||
username = ''
|
||
start = time.time()
|
||
self.viewMainWindow = mainview.MainWinController(username=username)
|
||
self.viewMainWindow.exitSignal.connect(self.close)
|
||
try:
|
||
self.viewMainWindow.setWindowTitle(f"留痕-{version}")
|
||
self.viewMainWindow.show()
|
||
end = time.time()
|
||
self.viewMainWindow.init_ui()
|
||
print('ok', '本次加载用了', end - start, 's')
|
||
|
||
except Exception as e:
|
||
print(f"Exception: {e}")
|
||
logger.error(traceback.format_exc())
|
||
|
||
def show_success(self):
|
||
QMessageBox.about(self, "解密成功", "数据库文件存储在\napp/DataBase/Msg\n文件夹下")
|
||
|
||
def close(self) -> bool:
|
||
close_db()
|
||
super().close()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
app = QApplication(sys.argv)
|
||
font = QFont('微软雅黑', 12) # 使用 Times New Roman 字体，字体大小为 14
|
||
app.setFont(font)
|
||
view = ViewController()
|
||
widget = view.viewMainWindow
|
||
try:
|
||
# view.loadPCDecryptView()
|
||
view.loadMainWinView()
|
||
# view.show()
|
||
# view.show_success()
|
||
sys.exit(app.exec_())
|
||
except Exception as e:
|
||
print(f"Exception: {e}")
|
||
logger.error(traceback.format_exc())
|