From a553df3b036a020152dd5f383229062ef1daf593 Mon Sep 17 00:00:00 2001
From: Laurens Valk <laurens@pybricks.com>
Date: Thu, 23 Feb 2023 16:30:18 +0100
Subject: [PATCH] pybricks.iodevices.PUPDevice: Add device class as comments.

---
 examples/pup/iodevices_pupdevice/port_info.py | 34 ++++++++++++-------
 1 file changed, 21 insertions(+), 13 deletions(-)

diff --git a/examples/pup/iodevices_pupdevice/port_info.py b/examples/pup/iodevices_pupdevice/port_info.py
index ff77ee2..fbb437f 100644
--- a/examples/pup/iodevices_pupdevice/port_info.py
+++ b/examples/pup/iodevices_pupdevice/port_info.py
@@ -3,27 +3,35 @@ from pybricks.parameters import Port
 from uerrno import ENODEV
 
 # Dictionary of device identifiers along with their name.
-# Also check https://github.com/pybricks/technical-info/blob/master/assigned-numbers.md#io-device-type-ids
-# for retired/internal
 device_names = {
-    1: "Wedo 2.0 Medium Motor", #DCMotor, analog ID
-    2: "PUP Train Motor",       #DCMotor, analog ID
-    8: "PUP Light",             #Light, analog ID
-    34: "Wedo 2.0 Tilt Sensor",
-    35: "Wedo 2.0 Infrared Sensor", # aka WeDo 2.0 Motion Sensor
-    37: "BOOST Color Distance Sensor",
+    # pybricks.pupdevices.DCMotor
+    1: "Wedo 2.0 Medium Motor",
+    2: "Powered Up Train Motor",
+    # pybricks.pupdevices.Light
+    8: "Powered Up Light",
+    # pybricks.pupdevices.Motor
     38: "BOOST Interactive Motor",
     46: "Technic Large Motor",
     47: "Technic Extra Large Motor",
     48: "SPIKE Medium Angular Motor",
     49: "SPIKE Large Angular Motor",
-    61: "SPIKE Color Sensor",
-    62: "SPIKE Ultrasonic Sensor",
-    63: "SPIKE Force Sensor",
-	64: "SPIKE 3x3 Color Light Matrix",
-	65: "SPIKE Small Angular Motor",
+    65: "SPIKE Small Angular Motor",
     75: "Technic Medium Angular Motor",
     76: "Technic Large Angular Motor",
+    # pybricks.pupdevices.TiltSensor
+    34: "Wedo 2.0 Tilt Sensor",
+    # pybricks.pupdevices.InfraredSensor
+    35: "Wedo 2.0 Infrared Motion Sensor",
+    # pybricks.pupdevices.ColorDistanceSensor
+    37: "BOOST Color Distance Sensor",
+    # pybricks.pupdevices.ColorSensor
+    61: "SPIKE Color Sensor",
+    # pybricks.pupdevices.UltrasonicSensor
+    62: "SPIKE Ultrasonic Sensor",
+    # pybricks.pupdevices.ForceSensor
+    63: "SPIKE Force Sensor",
+    # pybricks.pupdevices.ColorLightMatrix
+    64: "SPIKE 3x3 Color Light Matrix",
 }
 
 # Make a list of known ports.