from __future__ import print_function

import hid
import time

# enumerate USB devices

for d in hid.enumerate():
 keys = list(d.keys())
 keys.sort()
 for key in keys:
 print("%s : %s" % (key, d[key]))
 print()

# try opening a device, then perform write and read
h = hid.device()
try:
 print("Opening the device")
 h.open(0x17ef, 0xa080) # TREZOR VendorID/ProductID

 print("Manufacturer: %s" % h.get_manufacturer_string())
 print("Product: %s" % h.get_product_string())
 print("Serial No: %s" % h.get_serial_number_string())

 # enable non-blocking mode
 h.set_nonblocking(1)

 # mute
 print("Write the data")
 h.write([2, 3, 0, 0] + [0] * 61)

 # wait
 time.sleep(1)
 # unmute
 print("Write the data")
 h.write([2, 1, 0, 0] + [0] * 61)

 # wait
 time.sleep(1)
 # read back the answer
 print("Read the data")
 while True:
 d = h.read(64)
 if d:
 print(d)
 else:
 break

 print("Closing the device")
 h.close()

except IOError as ex:
 print(ex)
 print("hid error:")
 print(h.error())
 print("")
 print("You probably don't have the hard-coded device.")
 print("Update the h.open() line in this script with the one")
 print("from the enumeration list output above and try again.")

print("Done")