Add missing __call__ methods to Service classes

homeassistant_api/models/domains.py

@@ -630,6 +630,18 @@ def trigger(
                 **service_data,
             )
 
+    def __call__(
+        self,
+        **service_data: Any,
+    ) -> (
+        tuple[State, ...]
+        | tuple[tuple[State, ...], dict[str, Any]]
+        | dict[str, Any]
+        | None
+    ):
+        """Calls the service associated with this object."""
+        return self.trigger(**service_data)
+
 
 class AsyncService(BaseService):
     """Async service with async trigger method."""
@@ -658,3 +670,15 @@ async def trigger(
                 self.service_id,
                 **service_data,
             )
+
+    async def __call__(
+        self,
+        **service_data: Any,
+    ) -> (
+        tuple[State, ...]
+        | tuple[tuple[State, ...], dict[str, Any]]
+        | dict[str, Any]
+        | None
+    ):
+        """Calls the service associated with this object."""
+        return await self.trigger(**service_data)

pyproject.toml

@@ -4,7 +4,7 @@ requires = ["hatchling"]
 
 [project]
 name = "HomeAssistant-API"
-version = "6.0.0"
+version = "6.0.1"
 description = "Python Wrapper for Homeassistant's REST API"
 readme = "README.md"
 license = "GPL-3.0-or-later"

tests/test_endpoints.py

@@ -499,6 +499,17 @@ def test_trigger_service(cached_client: Client) -> None:
     logger.info(resp)
     assert isinstance(resp, tuple)
 
+def test_call_service(cached_client: Client) -> None:
+    """Tests the `POST /api/services/<domain>/<service>` endpoint."""
+    notify = cached_client.get_domain("notify")
+    assert notify is not None
+    resp = notify.persistent_notification(
+        message="Your API Test Suite just said hello!",
+        title="Test Suite Notifcation",
+    )
+    logger.info(resp)
+    assert isinstance(resp, tuple)
+
 
 async def test_async_trigger_service(async_cached_client: AsyncClient) -> None:
     """Tests the `POST /api/services/<domain>/<service>` endpoint."""
@@ -511,6 +522,17 @@ async def test_async_trigger_service(async_cached_client: AsyncClient) -> None:
     assert isinstance(resp, tuple)
 
 
+async def test_async_call_service(async_cached_client: AsyncClient) -> None:
+    """Tests the `POST /api/services/<domain>/<service>` endpoint."""
+    notify = await async_cached_client.get_domain("notify")
+    assert notify is not None
+    resp = await notify.persistent_notification(
+        message="Your API Test Suite just said hello!",
+        title="Test Suite Notifcation (Async)",
+    )
+    assert isinstance(resp, tuple)
+
+
 def test_websocket_trigger_service(websocket_client: WebsocketClient) -> None:
     """Tests the `"type": "trigger_service"` websocket command."""
     notify = websocket_client.get_domain("notify")