Allow bytearray for mqtt payload type

cdce8p · Dec 11, 2024 · bc64d09 · bc64d09
1 parent 9c9e82a
commit bc64d09
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/homeassistant/components/mqtt/client.py b/homeassistant/components/mqtt/client.py
@@ -119,7 +119,7 @@
 
 type SocketType = socket.socket | ssl.SSLSocket | mqtt.WebsocketWrapper | Any
 
-type SubscribePayloadType = str | bytes  # Only bytes if encoding is None
+type SubscribePayloadType = str | bytes | bytearray  # Only bytes if encoding is None
 
 
 def publish(

diff --git a/homeassistant/components/mqtt/switch.py b/homeassistant/components/mqtt/switch.py
@@ -91,7 +91,7 @@ class MqttSwitch(MqttEntity, SwitchEntity, RestoreEntity):
     _entity_id_format = switch.ENTITY_ID_FORMAT
 
     _optimistic: bool
-    _is_on_map: dict[str | bytes, bool | None]
+    _is_on_map: dict[str | bytes | bytearray, bool | None]
     _command_template: Callable[[PublishPayloadType], PublishPayloadType]
     _value_template: Callable[[ReceivePayloadType], ReceivePayloadType]
 

diff --git a/homeassistant/helpers/service_info/mqtt.py b/homeassistant/helpers/service_info/mqtt.py
@@ -4,7 +4,7 @@
 
 from homeassistant.data_entry_flow import BaseServiceInfo
 
-type ReceivePayloadType = str | bytes
+type ReceivePayloadType = str | bytes | bytearray
 
 
 @dataclass(slots=True)