Fix zwave_js api typing (#50923)

This commit is contained in:
Martin Hjelmare 2021-05-21 14:20:58 +02:00 committed by GitHub
parent 07e2f53b37
commit b4bb7c38ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -142,14 +142,14 @@ def async_register_api(hass: HomeAssistant) -> None:
hass, websocket_update_data_collection_preference hass, websocket_update_data_collection_preference
) )
websocket_api.async_register_command(hass, websocket_data_collection_status) websocket_api.async_register_command(hass, websocket_data_collection_status)
hass.http.register_view(DumpView) # type: ignore hass.http.register_view(DumpView())
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{vol.Required(TYPE): "zwave_js/network_status", vol.Required(ENTRY_ID): str} {vol.Required(TYPE): "zwave_js/network_status", vol.Required(ENTRY_ID): str}
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_network_status( async def websocket_network_status(
hass: HomeAssistant, hass: HomeAssistant,
@ -177,7 +177,6 @@ async def websocket_network_status(
) )
@websocket_api.async_response # type: ignore
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/node_status", vol.Required(TYPE): "zwave_js/node_status",
@ -185,6 +184,7 @@ async def websocket_network_status(
vol.Required(NODE_ID): int, vol.Required(NODE_ID): int,
} }
) )
@websocket_api.async_response
@async_get_node @async_get_node
async def websocket_node_status( async def websocket_node_status(
hass: HomeAssistant, hass: HomeAssistant,
@ -206,8 +206,7 @@ async def websocket_node_status(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/add_node", vol.Required(TYPE): "zwave_js/add_node",
@ -215,6 +214,7 @@ async def websocket_node_status(
vol.Optional("secure", default=False): bool, vol.Optional("secure", default=False): bool,
} }
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_add_node( async def websocket_add_node(
hass: HomeAssistant, hass: HomeAssistant,
@ -300,14 +300,14 @@ async def websocket_add_node(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/stop_inclusion", vol.Required(TYPE): "zwave_js/stop_inclusion",
vol.Required(ENTRY_ID): str, vol.Required(ENTRY_ID): str,
} }
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_stop_inclusion( async def websocket_stop_inclusion(
hass: HomeAssistant, hass: HomeAssistant,
@ -325,14 +325,14 @@ async def websocket_stop_inclusion(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/stop_exclusion", vol.Required(TYPE): "zwave_js/stop_exclusion",
vol.Required(ENTRY_ID): str, vol.Required(ENTRY_ID): str,
} }
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_stop_exclusion( async def websocket_stop_exclusion(
hass: HomeAssistant, hass: HomeAssistant,
@ -350,14 +350,14 @@ async def websocket_stop_exclusion(
) )
@websocket_api.require_admin # type:ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/remove_node", vol.Required(TYPE): "zwave_js/remove_node",
vol.Required(ENTRY_ID): str, vol.Required(ENTRY_ID): str,
} }
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_remove_node( async def websocket_remove_node(
hass: HomeAssistant, hass: HomeAssistant,
@ -409,8 +409,7 @@ async def websocket_remove_node(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/refresh_node_info", vol.Required(TYPE): "zwave_js/refresh_node_info",
@ -418,6 +417,7 @@ async def websocket_remove_node(
vol.Required(NODE_ID): int, vol.Required(NODE_ID): int,
}, },
) )
@websocket_api.async_response
@async_get_node @async_get_node
async def websocket_refresh_node_info( async def websocket_refresh_node_info(
hass: HomeAssistant, hass: HomeAssistant,
@ -459,8 +459,7 @@ async def websocket_refresh_node_info(
connection.send_result(msg[ID], result) connection.send_result(msg[ID], result)
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/refresh_node_values", vol.Required(TYPE): "zwave_js/refresh_node_values",
@ -468,6 +467,7 @@ async def websocket_refresh_node_info(
vol.Required(NODE_ID): int, vol.Required(NODE_ID): int,
}, },
) )
@websocket_api.async_response
@async_get_node @async_get_node
async def websocket_refresh_node_values( async def websocket_refresh_node_values(
hass: HomeAssistant, hass: HomeAssistant,
@ -480,8 +480,7 @@ async def websocket_refresh_node_values(
connection.send_result(msg[ID]) connection.send_result(msg[ID])
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/refresh_node_cc_values", vol.Required(TYPE): "zwave_js/refresh_node_cc_values",
@ -490,6 +489,7 @@ async def websocket_refresh_node_values(
vol.Required(COMMAND_CLASS_ID): int, vol.Required(COMMAND_CLASS_ID): int,
}, },
) )
@websocket_api.async_response
@async_get_node @async_get_node
async def websocket_refresh_node_cc_values( async def websocket_refresh_node_cc_values(
hass: HomeAssistant, hass: HomeAssistant,
@ -512,8 +512,7 @@ async def websocket_refresh_node_cc_values(
connection.send_result(msg[ID]) connection.send_result(msg[ID])
@websocket_api.require_admin # type:ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/set_config_parameter", vol.Required(TYPE): "zwave_js/set_config_parameter",
@ -524,6 +523,7 @@ async def websocket_refresh_node_cc_values(
vol.Required(VALUE): int, vol.Required(VALUE): int,
} }
) )
@websocket_api.async_response
@async_get_node @async_get_node
async def websocket_set_config_parameter( async def websocket_set_config_parameter(
hass: HomeAssistant, hass: HomeAssistant,
@ -563,8 +563,7 @@ async def websocket_set_config_parameter(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/get_config_parameters", vol.Required(TYPE): "zwave_js/get_config_parameters",
@ -572,6 +571,7 @@ async def websocket_set_config_parameter(
vol.Required(NODE_ID): int, vol.Required(NODE_ID): int,
} }
) )
@websocket_api.async_response
@async_get_node @async_get_node
async def websocket_get_config_parameters( async def websocket_get_config_parameters(
hass: HomeAssistant, connection: ActiveConnection, msg: dict, node: Node hass: HomeAssistant, connection: ActiveConnection, msg: dict, node: Node
@ -613,14 +613,14 @@ def filename_is_present_if_logging_to_file(obj: dict) -> dict:
return obj return obj
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/subscribe_logs", vol.Required(TYPE): "zwave_js/subscribe_logs",
vol.Required(ENTRY_ID): str, vol.Required(ENTRY_ID): str,
} }
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_subscribe_logs( async def websocket_subscribe_logs(
hass: HomeAssistant, hass: HomeAssistant,
@ -660,8 +660,7 @@ async def websocket_subscribe_logs(
connection.send_result(msg[ID]) connection.send_result(msg[ID])
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/update_log_config", vol.Required(TYPE): "zwave_js/update_log_config",
@ -688,6 +687,7 @@ async def websocket_subscribe_logs(
), ),
}, },
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_update_log_config( async def websocket_update_log_config(
hass: HomeAssistant, hass: HomeAssistant,
@ -703,14 +703,14 @@ async def websocket_update_log_config(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/get_log_config", vol.Required(TYPE): "zwave_js/get_log_config",
vol.Required(ENTRY_ID): str, vol.Required(ENTRY_ID): str,
}, },
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_get_log_config( async def websocket_get_log_config(
hass: HomeAssistant, hass: HomeAssistant,
@ -727,8 +727,7 @@ async def websocket_get_log_config(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/update_data_collection_preference", vol.Required(TYPE): "zwave_js/update_data_collection_preference",
@ -736,6 +735,7 @@ async def websocket_get_log_config(
vol.Required(OPTED_IN): bool, vol.Required(OPTED_IN): bool,
}, },
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_update_data_collection_preference( async def websocket_update_data_collection_preference(
hass: HomeAssistant, hass: HomeAssistant,
@ -758,14 +758,14 @@ async def websocket_update_data_collection_preference(
) )
@websocket_api.require_admin # type: ignore @websocket_api.require_admin
@websocket_api.async_response
@websocket_api.websocket_command( @websocket_api.websocket_command(
{ {
vol.Required(TYPE): "zwave_js/data_collection_status", vol.Required(TYPE): "zwave_js/data_collection_status",
vol.Required(ENTRY_ID): str, vol.Required(ENTRY_ID): str,
}, },
) )
@websocket_api.async_response
@async_get_entry @async_get_entry
async def websocket_data_collection_status( async def websocket_data_collection_status(
hass: HomeAssistant, hass: HomeAssistant,