Add zwave to zwave_js migration

homeassistant/components/ozw/config_flow.py

@@ -35,15 +35,6 @@ def __init__(self):
         self.integration_created_addon = False
         self.install_task = None
 
-    async def async_step_import(self, data):
-        """Handle imported data.
-
-        This step will be used when importing data during zwave to ozw migration.
-        """
-        self.network_key = data.get(CONF_NETWORK_KEY)
-        self.usb_path = data.get(CONF_USB_PATH)
-        return await self.async_step_user()
-
     async def async_step_user(self, user_input=None):
         """Handle the initial step."""
         if self._async_current_entries():

homeassistant/components/ozw/websocket_api.py

@@ -25,7 +25,6 @@
 
 from .const import ATTR_CONFIG_PARAMETER, ATTR_CONFIG_VALUE, DOMAIN, MANAGER
 from .lock import ATTR_USERCODE
-from .migration import async_get_migration_data, async_migrate, map_node_values
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -58,7 +57,6 @@
 @callback
 def async_register_api(hass):
     """Register all of our api endpoints."""
-    websocket_api.async_register_command(hass, websocket_migrate_zwave)
     websocket_api.async_register_command(hass, websocket_get_instances)
     websocket_api.async_register_command(hass, websocket_get_nodes)
     websocket_api.async_register_command(hass, websocket_network_status)
@@ -168,63 +166,6 @@ def _get_config_params(node, *args):
     return config_params
 
 
-@websocket_api.require_admin
-@websocket_api.async_response
-@websocket_api.websocket_command(
-    {
-        vol.Required(TYPE): "ozw/migrate_zwave",
-        vol.Optional(DRY_RUN, default=True): bool,
-    }
-)
-async def websocket_migrate_zwave(hass, connection, msg):
-    """Migrate the zwave integration device and entity data to ozw integration."""
-    if "zwave" not in hass.config.components:
-        _LOGGER.error("Can not migrate, zwave integration is not loaded")
-        connection.send_message(
-            websocket_api.error_message(
-                msg["id"], "zwave_not_loaded", "Integration zwave is not loaded"
-            )
-        )
-        return
-
-    zwave = hass.components.zwave
-    zwave_data = await zwave.async_get_ozw_migration_data(hass)
-    _LOGGER.debug("Migration zwave data: %s", zwave_data)
-
-    ozw_data = await async_get_migration_data(hass)
-    _LOGGER.debug("Migration ozw data: %s", ozw_data)
-
-    can_migrate = map_node_values(zwave_data, ozw_data)
-
-    zwave_entity_ids = [
-        entry["entity_entry"].entity_id for entry in zwave_data.values()
-    ]
-    ozw_entity_ids = [entry["entity_entry"].entity_id for entry in ozw_data.values()]
-    migration_device_map = {
-        zwave_device_id: ozw_device_id
-        for ozw_device_id, zwave_device_id in can_migrate["device_entries"].items()
-    }
-    migration_entity_map = {
-        zwave_entry["entity_entry"].entity_id: ozw_entity_id
-        for ozw_entity_id, zwave_entry in can_migrate["entity_entries"].items()
-    }
-    _LOGGER.debug("Migration entity map: %s", migration_entity_map)
-
-    if not msg[DRY_RUN]:
-        await async_migrate(hass, can_migrate)
-
-    connection.send_result(
-        msg[ID],
-        {
-            "migration_device_map": migration_device_map,
-            "zwave_entity_ids": zwave_entity_ids,
-            "ozw_entity_ids": ozw_entity_ids,
-            "migration_entity_map": migration_entity_map,
-            "migrated": not msg[DRY_RUN],
-        },
-    )
-
-
 @websocket_api.websocket_command({vol.Required(TYPE): "ozw/get_instances"})
 def websocket_get_instances(hass, connection, msg):
     """Get a list of OZW instances."""