mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-08-29 22:19:37 +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): async def async_setup_entry(hass, config_entry):
"""Set up Mikrotik Router as config entry.""" """Set up Mikrotik Router as config entry."""
mikrotik_controller = MikrotikControllerData(hass, config_entry) controller = MikrotikControllerData(hass, config_entry)
await mikrotik_controller.async_hwinfo_update() 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() raise ConfigEntryNotReady()
await mikrotik_controller.async_init() await controller.async_init()
hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] = mikrotik_controller hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] = controller
hass.async_create_task( hass.async_create_task(
hass.config_entries.async_forward_entry_setup(config_entry, "sensor") 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 = await hass.helpers.device_registry.async_get_registry()
device_registry.async_get_or_create( device_registry.async_get_or_create(
config_entry_id=config_entry.entry_id, config_entry_id=config_entry.entry_id,
manufacturer=mikrotik_controller.data["resource"]["platform"], manufacturer=controller.data["resource"]["platform"],
model=mikrotik_controller.data["routerboard"]["model"], model=controller.data["routerboard"]["model"],
name=mikrotik_controller.data["routerboard"]["model"], name=controller.data["routerboard"]["model"],
sw_version=mikrotik_controller.data["resource"]["version"], sw_version=controller.data["resource"]["version"],
) )
return True return True
@ -74,13 +74,13 @@ async def async_setup_entry(hass, config_entry):
# --------------------------- # ---------------------------
async def async_unload_entry(hass, config_entry): async def async_unload_entry(hass, config_entry):
"""Unload a 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, "sensor")
await hass.config_entries.async_forward_entry_unload(config_entry, await hass.config_entries.async_forward_entry_unload(config_entry,
"binary_sensor") "binary_sensor")
await hass.config_entries.async_forward_entry_unload(config_entry, await hass.config_entries.async_forward_entry_unload(config_entry,
"device_tracker") "device_tracker")
await hass.config_entries.async_forward_entry_unload(config_entry, "switch") 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) hass.data[DOMAIN][DATA_CLIENT].pop(config_entry.entry_id)
return True return True