PYDFU: Fix listing of all devices

micropython · May 18, 2021 · f008a24 · f008a24
1 parent 452fa3f
commit f008a24
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/tools/pydfu.py b/tools/pydfu.py
@@ -572,8 +572,8 @@ def main():
     parser.add_argument(
         "-l", "--list", help="list available DFU devices", action="store_true", default=False
     )
-    parser.add_argument("--vid", help="USB Vendor ID", type=lambda x: int(x, 0), default=__VID)
-    parser.add_argument("--pid", help="USB Product ID", type=lambda x: int(x, 0), default=__PID)
+    parser.add_argument("--vid", help="USB Vendor ID", type=lambda x: int(x, 0), default=None)
+    parser.add_argument("--pid", help="USB Product ID", type=lambda x: int(x, 0), default=None)
     parser.add_argument(
         "-m", "--mass-erase", help="mass erase device", action="store_true", default=False
     )
@@ -588,11 +588,17 @@ def main():
 
     __verbose = args.verbose
 
-    __VID = args.vid
-    __PID = args.pid
+    kwargs = {}
+    if args.vid:
+        __VID = args.vid
+        kwargs["idVendor"] = __VID
+
+    if args.pid:
+        __PID = args.pid
+        kwargs["idProduct"] = __PID
 
     if args.list:
-        list_dfu_devices(idVendor=__VID, idProduct=__PID)
+        list_dfu_devices(**kwargs)
         return
 
     init()