Add get entry subentry method

Author: mj23000

homeassistant_api/models/__init__.py

@@ -6,6 +6,7 @@
     ConfigEntryDisabler,
     ConfigEntryState,
     ConfigFlowContext,
+    ConfigSubEntry,
     DisableEnableResult,
     DiscoveryKey,
     FlowContext,
@@ -43,4 +44,5 @@
     "ConfigEntryDisabler",
     "ConfigEntryState",
     "ConfigEntry",
+    "ConfigSubEntry",
 )

homeassistant_api/models/config_entries.py

@@ -2,7 +2,7 @@
 
 import asyncio
 from enum import Enum
-from typing import Any, Container, Dict, Tuple, Union
+from typing import Any, Container, Dict, Optional, Tuple, Union
 
 from .base import BaseModel
 
@@ -134,3 +134,12 @@ class ConfigEntry(BaseModel):
     error_reason_translation_key: str | None
     error_reason_translation_placeholders: dict[str, Any] | None
     num_subentries: int
+
+
+class ConfigSubEntry(BaseModel):
+    """A configuration sub-entry. This is the model that Home Assistant returns, but not what is used internally."""
+
+    subentry_id: str
+    subentry_type: str
+    title: str
+    unique_id: Optional[str]

homeassistant_api/websocket.py

@@ -5,6 +5,7 @@
 
 from homeassistant_api.models import (
     ConfigEntry,
+    ConfigSubEntry,
     DisableEnableResult,
     Domain,
     Entity,
@@ -266,6 +267,24 @@ def get_config_entries(
             )
         )
 
+    # TODO: config_entries/subscribe
+
+    # UNTESTED
+    def get_entry_subentries(self, entry_id: str) -> Tuple[ConfigSubEntry, ...]:
+        """Get an entry's sub-entries."""
+        return tuple(
+            ConfigSubEntry.from_json(sub_entry)
+            for sub_entry in cast(
+                list[dict[str, JSONType]],
+                cast(
+                    ResultResponse,
+                    self.recv(
+                        self.send("config_entries/subentries/list", entry_id=entry_id)
+                    ),
+                ).result,
+            )
+        )
+
     def trigger_service(
         self,
         domain: str,