Python串列埠通訊助手
阿新 • • 發佈：2020-10-09
1 # -*- coding: utf-8 -*- 2 import sys 3 import serial 4 import serial.tools.list_ports 5 import threading 6 import signal 7 import binascii 8 9 #import strings_zh as strings 10 APP_NAME = '串列埠除錯助手' 11 APP_VER = '1.0' 12 AUTHOR = 'txqdm' 13 14 recv_buf_text = "接收緩衝區" 15 text_mode_text = "文字模式" 16 hex_mode_text = "16進位制模式" 17 clear_recv_text = "清空接收" 18 send_buf_text = "傳送緩衝區" 19 line_break_text = "換行符:'\\r\\n'" 20 clear_send_text = "清空傳送" 21 send_btn_text = "傳送資料" 22 port_lbl_text = "串列埠:" 23 baud_lbl_text = "波特率:" 24 size_lbl_text = "資料位:" 25 parity_lbl_text = "校驗位:" 26 stopbit_lbl_text = "停止位:" 27 send_lbl_text = "SEND" 28 recv_lbl_text = "RECV" 29 clear_btn_text = "清除" 30 opening = "已開啟" 31 closed = "已關閉" 32 select_port = "請選擇串列埠" 33 start_port_text = "開啟串列埠" 34 close_port_text = "關閉串列埠" 35 36 if sys.version_info.major == 2: 37 import Tkinter as tk 38 import tkMessageBox as msgbox39 elif sys.version_info.major == 3: 40 import tkinter as tk 41 from tkinter import messagebox as msgbox 42 43 44 WINDOWSIZE = '710x510+50+50' 45 # Common baudrate 46 BAUDRATES = (1200, 2400, 4800, 9600, 14400, 19200, 38400, 57600, 115200) 47 BYTESIZES = (5, 6, 7, 8) 48 PARITIES = {'None': 'N', 'Even': 'E', 'Odd': 'O', 'Mark': 'M', 'Space': 'S'} 49 STOPBITS = (1, 1.5, 2) 50 BYTETYPES = ('1 Byte', '2 Bytes', '4 Bytes') 51 52 TIMEOUT = 0.015 53 54 font = ("宋體", 10, "normal") 55 font_text = ("Courier", 9, "normal") 56 font_button = ("黑體", 12, "normal") 57 58 59 def format_data(data): 60 li = data.split(' ') 61 result = [] 62 k = 0 63 for each in li: 64 if len(each) <= 2: 65 result.append(each) 66 67 if len(each) > 2: 68 while k < len(each): 69 result.append(each[k: k + 2]) 70 k = k + 2 71 k = 0 72 73 for i in range(len(result)): 74 if len(result[i]) == 1: 75 result[i] = '0' + result[i] 76 77 return result 78 79 80 def getAvailabelSerialPort(): 81 available_ports = [] 82 coms = serial.tools.list_ports.comports() 83 if coms is not None: 84 for com in coms: 85 if sys.platform == 'win32': 86 available_ports.append(com.device) 87 elif sys.platform == 'linux2': 88 if com[2] != 'n/a': 89 available_ports.append(com[0]) 90 return tuple(available_ports) 91 92 93 if sys.platform == 'win32': 94 PORTS = sorted(getAvailabelSerialPort(), key=lambda n: int(n[3:])) 95 else: 96 PORTS = getAvailabelSerialPort() 97 98 99 class SerialAssistantGUI(object): 100 def __init__(self): 101 self.root = tk.Tk() 102 self.root.title('%s v%s by %s' % (APP_NAME, APP_VER, 103 AUTHOR)) 104 self.root.geometry(WINDOWSIZE) 105 self.root.resizable(width=False, height=False) 106 107 self.__recv_area() 108 self.__send_area() 109 self.__cmd_area() 110 self.__opt_area() 111 112 self.root.mainloop() 113 114 def __recv_area(self): 115 recv_lframe = tk.LabelFrame(self.root, 116 text=recv_buf_text, 117 height=250) 118 recv_lframe.pack(fill=tk.X, padx=5) 119 120 recv_optframe = tk.Frame(recv_lframe) 121 recv_txtframe = tk.Frame(recv_lframe) 122 recv_optframe.pack(fill=tk.Y, side=tk.LEFT, padx=5, pady=5) 123 recv_txtframe.pack(fill=tk.Y, side=tk.RIGHT, padx=5, pady=5) 124 125 self.recv_mode = tk.IntVar() 126 self.recv_mode.set(0) 127 recv_radbtn1 = tk.Radiobutton(recv_optframe, 128 text=text_mode_text, 129 font=font, 130 variable=self.recv_mode, value=0) 131 recv_radbtn2 = tk.Radiobutton(recv_optframe, 132 text=hex_mode_text, 133 font=font, 134 variable=self.recv_mode, value=1) 135 recv_clrbtn = tk.Button(recv_optframe, width=15, 136 text=clear_recv_text, 137 font=font, 138 command=self.clear_received) 139 recv_radbtn1.pack(anchor=tk.W) 140 recv_radbtn2.pack(anchor=tk.W) 141 recv_clrbtn.pack(fill=tk.X) 142 143 self.recv_txtarea = tk.Text(recv_txtframe, height=15, width=74, 144 font=font_text) 145 self.recv_txtarea.pack(side=tk.LEFT, fill=tk.X) 146 recv_scrbar = tk.Scrollbar(recv_txtframe) 147 recv_scrbar.pack(side=tk.RIGHT, fill=tk.Y) 148 self.recv_txtarea['yscrollcommand'] = recv_scrbar.set 149 recv_scrbar['command'] = self.recv_txtarea.yview 150 151 def __send_area(self): 152 send_lframe = tk.LabelFrame(self.root, 153 text=send_buf_text, 154 height=100) 155 send_lframe.pack(fill=tk.X, padx=5) 156 157 send_optframe = tk.Frame(send_lframe) 158 send_txtframe = tk.Frame(send_lframe) 159 send_optframe.pack(fill=tk.Y, side=tk.LEFT, padx=5, pady=5) 160 send_txtframe.pack(fill=tk.Y, side=tk.RIGHT, padx=5, pady=5) 161 162 self.send_mode = tk.IntVar() 163 self.send_mode.set(0) 164 send_radbtn1 = tk.Radiobutton(send_optframe, 165 text=text_mode_text, 166 font=font, 167 variable=self.send_mode, value=0) 168 send_radbtn2 = tk.Radiobutton(send_optframe, 169 text=hex_mode_text, 170 font=font, 171 variable=self.send_mode, value=1) 172 self.linebreak = tk.IntVar() 173 if sys.platform == 'win32': 174 self.linebreak.set(1) 175 elif sys.platform == 'linux2': 176 self.linebreak.set(0) 177 send_chkbtn = tk.Checkbutton(send_optframe, 178 text=line_break_text, 179 font=font, 180 variable=self.linebreak) 181 send_clrbtn = tk.Button(send_optframe, 182 text=clear_send_text, 183 font=font, 184 width=15, 185 command=self.clear_sent) 186 send_radbtn1.pack(anchor=tk.W) 187 send_radbtn2.pack(anchor=tk.W) 188 send_chkbtn.pack(anchor=tk.W) 189 send_clrbtn.pack(fill=tk.X) 190 191 self.send_txtarea = tk.Text(send_txtframe, height=7, width=74, 192 font=font_text) 193 self.send_txtarea.pack(side=tk.LEFT, fill=tk.X) 194 send_scrbar = tk.Scrollbar(send_txtframe) 195 send_scrbar.pack(side=tk.RIGHT, fill=tk.Y) 196 self.send_txtarea['yscrollcommand'] = send_scrbar.set 197 send_scrbar['command'] = self.send_txtarea.yview 198 199 def __cmd_area(self): 200 cmd_frame = tk.Frame(self.root) 201 cmd_frame.pack(fill=tk.X, padx=5, pady=5) 202 #cmd_btn = tk.Button(cmd_frame, 203 # text=send_btn_text, 204 # font=font, 205 # command=self.send_data) 206 #cmd_btn.pack(side=tk.LEFT) 207 208 def __opt_area(self): 209 opt_frame = tk.Frame(self.root) 210 opt_frame.pack(fill=tk.X, padx=5, pady=5) 211 212 # Serial port setting 213 port_label = tk.Label(opt_frame, 214 text=port_lbl_text, 215 font=font) 216 217 self.port_var = tk.StringVar() 218 self.port_var.set(select_port) 219 220 if len(PORTS) == 0: 221 port_opmenu = tk.OptionMenu(opt_frame, self.port_var, '') 222 else: 223 port_opmenu = tk.OptionMenu(opt_frame, self.port_var, *PORTS) 224 port_opmenu.config(anchor=tk.W, 225 width=10, 226 font=font) 227 port_label.pack(side=tk.LEFT) 228 port_opmenu.pack(side=tk.LEFT) 229 230 # Baudrate setting 231 brt_label = tk.Label(opt_frame, 232 text=baud_lbl_text, 233 font=font) 234 self.brt_var = tk.StringVar() 235 self.brt_var.set(BAUDRATES[3]) 236 brt_opmenu = tk.OptionMenu(opt_frame, self.brt_var, *BAUDRATES) 237 brt_opmenu.config(anchor=tk.W, 238 width=6, 239 font=font) 240 brt_label.pack(side=tk.LEFT) 241 brt_opmenu.pack(side=tk.LEFT) 242 243 # Bytesize setting 244 size_label = tk.Label(opt_frame, 245 text=size_lbl_text, 246 font=font) 247 self.size_var = tk.StringVar() 248 self.size_var.set(BYTESIZES[3]) 249 size_opmenu = tk.OptionMenu(opt_frame, self.size_var, *BYTESIZES) 250 size_opmenu.config(anchor=tk.W, 251 width=3, 252 font=font) 253 size_label.pack(side=tk.LEFT) 254 size_opmenu.pack(side=tk.LEFT) 255 256 # Parity setting 257 parity_label = tk.Label(opt_frame, 258 text=parity_lbl_text, 259 font=font) 260 self.parity_var = tk.StringVar() 261 self.parity_var.set('None') 262 parity_opmenu = tk.OptionMenu(opt_frame, self.parity_var, 263 *PARITIES) 264 parity_opmenu.config(anchor=tk.W, 265 width=4, 266 font=font) 267 parity_label.pack(side=tk.LEFT) 268 parity_opmenu.pack(side=tk.LEFT) 269 270 # Stopbit setting 271 stop_label = tk.Label(opt_frame, 272 text=stopbit_lbl_text, 273 font=font) 274 self.stop_var = tk.StringVar() 275 self.stop_var.set(STOPBITS[0]) 276 stop_opmenu = tk.OptionMenu(opt_frame, self.stop_var, *STOPBITS) 277 stop_opmenu.config(anchor=tk.W, 278 width=3, 279 font=font) 280 stop_label.pack(side=tk.LEFT) 281 stop_opmenu.pack(side=tk.LEFT) 282 283 # Set buttons 284 control_frame = tk.Frame(self.root, width=300) 285 status_frame = tk.Frame(self.root, width=300) 286 control_frame.pack(side=tk.LEFT) 287 status_frame.pack(side=tk.RIGHT) 288 289 start_btn = tk.Button(control_frame, 290 text=start_port_text, 291 width=12,font=font_button, 292 command=self.start_port) 293 close_btn = tk.Button(control_frame, 294 text=close_port_text, 295 width=12,font=font_button, 296 command=self.close_port) 297 cmd_btn = tk.Button(control_frame, 298 text=send_btn_text, 299 width=12,font=font_button, 300 command=self.send_data) 301 start_btn.pack(side=tk.LEFT, padx=5) 302 close_btn.pack(side=tk.LEFT, padx=5) 303 cmd_btn.pack(side=tk.LEFT, padx=5) 304 self.state_lbl = tk.Label(control_frame, text='') 305 self.state_lbl.pack(side=tk.LEFT, padx=5) 306 307 # Status frame widgets 308 send_cnt_label = tk.Label(status_frame, 309 text=send_lbl_text, 310 font=font) 311 self.send_cnt = tk.StringVar() 312 self.send_cnt.set(self.TX) 313 send_cnt_entry = tk.Entry(status_frame, 314 textvariable=self.send_cnt, 315 width=10,font=font, 316 relief=tk.SUNKEN, 317 state=tk.DISABLED, 318 justify=tk.RIGHT) 319 send_cnt_label.pack(side=tk.LEFT) 320 send_cnt_entry.pack(side=tk.LEFT) 321 322 recv_cnt_label = tk.Label(status_frame, 323 text=recv_lbl_text, 324 font=font) 325 self.recv_cnt = tk.StringVar() 326 self.recv_cnt.set(self.RX) 327 recv_cnt_entry = tk.Entry(status_frame, 328 textvariable=self.recv_cnt, 329 width=10,font=font, 330 relief=tk.SUNKEN, 331 state=tk.DISABLED, 332 justify=tk.RIGHT) 333 recv_cnt_label.pack(side=tk.LEFT) 334 recv_cnt_entry.pack(side=tk.LEFT) 335 336 clr_btn = tk.Button(status_frame, 337 text=clear_btn_text, 338 width=12,font=font_button, 339 command=self.clear_count) 340 #clr_btn.pack() 341 clr_btn.pack(side=tk.LEFT, padx=10) 342 343 def clear_received(self): 344 self.recv_txtarea.delete(0.0, tk.END) 345 346 def clear_sent(self): 347 self.send_txtarea.delete(0.0, tk.END) 348 349 def clear_count(self): 350 pass 351 352 def start_port(self): 353 pass 354 355 def send_data(self): 356 pass 357 358 def recv_data(self): 359 pass 360 361 362 class SerialAssistant(SerialAssistantGUI): 363 364 portisopen = 0 365 TX = 0 366 RX = 0 367 368 def __init__(self): 369 super(SerialAssistant, self).__init__() 370 371 def clear_count(self): 372 self.RX = 0 373 self.TX = 0 374 self.send_cnt.set(self.RX) 375 self.recv_cnt.set(self.TX) 376 377 def start_port(self): 378 port = self.port_var.get() 379 baudrate = int(self.brt_var.get()) 380 bytesize = int(self.size_var.get()) 381 parity = PARITIES[self.parity_var.get()] 382 stopbits = float(self.stop_var.get()) 383 384 try: 385 self.s = serial.Serial(port=port, 386 baudrate=baudrate, 387 bytesize=bytesize, 388 parity=parity, 389 stopbits=stopbits, 390 timeout=TIMEOUT) 391 except serial.SerialException as e: 392 msgbox.showerror("OpenError", e) 393 except serial.SerialTimeoutException as e: 394 msgbox.showerror("OpenError", e) 395 else: 396 self.portisopen = 1 397 self.state_lbl.config(background='green', 398 text=port + opening) 399 self.th = threading.Thread(target=self.recv_data) 400 self.th.daemon = True 401 self.th.start() 402 403 def send_data(self): 404 if self.portisopen: 405 data_fmt = [] 406 data = self.send_txtarea.get(0.0, tk.END) 407 408 if len(data) == 1: 409 return 410 411 if self.send_mode.get(): 412 if data[-1] == '\x0a': 413 data = data[0:-1] 414 data_fmt = format_data(data) 415 for each in data_fmt: 416 try: 417 x = binascii.a2b_hex(each) 418 except TypeError: 419 x = '\x00' 420 421 self.s.write(x) 422 self.TX += 1 423 else: 424 if data[-1] == '\n': 425 data = data[0:-1] 426 if self.linebreak.get(): 427 data = data.replace('\n', '\r\n') 428 429 self.s.write(data.encode('utf-8')) 430 self.TX += len(data) 431 self.send_cnt.set(self.TX) 432 else: 433 msgbox.showerror("Error", "Port NOT open!") 434 return 435 436 def recv_data(self): 437 while self.portisopen: 438 data = self.s.read() 439 if len(data) != 0: 440 if self.recv_mode.get() == 1: 441 data = binascii.b2a_hex(data) 442 if sys.version_info.major == 3: 443 data += b' ' 444 elif sys.version_info.major == 2: 445 data += ' ' 446 self.RX += 1 447 else: 448 self.RX += len(data) 449 self.recv_cnt.set(self.RX) 450 if data != '\x0d': 451 self.recv_txtarea.insert(tk.END, data) 452 self.recv_txtarea.see(tk.END) 453 return 454 455 def close_port(self): 456 if self.portisopen: 457 self.portisopen = 0 458 self.s.close() 459 self.state_lbl.config(background='red', 460 text=self.port_var.get() + closed) 461 else: 462 return 463 464 def handler(self, signum, frame): 465 self.portisopen = 0 466 self.th.join() 467 468 def __del__(self): 469 self.close_port() 470 signal.signal(signal.SIGINT, self.handler) 471 472 473 def main(): 474 SerialAssistant() 475 476 477 if __name__ == '__main__': 478 main()