from PyQt5.QtCore import (QCoreApplication, Qt) from PyQt5.QtWidgets import (QComboBox, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QPushButton, QListView, QWidget, QMessageBox) from utils.DBModels.ProtocolModel import RTUSetting class CustomBox(QComboBox): def __init__(self): super(CustomBox, self).__init__() self.setCursor(Qt.PointingHandCursor) self.setView(QListView()) def showPopup(self): QComboBox.showPopup(self) pop = self.children()[1] pop.move(pop.x(), pop.y() + 51) class RTUSettingWidget(QWidget): def __init__(self, rtuType): super(RTUSettingWidget, self).__init__() self.rtuType = rtuType self.setObjectName("RTUSettingWidget") self.setupUI() def setupUI(self): # 主布局 - 垂直布局 main_layout = QVBoxLayout(self) main_layout.setContentsMargins(0, 0, 0, 0) main_layout.setSpacing(7) # 第一行：端口号和波特率 row1_layout = QHBoxLayout() row1_layout.setSpacing(15) # 端口号列 port_column = QVBoxLayout() port_column.setSpacing(4) self.label = QLabel('端口号') self.label.setObjectName("setlabel") self.portEdit = QLineEdit() self.portEdit.setObjectName("setEdit") port_column.addWidget(self.label) port_column.addWidget(self.portEdit) # 波特率列 baud_column = QVBoxLayout() baud_column.setSpacing(4) self.label_5 = QLabel('波特率') self.label_5.setObjectName("setlabel") self.baudrateEdit = QLineEdit() self.baudrateEdit.setObjectName("setEdit") baud_column.addWidget(self.label_5) baud_column.addWidget(self.baudrateEdit) row1_layout.addLayout(port_column) row1_layout.addLayout(baud_column) main_layout.addLayout(row1_layout) # 第二行：字节长度和停止位 row2_layout = QHBoxLayout() row2_layout.setSpacing(15) # 字节长度列 byte_column = QVBoxLayout() byte_column.setSpacing(4) self.label_2 = QLabel('字节长度') self.label_2.setObjectName("setlabel") self.byteSizeBox = CustomBox() self.byteSizeBox.addItem("8") self.byteSizeBox.addItem("7") self.byteSizeBox.setObjectName("setBox") byte_column.addWidget(self.label_2) byte_column.addWidget(self.byteSizeBox) # 停止位列 stop_column = QVBoxLayout() stop_column.setSpacing(4) self.label_4 = QLabel('停止位') self.label_4.setObjectName("setlabel") self.stopbitsBox = CustomBox() self.stopbitsBox.addItem("1") self.stopbitsBox.addItem("2") self.stopbitsBox.setObjectName("setBox") stop_column.addWidget(self.label_4) stop_column.addWidget(self.stopbitsBox) row2_layout.addLayout(byte_column) row2_layout.addLayout(stop_column) main_layout.addLayout(row2_layout) # 第三行：奇偶校验（单独一行） parity_layout = QVBoxLayout() parity_layout.setSpacing(4) self.label_3 = QLabel('奇偶校验') self.label_3.setObjectName("setlabel") self.parityBox = CustomBox() self.parityBox.addItem("N 无") self.parityBox.addItem("O 奇") self.parityBox.addItem("E 偶") self.parityBox.setObjectName("setBox") parity_layout.addWidget(self.label_3) parity_layout.addWidget(self.parityBox) main_layout.addLayout(parity_layout) # 第四行：通信频率和起始值 row4_layout = QHBoxLayout() row4_layout.setSpacing(15) # 通信频率列 freq_column = QVBoxLayout() freq_column.setSpacing(4) self.label_6 = QLabel('通信频率') self.label_6.setObjectName("setlabel") self.freEdit = QLineEdit() self.freEdit.setObjectName("setEdit") freq_column.addWidget(self.label_6) freq_column.addWidget(self.freEdit) # 起始值列 offset_column = QVBoxLayout() offset_column.setSpacing(4) self.label_7 = QLabel('起始值') self.label_7.setObjectName('setlabel') self.offsetBox = CustomBox() self.offsetBox.addItem("1") self.offsetBox.addItem("0") self.offsetBox.setObjectName("setBox") offset_column.addWidget(self.label_7) offset_column.addWidget(self.offsetBox) row4_layout.addLayout(freq_column) row4_layout.addLayout(offset_column) main_layout.addLayout(row4_layout) # 保存按钮 self.saveButton = QPushButton('保存') if self.rtuType == 'master': self.saveButton.setObjectName('rtuSaveButton') else: self.saveButton.setObjectName('setButton') self.saveButton.clicked.connect(self.saveSetting) main_layout.addWidget(self.saveButton) # 添加弹性空间 main_layout.addStretch() def loadSettings(self): """加载设置数据""" try: if self.rtuType == 'master': query = RTUSetting.get_by_id(1) else: query = RTUSetting.get_by_id(2) port = str(query.port) byteSize = str(query.byteSize) baudrate = str(query.baudrate) stopbits = str(query.stopbits) parity = str(query.parity) frequency = str(query.frequency) offset = str(query.offset) self.portEdit.setText(port) self.byteSizeBox.setCurrentText(byteSize) self.baudrateEdit.setText(baudrate) self.stopbitsBox.setCurrentText(stopbits) self.freEdit.setText(frequency) self.offsetBox.setCurrentText(offset) self.parityBox.setCurrentText(parity) except Exception as e: print(f"加载RTU设置失败: {e}") # 设置默认值 self.portEdit.setText("COM1") self.byteSizeBox.setCurrentText("8") self.baudrateEdit.setText("9600") self.stopbitsBox.setCurrentText("1") self.freEdit.setText("1") self.offsetBox.setCurrentText("1") self.parityBox.setCurrentText("N 无") def saveSetting(self): """保存设置""" try: port = self.portEdit.text() byteSize = self.byteSizeBox.currentText() baudrate = self.baudrateEdit.text() stopbits = self.stopbitsBox.currentText() parity = self.parityBox.currentText() frequency = self.freEdit.text() offset = self.offsetBox.currentText() if self.rtuType == 'master': RTUSetting.update(port=port, byteSize=byteSize, baudrate=baudrate, stopbits=stopbits, parity=parity, frequency=frequency, offset=offset).where(RTUSetting.tcpType == 'master').execute() else: RTUSetting.update(port=port, byteSize=byteSize, baudrate=baudrate, stopbits=stopbits, parity=parity, frequency=frequency, offset=offset).where(RTUSetting.tcpType == 'slave').execute() # 创建消息框 QMessageBox.information(self, '保存成功', f'RTU{self.rtuType}站配置已保存！') except Exception as e: QMessageBox.critical(self, '保存失败', f'保存RTU设置时发生错误：{str(e)}')