jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-05-02 05:07:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add type hint on inherrited attribute _message_callback for MQTT mixin classes (#142011)

Browse Source
This commit is contained in:
Jan Bouwhuis 2025-04-01 20:28:11 +02:00 committed by GitHub
parent e7fadcda7b
commit 177fff3ff0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/mqtt/entity.py
View File

@ -399,6 +399,9 @@ class MqttAttributesMixin(Entity):
_attributes_extra_blocked: frozenset[str] = frozenset() _attributes_extra_blocked: frozenset[str] = frozenset()
_attr_tpl: Callable[[ReceivePayloadType], ReceivePayloadType] | None = None _attr_tpl: Callable[[ReceivePayloadType], ReceivePayloadType] | None = None
_message_callback: Callable[
[MessageCallbackType, set[str] | None, ReceiveMessage], None
]
def __init__(self, config: ConfigType) -> None: def __init__(self, config: ConfigType) -> None:
"""Initialize the JSON attributes mixin.""" """Initialize the JSON attributes mixin."""
@ -433,7 +436,7 @@ class MqttAttributesMixin(Entity):
CONF_JSON_ATTRS_TOPIC: { CONF_JSON_ATTRS_TOPIC: {
"topic": self._attributes_config.get(CONF_JSON_ATTRS_TOPIC), "topic": self._attributes_config.get(CONF_JSON_ATTRS_TOPIC),
"msg_callback": partial( "msg_callback": partial(
self._message_callback, # type: ignore[attr-defined] self._message_callback,
self._attributes_message_received, self._attributes_message_received,
{"_attr_extra_state_attributes"}, {"_attr_extra_state_attributes"},
), ),
@ -482,6 +485,10 @@ class MqttAttributesMixin(Entity):
class MqttAvailabilityMixin(Entity): class MqttAvailabilityMixin(Entity):
"""Mixin used for platforms that report availability.""" """Mixin used for platforms that report availability."""
_message_callback: Callable[
[MessageCallbackType, set[str] | None, ReceiveMessage], None
]
def __init__(self, config: ConfigType) -> None: def __init__(self, config: ConfigType) -> None:
"""Initialize the availability mixin.""" """Initialize the availability mixin."""
self._availability_sub_state: dict[str, EntitySubscription] = {} self._availability_sub_state: dict[str, EntitySubscription] = {}
@ -547,7 +554,7 @@ class MqttAvailabilityMixin(Entity):
f"availability_{topic}": { f"availability_{topic}": {
"topic": topic, "topic": topic,
"msg_callback": partial( "msg_callback": partial(
self._message_callback, # type: ignore[attr-defined] self._message_callback,
self._availability_message_received, self._availability_message_received,
{"available"}, {"available"},
), ),