device: Add second generation HomePod

postlund · Jun 17, 2023 · 02a8cab · 02a8cab
1 parent 51dffc9
commit 02a8cab
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/pyatv/const.py b/pyatv/const.py
@@ -186,6 +186,9 @@ class DeviceModel(Enum):
     AppleTV4KGen3 = 11
     """Device model is seventh generation Apple TV (Apple TV 4K gen 3)."""
 
+    HomePodGen2 = 12
+    """Device model is HomePod (second generation)."""
+
 
 class InputAction(Enum):
     """Type of input when pressing a button."""

diff --git a/pyatv/convert.py b/pyatv/convert.py
@@ -73,7 +73,8 @@ def model_str(device_model: DeviceModel) -> str:
         DeviceModel.HomePodMini: "HomePod Mini",
         DeviceModel.AirPortExpress: "AirPort Express (gen 1)",
         DeviceModel.AirPortExpressGen2: "AirPort Express (gen 2)",
-        DeviceModel.AppleTV4KGen2: "Apple TV 4K (gen2)",
+        DeviceModel.AppleTV4KGen2: "Apple TV 4K (gen 2)",
         DeviceModel.Music: "Music/iTunes",
-        DeviceModel.AppleTV4KGen3: "Apple TV 4K (gen3)",
+        DeviceModel.AppleTV4KGen3: "Apple TV 4K (gen 3)",
+        DeviceModel.HomePodGen2: "HomePod (gen 2)",
     }.get(device_model, "Unknown")
diff --git a/pyatv/support/device_info.py b/pyatv/support/device_info.py
@@ -18,6 +18,7 @@
     "AudioAccessory1,1": DeviceModel.HomePod,
     "AudioAccessory1,2": DeviceModel.HomePod,
     "AudioAccessory5,1": DeviceModel.HomePodMini,
+    "AudioAccessory6,1": DeviceModel.HomePodGen2,
 }
 
 

diff --git a/tests/test_convert.py b/tests/test_convert.py
@@ -95,9 +95,10 @@ def test_protocol_str(protocol, output):
         (DeviceModel.HomePodMini, "HomePod Mini"),
         (DeviceModel.AirPortExpress, "AirPort Express (gen 1)"),
         (DeviceModel.AirPortExpressGen2, "AirPort Express (gen 2)"),
-        (DeviceModel.AppleTV4KGen2, "Apple TV 4K (gen2)"),
+        (DeviceModel.AppleTV4KGen2, "Apple TV 4K (gen 2)"),
         (DeviceModel.Music, "Music/iTunes"),
-        (DeviceModel.AppleTV4KGen3, "Apple TV 4K (gen3)"),
+        (DeviceModel.AppleTV4KGen3, "Apple TV 4K (gen 3)"),
+        (DeviceModel.HomePodGen2, "HomePod (gen 2)"),
         (1234, "Unknown"),
     ],
 )