ucapi-framework 1.2.3__tar.gz → 1.3.0__tar.gz

{ucapi_framework-1.2.3 → ucapi_framework-1.3.0}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: ucapi-framework
-Version: 1.2.3
+Version: 1.3.0
 Summary: ucapi framework that provides core functionality for building integrations.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: pyee>=9.0.0
-Requires-Dist: ucapi>=0.4.0
+Requires-Dist: ucapi>=0.5.0
 Requires-Dist: aiohttp>=3.9.0
 
 [![Tests](https://github.com/jackjpowell/ucapi-framework/actions/workflows/test.yml/badge.svg)](https://github.com/jackjpowell/ucapi-framework/actions/workflows/test.yml)

{ucapi_framework-1.2.3 → ucapi_framework-1.3.0}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "ucapi-framework"
-version = "1.2.3"
+version = "1.3.0"
 description = "ucapi framework that provides core functionality for building integrations."
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "pyee>=9.0.0",
-    "ucapi>=0.4.0",
+    "ucapi>=0.5.0",
     "aiohttp>=3.9.0",
 ]
 

{ucapi_framework-1.2.3 → ucapi_framework-1.3.0}/tests/test_driver.py

@@ -1958,6 +1958,7 @@ class TestDeviceEventHandlersEntityTypes:
             EntityTypes.SENSOR,
             EntityTypes.SWITCH,
             EntityTypes.IR_EMITTER,
+            EntityTypes.VOICE_ASSISTANT,
         ]:
             mock_entity = MagicMock()
             mock_entity.entity_type = entity_type
@@ -1985,6 +1986,7 @@ class TestDeviceEventHandlersEntityTypes:
             EntityTypes.SENSOR,
             EntityTypes.SWITCH,
             EntityTypes.IR_EMITTER,
+            EntityTypes.VOICE_ASSISTANT,
         ]:
             mock_entity = MagicMock()
             mock_entity.entity_type = entity_type
@@ -2012,6 +2014,7 @@ class TestDeviceEventHandlersEntityTypes:
             EntityTypes.SENSOR,
             EntityTypes.SWITCH,
             EntityTypes.IR_EMITTER,
+            EntityTypes.VOICE_ASSISTANT,
         ]:
             mock_entity = MagicMock()
             mock_entity.entity_type = entity_type
@@ -2200,6 +2203,23 @@ class TestOnDeviceUpdateEntityTypes:
 
         driver.api.configured_entities.update_attributes.assert_called()
 
+    @pytest.mark.asyncio
+    async def test_on_device_update_voice_assistant(self):
+        """Test on_device_update for voice assistant entity."""
+        driver = self._create_driver()
+        config = DeviceConfigForTests("dev1", "Device 1", "192.168.1.1")
+        driver.add_configured_device(config, connect=False)
+
+        mock_entity = MagicMock()
+        mock_entity.entity_type = EntityTypes.VOICE_ASSISTANT
+        driver.api.configured_entities.get = MagicMock(return_value=mock_entity)
+        driver.api.configured_entities.contains = MagicMock(return_value=True)
+        driver.api.configured_entities.update_attributes = MagicMock()
+
+        await driver.on_device_update("dev1", {"state": "on"})
+
+        driver.api.configured_entities.update_attributes.assert_called()
+
     @pytest.mark.asyncio
     async def test_on_device_update_unknown_entity_type(self, caplog):
         """Test on_device_update with unknown entity type logs warning."""

{ucapi_framework-1.2.3 → ucapi_framework-1.3.0}/ucapi_framework/driver.py

@@ -25,6 +25,7 @@ from ucapi import (
     remote,
     sensor,
     switch,
+    voice_assistant,
 )
 
 from ucapi_framework.config import BaseConfigManager
@@ -542,6 +543,10 @@ class BaseIntegrationDriver(ABC, Generic[DeviceT, ConfigT]):
                 self.api.configured_entities.update_attributes(
                     entity_id, {remote.Attributes.STATE: state}
                 )
+            case EntityTypes.VOICE_ASSISTANT:
+                self.api.configured_entities.update_attributes(
+                    entity_id, {voice_assistant.Attributes.STATE: state}
+                )
 
     # ========================================================================
     # Device Lifecycle Management
@@ -782,6 +787,10 @@ class BaseIntegrationDriver(ABC, Generic[DeviceT, ConfigT]):
                     self.api.configured_entities.update_attributes(
                         entity_id, {remote.Attributes.STATE: state}
                     )
+                case EntityTypes.VOICE_ASSISTANT:
+                    self.api.configured_entities.update_attributes(
+                        entity_id, {voice_assistant.Attributes.STATE: state}
+                    )
 
     async def on_device_disconnected(self, device_id: str) -> None:
         """
@@ -847,6 +856,13 @@ class BaseIntegrationDriver(ABC, Generic[DeviceT, ConfigT]):
                         entity_id,
                         {remote.Attributes.STATE: media_player.States.UNAVAILABLE},
                     )
+                case EntityTypes.VOICE_ASSISTANT:
+                    self.api.configured_entities.update_attributes(
+                        entity_id,
+                        {
+                            voice_assistant.Attributes.STATE: media_player.States.UNAVAILABLE
+                        },
+                    )
 
     async def on_device_connection_error(self, device_id: str, message: str) -> None:
         """
@@ -913,6 +929,13 @@ class BaseIntegrationDriver(ABC, Generic[DeviceT, ConfigT]):
                         entity_id,
                         {remote.Attributes.STATE: media_player.States.UNAVAILABLE},
                     )
+                case EntityTypes.VOICE_ASSISTANT:
+                    self.api.configured_entities.update_attributes(
+                        entity_id,
+                        {
+                            voice_assistant.Attributes.STATE: media_player.States.UNAVAILABLE
+                        },
+                    )
 
     async def on_device_update(
         self,
@@ -1086,6 +1109,14 @@ class BaseIntegrationDriver(ABC, Generic[DeviceT, ConfigT]):
                     state = self.map_device_state(update[remote.Attributes.STATE.value])
                     attributes[remote.Attributes.STATE] = state
 
+            case EntityTypes.VOICE_ASSISTANT:
+                # Voice Assistant entities: STATE
+                if voice_assistant.Attributes.STATE.value in update:
+                    state = self.map_device_state(
+                        update[voice_assistant.Attributes.STATE.value]
+                    )
+                    attributes[voice_assistant.Attributes.STATE] = state
+
             case _:
                 # Unknown entity type - log warning
                 _LOG.warning(

{ucapi_framework-1.2.3 → ucapi_framework-1.3.0}/ucapi_framework.egg-info/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: ucapi-framework
-Version: 1.2.3
+Version: 1.3.0
 Summary: ucapi framework that provides core functionality for building integrations.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: pyee>=9.0.0
-Requires-Dist: ucapi>=0.4.0
+Requires-Dist: ucapi>=0.5.0
 Requires-Dist: aiohttp>=3.9.0
 
 [![Tests](https://github.com/jackjpowell/ucapi-framework/actions/workflows/test.yml/badge.svg)](https://github.com/jackjpowell/ucapi-framework/actions/workflows/test.yml)

{ucapi_framework-1.2.3 → ucapi_framework-1.3.0}/ucapi_framework.egg-info/requires.txt

@@ -1,3 +1,3 @@
 pyee>=9.0.0
-ucapi>=0.4.0
+ucapi>=0.5.0
 aiohttp>=3.9.0