bd335e1ac1
* First draft of area registry * Refactor based on input * Add tests for areas Add tests for updating device * Updating a device shouldn't require area * Fix Martins comment * Require admin * Save after deleting * Rename read to list_areas Fix device entry_dict Remove area id from device when deleting area * Fix tests
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
"""HTTP views to interact with the device registry."""
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components import websocket_api
|
|
from homeassistant.components.websocket_api.decorators import (
|
|
async_response, require_admin)
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers.device_registry import async_get_registry
|
|
|
|
DEPENDENCIES = ['websocket_api']
|
|
|
|
WS_TYPE_LIST = 'config/device_registry/list'
|
|
SCHEMA_WS_LIST = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
|
|
vol.Required('type'): WS_TYPE_LIST,
|
|
})
|
|
|
|
WS_TYPE_UPDATE = 'config/device_registry/update'
|
|
SCHEMA_WS_UPDATE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
|
|
vol.Required('type'): WS_TYPE_UPDATE,
|
|
vol.Required('device_id'): str,
|
|
vol.Optional('area_id'): str,
|
|
})
|
|
|
|
|
|
async def async_setup(hass):
|
|
"""Enable the Device Registry views."""
|
|
hass.components.websocket_api.async_register_command(
|
|
WS_TYPE_LIST, websocket_list_devices,
|
|
SCHEMA_WS_LIST
|
|
)
|
|
hass.components.websocket_api.async_register_command(
|
|
WS_TYPE_UPDATE, websocket_update_device, SCHEMA_WS_UPDATE
|
|
)
|
|
return True
|
|
|
|
|
|
@async_response
|
|
async def websocket_list_devices(hass, connection, msg):
|
|
"""Handle list devices command."""
|
|
registry = await async_get_registry(hass)
|
|
connection.send_message(websocket_api.result_message(
|
|
msg['id'], [_entry_dict(entry) for entry in registry.devices.values()]
|
|
))
|
|
|
|
|
|
@require_admin
|
|
@async_response
|
|
async def websocket_update_device(hass, connection, msg):
|
|
"""Handle update area websocket command."""
|
|
registry = await async_get_registry(hass)
|
|
|
|
entry = registry.async_update_device(
|
|
msg['device_id'], area_id=msg['area_id'])
|
|
|
|
connection.send_message(websocket_api.result_message(
|
|
msg['id'], _entry_dict(entry)
|
|
))
|
|
|
|
|
|
@callback
|
|
def _entry_dict(entry):
|
|
"""Convert entry to API format."""
|
|
return {
|
|
'config_entries': list(entry.config_entries),
|
|
'connections': list(entry.connections),
|
|
'manufacturer': entry.manufacturer,
|
|
'model': entry.model,
|
|
'name': entry.name,
|
|
'sw_version': entry.sw_version,
|
|
'id': entry.id,
|
|
'hub_device_id': entry.hub_device_id,
|
|
'area_id': entry.area_id,
|
|
}
|