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

ps - fix websocket yielding pt2 (#7434)

Browse Source
This commit is contained in:
Paulus Schoutsen 2017-05-03 21:12:08 -07:00 committed by GitHub
parent ad5a9bf5ac
commit fab533de87
1 changed files with 45 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
homeassistant/components/websocket_api.py
View File

@ -289,7 +289,7 @@ class ActiveConnection:
else: else:
handler_name = 'handle_{}'.format(msg['type']) handler_name = 'handle_{}'.format(msg['type'])
getattr(self, handler_name)(msg) yield from getattr(self, handler_name)(msg)
last_id = cur_id last_id = cur_id
msg = yield from wsock.receive_json() msg = yield from wsock.receive_json()
@ -348,18 +348,19 @@ class ActiveConnection:
return wsock return wsock
@asyncio.coroutine
def handle_subscribe_events(self, msg): def handle_subscribe_events(self, msg):
"""Handle subscribe events command.""" """Handle subscribe events command."""
msg = SUBSCRIBE_EVENTS_MESSAGE_SCHEMA(msg) msg = SUBSCRIBE_EVENTS_MESSAGE_SCHEMA(msg)
@callback @asyncio.coroutine
def forward_events(event): def forward_events(event):
"""Forward events to websocket.""" """Forward events to websocket."""
if event.event_type == EVENT_TIME_CHANGED: if event.event_type == EVENT_TIME_CHANGED:
return return
try: try:
self.send_message(event_message(msg['id'], event)) yield from self.send_message(event_message(msg['id'], event))
except RuntimeError: except RuntimeError:
# Socket has been closed. # Socket has been closed.
pass pass
@ -367,24 +368,31 @@ class ActiveConnection:
self.event_listeners[msg['id']] = self.hass.bus.async_listen( self.event_listeners[msg['id']] = self.hass.bus.async_listen(
msg['event_type'], forward_events) msg['event_type'], forward_events)
self.send_message(result_message(msg['id'])) return self.send_message(result_message(msg['id']))
def handle_unsubscribe_events(self, msg): def handle_unsubscribe_events(self, msg):
"""Handle unsubscribe events command.""" """Handle unsubscribe events command.
Returns a coroutine object.
"""
msg = UNSUBSCRIBE_EVENTS_MESSAGE_SCHEMA(msg) msg = UNSUBSCRIBE_EVENTS_MESSAGE_SCHEMA(msg)
subscription = msg['subscription'] subscription = msg['subscription']
if subscription not in self.event_listeners: if subscription in self.event_listeners:
self.send_message(error_message( self.event_listeners.pop(subscription)()
return self.send_message(result_message(msg['id']))
else:
return self.send_message(error_message(
msg['id'], ERR_NOT_FOUND, msg['id'], ERR_NOT_FOUND,
'Subscription not found.')) 'Subscription not found.'))
else:
self.event_listeners.pop(subscription)()
self.send_message(result_message(msg['id']))
@asyncio.coroutine
def handle_call_service(self, msg): def handle_call_service(self, msg):
"""Handle call service command.""" """Handle call service command.
This is a coroutine.
"""
msg = CALL_SERVICE_MESSAGE_SCHEMA(msg) msg = CALL_SERVICE_MESSAGE_SCHEMA(msg)
@asyncio.coroutine @asyncio.coroutine
@ -393,7 +401,7 @@ class ActiveConnection:
yield from self.hass.services.async_call( yield from self.hass.services.async_call(
msg['domain'], msg['service'], msg['service_data'], True) msg['domain'], msg['service'], msg['service_data'], True)
try: try:
self.send_message(result_message(msg['id'])) yield from self.send_message(result_message(msg['id']))
except RuntimeError: except RuntimeError:
# Socket has been closed. # Socket has been closed.
pass pass
@ -401,33 +409,48 @@ class ActiveConnection:
self.hass.async_add_job(call_service_helper(msg)) self.hass.async_add_job(call_service_helper(msg))
def handle_get_states(self, msg): def handle_get_states(self, msg):
"""Handle get states command.""" """Handle get states command.
Returns a coroutine object.
"""
msg = GET_STATES_MESSAGE_SCHEMA(msg) msg = GET_STATES_MESSAGE_SCHEMA(msg)
self.send_message(result_message( return self.send_message(result_message(
msg['id'], self.hass.states.async_all())) msg['id'], self.hass.states.async_all()))
def handle_get_services(self, msg): def handle_get_services(self, msg):
"""Handle get services command.""" """Handle get services command.
Returns a coroutine object.
"""
msg = GET_SERVICES_MESSAGE_SCHEMA(msg) msg = GET_SERVICES_MESSAGE_SCHEMA(msg)
self.send_message(result_message( return self.send_message(result_message(
msg['id'], self.hass.services.async_services())) msg['id'], self.hass.services.async_services()))
def handle_get_config(self, msg): def handle_get_config(self, msg):
"""Handle get config command.""" """Handle get config command.
Returns a coroutine object.
"""
msg = GET_CONFIG_MESSAGE_SCHEMA(msg) msg = GET_CONFIG_MESSAGE_SCHEMA(msg)
self.send_message(result_message( return self.send_message(result_message(
msg['id'], self.hass.config.as_dict())) msg['id'], self.hass.config.as_dict()))
def handle_get_panels(self, msg): def handle_get_panels(self, msg):
"""Handle get panels command.""" """Handle get panels command.
Returns a coroutine object.
"""
msg = GET_PANELS_MESSAGE_SCHEMA(msg) msg = GET_PANELS_MESSAGE_SCHEMA(msg)
self.send_message(result_message( return self.send_message(result_message(
msg['id'], self.hass.data[frontend.DATA_PANELS])) msg['id'], self.hass.data[frontend.DATA_PANELS]))
def handle_ping(self, msg): def handle_ping(self, msg):
"""Handle ping command.""" """Handle ping command.
self.send_message(pong_message(msg['id']))
Returns a coroutine object.
"""
return self.send_message(pong_message(msg['id']))