358 lines
16 KiB
Python
358 lines
16 KiB
Python
import os
|
||
import sys
|
||
import csv
|
||
import socket, struct
|
||
import time
|
||
|
||
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, 1093, 737)
|
||
|
||
# 初始化分页
|
||
self.current_page = 0
|
||
self.rows_per_page = 20
|
||
|
||
# 初始化数据
|
||
self.data = []
|
||
|
||
# 初始化UI
|
||
self.init_ui()
|
||
|
||
def init_ui(self):
|
||
# 主布局
|
||
main_layout = QVBoxLayout()
|
||
|
||
# 创建表格（8列，初始行数可以设置为20）
|
||
self.table_widget = QTableWidget(self.rows_per_page, 8) # 默认显示20行
|
||
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.prev_page_button = QPushButton("上一页")
|
||
self.next_page_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)
|
||
self.prev_page_button.clicked.connect(self.prev_page)
|
||
self.next_page_button.clicked.connect(self.next_page)
|
||
|
||
# 将按钮加入布局
|
||
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)
|
||
button_layout.addWidget(self.prev_page_button)
|
||
button_layout.addWidget(self.next_page_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):
|
||
# 遍历表格的第1列，获取选中的行的IP地址
|
||
for row in range(self.table_widget.rowCount()):
|
||
item = self.table_widget.item(row, 0)
|
||
if item.checkState() == Qt.Checked: # 如果该行被选中
|
||
ip = self.table_widget.item(row, 1).text() # 获取IP地址
|
||
if ip:
|
||
try:
|
||
# 创建TCP连接到IP地址的12345端口
|
||
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
||
s.settimeout(5) # 设置连接超时时间为5秒
|
||
s.connect((ip, 12345))
|
||
|
||
# 发送消息
|
||
message = bytes([0xAA, 0x55, 0xAA, 0x0F, 0x01, 0x00])
|
||
s.sendall(message)
|
||
|
||
# 接收返回的数据（假设返回数据至少有8个字节）
|
||
response = s.recv(1024)
|
||
if len(response) >= 8:
|
||
# 解析第7、8字节作为版本号
|
||
version = f"{response[6]}-{response[7]}"
|
||
# 将版本号写入到第4列（版本号列）
|
||
self.table_widget.setItem(row, 3, QTableWidgetItem(version))
|
||
else:
|
||
print(f"接收到的数据长度不足：{len(response)}，无法提取版本号")
|
||
except Exception as e:
|
||
print(f"与IP {ip} 连接失败: {e}")
|
||
self.table_widget.setItem(row, 3, QTableWidgetItem("连接失败"))
|
||
|
||
def get_mac(self):
|
||
for row in range(self.table_widget.rowCount()):
|
||
item = self.table_widget.item(row, 0)
|
||
if item.checkState() == Qt.Checked: # 如果该行被选中
|
||
ip = self.table_widget.item(row, 1).text() # 获取IP地址
|
||
if ip:
|
||
try:
|
||
# 创建TCP连接到IP地址的12345端口
|
||
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
||
s.settimeout(5) # 设置连接超时时间为5秒
|
||
s.connect((ip, 12345))
|
||
|
||
# 发送消息
|
||
message = bytes([0xAA, 0x55, 0xAA, 0x12, 0x01, 0x00])
|
||
s.sendall(message)
|
||
|
||
# 接收返回的数据（假设返回数据至少有8个字节）
|
||
response = s.recv(1024)
|
||
if len(response) >= 8:
|
||
# 解析第7、8字节作为mac后2字节
|
||
mac = f"50-29-4D-20-{response[6]}-{response[7]}"
|
||
# 将版本号写入到第3列（MAC列）
|
||
self.table_widget.setItem(row, 2, QTableWidgetItem(mac))
|
||
else:
|
||
print(f"接收到的数据长度不足：{len(response)}，无法提取版本号")
|
||
except Exception as e:
|
||
print(f"与IP {ip} 连接失败: {e}")
|
||
self.table_widget.setItem(row, 3, QTableWidgetItem("连接失败"))
|
||
|
||
def calculate_crc(self, data: bytes) -> int:
|
||
"""计算数据的累加和CRC"""
|
||
crc = 0
|
||
for byte in data:
|
||
crc += byte
|
||
return crc & 0xFF # 只保留最低字节
|
||
|
||
def upgrade_node(self):
|
||
"""打开文件选择对话框"""
|
||
file_path, _ = QFileDialog.getOpenFileName(
|
||
self, "选择升级文件", "", "二进制文件 (*.bin);;所有文件 (*)"
|
||
)
|
||
|
||
if file_path:
|
||
file_name = os.path.basename(file_path)
|
||
# self.status_bar.showMessage(f"已选择: {file_name}")
|
||
|
||
try:
|
||
print("Starting upgrade process...")
|
||
if not os.path.exists(file_path):
|
||
raise FileNotFoundError(f"Upgrade file {file_path} not found")
|
||
|
||
with open(file_path, "rb") as f:
|
||
package_data = f.read()
|
||
|
||
if not package_data:
|
||
raise ValueError("Upgrade file is empty")
|
||
|
||
print(f"Read upgrade package, size: {len(package_data)} bytes")
|
||
|
||
upgrade_len = len(package_data)
|
||
crc = self.calculate_crc(package_data)
|
||
upgrade_req = struct.pack("<IB", upgrade_len, crc) + package_data
|
||
|
||
header_magic = bytes([0xAA, 0x55, 0xAA])
|
||
cmd = 0x05
|
||
version = 1
|
||
result_code = 0
|
||
|
||
package_head = struct.pack("<3sBBB", header_magic, cmd, version, result_code)
|
||
full_packet = package_head + upgrade_req
|
||
|
||
print("Full packet prepared:")
|
||
print(f"Header magic: {package_head[:3].hex()}")
|
||
print(f"Command: {cmd}, Version: {version}")
|
||
print(f"Upgrade length: {upgrade_len}, CRC: {crc}")
|
||
print(f"Total packet size: {len(full_packet)} bytes")
|
||
|
||
# 分包处理：每包最大1500字节
|
||
max_packet_size = 1500
|
||
total_size = len(full_packet)
|
||
num_packets = (total_size // max_packet_size) + (1 if total_size % max_packet_size != 0 else 0)
|
||
|
||
print(f"Total packets: {num_packets}, each up to {max_packet_size} bytes")
|
||
|
||
for row in range(self.table_widget.rowCount()):
|
||
item = self.table_widget.item(row, 0)
|
||
if item.checkState() == Qt.Checked: # 如果该行被选中
|
||
ip = self.table_widget.item(row, 1).text() # 获取IP地址
|
||
if ip:
|
||
try:
|
||
# 创建TCP连接到IP地址的12345端口
|
||
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
||
s.settimeout(5) # 设置连接超时时间为5秒
|
||
s.connect((ip, 12345))
|
||
|
||
for i in range(num_packets):
|
||
start_index = i * max_packet_size
|
||
end_index = min((i + 1) * max_packet_size, total_size)
|
||
packet_chunk = full_packet[start_index:end_index]
|
||
|
||
print(f"Sending packet {i + 1}/{num_packets}, size: {len(packet_chunk)} bytes")
|
||
s.sendall(packet_chunk)
|
||
print(f"Packet {i + 1} sent successfully")
|
||
time.sleep(0.03)
|
||
except Exception as e:
|
||
print(f"与IP {ip} 连接失败: {e}")
|
||
# self.table_widget.setItem(row, 3, QTableWidgetItem("连接失败"))
|
||
QMessageBox.information(self, "升级", "升级全部完成")
|
||
except Exception as e:
|
||
print(f"Upgrade failed: {str(e)}")
|
||
QMessageBox.critical(
|
||
self,
|
||
"升级",
|
||
"升级包上传失败"
|
||
)
|
||
|
||
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[1:])
|
||
# 写入表格内容
|
||
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(1, self.table_widget.columnCount())] # Start from col=1 to skip the first column
|
||
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地址加载到表格
|
||
self.table_widget.setRowCount(len(lines))
|
||
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) # 跳过表头
|
||
self.data = [row for row in reader]
|
||
if len(self.data) <= 20:
|
||
self.table_widget.setRowCount(len(self.data))
|
||
# 确保加载数据后更新分页
|
||
self.current_page = 0 # 每次加载新数据，默认回到第一页
|
||
self.update_table()
|
||
|
||
def update_table(self):
|
||
# 获取当前页的数据
|
||
start_row = self.current_page * self.rows_per_page
|
||
end_row = start_row + self.rows_per_page
|
||
rows_to_display = self.data[start_row:end_row]
|
||
|
||
# 更新表格内容
|
||
for row_num, row_data in enumerate(rows_to_display):
|
||
for col_num, cell_data in enumerate(row_data):
|
||
self.table_widget.setItem(row_num, col_num + 1, QTableWidgetItem(cell_data)) # +1跳过“选择”列
|
||
|
||
# 根据数据量调整上一页和下一页按钮的状态
|
||
self.prev_page_button.setEnabled(self.current_page > 0)
|
||
self.next_page_button.setEnabled((self.current_page + 1) * self.rows_per_page < len(self.data))
|
||
|
||
def prev_page(self):
|
||
if self.current_page > 0:
|
||
self.current_page -= 1
|
||
self.update_table()
|
||
|
||
def next_page(self):
|
||
if (self.current_page + 1) * self.rows_per_page < len(self.data):
|
||
self.current_page += 1
|
||
self.update_table()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app = QApplication(sys.argv)
|
||
dialog = BatchOperationDialog()
|
||
dialog.show()
|
||
sys.exit(app.exec_())
|