Allow arbitrary base classes for hid devices

Yubico · Jan 6, 2016 · 47b965a · 47b965a
1 parent a5e4ee7
commit 47b965a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/u2flib_host/hid_transport.py b/u2flib_host/hid_transport.py
@@ -65,19 +65,20 @@
 TIMEOUT = 1000
 
 
-def list_devices():
+def list_devices(dev_class=None):
+    dev_class = dev_class or HIDDevice
     devices = []
     for d in hid.enumerate(0, 0):
         usage_page = d['usage_page']
         if usage_page == 0xf1d0 and d['usage'] == 1:
-            devices.append(HIDDevice(d['path']))
+            devices.append(dev_class(d['path']))
         # Usage page doesn't work on Linux
         elif (d['vendor_id'], d['product_id']) in DEVICES:
             device = HIDDevice(d['path'])
             try:
                 device.open()
                 device.close()
-                devices.append(HIDDevice(d['path']))
+                devices.append(dev_class(d['path']))
             except exc.DeviceError:
                 pass
     return devices