Ver Fonte

refine service registration

Noah Vogt há 2 meses atrás
pai
commit
1985f7a578

+ 10 - 1
custom_components/pjlink2/__init__.py

@@ -1 +1,10 @@
-"""PJLink2 Custom Component."""
+"""PJLink2 Custom Component."""
+from __future__ import annotations
+
+import logging
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.typing import ConfigType
+
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
+    """Set up the PJLink2 component."""
+    return True

+ 11 - 8
custom_components/pjlink2/media_player.py

@@ -79,6 +79,9 @@ async def async_setup_platform(
     async_add_entities: Callable,
     discovery_info: DiscoveryInfoType | None = None,
 ) -> None:
+    _LOGGER.debug("Setting up PJLink2 platform")
+    platform = entity_platform.async_get_current_platform()
+
     host = config.get(CONF_HOST)
     port = config.get(CONF_PORT)
     password = config.get(CONF_PASSWORD)
@@ -89,14 +92,14 @@ async def async_setup_platform(
     devices = [PJLink2MediaPlayer(pjl, name, sources)]
     async_add_entities(devices, update_before_add=False)
 
-    platform = entity_platform.async_get_current_platform()
-    platform.async_register_entity_service(
-        "freeze",
-        {
-            vol.Required("freeze"): cv.boolean,
-        },
-        "async_freeze",
-    )
+    if platform:
+        platform.async_register_entity_service(
+            "freeze",
+            {
+                vol.Required("freeze"): cv.boolean,
+            },
+            "async_freeze",
+        )
 
 
 class PJLink2MediaPlayer(MediaPlayerEntity):