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

Convert WebSocket messages to bytes before passing them to send_message (#129300)

Browse Source
This commit is contained in:
J. Nick Koston 2024-10-28 01:21:12 -10:00 committed by GitHub
parent f7ad40263b
commit 4749af6e90
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
homeassistant/components/websocket_api/connection.py
View File

@ -16,6 +16,12 @@ from homeassistant.helpers.http import current_request
from homeassistant.util.json import JsonValueType from homeassistant.util.json import JsonValueType
from . import const, messages from . import const, messages
from .messages import (
error_message,
event_message,
message_to_json_bytes,
result_message,
)
from .util import describe_request from .util import describe_request
if TYPE_CHECKING: if TYPE_CHECKING:
@ -126,12 +132,12 @@ class ActiveConnection:
@callback @callback
def send_result(self, msg_id: int, result: Any | None = None) -> None: def send_result(self, msg_id: int, result: Any | None = None) -> None:
"""Send a result message.""" """Send a result message."""
self.send_message(messages.result_message(msg_id, result)) self.send_message(message_to_json_bytes(result_message(msg_id, result)))
@callback @callback
def send_event(self, msg_id: int, event: Any | None = None) -> None: def send_event(self, msg_id: int, event: Any | None = None) -> None:
"""Send a event message.""" """Send a event message."""
self.send_message(messages.event_message(msg_id, event)) self.send_message(message_to_json_bytes(event_message(msg_id, event)))
@callback @callback
def send_error( def send_error(
@ -145,13 +151,15 @@ class ActiveConnection:
) -> None: ) -> None:
"""Send an error message.""" """Send an error message."""
self.send_message( self.send_message(
messages.error_message( message_to_json_bytes(
msg_id, error_message(
code, msg_id,
message, code,
translation_key=translation_key, message,
translation_domain=translation_domain, translation_key=translation_key,
translation_placeholders=translation_placeholders, translation_domain=translation_domain,
translation_placeholders=translation_placeholders,
)
) )
) )