PROFIBUS/UI/MainWindow.py

import os
import sys
import time
import win32con
import win32gui
import win32process
import subprocess
import qtawesome

from PyQt5.QtWidgets import QApplication, QPushButton, QMainWindow, QDockWidget, QToolBar, QAction, QTabWidget, QGridLayout, QWidget, QHBoxLayout, QStackedWidget\
                            ,QVBoxLayout, QProgressBar, QLabel
from PyQt5.QtCore import Qt, QTimer, QSize
from PyQt5.QtGui import QIcon, QWindow, QPixmap
from utils.DBModels.BaseModel import *
from model.ClientModel.Client import Client
from UI.DeviceWidget import DeviceTab
from model.ProjectModel.DeviceManage import DevicesManange

from protocol.BatterySerial.BatteryProtocol import BatteryManange

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QProgressBar, QVBoxLayout, QWidget
from PyQt5.QtCore import Qt, QRect, QSize
from PyQt5.QtGui import QPainter, QFont, QColor
 
from utils import Globals


class CustomProgressBar(QProgressBar):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setTextVisible(False)
        self.setFont(QFont("Arial", 10))  # 设置字体大小和类型
       

    def paintEvent(self, event):
        super().paintEvent(event)
        painter = QPainter(self)
        value = self.value()
        percentage = round(value / self.maximum() * 100, 1)
        text = f"剩余电量{percentage}%"
        painter.drawText(self.rect(), Qt.AlignCenter, text)




def getHwndByPid(pid):
    def callback(hwnd, hwnds):
        if hwnd is not None and win32gui.IsWindowVisible(hwnd):
            _, found_pid = win32process.GetWindowThreadProcessId(hwnd)
            if found_pid == pid:
                hwnds.append(hwnd)

    hwnds = []
    # win32gui.EnumWindows(callback, hwnds)
    while True:
        win32gui.EnumWindows(callback, hwnds)
        if hwnds:
            return hwnds[0]
        # 如果没有找到窗口，则等待一段时间再尝试
        time.sleep(0.1)
        # return hwnds
    # return hwnds[0]

class CommonHelper:
    def __init__(self):
        pass

    @staticmethod
    def readQss(style):
        with open(style,"r") as f:
            return f.read()


