mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-08-03 09:24:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

changed controller object name

Browse source
This commit is contained in:
tomaae 2020-04-11 05:35:40 +02:00
parent 6c88cd9b06
commit 8f89051be6
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
custom_components/mikrotik_router/__init__.py
View file

@ -28,16 +28,16 @@ async def async_setup(hass, _config):
# ---------------------------
async def async_setup_entry(hass, config_entry):
"""Set up Mikrotik Router as config entry."""
mikrotik_controller = MikrotikControllerData(hass, config_entry)
await mikrotik_controller.async_hwinfo_update()
controller = MikrotikControllerData(hass, config_entry)
await controller.async_hwinfo_update()
await mikrotik_controller.async_update()
await controller.async_update()
if not mikrotik_controller.data:
if not controller.data:
raise ConfigEntryNotReady()
await mikrotik_controller.async_init()
hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] = mikrotik_controller
await controller.async_init()
hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] = controller
hass.async_create_task(
hass.config_entries.async_forward_entry_setup(config_entry, "sensor")
@ -60,10 +60,10 @@ async def async_setup_entry(hass, config_entry):
device_registry = await hass.helpers.device_registry.async_get_registry()
device_registry.async_get_or_create(
config_entry_id=config_entry.entry_id,
manufacturer=mikrotik_controller.data["resource"]["platform"],
model=mikrotik_controller.data["routerboard"]["model"],
name=mikrotik_controller.data["routerboard"]["model"],
sw_version=mikrotik_controller.data["resource"]["version"],
manufacturer=controller.data["resource"]["platform"],
model=controller.data["routerboard"]["model"],
name=controller.data["routerboard"]["model"],
sw_version=controller.data["resource"]["version"],
)
return True
@ -74,13 +74,13 @@ async def async_setup_entry(hass, config_entry):
# ---------------------------
async def async_unload_entry(hass, config_entry):
"""Unload a config entry."""
mikrotik_controller = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id]
controller = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id]
await hass.config_entries.async_forward_entry_unload(config_entry, "sensor")
await hass.config_entries.async_forward_entry_unload(config_entry,
"binary_sensor")
await hass.config_entries.async_forward_entry_unload(config_entry,
"device_tracker")
await hass.config_entries.async_forward_entry_unload(config_entry, "switch")
await mikrotik_controller.async_reset()
await controller.async_reset()
hass.data[DOMAIN][DATA_CLIENT].pop(config_entry.entry_id)
return True