import sys import csv from PyQt5.QtWidgets import QApplication, QDialog, QVBoxLayout, QHBoxLayout, QTableWidget, QTableWidgetItem, \ QPushButton, QCheckBox, QFileDialog, QMessageBox from PyQt5.QtCore import Qt class BatchOperationDialog(QDialog): def __init__(self): super().__init__() self.setWindowTitle("批量操作节点") self.setGeometry(100, 100, 800, 400) # 初始化UI self.init_ui() def init_ui(self): # 主布局 main_layout = QVBoxLayout() # 创建表格（8列，初始行数可以设置为10） self.table_widget = QTableWidget(10, 8) # 10行8列 self.table_widget.setHorizontalHeaderLabels( ["选择", "IP地址", "MAC地址", "版本号", "通道1名称", "通道1绑定设备", "通道2名称", "通道2绑定设备"] ) # 使通道列可编辑 for row in range(self.table_widget.rowCount()): self.table_widget.setItem(row, 4, QTableWidgetItem()) # 通道1名称 self.table_widget.setItem(row, 5, QTableWidgetItem()) # 通道1绑定设备 self.table_widget.setItem(row, 6, QTableWidgetItem()) # 通道2名称 self.table_widget.setItem(row, 7, QTableWidgetItem()) # 通道2绑定设备 # 添加勾选框到第1列 for row in range(self.table_widget.rowCount()): check_box_item = QTableWidgetItem() check_box_item.setCheckState(0) # 初始状态为未选中 self.table_widget.setItem(row, 0, check_box_item) # 按钮布局 button_layout = QHBoxLayout() # 创建按钮 self.select_all_button = QPushButton("全选") self.select_none_button = QPushButton("全不选") self.get_version_button = QPushButton("获取版本") self.get_mac_button = QPushButton("获取MAC") self.upgrade_button = QPushButton("升级节点") self.load_ips_button = QPushButton("加载IP列表") self.load_all_button = QPushButton("加载完整信息列表") self.save_button = QPushButton("保存表数据") # 按钮连接 self.select_all_button.clicked.connect(self.select_all) self.select_none_button.clicked.connect(self.select_none) self.get_version_button.clicked.connect(self.get_version) self.get_mac_button.clicked.connect(self.get_mac) self.upgrade_button.clicked.connect(self.upgrade_node) self.save_button.clicked.connect(self.save_data) self.load_ips_button.clicked.connect(self.load_ips) self.load_all_button.clicked.connect(self.load_all_info) # 将按钮加入布局 button_layout.addWidget(self.select_all_button) button_layout.addWidget(self.select_none_button) button_layout.addWidget(self.get_version_button) button_layout.addWidget(self.get_mac_button) button_layout.addWidget(self.upgrade_button) button_layout.addWidget(self.load_ips_button) button_layout.addWidget(self.load_all_button) button_layout.addWidget(self.save_button) # 将表格和按钮布局添加到主布局 main_layout.addWidget(self.table_widget) main_layout.addLayout(button_layout) # 设置对话框的布局 self.setLayout(main_layout) # 允许最大化 # self.setWindowState(Qt.WindowMaximized) def select_all(self): for row in range(self.table_widget.rowCount()): item = self.table_widget.item(row, 0) item.setCheckState(2) # 选中 def select_none(self): for row in range(self.table_widget.rowCount()): item = self.table_widget.item(row, 0) item.setCheckState(0) # 未选中 def get_version(self): print("获取版本按钮被点击") def get_mac(self): print("获取MAC按钮被点击") def upgrade_node(self): print("升级节点按钮被点击") def save_data(self): # 打开文件保存对话框 options = QFileDialog.Options() file_path, _ = QFileDialog.getSaveFileName(self, "保存表数据", "", "CSV Files (*.csv);;All Files (*)", options=options) if file_path: # 保存表格数据到CSV文件 with open(file_path, mode='w', newline='', encoding='utf-8') as file: writer = csv.writer(file) # 写入表头 headers = [self.table_widget.horizontalHeaderItem(i).text() for i in range(self.table_widget.columnCount())] writer.writerow(headers) # 写入表格内容 for row in range(self.table_widget.rowCount()): row_data = [self.table_widget.item(row, col).text() if self.table_widget.item(row, col) else "" for col in range(self.table_widget.columnCount())] writer.writerow(row_data) print(f"数据已保存到 {file_path}") def load_ips(self): # 打开文件选择对话框 options = QFileDialog.Options() file_path, _ = QFileDialog.getOpenFileName(self, "选择IP地址文件", "", "Text Files (*.txt);;All Files (*)", options=options) if file_path: # 读取IP地址 with open(file_path, 'r') as file: lines = file.readlines() invalid_ips = [] for i, line in enumerate(lines): ip = line.strip() # 验证IP地址格式 if not self.validate_ip(ip): invalid_ips.append(i + 1) if invalid_ips: # 如果有无效IP，弹出提示框 QMessageBox.warning(self, "错误", f"以下行的IP地址无效：{', '.join(map(str, invalid_ips))}") else: # 将IP地址加载到表格 for i, line in enumerate(lines): ip = line.strip() if i < self.table_widget.rowCount(): self.table_widget.setItem(i, 1, QTableWidgetItem(ip)) # 设置IP地址列 print(f"加载了{len(lines)}个IP地址") def validate_ip(self, ip): # IP地址验证正则表达式 import re pattern = r"^(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$" return bool(re.match(pattern, ip)) def load_all_info(self): # 打开CSV文件选择对话框 options = QFileDialog.Options() file_path, _ = QFileDialog.getOpenFileName(self, "选择CSV文件", "", "CSV Files (*.csv);;All Files (*)", options=options) if file_path: # 读取CSV文件 with open(file_path, mode='r', encoding='utf-8') as file: reader = csv.reader(file) headers = next(reader) # 跳过表头 for row_num, row_data in enumerate(reader): if row_num < self.table_widget.rowCount(): for col_num, cell_data in enumerate(row_data): self.table_widget.setItem(row_num, col_num+1, QTableWidgetItem(cell_data)) print(f"加载了CSV文件中的内容：{file_path}") if __name__ == "__main__": app = QApplication(sys.argv) dialog = BatchOperationDialog() dialog.show() sys.exit(app.exec_())