TG-PlatformPlus/main.py

# #!/opt/homebrew/bin/python3
# # -*- coding:utf-8 -*-

import sys
import traceback
import logging
import os
from PyQt6 import *
from PyQt6.QtCore import *
from PyQt6.QtGui import *
from PyQt6.QtWidgets import *
from PyQt6.QtWebEngineWidgets import QWebEngineView
from mainWindow import MainWindow
from dog import dog
from dog import suDog
import xlwings
import psutil
import matplotlib
import asyncio
from bleak import BleakScanner
import nidaqmx
import pyvisa
import time
from PyQt6.QtQml import qmlRegisterType
from CustomPlotItem import CustomPlotItem
from shutil import copyfile
import datetime
import threading


# # 自定义消息处理器
# def myMsgHandler(msg_type, context, msg):
#     try:
#         # 线程安全的锁
#         lock = threading.Lock()
#         lock.acquire()

#         # 获取当前时间
#         current_time = datetime.datetime.now().strftime("%H")
        
#         # 根据消息类型设置日志级别
#         if msg_type == QtMsgType.QtDebugMsg:
#             log_level = "Debug"
#         elif msg_type == QtMsgType.QtWarningMsg:
#             log_level = "Warning"
#         elif msg_type == QtMsgType.QtCriticalMsg:
#             log_level = "Critical"
#         elif msg_type == QtMsgType.QtFatalMsg:
#             log_level = "Fatal"
#         else:
#             log_level = "Unknown"

#         # 构建日志消息
#         log_message = f"{current_time} [{log_level}]: {msg}"
        
#         # 写入到日志文件
#         log_filename = f"{LOGDIR}{current_time}_log.txt"
#         with open(log_filename, "a", encoding="utf-8") as file:
#             file.write(log_message + "\n")

#         # 释放锁
#         lock.release()
#     except:
#         pass

# # 安装自定义的消息处理器
# qInstallMessageHandler(myMsgHandler)

# 自定义的异常钩子
def my_exception_hook(exctype, value, traceback_obj):
    LOGDIR = os.path.join(os.path.dirname(__file__), 'logFile', datetime.datetime.now().strftime("%Y-%m-%d"))
    if not os.path.exists(LOGDIR):
        os.makedirs(LOGDIR)
    with open(os.path.join(LOGDIR,"error_log.txt"), "a") as f:
        # 写入异常信息
        f.write(f"Unhandled exception occurred at {datetime.datetime.now()}\n")
        if exctype:
            f.write(f"Exception type: {exctype.__name__}\n")
        else:
            f.write("Exception type: None\n")
        
        if value:
            f.write(f"Exception value: {value}\n")
        else:
            f.write("Exception value: None\n")
        
        # 写入堆栈跟踪信息
        for line in traceback.format_exception(exctype, value, traceback_obj):
            f.write(line)
        
        # 如果需要，可以添加额外的文件名和行号信息
        if traceback_obj:
            f.write(f"Exception occurred in file '{traceback_obj.tb_frame.f_code.co_filename}' at line {traceback_obj.tb_lineno}\n")

# 设置自定义的异常钩子
sys.excepthook = my_exception_hook

class ThemeGuard(QObject):
    def eventFilter(self, obj, event):
        if event.type() == QEvent.Type.PaletteChange:
            enforce_theme()  # 拦截主题变更事件 
        return False 

def enforce_theme():
    # 1. 强制设定调色板 
    palette = app.palette()
    palette.setColor(QPalette.ColorRole.Window, QColor(240, 240, 240))       # 主背景色 
    palette.setColor(QPalette.ColorRole.WindowText, QColor(30, 30, 30))      # 文本色 
    palette.setColor(QPalette.ColorRole.Base, QColor(255, 255, 255))         # 输入框背景 
    palette.setColor(QPalette.ColorRole.Text, QColor(0, 0, 0))               # 输入文本 
    app.setPalette(palette) 
 
if __name__ == "__main__":
    app = QApplication(sys.argv)
    enforce_theme()
    # app.installEventFilter(ThemeGuard())
    # 不需要手动设置高DPI，自动启用缩放 
    QApplication.setHighDpiScaleFactorRoundingPolicy(Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
    screen = app.primaryScreen()
    dogErrorIndex = 0
    for i in range(3):
        state = dog.isDogAlive()
        if state != 1:
            if suDog.isDogAlive() != 1:
                dogErrorIndex += 1
                print("dogErrorIndex:", dogErrorIndex)
            else:
                dogErrorIndex = 0
        else:
            dogErrorIndex = 0
            print("dogErrorIndex:", dogErrorIndex)
                    
    if dogErrorIndex >= 3:
        reply = QMessageBox.critical(None, "Error", "请插入加密狗", QMessageBox.StandardButton.Ok)
        if reply == QMessageBox.StandardButton.Ok:
            sys.exit(1)
    if not os.path.exists("data.db"):
        try:
            copyfile("default_data.db", "data.db")
        except IOError as e:
            print("Unable to copy file. %s" % e)
        except:
            print("Unexpected error:", sys.exc_info())
    nidaqmx.annotations.getOptionalRelease()
    app.setStyle("Fusion")
    # app.style().metaObject().className = lambda: "FusionStyle"  # 欺骗系统主题检测 
    # app.paletteChanged = lambda *args: app.setPalette(app.palette())  # 冻结调色板 
    app.setWindowIcon(QIcon(os.path.join(os.path.dirname(__file__), 'logo','logo.ico')))
    qmlRegisterType(CustomPlotItem, "CustomPlot", 1, 0, "CustomPlot")
    widget = MainWindow()
    # widget.show()
    sys.exit(app.exec())