class MainWindow(QWidget):
    def __init__(self):
        super().__init__()
        self.setObjectName("MainWindow")
        self.devicesManange = DevicesManange()
        self.batteryManange = BatteryManange()
        self.process = None
        self.initUI()

    def initUI(self):
        self.protocolTimer = QTimer()
        self.protocolTimer.timeout.connect(self.readValues)

        self.batteryTimer = QTimer()
        self.batteryTimer.timeout.connect(self.refreshProgressBar)
        self.batteryTimer.start(3000)

        # self.toolbarWidget = QWidget()
        # self.addToolBar(Qt.LeftToolBarArea, self.toolbar)

        toolbarLayout = QHBoxLayout()
        
        self.startProtocolBtn = QPushButton()
        self.startProtocolBtn.setObjectName("startProtocolBtn")
        # self.startProtocolBtn.setIcon(QIcon('Static/startProtocol.png'))
        self.startProtocolBtn.setText('开始通讯')
        self.startProtocolBtn.setIcon(QIcon('Static/start.png'))
        self.startProtocolBtn.setIconSize(QSize(23, 23))
        self.startProtocolBtn.setCheckable(True)
        self.startProtocolBtn.clicked.connect(self.startProtocol)

        self.switchBtn = QPushButton('通讯组态')
        self.switchBtn.setObjectName("switchBtn")
        self.switchBtn.setIcon(qtawesome.icon('fa.exchange', color='#1fbb6f'))
        self.switchBtn.setIconSize(QSize(25, 25))
        self.switchBtn.setCheckable(True)
        self.switchBtn.clicked.connect(self.switchWidget)

        self.batteryProBar = CustomProgressBar()
        self.batteryProBar.setRange(0, 100)

        self.batteryStateLabel = QLabel()
        self.batteryStateLabel.setObjectName('batteryLabel')

        self.switchTouchBtn = QPushButton("触控模式")
        self.switchTouchBtn.setIconSize(QSize(25, 25))
        self.switchTouchBtn.setObjectName('switchTouchBtn')
        self.switchTouchBtn.setIcon(qtawesome.icon('fa.keyboard-o', color='#1fbb6f'))
        self.switchTouchBtn.clicked.connect(self.switchTouchMode)
        self.switchTouchBtn.setCheckable(True)

        # 创建按钮
        self.minimizeButton = QPushButton(QIcon('Static/min.png'), "")
        self.minimizeButton.setObjectName('minButton')
        self.closeButton = QPushButton(QIcon('Static/close.png'), "")
        self.closeButton.setObjectName('closeButton')
        
        # 按钮点击事件连接
        self.minimizeButton.clicked.connect(self.showMinimized)
        self.closeButton.clicked.connect(self.close)
        hLayout = QHBoxLayout()
        hLayout.addWidget(self.minimizeButton)
        hLayout.addWidget(self.closeButton)

        iconLabel = QLabel()
        pix = QPixmap('Static/Hicent.png')
        scaledPixmap = pix.scaled(168, 28, Qt.KeepAspectRatio)
        iconLabel.setPixmap(scaledPixmap)
        iconLabel.setScaledContents(True)

        toolbarLayout.addWidget(iconLabel, 1)
        toolbarLayout.addWidget(QWidget(), 1)
        toolbarLayout.addWidget(self.startProtocolBtn, 1)
        toolbarLayout.addWidget(self.switchBtn, 1)
        toolbarLayout.addWidget(self.switchTouchBtn, 1)
        toolbarLayout.addWidget(QWidget(), 20)
        # toolbarLayout.addWidget(QLabel('电量:'), 1)
        toolbarLayout.addWidget(self.batteryProBar, 1)
        toolbarLayout.addWidget(self.batteryStateLabel, 1)
        toolbarLayout.addWidget(QWidget(), 1)
        toolbarLayout.addLayout(hLayout, 1)

        toolbarLayout.setSpacing(20)
        toolbarLayout.setContentsMargins(0, 0, 0, 0)

        dpMasterDockWidget = QDockWidget('DP主站')
        dpMasterDockWidget.setWidget(DeviceTab(dpMasterDockWidget, self.devicesManange))
        dpMasterDockWidget.setFeatures(QDockWidget.DockWidgetMovable | QDockWidget.DockWidgetFloatable) # type: ignore

        dpSlaveDockWidget = QDockWidget('DP从站')
        dpSlaveDockWidget.setWidget(DeviceTab(dpSlaveDockWidget, self.devicesManange))
        dpSlaveDockWidget.setFeatures(QDockWidget.DockWidgetMovable | QDockWidget.DockWidgetFloatable) # type: ignore

        paMasterDockWidget = QDockWidget('PA主站')
        paMasterDockWidget.setWidget(DeviceTab(paMasterDockWidget, self.devicesManange))
        paMasterDockWidget.setFeatures(QDockWidget.DockWidgetMovable | QDockWidget.DockWidgetFloatable) # type: ignore

        paSlaveDockWidget = QDockWidget('PA从站')
        paSlaveDockWidget.setWidget(DeviceTab(paSlaveDockWidget, self.devicesManange))
        paSlaveDockWidget.setFeatures(QDockWidget.DockWidgetMovable | QDockWidget.DockWidgetFloatable) # type: ignore


        self.upperWidget = QMainWindow()
        self.upperWidget.addDockWidget(Qt.TopDockWidgetArea, dpMasterDockWidget) # type: ignore
        self.upperWidget.addDockWidget(Qt.TopDockWidgetArea, dpSlaveDockWidget) # type: ignore
        self.upperWidget.addDockWidget(Qt.BottomDockWidgetArea, paMasterDockWidget) # type: ignore
        self.upperWidget.addDockWidget(Qt.BottomDockWidgetArea, paSlaveDockWidget) # type: ignore

        self.stackWidget = QStackedWidget()

        self.stackWidget.addWidget(self.upperWidget)
        self.stackWidget.addWidget(QWidget())
        
        self.mainLayout = QVBoxLayout(self)
        self.mainLayout.addLayout(toolbarLayout, 1)
        self.mainLayout.addWidget(self.stackWidget, 50)
        self.setLayout(self.mainLayout)
        # self.setCentralWidget(self.stackWidget)

        self.setWindowIcon(QIcon('Static/Hicent.jpg'))

        self.setWindowTitle("PROFIBUS总线测试工具")

        self.refreshProgressBar()

        self.devicesManange.connect()

        self.setWindowFlags(Qt.FramelessWindowHint)
   
        # self.resize(800, 600)
        # self.showMaximized()

    def startProtocol(self):
        if self.startProtocolBtn.isChecked():
            
            self.startProtocolBtn.setText('停止通讯')
            self.startProtocolBtn.setIcon(QIcon('Static/pause.png'))
            self.startProtocolBtn.setIconSize(QSize(22, 22))
            self.protocolTimer.start(500)

        else:
            self.startProtocolBtn.setText('开始通讯')
            self.startProtocolBtn.setIcon(QIcon('Static/start.png'))
            self.protocolTimer.stop()

    def readValues(self):
        self.devicesManange.readAreas()
        dockWidgets = self.findChildren(QDockWidget) #找到四个dockWidget窗口
        for dockWidget in dockWidgets:
            if dockWidget.widget().currentWidget().objectName() == 'initWidget':
                # print(dockWidget.widget().currentWidget().objectName())
                continue

            areaTabWidget = dockWidget.widget().currentWidget()#判断四个窗口里是否
            if areaTabWidget.currentWidget().objectName() == 'initWidget':
                # print(2)
                continue
            if areaTabWidget.currentWidget().state:
                # print(3)
                continue

            # masterAndSlave = dockWidget.windowTitle()[2:5]
            # proType = dockWidget.windowTitle()
            # dataType = dataTypeCombox.currentText()
            # dataTypeAndModel = masterAndSlave + dataType 
    
            # if dataTypeAndModel in ['主站AI', '主站DI', '从站AO', '从站DO']:
            #     devicecurIndex =.currentIndex()
            #     deviceTabText =.tabText(devicecurIndex)
            # try:
            areacurindex = areaTabWidget.currentIndex()
            #     deviceName = deviceTabText + proType
            buttonlayoutWidget = areaTabWidget.currentWidget().rightAreaWidgets#获取buttonlayout
            buttonlayoutWidget.readValues(curIndex = areacurindex)
            # except Exception as e:
            #     print(e)

    def refreshProgressBar(self):
        # self.temp = temp / 10 if temp > -3276.8 else float('nan')  # 假设温度值需要除以10得到实际℃数
        # self.current = current  # mA
        # self.volt = volt  # mV
        # self.remaintime = remaintime  # 分钟
        # self.capacity = capacity  # mAH
        # self.cappercent = cappercent  # 百分比%
        # self.state = state  # 充放电状态
        result = self.batteryManange.readBatteryInfo()
        self.batteryProBar.setValue(result.cappercent)
        self.batteryStateLabel.setText(result.chargingStatus)

       

    def switchWidget(self):
        if self.switchBtn.isChecked():
            if self.process:
                self.stackWidget.setCurrentIndex(1)
                QTimer.singleShot(50, lambda:self.stackWidget.setCurrentIndex(2))
            else:
                self.stackWidget.setCurrentIndex(1)
                startupInfo = subprocess.STARTUPINFO()
                startupInfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
                startupInfo.wShowWindow = 2
                self.process = subprocess.Popen("D:\\EnTalk PROFIBUS Manager\\DP.exe",startupinfo=startupInfo)
                QTimer.singleShot(500, lambda:self.showLowerWidget(self.process))
            self.switchBtn.setText('变量读写')
            self.showMaximized()
            # self.switchBtn.setIcon(QIcon('Static/varMagH.png'))
        else:
            # self.stackWidget.setCurrentIndex(1)
            self.stackWidget.setCurrentIndex(0)
            self.switchBtn.setText('通讯组态')
            # self.switchBtn.setIcon(QIcon('Static/newH.png'))

     

    def showLowerWidget(self, process):
        pid = process.pid
        hwnd = int(getHwndByPid(pid))
        style = win32gui.GetWindowLong(hwnd, win32con.GWL_STYLE)
        exstyle = win32gui.GetWindowLong(hwnd, win32con.GWL_EXSTYLE)
        wrect = win32gui.GetWindowRect(hwnd)[:2] + win32gui.GetClientRect(hwnd)[2:]
        # print('save', hwnd, style, exstyle, wrect)
        widget = QWidget.createWindowContainer(QWindow.fromWinId(hwnd))
        # print(type(widget))
        widget.hwnd = hwnd  # 窗口句柄
        widget.phwnd = 0  # 父窗口句柄
        widget.style = style  # 窗口样式
        widget.exstyle = exstyle  # 窗口额外样式
        widget.wrect = wrect  # 窗口位置
     
        
        self.stackWidget.addWidget(widget)
        widget.setWindowFlags(Qt.WindowStaysOnTopHint | Qt.FramelessWindowHint | Qt.WindowFullScreen)

        win32gui.SetParent(hwnd, int(self.winId()))
        
        self.stackWidget.setCurrentIndex(2)

    def closeEvent(self, event):
        if self.process:
            self.process.kill()
        self.batteryManange.close()

    def switchTouchMode(self):
        if self.switchTouchBtn.isChecked():
            Globals.setValue('_touchMode', 1)
            self.switchTouchBtn.setText('键盘模式')
        else:
            Globals.setValue('_touchMode', 0)
            self.switchTouchBtn.setText('触控模式')

    def toggleMaximize(self):
        if self.isMaximized():
            self.showNormal()
            self.maxRestoreButton.setIcon(QIcon('maximizeIcon.png'))
        else:
            self.showMaximized()
            self.maxRestoreButton.setIcon(QIcon('restoreIcon.png'))  # 替换为合适的还原图标路径

    

if __name__ == '__main__':
    app = QApplication(sys.argv)
    Client.initDB()
    window = MainWindow()

    window.show()
    sys.exit(app.exec_())