Files
scomm/scomm.py
2021-08-24 19:15:58 +08:00

368 lines
15 KiB
Python
Executable File

#! /usr/bin/env python3
import os,sys,json
import serial
import serial.tools.list_ports
import tkinter
import threading
import tkgen.gengui
import tkinter.scrolledtext
import tkinter.filedialog
import time, datetime
def tsnow():
return int(time.time()*1000)
def strnow():
return datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')[:-3]
def tohex(b):
return ' '.join(['%02X'%x for x in b])
class UIproc():
def __init__(self, app):
self.root = app
self.combobox_port = self.root.get('cbbox-com')
self.entry_baud = self.root.get('entry-baud')
self.entry_split = self.root.get('entry-split')
self.entry_cycle = self.root.get('entry-cycle')
self.entry_encoding = self.root.get('entry-encoding')
self.entry_sendText = self.root.get('entry-sendText')
self.btn_onoff = self.root.get('btn-onoff')
self.canvas_led = self.root.get('canvas-led')
self.label_status = self.root.get('label-status')
self.ckbtn_shex = self.root.get('ckbtn-shex')
self.ckbtn_rhex = self.root.get('ckbtn-rhex')
self.ckbtn_sendshow = self.root.get('ckbtn-sendshow')
self.ckbtn_time = self.root.get('ckbtn-time')
self.text_recv = self.root.get('text-recv')
self.entry_split = self.root.get('entry-split')
self.ckbtn_0d = self.root.get('ckbtn-0d')
self.ckbtn_0a = self.root.get('ckbtn-0a')
self.lastRecvTicks = 0
self.lastCursor = "end"
self.lastRecvData = b''
self.event_init()
def getSendData(self):
data = self.entry_sendText.var.get()
encoding = self.entry_encoding.var.get()
dat = self.ckbtn_shex.var.get() and bytes.fromhex(data) or data.encode(encoding, 'ignore')
dat += self.ckbtn_0d.var.get() and b'\r' or b''
dat += self.ckbtn_0a.var.get() and b'\n' or b''
return dat
def dmesg(self, cate, data):
text = self.ckbtn_time.var.get() and '[%s]'%strnow() or ''
encoding = self.entry_encoding.var.get()
splitms = int(self.entry_split.var.get().replace('ms',''))
ts = tsnow()
_i0 = self.text_recv.index('end');bg,fg='','black'
if cate == 'send' and self.ckbtn_sendshow.var.get():
text += '> '
text += self.ckbtn_rhex.var.get() and tohex(data) or data.decode(encoding, 'ignore')
self.text_recv.insert('end', '\n'+text)
self.lastRecvTicks = 0
self.lastCursor = "end"
self.lastRecvData = b''
elif cate == 'recv':
_ic = 'end'
if(ts-self.lastRecvTicks>splitms):
self.lastCursor = self.text_recv.index('end')
self.lastRecvData = data
text += '< '
text += self.ckbtn_rhex.var.get() and tohex(data) or data.decode(encoding, 'ignore')
#self.text_recv.insert('end', '\n'+text)
#print('##### todo:recv1 = %sbytes'%len(data))
else:
data = self.lastRecvData + data
self.lastRecvData = data
text += '< '
text += self.ckbtn_rhex.var.get() and tohex(data) or data.decode(encoding, 'ignore')
_i0 = self.lastCursor; _ic = _i0
self.text_recv.delete(_i0,'end')
#self.text_recv.insert(_i0, '\n'+text)
#print('##### todo:recv2 = %sbytes'%len(data))
for cb in self.root.unpack.values():
try:
if cb: text += '\n[%s] %s'%(cb['title'],eval(cb['value'], {"data":data}))
except: pass
self.text_recv.insert(_ic, '\n'+text)
self.lastRecvTicks = ts
bg,fg='','blue'
_i1 = self.text_recv.index('end')
self.text_recv.tag_add('%s'%ts, _i0, _i1)
self.text_recv.tag_config('%s'%ts,background=bg,foreground=fg)
self.text_recv.yview('end')
def serial_open(self):
self.entry_baud.configure(state='disabled')
self.combobox_port.configure(state='disabled')
self.btn_onoff.configure(text='关闭串口')
self.canvas_led.create_oval(4,4,19,19,fill='green')
def serial_close(self):
self.entry_baud.configure(state='normal')
self.combobox_port.configure(state='normal')
self.btn_onoff.configure(text='打开串口')
self.canvas_led.create_oval(4,4,19,19,fill='red')
def read_serial_port(self):
return self.combobox_port.get()
def read_serial_baud(self):
return self.entry_baud.var.get()
def set_serial_port_list(self, portList):
currText = self.combobox_port.get()
_ports = []
for i in portList:
_ports.append(str(i[0]))
self.combobox_port['values'] = _ports
if currText in _ports:
self.combobox_port.set(currText)
else:
self.combobox_port.set(_ports[-1])
def clear_recvtext(self):
self.text_recv.delete('1.0','end')
def save_recvtext(self):
f = tkinter.filedialog.asksaveasfile(mode='w', defaultextension='.txt', initialfile='scommlog-%s'%tsnow())
if f:
f.write(self.text_recv.get('1.0','end'))
f.close()
def log(self, s):
print('[sys.log]: %s'%s)
self.label_status.var.set(str(s))
def event_init(self):
self.combobox_port.bind("<<ComboboxSelected>>", lambda x:self.log(self.combobox_port.get()))
class SerComm():
#root = None
def __init__(self, app):
self.ui = UIproc(app)
self.com = serial.Serial()
self.thread = None
self.alive = threading.Event()
self.setting = None
self.isDetectSerialPort = False
self.receiveProgressStop = True
self.sendCount = 0
self.recvCount = 0
def detectSerialPort(self):
if not self.isDetectSerialPort:
self.isDetectSerialPort = True
t = threading.Thread(target=self.detectSerialPortProcess)
t.setDaemon(True)
t.start()
def openCloseSerial(self):
self.ui.log('Serial Port waitting...')
t = threading.Thread(target=self.openCloseSerialProcess)
t.setDaemon(True)
t.start()
def receiveData(self):
while not self.receiveProgressStop:
try:
data = self.com.read(self.com.in_waiting)
if data:
self.ui.log('%s: recv %s bytes: %s...' % (self.com.port,len(data),str(data)[:16]))
self.ui.dmesg('recv', data)
time.sleep(0.01)
except Exception as e:
self.ui.log('%s: receive trace: %s' % (self.com.port,str(e)))
def sendData(self):
#try:
if True:
if not self.com.is_open:
self.ui.log('Serial Port not open')
else:
data = self.ui.getSendData()
if data and len(data) > 0:
self.com.write(data)
self.sendCount += len(data)
self.ui.dmesg('send', data)
self.ui.log('%s: send %s bytes: %s...' % (self.com.port,len(data),str(data)[:16]))
# scheduled send
#if self.sendSettingsScheduledCheckBox.isChecked():
# if not self.isScheduledSending:
# t = threading.Thread(target=self.scheduledSend)
# t.setDaemon(True)
# t.start()
#except Exception as e:
# self.ui.log('%s: send trace: %s' % (self.com.port,str(e)))
def openCloseSerialProcess(self):
try:
if self.com.is_open:
self.receiveProgressStop = True
self.com.close()
self.ui.serial_close()
self.ui.log('%s: closed' % self.com.port)
else:
try:
self.com.baudrate = int(self.ui.read_serial_baud())
self.com.port = self.ui.read_serial_port()
self.com.bytesize = 8
self.com.parity = 'N'
self.com.stopbits = 1
self.com.timeout = 0
self.com.write_timeout = 0
self.com.inter_byte_timeout = 0
self.com.open()
self.ui.serial_open()
self.ui.log('%s: open success' % self.com.port)
self.receiveProgressStop = False
self.receiveProcess = threading.Thread(target=self.receiveData)
self.receiveProcess.setDaemon(True)
self.receiveProcess.start()
except Exception as e:
self.com.close()
self.ui.serial_close()
self.receiveProgressStop = True
self.ui.log('%s: open failed: %s' % (self.com.port,str(e)))
except Exception as e:
self.ui.log('%s: openClose trace: %s' % (self.com.port,str(e)))
def clearWin(self):
self.ui.clear_recvtext()
def saveFile(self):
self.ui.save_recvtext()
def findSerialPort(self):
self.port_list = list(serial.tools.list_ports.comports())
return self.port_list
def detectSerialPortProcess(self):
while(1):
portList = self.findSerialPort()
if len(portList)>0:
self.ui.log('detectSerialPort = %s' % len(portList))
self.ui.set_serial_port_list(portList)
break
time.sleep(1)
self.isDetectSerialPort = False
def safe_exit(self):
self.com.close()
sys.exit(0)
class TopWin():
def __init__(self, root):
self.root = root
self.root.unpack = {}
self.WinData = None
self.WinPack = None
self.WinUnpack = None
def set_send_data(self, btn):
_cfg = self.root.usercfg.get(btn, {})
if _cfg:
val = _cfg.get('value')
self.root.get('entry-sendText').var.set(val)
self.root.get('ckbtn-shex').var.set(_cfg.get('hex') and 1 or 0)
self.root.get('btn-send').invoke()
def set_unpack(self, btn):
self.root.unpack[btn] = self.root.get(btn).var.get() and self.root.usercfg.get(btn) or None
def save_cfg(self, btn, dat):
with open('usercfg.json', 'wb+') as f:
self.root.usercfg[btn] = dat
encoding = self.root.get('entry-encoding').var.get()
f.write(json.dumps(self.root.usercfg,indent=4,sort_keys=True).encode(encoding,'ignore'))
def win_data(self, event):
def _save(w):
dat = {'title':self.root.get('entry-dfile').var.get()}
dat['value'] = self.root.get('text-dsetting').get('1.0','end -1 chars')
dat['hex'] = self.root.get('ckbtn-dhex').var.get() and 1 or 0
self.save_cfg(w, dat)
self.WinData.destroy()
if self.WinData: self.WinData.destroy()
self.WinData = self.root.toplevel('data.ui', title='预置数据')
self.WinData.configure(bg='#e8e8e8')
btn = event.widget._name
_cfg = self.root.usercfg.get(btn, {})
self.root.entry('entry-dfile').set(_cfg.get('title', btn))
self.root.get('text-dsetting').insert('end', _cfg.get('value',''))
self.root.checkbox('ckbtn-dhex').set(_cfg.get('hex') and 1 or 0)
self.root.button('btn-dsave', cmd=lambda x=btn:_save(x), focus=True)
def win_unpack(self, event):
def _save(w):
dat = {'title':self.root.get('entry-ufile').var.get()}
dat['value'] = self.root.get('text-usetting').get('1.0','end -1 chars')
self.save_cfg(w,dat)
self.WinUnpack.destroy()
if self.WinUnpack: self.WinUnpack.destroy()
self.WinUnpack = self.root.toplevel('unpack.ui', title='解析脚本')
self.WinUnpack.configure(bg='#e8e8e8')
btn = event.widget._name
_cfg = self.root.usercfg.get(btn, {})
self.root.entry('entry-ufile').set(_cfg.get('title', btn))
self.root.get('text-usetting').insert('end', _cfg.get('value',''))
self.root.button('btn-usave', cmd=lambda x=btn:_save(x), focus=True)
if __name__ == '__main__':
tkinter.ScrolledText = tkinter.scrolledtext.ScrolledText
root = tkgen.gengui.TkJson('app.ui', title='scomm串口调试助手')
comm = SerComm(root)
wm = TopWin(root)
# 读取用户数据文件
root.usercfg = json.load(open('usercfg.json')) if os.path.isfile('usercfg.json') else {}
# 预置数据回调函数
for i in range(10):
name = 'btn-data%02d'%(i+1)
try:
btn = root.get(name)
root.button(name, cmd=lambda x=name: wm.set_send_data(x))
btn.bind('<Button-2>', wm.win_data)
btn.bind('<Button-3>', wm.win_data)
_cfg = root.usercfg.get(name)
if _cfg and btn:
btn.config(text=_cfg.get('title',name))
except:
pass
# 解析脚本回调函数
var = tkinter.IntVar()
for i in range(10):
name = 'btn-unpack%02d'%(i+1)
try:
btn = root.get(name)
root.button(name, cmd=lambda x=name: wm.set_unpack(x))
btn.bind('<Button-2>', wm.win_unpack)
btn.bind('<Button-3>', wm.win_unpack)
btn.var = var
_cfg = root.usercfg.get(name)
if _cfg and btn:
btn.config(text=_cfg.get('title',name))
root.checkbox(name).set(0)
except:
pass
# UI相关设置
root.get('canvas-led').create_oval(4,4,19,19,fill='gray')
root.checkbox('ckbtn-rhex').set(0)
root.checkbox('ckbtn-shex').set(0)
root.checkbox('ckbtn-0d').set(0)
root.checkbox('ckbtn-0a').set(0)
root.checkbox('ckbtn-split').set(1)
root.checkbox('ckbtn-cycle').set(0)
root.checkbox('ckbtn-time').set(1)
root.checkbox('ckbtn-sendshow').set(1)
root.entry('entry-split').set('99ms')
root.entry('entry-cycle').set('1024ms')
root.entry('entry-baud').set('9600')
root.entry('entry-encoding').set('gbk')
root.entry('entry-sendText', key='<Return>', cmd=lambda x:comm.sendData()).set('')
root.button('btn-scan', cmd=lambda:comm.detectSerialPort())
root.button('btn-onoff', cmd=lambda:comm.openCloseSerial())
root.button('btn-send', cmd=lambda:comm.sendData())
root.button('btn-clear', cmd=lambda:comm.clearWin())
root.button('btn-savefile', cmd=lambda:comm.saveFile())
_stvar = tkinter.StringVar()
root.label('label-status').textvariable=_stvar
root.label('label-status').var =_stvar
#root.get('text-recv').configure(state=tkinter.DISABLED)
# 其他设置
root.configure(bg='#e8e8e8')
root.lift() # 把主窗口置于最前面
#root.wm_state('zoomed') # 最大化窗口
root.protocol("WM_DELETE_WINDOW", lambda:comm.safe_exit())
# 启动任务
comm.detectSerialPort()
root.mainloop()