T1L_Config/batch_dialog.py

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_())