506 lines
21 KiB
Python
506 lines
21 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
Created on Sun Sep 29 09:01:22 2024
|
||
|
||
@author: WANGXIBAO
|
||
"""
|
||
import sys,os
|
||
import re,socket
|
||
|
||
import time,datetime
|
||
|
||
from PyQt5 import QtWidgets
|
||
from PyQt5.QtWidgets import QMessageBox ,QFileDialog,QInputDialog
|
||
from PyQt5.QtCore import QTimer
|
||
from PyNetUi import Ui_UartAssistant
|
||
|
||
from configparser import ConfigParser
|
||
|
||
class Pyqt5Net(QtWidgets.QWidget,Ui_UartAssistant):
|
||
# %%初始化程序
|
||
def __init__(self):
|
||
super(Pyqt5Net, self).__init__()
|
||
self.setupUi(self)
|
||
self.quick_num = 99
|
||
self.init() #信号和槽
|
||
self.IniPath = "PyNet.ini"
|
||
self.log_time = 10
|
||
self.CheckCfgIniData() #更改按钮的文字
|
||
self.log_count = 0
|
||
self.log_buffer = {} #定义成一个字典，键--文件名，值--对于收到的字符串列表
|
||
|
||
# 设置Logo和标题
|
||
#self.setWindowIcon(QIcon('favicon.ico'))
|
||
self.setWindowTitle("网口调试助手")
|
||
# 设置禁止拉伸窗口大小
|
||
#self.setFixedSize(self.width(), self.height())
|
||
|
||
# 发送数据和接收数据数目置零
|
||
self.data_num_sended = 0
|
||
self.lineEditSendNum.setText(str(self.data_num_sended))
|
||
self.data_num_received = 0
|
||
self.lineEditReceiveNum.setText(str(self.data_num_received))
|
||
|
||
# 串口关闭按钮使能关闭
|
||
self.pushButton_closeNet.setEnabled(False)
|
||
|
||
# 发送框、文本框清除
|
||
self.textEditReceive.setText("")
|
||
self.textEditSend.setText("")
|
||
|
||
self.get_all_local_ips()
|
||
|
||
# 定时器接收数据
|
||
self.timer = QTimer()
|
||
self.timer.timeout.connect(self.data_receive)
|
||
#加载快捷指令
|
||
self.widget_6.hide()
|
||
#加载快捷指令的按键值
|
||
|
||
|
||
#建立信号与槽
|
||
def init(self):
|
||
self.pushButton_openNet.clicked.connect(self.open_net)
|
||
self.pushButton_closeNet.clicked.connect(self.close_net)
|
||
self.pushButtonSend.clicked.connect(lambda:self.data_send(text_quick= None))
|
||
# 清除发送按钮
|
||
self.pushButtonClearSend.clicked.connect(self.send_data_clear)
|
||
# 清除接收按钮
|
||
self.pushButtonClearReceive.clicked.connect(self.receive_data_clear)
|
||
|
||
# 定时发送数据
|
||
self.timer_send = QTimer()
|
||
self.timer_send.timeout.connect(self.data_send)
|
||
self.checkBoxReapitSend.stateChanged.connect(self.data_send_timer)
|
||
# 快捷指令扩展区域
|
||
self.pushButton_expend.clicked.connect(self.adjust_sidebar)
|
||
# 清空网络通道列表
|
||
self.pushButtonClearChannel.clicked.connect(self.channel_clear)
|
||
# 动态创建控件并存储引用
|
||
for i in range(self.quick_num): # 从0到20
|
||
index_str = f"{i:02}" # 确保编号为两位数字形式
|
||
horizontalLayoutName = f"horizontalLayoutQuick_{index_str}"
|
||
horizontalLayout = QtWidgets.QHBoxLayout()
|
||
horizontalLayout.setObjectName(horizontalLayoutName)
|
||
|
||
# 创建 QLineEdit 并设置动态属性
|
||
lineEditName = f"lineEditQuick_{index_str}"
|
||
setattr(self, lineEditName, QtWidgets.QLineEdit(self.layoutWidget1))
|
||
getattr(self, lineEditName).setObjectName(lineEditName)
|
||
horizontalLayout.addWidget(getattr(self, lineEditName))
|
||
# 创建 QPushButton 并设置动态属性
|
||
buttonName = f"pushButtonQuick_{index_str}"
|
||
setattr(self, buttonName, QtWidgets.QPushButton(self.layoutWidget1))
|
||
getattr(self, buttonName).setObjectName(buttonName)
|
||
horizontalLayout.addWidget(getattr(self, buttonName))
|
||
self.verticalLayout_8.addLayout(horizontalLayout)
|
||
# 连接按钮点击事件到槽函数
|
||
button = getattr(self, buttonName)
|
||
if button:
|
||
button.clicked.connect(lambda checked, idx=i: self.onButtonClick(idx))
|
||
self.double_click_timers = {} # 存储双击定时器
|
||
|
||
|
||
#%% 重写关闭按钮
|
||
def closeEvent(self, event):
|
||
|
||
|
||
#关闭界面前保存快捷区域的命令和名称
|
||
for i in range(self.quick_num): # 假设有20个按钮
|
||
index_str = f"{i:02}" # 确保编号为两位数字形式
|
||
lineEditName = f"lineEditQuick_{index_str}"
|
||
buttonName = f"pushButtonQuick_{index_str}" # 格式化按钮名称，确保两位数
|
||
button_name = f"button{index_str}"
|
||
set_text = getattr(self,lineEditName).text()+ "|" + getattr(self,buttonName).text()
|
||
self.SetCfgIniData(button_name, set_text)
|
||
self.close_net()
|
||
|
||
# 调用父类的关闭事件处理函数
|
||
super().closeEvent(event)
|
||
|
||
|
||
#加载本地网络地址
|
||
def get_all_local_ips(self):
|
||
ip_addresses = []
|
||
for interface in socket.getaddrinfo(socket.gethostname(), None):
|
||
address = interface[4][0]
|
||
if ':' not in address: # IPv6地址包含冒号
|
||
ip_addresses.append(address)
|
||
print(ip_addresses)
|
||
for ip in ip_addresses:
|
||
self.comboBox_localAddr.insertItem(0,ip)
|
||
self.comboBox_localAddr.setCurrentIndex(0)
|
||
|
||
# 打开网络连接
|
||
def open_net(self):
|
||
IP = self.comboBox_localAddr.currentText()
|
||
port = self.lineEdit_port.text()
|
||
udp_addr = (IP, int(port))
|
||
self.udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #创建接收空socket
|
||
print("绑定地址", udp_addr)
|
||
try:
|
||
self.udp_socket.bind(udp_addr)
|
||
print("开始监听。。。")
|
||
# 设置套接字为非阻塞模式
|
||
self.udp_socket.setblocking(False)
|
||
except Exception as e:
|
||
print(f"Error reading configuration: {e}")
|
||
return None
|
||
|
||
|
||
#使能相关按钮
|
||
self.pushButton_closeNet.setEnabled(True)
|
||
self.pushButton_openNet.setEnabled(False)
|
||
|
||
|
||
# 打开串口接收定时器，周期为1ms
|
||
self.timer.start(20)
|
||
|
||
|
||
def close_net(self):
|
||
if hasattr(self, 'udp_socket'):
|
||
self.timer.stop()
|
||
self.udp_socket.close()
|
||
#使能相关按钮
|
||
self.pushButton_closeNet.setEnabled(False)
|
||
self.pushButton_openNet.setEnabled(True)
|
||
print("关闭网络连接")
|
||
#接收数据
|
||
def data_receive(self):
|
||
try:
|
||
recv_data = self.udp_socket.recvfrom(1024) # 1024表示本次接收的最大字节数
|
||
out_s = ''
|
||
# 打印接收到的数据
|
||
if self.checkBoxHexReceive.checkState():
|
||
line_utf8 = recv_data[0].hex()
|
||
else:
|
||
line_utf8 = recv_data[0].decode('utf-8',errors='replace')
|
||
|
||
print("[From %s:%d]:%s" % (recv_data[1][0], recv_data[1][1], line_utf8))
|
||
|
||
num = len(line_utf8)
|
||
# 接收窗口中增加来源地址显示
|
||
recv_addr = recv_data[1][0]+':'+str(recv_data[1][1])
|
||
if not self.is_in_comboBox_channel(recv_addr):
|
||
self.comboBox_channel.addItem(recv_addr)
|
||
|
||
if recv_addr==self.comboBox_channel.currentText() or self.comboBox_channel.currentText()=="ALL":
|
||
#self.textEditReceive.insertPlainText('<' + recv_addr+ '> ' )
|
||
|
||
if self.checkBoxAddDate.isChecked():
|
||
nowTime = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")
|
||
nowTime = nowTime[:-3]
|
||
#self.textEditReceive.insertPlainText(nowTime + " ")
|
||
self.add_line_to_textedit(nowTime + " ")
|
||
# HEX显示数据
|
||
if self.checkBoxHexReceive.checkState():
|
||
# line_utf8_bytes = line_utf8.hex() # 编码为 UTF-8 字节串
|
||
|
||
# for byte in line_utf8_bytes:
|
||
# out_s += '{:02X} '.format(byte) # 使用大写字母 X
|
||
|
||
out_s = ' '.join(['{:02X}'.format(b) for b in bytes.fromhex(line_utf8)])
|
||
|
||
# self.textEditReceive.insertPlainText(out_s)
|
||
# self.textEditReceive.insertPlainText('\r\n')
|
||
self.add_line_to_textedit('<' + recv_addr+ '> ' + out_s)
|
||
# ASCII显示数据
|
||
else:
|
||
#print("解码前",datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f"))
|
||
#self.textEditReceive.insertPlainText(line_utf8)
|
||
self.add_line_to_textedit('<' + recv_addr+ '> ' + line_utf8)
|
||
#print("解码数据",datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f"))
|
||
# 接收换行
|
||
if self.checkBoxCRLF.isChecked():
|
||
self.textEditReceive.insertPlainText('\r\n')
|
||
|
||
|
||
|
||
|
||
# 获取到text光标
|
||
textCursor = self.textEditReceive.textCursor()
|
||
# 滚动到底部
|
||
textCursor.movePosition(textCursor.End)
|
||
# 设置光标到text中去
|
||
self.textEditReceive.setTextCursor(textCursor)
|
||
|
||
# 统计接收字符的数量
|
||
self.data_num_received += num
|
||
self.lineEditReceiveNum.setText(str(self.data_num_received))
|
||
|
||
# 自动保存日志
|
||
if self.checkBoxAutoSaveLog.isChecked():
|
||
line = line_utf8.split(',')
|
||
if len(line) ==2:
|
||
sav_name = line[0] +'_'+ datetime.datetime.now().strftime("%Y-%m-%d")+'.txt'
|
||
sav_str = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%f")[:23] +" " +line[1]
|
||
|
||
else:
|
||
sav_name = datetime.datetime.now().strftime("%Y-%m-%d")+'.log'
|
||
if self.checkBoxHexReceive.checkState():
|
||
sav_str = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%f")[:23] +" " +out_s +"\r\n"
|
||
else:
|
||
sav_str = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%f")[:23] +" " +line_utf8
|
||
|
||
|
||
# 将日志信息添加到缓冲区
|
||
if sav_name not in self.log_buffer:
|
||
self.log_buffer[sav_name] = []
|
||
self.log_buffer[sav_name].append(sav_str)
|
||
|
||
# 当缓冲区中的记录达到10条时，写入文件
|
||
if len(self.log_buffer[sav_name]) >= self.log_time:
|
||
with open(sav_name, mode='a', newline='',encoding='utf-8', errors='replace') as file:
|
||
file.writelines(self.log_buffer[sav_name])
|
||
self.log_buffer[sav_name].clear()
|
||
|
||
except socket.error as e:
|
||
#print(f"Socket error: {e}")
|
||
pass
|
||
|
||
def data_send(self,text_quick = None):
|
||
if text_quick== None:
|
||
input_s = self.textEditSend.toPlainText()
|
||
else:
|
||
input_s = text_quick
|
||
|
||
try:
|
||
|
||
if input_s != "":
|
||
# 时间显示
|
||
if self.checkBoxAddDate.isChecked():
|
||
self.textEditReceive.insertPlainText((time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())) + " ")
|
||
|
||
# HEX发送
|
||
if self.checkBoxHexSend.isChecked():
|
||
input_s = input_s.strip()
|
||
send_list = []
|
||
while input_s != '':
|
||
try:
|
||
num = int(input_s[0:2], 16)
|
||
except ValueError:
|
||
QMessageBox.critical(self, '串口异常', '请输入规范十六进制数据，以空格分开！')
|
||
return None
|
||
|
||
input_s = input_s[2:].strip()
|
||
send_list.append(num)
|
||
|
||
input_s = bytes(send_list)
|
||
# ASCII发送
|
||
else:
|
||
input_s = (input_s).encode('utf-8')
|
||
input_s = re.sub(b'(?<!\r)\n', b'\r\n', input_s)
|
||
|
||
# HEX接收显示
|
||
if self.checkBoxHexReceive.isChecked():
|
||
out_s = ''
|
||
for i in range(0, len(input_s)):
|
||
out_s = out_s + '{:02X}'.format(input_s[i]) + ' '
|
||
print(out_s)
|
||
self.textEditReceive.insertPlainText(out_s)
|
||
# ASCII接收显示
|
||
else:
|
||
self.textEditReceive.insertPlainText(input_s.decode('utf-8'))
|
||
|
||
# 接收换行
|
||
if self.checkBoxCRLF.isChecked():
|
||
self.textEditReceive.insertPlainText('\r\n')
|
||
|
||
# 获取到Text光标
|
||
textCursor = self.textEditReceive.textCursor()
|
||
# 滚动到底部
|
||
textCursor.movePosition(textCursor.End)
|
||
# 设置光标到Text中去
|
||
self.textEditReceive.setTextCursor(textCursor)
|
||
|
||
#定义服务器的IP地址和端口号
|
||
sever = self.comboBox_channel.currentText().split(':')
|
||
server_ip = sever[0]
|
||
server_port = int(sever[1])
|
||
server_address = (server_ip, server_port)
|
||
# 将消息发送到服务器
|
||
num = self.udp_socket.sendto(input_s, server_address)
|
||
|
||
# 统计发送字符数量
|
||
self.data_num_sended += num
|
||
self.lineEditSendNum.setText(str(self.data_num_sended))
|
||
|
||
except:
|
||
pass
|
||
|
||
# %%定时发送数据
|
||
def data_send_timer(self):
|
||
try:
|
||
if 1<= int(self.lineEditTime.text()) <= 30000: # 定时时间1ms~30s内
|
||
if self.checkBoxReapitSend.isChecked():
|
||
self.timer_send.start(int(self.lineEditTime.text()))
|
||
self.lineEditTime.setEnabled(False)
|
||
else:
|
||
self.timer_send.stop()
|
||
self.lineEditTime.setEnabled(True)
|
||
else:
|
||
QMessageBox.critical(self, '定时发送数据异常', '定时发送数据周期仅可设置在30秒内！')
|
||
except:
|
||
QMessageBox.critical(self, '定时发送数据异常', '请设置正确的数值类型！')
|
||
|
||
# %%保存日志
|
||
def savefiles(self):
|
||
dlg = QFileDialog()
|
||
filenames = dlg.getSaveFileName(None, "保存日志文件", None, "Txt files(*.txt)")
|
||
|
||
try:
|
||
with open(file = filenames[0], mode='w', encoding='utf-8') as file:
|
||
file.write(self.textEditReceive.toPlainText())
|
||
except:
|
||
QMessageBox.critical(self, '日志异常', '保存日志文件失败！')
|
||
|
||
#%% 加载日志
|
||
def openfiles(self):
|
||
dlg = QFileDialog()
|
||
filenames = dlg.getOpenFileName(None, "加载日志文件", None, "Txt files(*.txt)")
|
||
|
||
try:
|
||
with open(file = filenames[0], mode='r', encoding='utf-8') as file:
|
||
self.textEditSend.setPlainText(file.read())
|
||
except:
|
||
QMessageBox.critical(self, '日志异常', '加载日志文件失败！')
|
||
#%% 清除计数
|
||
def send_data_clear(self):
|
||
self.textEditSend.setText("")
|
||
|
||
self.data_num_sended = 0
|
||
self.lineEditSendNum.setText(str(self.data_num_sended))
|
||
|
||
# 清除接收数据显示
|
||
def receive_data_clear(self):
|
||
self.textEditReceive.setText("")
|
||
|
||
self.data_num_received = 0
|
||
self.lineEditSendNum.setText(str(self.data_num_received))
|
||
|
||
#设置接受区域显示条数
|
||
def add_line_to_textedit(self, new_line):
|
||
# 获取当前文本并按行分割
|
||
current_lines = self.textEditReceive.toPlainText().splitlines()
|
||
# 添加新的行到列表中
|
||
current_lines.append(new_line)
|
||
# 如果行数超过100，则删除最早的行，直到剩下100行
|
||
if len(current_lines) > 1000:
|
||
current_lines = current_lines[-1000:] # 只保留最新的100行
|
||
# 将更新后的行列表转换回字符串并设置为textEdit的内容
|
||
self.textEditReceive.setPlainText('\n'.join(current_lines))
|
||
|
||
#检查是否在comboBox_channel中
|
||
def is_in_comboBox_channel(self,addr_rece):
|
||
for i in range(self.comboBox_channel.count()):
|
||
if addr_rece == self.comboBox_channel.itemText(i):
|
||
return True
|
||
return False
|
||
|
||
def channel_clear(self):
|
||
self.comboBox_channel.lineEdit().clear()
|
||
for i in range(self.comboBox_channel.count()-1):
|
||
self.comboBox_channel.removeItem(1)
|
||
self.comboBox_channel.setCurrentIndex(0)
|
||
|
||
#开关快捷指令栏
|
||
def adjust_sidebar(self):
|
||
if self.widget_6.isHidden():
|
||
self.widget_6.show()
|
||
else:
|
||
self.widget_6.hide()
|
||
|
||
def onPushButtonQuickClicked(self, line_edit):
|
||
text = getattr(self, line_edit).text()
|
||
#print(f"Button clicked: {text}")
|
||
if self.checkBox_return.isChecked():
|
||
text = text + "\r\n"
|
||
self.data_send(text)
|
||
|
||
def CheckCfgIniData(self):
|
||
if not os.path.exists(self.IniPath):
|
||
config = ConfigParser()
|
||
config.add_section('Quick_config')
|
||
config.set('Quick_config', 'log_time', '10')
|
||
for i in range(self.quick_num):
|
||
idx = f'{i:02}'
|
||
button_name = f'Button{idx}' # 格式化按钮名称，确保两位数
|
||
config.set('Quick_config', button_name, '')
|
||
with open(self.IniPath, 'w' ,encoding='utf-8') as f:
|
||
config.write(f)
|
||
|
||
config = ConfigParser()
|
||
config.read(self.IniPath, encoding='utf-8')
|
||
try:
|
||
|
||
# 创建一个空字典来存储按钮名称和对应的配置
|
||
self.log_time = int(config.get('Quick_config', 'log_time'))
|
||
# 循环遍历按钮编号，从0到19
|
||
for i in range(self.quick_num):
|
||
idx = f'{i:02}'
|
||
button_name = f'Button{idx}' # 格式化按钮名称，确保两位数
|
||
# 使用 get 方法安全地获取配置，如果不存在则返回空字符串
|
||
config_value = config.get('Quick_config', button_name, fallback='')
|
||
config_value_split = config_value.split('|')
|
||
# 将按钮名称和配置信息存储在字典中
|
||
if len(config_value_split) == 2:
|
||
# 打印字典查看结果,并赋值
|
||
print(button_name,config_value_split[0],config_value_split[1])
|
||
getattr(self, f'pushButtonQuick_{idx}').setText(config_value_split[1])
|
||
getattr(self, f'lineEditQuick_{idx}').setText(config_value_split[0])
|
||
|
||
except Exception as e:
|
||
print(f"Error reading configuration: {e}")
|
||
|
||
|
||
def SetCfgIniData(self,button_name,set_text):
|
||
config = ConfigParser()
|
||
config.read(self.IniPath, encoding='utf-8')
|
||
config.set('Quick_config', button_name, set_text)
|
||
with open(self.IniPath, 'w' ,encoding='utf-8') as f:
|
||
config.write(f)
|
||
|
||
|
||
def onButtonClick(self, idx):
|
||
# 槽函数处理按钮点击事件
|
||
|
||
if idx not in self.double_click_timers:
|
||
self.double_click_timers[idx] = None
|
||
|
||
if self.double_click_timers[idx] is None:
|
||
self.double_click_timers[idx] = True
|
||
QTimer.singleShot(200, lambda: self.onButtonSigleClick(idx))
|
||
else:
|
||
self.onButtonDoubleClick(idx)
|
||
|
||
def onButtonSigleClick(self, idx):
|
||
index_str = f"{idx:02}"
|
||
if self.double_click_timers[idx]:
|
||
self.double_click_timers[idx] = None
|
||
lineEdit = getattr(self, f"lineEditQuick_{index_str}")
|
||
print(f"Text in lineEdit_{idx}: {lineEdit.text()}")
|
||
text = lineEdit.text()
|
||
if self.checkBox_return.isChecked():
|
||
text = text + "\r\n"
|
||
self.data_send(text)
|
||
|
||
def onButtonDoubleClick(self, idx):
|
||
# 槽函数处理按钮双击事件
|
||
self.double_click_timers[idx] = None
|
||
index_str = f"{idx:02}"
|
||
print(f"Double click detected on Button {index_str}.")
|
||
button = getattr(self, f"pushButtonQuick_{index_str}")
|
||
new_name, ok = QInputDialog.getText(self, 'Button Rename', 'Enter new button name:')
|
||
if ok and new_name:
|
||
button.setText(new_name)
|
||
|
||
|
||
|
||
|
||
|
||
#执行
|
||
if __name__ == '__main__':
|
||
app = QtWidgets.QApplication(sys.argv)
|
||
myshow = Pyqt5Net()
|
||
myshow.show()
|
||
sys.exit(app.exec_()) |