99 lines
2.4 KiB
Python
99 lines
2.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Python 2 only for now.
|
|
|
|
import time
|
|
from binascii import hexlify, unhexlify
|
|
|
|
|
|
def read_next(handle, timeout=1000, ignore_nodata=False):
|
|
reply = hidapi.read(handle, 128, timeout)
|
|
if reply is None:
|
|
print "!! Read failed, aborting"
|
|
raise Exception()
|
|
if reply:
|
|
hexs = hexlify(reply)
|
|
print ">> [%s %s %s %s] %s" % (hexs[0:2], hexs[2:4], hexs[4:8], hexs[8:], repr(reply))
|
|
return True
|
|
|
|
if not ignore_nodata:
|
|
print ">> []"
|
|
return False
|
|
|
|
|
|
def console_cycle(handle):
|
|
last_data = None
|
|
|
|
while True:
|
|
if read_next(handle, timeout=100, ignore_nodata=True):
|
|
continue
|
|
|
|
line = raw_input('!! Command: ')
|
|
line = line.strip().replace(' ', '').replace('-', '')
|
|
if not line:
|
|
continue
|
|
|
|
data = None
|
|
|
|
if line == 'h':
|
|
print 'Commands:'
|
|
print ' <hex bytes> - send a packet to the device'
|
|
print ' r - re-send last packet'
|
|
print ' w<float> - listen for events for <float> seconds'
|
|
print ' h - this help screen'
|
|
print ' ^C - exit'
|
|
elif line == 'r':
|
|
data = last_data
|
|
elif line[0] == 'w':
|
|
line = line[1:].strip()
|
|
try:
|
|
seconds = float(line)
|
|
except:
|
|
print "!! Bad number <" + line + ">"
|
|
else:
|
|
count = 0
|
|
start_time = time.time()
|
|
while time.time() - start_time < seconds:
|
|
if read_next(handle, timeout=100, ignore_nodata=True):
|
|
count += 1
|
|
print "!! Got %d events" % count
|
|
else:
|
|
try:
|
|
data = unhexlify(line)
|
|
except:
|
|
print "!! Invalid input."
|
|
continue
|
|
|
|
if data:
|
|
hexs = hexlify(data)
|
|
print "<< [%s %s %s %s] %s" % (hexs[0:2], hexs[2:4], hexs[4:8], hexs[8:], repr(data))
|
|
last_data = data
|
|
hidapi.write(handle, data)
|
|
read_next(handle)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
arg_parser = argparse.ArgumentParser()
|
|
arg_parser.add_argument('device', default=None,
|
|
help='linux device to connect to')
|
|
args = arg_parser.parse_args()
|
|
|
|
import hidapi
|
|
print "!! Opening device ", args.device
|
|
handle = hidapi.open_path(args.device)
|
|
if handle:
|
|
print "!! Opened %x" % handle
|
|
print "!! vendor %s product %s serial %s" % (
|
|
repr(hidapi.get_manufacturer(handle)),
|
|
repr(hidapi.get_product(handle)),
|
|
repr(hidapi.get_serial(handle)))
|
|
print "!! Type 'h' for help."
|
|
try:
|
|
console_cycle(handle)
|
|
except:
|
|
print "!! Closing handle %x" % handle
|
|
hidapi.close(handle)
|
|
else:
|
|
print "!! Failed to open %s, aborting" % args.device
|