Add new stream filter

Teslemetry · Mar 18, 2024 · 110e544 · 110e544
1 parent f5056ae
commit 110e544
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/config/configuration.yaml b/config/configuration.yaml
@@ -1,6 +1,8 @@
 # https://www.home-assistant.io/integrations/default_config/
 default_config:
 
+ffmpeg:
+
 # https://www.home-assistant.io/integrations/logger/
 logger:
   default: info

diff --git a/custom_components/teslemetry/entity.py b/custom_components/teslemetry/entity.py
@@ -57,7 +57,10 @@ async def async_added_to_hass(self) -> None:
         await super().async_added_to_hass()
         if self.stream.server:
             self.async_on_remove(
-                self.stream.async_add_listener(self._handle_stream_update)
+                self.stream.async_add_listener(
+                    self._handle_stream_update,
+                    {"vin": self.vin, "data": {self.streaming_key: None}},
+                )
             )
 
     def _handle_stream_update(self, data: dict[str, Any]) -> None:
@@ -179,13 +182,14 @@ async def async_added_to_hass(self) -> None:
         await super().async_added_to_hass()
         if self.stream.server and self.streaming_key:
             self.async_on_remove(
-                self.stream.async_add_listener(self._handle_stream_update)
+                self.stream.async_add_listener(
+                    self._handle_stream_update,
+                    {"vin": self.vin, "data": {self.streaming_key: None}},
+                )
             )
 
     def _handle_stream_update(self, data: dict[str, Any]) -> None:
         """Handle updated data from the stream."""
-        if (value := data["data"].get(self.streaming_key)) is None:
-            return
         if data["timestamp"] < self._last_update:
             LOGGER.warning(
                 "Streaming data of %s was %s seconds older than polling data",
@@ -194,7 +198,7 @@ def _handle_stream_update(self, data: dict[str, Any]) -> None:
             )
             return
         self._last_update = data["timestamp"]
-        self._async_value_from_stream(value)
+        self._async_value_from_stream(data["data"][self.streaming_key])
         self.async_write_ha_state()
 
     def _handle_coordinator_update(self) -> None:

diff --git a/custom_components/teslemetry/manifest.json b/custom_components/teslemetry/manifest.json
@@ -14,7 +14,7 @@
   ],
   "requirements": [
     "tesla-fleet-api==0.4.9",
-    "teslemetry-stream==0.1.5"
+    "teslemetry-stream==0.2.0"
   ],
   "version": "1.5.6"
 }