forked from nicedayzhu/netAssist
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tcp_udp_ui.py
337 lines (310 loc) · 12.2 KB
/
tcp_udp_ui.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2019/1/10 16:00
# @Author : SeniorZhu1994
# @Site :
# @File : tcp_udp_ui.py
# @Software: PyCharm
from PyQt5 import QtGui, QtCore, QtWidgets
from PyQt5.QtCore import pyqtSignal
from netAssitui import Ui_NetAssist
from time import ctime
from PyQt5.QtWidgets import QFileDialog, QMessageBox, QDialog
import binascii
class Tcp_ucpUi(Ui_NetAssist):
# 主线程属性继承自Ui_NetAssist
# 信号槽机制：设置一个信号，用于触发接收区写入动作
signal_write_msg = pyqtSignal(str)
signal_status_connected = pyqtSignal(str)
signal_status_removed = pyqtSignal(str)
signal_add_clientstatus_info = pyqtSignal(str)
signal_messagebox_info = pyqtSignal(str)
# statusbar上添加的控件
# 使用字典方式进行管理
statusbar_dict = {}
rx_count = 0
tx_count = 0
# statusbar End
tail_ok = False
def __init__(self):
super(Tcp_ucpUi, self).__init__()
def custom_connect(self):
"""
控件信号-槽的设置
:param : QDialog类创建的对象
:return: None
"""
self.signal_write_msg.connect(self.write_msg)
self.signal_status_connected.connect(self.statusbar_connect)
self.signal_status_removed.connect(self.statusbar_remove)
self.signal_add_clientstatus_info.connect(self.add_clientstatus_plain)
self.signal_messagebox_info.connect(self.messagebox_info)
def send_fileload(self):
"""
载入文件功能
:return:
"""
if self.file_load.isChecked():
# 载入发送文件
send_file_name, sf_ok = QFileDialog.getOpenFileName(
self, u'保存文件', './', u'所有文件(*.*)')
if sf_ok:
self.statusbar.showMessage('文件载入成功', msecs=3000)
with open(send_file_name, 'rb') as send_f:
self.f_data = send_f.read()
print(self.f_data)
else:
self.file_load.setChecked(False)
self.statusbar.showMessage('文件载入失败', msecs=3000)
def add_clientstatus_plain(self, info):
# signal_add_clientstatus_info信号会触发本函数
"""
向接收框发送客户端连接信息
:param info:
:return:
"""
self.DataRecvtext.insertPlainText(info)
def messagebox_info(self, info):
# signal_messagebox_info信号会触发本函数
"""
弹出消息框
:param info:
:return:
"""
QMessageBox.critical(self, '错误', info)
def write_msg(self, msg):
# signal_write_msg信号会触发这个函数
"""
功能函数，向接收区写入数据的方法
信号-槽触发
tip：PyQt程序的子线程中，直接向主线程的界面传输字符是不符合安全原则的
:return: None
"""
# 为接收到的数据加上时间戳并且显示在接收框中
if self.timestamp.isChecked():
if self.newline.isChecked():
self.DataRecvtext.insertPlainText('[%s]\n' % ctime())
self.DataRecvtext.insertPlainText('%s' % msg)
else:
self.DataRecvtext.insertPlainText('[%s]' % ctime())
self.DataRecvtext.insertPlainText('%s' % msg)
else:
if self.newline.isChecked():
self.DataRecvtext.insertPlainText('%s\n' % msg)
else:
self.DataRecvtext.insertPlainText('%s' % msg)
# 滚动条移动到结尾
self.DataRecvtext.moveCursor(QtGui.QTextCursor.End)
def comboBox_removeItem_byName(self, combo, name):
'''QComboBox中删除特定名字的项目'''
for i in range(0, combo.count()):
if name == combo.itemText(i):
# 找到对应的项目
combo.removeItem(i)
def statusbar_connect(self, statusbar_client_info):
# signal_messagebox_info信号会触发本函数
self.statusbar.showMessage(
'客户端：%s 成功连接！' %
statusbar_client_info, msecs=2000)
def statusbar_remove(self, statusbar_client_info):
# signal_status_removed信号会触发本函数
self.statusbar.showMessage(
'客户端：%s 断开连接！' %
statusbar_client_info, msecs=2000)
def str_to_hex(self, s):
"""
字符串转16进制显示
:param s:
:return:
"""
return ' '.join([hex(ord(c)).replace('0x', '') for c in s])
def hex_to_str(self, s):
"""
16进制转字符串显示
:param s:
:return:
"""
return ''.join([chr(i) for i in [int(b, 16) for b in s.split(' ')]])
def str_to_bin(self, s):
"""
字符串转二进制显示
:param s:
:return:
"""
return ' '.join([bin(ord(c)).replace('0b', '') for c in s])
def bin_to_str(self, s):
"""
二进制转字符串显示
:param s:
:return:
"""
return ''.join([chr(i) for i in [int(b, 2) for b in s.split(' ')]])
def hex_show(self, str):
"""
将字符串转换为大写字母并每隔2个字符用空格分割处理后得到一个新字符串
如：faa5fbb5fcc5fdd5010200000028000001900000000a002d00000000017d7840000003e800005fa55fb55fc55fd5
FA A5 FB B5 FC C5 FD D5 01 02 00 00 00 28 00 00 01 90 00 00 00 0A 00 2D 00 00 00 00 01 7D 78 40 00 00 03 E8 00 00 5F A5 5F B5 5F C5 5F D5
:param str:
:return:
"""
t = str.upper()
return ' '.join([t[2 * i:2 * (i + 1)] for i in range(len(t) // 2)])
# / 是精确除法， // 是向下取整除法， % 是求模
def if_hex_send(self, pre_msg):
"""
判断是否以16进制发送并处理
:param pre_msg:
:return:
"""
try:
if self.hex_send.isChecked():
send_msg = pre_msg.replace(' ', '') # 删除无效的空格
if len(send_msg) % 2 != 0:
# 十六进制发送输入的长度必须是2的倍数
raise Exception('十六进制输入的长度必须是2的倍数')
send_msg = binascii.a2b_hex(send_msg)
else:
send_msg = pre_msg.encode('utf-8')
return send_msg
except binascii.Error as e:
QMessageBox.critical(self, '错误', '十六进制数中包含非法字符！')
except Exception as e:
QMessageBox.critical(self, '错误', '%s' % e)
def if_hex_show_tcpc_udp(self, pre_msg):
"""
判断是否以16进制显示并处理
:param pre_msg:
:return:
"""
if self.hex_recv.isChecked():
msg = binascii.b2a_hex(pre_msg).decode('utf-8')
print(msg, type(msg), len(msg)) # msg为 str 类型
msg = self.hex_show(msg) # 将解码后的16进制数据按照两个字符+'空字符'发送到接收框中显示
self.signal_write_msg.emit(msg)
else:
try:
# 尝试对接收到的数据解码，如果解码成功，即使解码后的数据是ascii可显示字符也直接发送，
msg = pre_msg.decode('utf-8')
print(msg)
self.signal_write_msg.emit(msg)
except Exception as ret:
# 如果出现解码错误，提示用户选中16进制显示
self.signal_messagebox_info.emit('解码错误，请尝试16进制显示')
def checksend_choose(self):
'''
勾选添加附加位toggle之后的动作
:return:
'''
if self.Sendcheck.isChecked():
self.checkDialog = QDialog()
self.checkDialog.resize(381, 200)
self.checkDialog.setMinimumSize(QtCore.QSize(381, 200))
self.checkDialog.setMaximumSize(QtCore.QSize(381, 200))
self.checkDialog.setWindowTitle('附加位设置')
self.widget = QtWidgets.QWidget(self.checkDialog)
self.widget.setGeometry(QtCore.QRect(90, 160, 195, 30))
self.horizontalLayout = QtWidgets.QHBoxLayout(self.widget)
self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
self.okBtn = QtWidgets.QPushButton(self.widget)
self.okBtn.setText('确定')
self.horizontalLayout.addWidget(self.okBtn)
self.cancelBtn = QtWidgets.QPushButton(self.widget)
self.cancelBtn.setText('取消')
self.horizontalLayout.addWidget(self.cancelBtn)
self.groupBox = QtWidgets.QGroupBox(self.checkDialog)
self.groupBox.setGeometry(QtCore.QRect(10, 10, 361, 141))
self.rBtn1 = QtWidgets.QRadioButton(self.groupBox)
self.rBtn1.setGeometry(QtCore.QRect(12, 27, 72, 19))
self.rBtn2 = QtWidgets.QRadioButton(self.groupBox)
self.rBtn2.setGeometry(QtCore.QRect(12, 53, 72, 19))
self.rBtn3 = QtWidgets.QRadioButton(self.groupBox)
self.rBtn3.setGeometry(QtCore.QRect(12, 80, 151, 19))
self.lineEdit = QtWidgets.QLineEdit(self.groupBox)
self.lineEdit.setGeometry(QtCore.QRect(170, 80, 41, 21))
self.lineEdit.setText('0d')
self.groupBox.setTitle("附加位设置")
self.rBtn1.setText("方法一")
self.rBtn2.setText("方法二")
self.rBtn3.setText("固定位 (16进制)")
self.okBtn.clicked.connect(self.ok)
self.cancelBtn.clicked.connect(self.cancel)
self.rBtn1.toggled.connect(self.settail_1)
self.rBtn2.toggled.connect(self.settail_2)
self.rBtn3.toggled.connect(self.settail_3)
self.checkDialog.exec_()
def ok(self):
'''
“确认”按钮按下后的动作
:return:
'''
if self.rBtn1.isChecked() or self.rBtn2.isChecked() or self.rBtn3.isChecked():
print('确定')
self.checkDialog.close()
else:
print('并未选择任何方法')
self.Sendcheck.setChecked(0)
self.checkDialog.close()
def cancel(self):
'''
“取消”按钮按下后的动作
:return:
'''
print('取消')
self.Sendcheck.setChecked(0)
self.checkDialog.close()
def settail_1(self):
'''
勾选方法一
:return:
'''
if self.rBtn1.isChecked():
# 添加附加位标志位的RadioButton勾选状态为Tue
self.tail_ok = True
# 添加附加位，当前设置为'f1'
self.append_tail = 'f1'
print('rBtn1 checked')
print(self.tail_ok)
def settail_2(self):
'''
勾选方法二
:return:
'''
if self.rBtn2.isChecked():
# 添加附加位标志位的RadioButton勾选状态为Tue
self.tail_ok = True
# 添加附加位，当前设置为'f2'
self.append_tail = 'f2'
print('rBtn2 checked')
print(self.tail_ok)
def settail_3(self):
'''
勾选方法三（固定位）
:return:
'''
if self.rBtn3.isChecked():
# 添加附加位标志位的RadioButton勾选状态为Tue
self.tail_ok = True
# 添加附加位，由用户自行添加
self.append_tail = self.lineEdit.text()
print('rBtn3 checked')
print(self.tail_ok)
def is_sendcheck_send(self, get_msg):
'''
判断是否进行附加位发送
:param get_msg:
:return:
'''
# 判断自动发送附加位combobox是否按下
if self.Sendcheck.isChecked():
# 判断附加位设置Dialog中的RadioButton是否选中
if self.tail_ok:
get_msg_c = get_msg + self.append_tail
print(get_msg_c)
# 返回加入附加位的新内容
return get_msg_c
else:
# 如果RadioButton未选中，返回初始消息
return get_msg
else:
# 如果自动发送附加位combobox未按下，返回初始消息
return get_msg