pybricksdev.dfu: Add SPIKE Essential USB product ID.

This adds the SPIKE Essential USB product ID for both DFU mode and
FS mode.

Author: dlech

CHANGELOG.md

@@ -8,6 +8,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 - Added `VOLUME` to `ble.lwp3.bytecodes.HubProperty` enum.
+- Added SPIKE Essential hub device IDs.
 
 ## [1.0.0-alpha.14] - 2021-08-27
 

pybricksdev/dfu.py

@@ -21,12 +21,12 @@
 FIRMWARE_SIZE = 1 * 1024 * 1024 - 32 * 1024  # 1MiB - 32KiB
 LEGO_VID = 0x0694
 SPIKE_PRIME_PID = 0x0008
+SPIKE_ESSENTIAL_PID = 0x000C
 MINDSTORMS_INVENTOR_PID = 0x0011
 
 
-SPIKE_PRIME_DEVICE = f"{LEGO_VID:04x}:{SPIKE_PRIME_PID:04x}"
-MINDSTORMS_INVENTOR_DEVICE = f"{LEGO_VID:04x}:{MINDSTORMS_INVENTOR_PID:04x}"
-ALL_DEVICES = [SPIKE_PRIME_DEVICE, MINDSTORMS_INVENTOR_DEVICE]
+ALL_PIDS = [SPIKE_PRIME_PID, SPIKE_ESSENTIAL_PID, MINDSTORMS_INVENTOR_PID]
+ALL_DEVICES = [f"{LEGO_VID:04x}:{pid:04x}" for pid in ALL_PIDS]
 
 
 def _get_dfu_util() -> ContextManager[os.PathLike]:
@@ -178,7 +178,7 @@ def flash_dfu(firmware_bin: bytes, metadata: dict) -> None:
                 exit(1)
 
             product_id = devices[0].idProduct
-            if product_id != SPIKE_PRIME_PID and product_id != MINDSTORMS_INVENTOR_PID:
+            if product_id not in ALL_PIDS:
                 print(f"Unknown USB product ID: {product_id:04X}", file=sys.stderr)
                 exit(1)
 

pybricksdev/resources/99-pybricksdev.rules

@@ -20,6 +20,12 @@ SUBSYSTEM=="usb", ATTRS{idVendor}=="0694", ATTRS{idProduct}=="0008", TAG+="uacce
 # SPIKE Prime hub
 SUBSYSTEM=="usb", ATTRS{idVendor}=="0694", ATTRS{idProduct}=="0009", TAG+="uaccess"
 
+# SPIKE Essential hub in firmware update mode (DFU mode)
+SUBSYSTEM=="usb", ATTRS{idVendor}=="0694", ATTRS{idProduct}=="000C", TAG+="uaccess"
+
+# SPIKE Essential hub
+SUBSYSTEM=="usb", ATTRS{idVendor}=="0694", ATTRS{idProduct}=="000D", TAG+="uaccess"
+
 # MINDSTORMS Inventor hub
 SUBSYSTEM=="usb", ATTRS{idVendor}=="0694", ATTRS{idProduct}=="0010", TAG+="uaccess"
 

pybricksdev/usbconnection.py

@@ -63,6 +63,7 @@ async def connect(self, product):
                 or (dev.pid == 0x5740 and dev.vid == 0x0483)
                 or (dev.vid == 0x0694 and dev.pid == 0x0010)
                 or (dev.vid == 0x0694 and dev.pid == 0x0009)
+                or (dev.vid == 0x0694 and dev.pid == 0x000D)
             ):
                 port = dev.device
                 break