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

converted switches

Browse source
This commit is contained in:
Tomaae 2023-08-09 09:37:48 +02:00
parent 2ea35dec75
commit 727a075dbb
No known key found for this signature in database
GPG key ID: 2F97770867DAA4E6
3 changed files with 39 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

29
custom_components/mikrotik_router/switch.py
View file

@ -1,12 +1,18 @@
"""Support for the Mikrotik Router switches."""
from __future__ import annotations
import logging
from typing import Any, Optional
from logging import getLogger
from collections.abc import Mapping
from typing import Any, Optional
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .entity import MikrotikEntity, async_add_entities
from .helper import format_attribute
from .entity import model_async_setup_entry, MikrotikEntity
from .switch_types import (
SENSOR_TYPES,
SENSOR_SERVICES,
@ -15,13 +21,17 @@ from .switch_types import (
DEVICE_ATTRIBUTES_IFACE_WIRELESS,
)
_LOGGER = logging.getLogger(__name__)
_LOGGER = getLogger(__name__)
# ---------------------------
# async_setup_entry
# ---------------------------
async def async_setup_entry(hass, config_entry, async_add_entities):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
_async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up entry for component"""
dispatcher = {
"MikrotikSwitch": MikrotikSwitch,
@ -32,14 +42,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
"MikrotikQueueSwitch": MikrotikQueueSwitch,
"MikrotikKidcontrolPauseSwitch": MikrotikKidcontrolPauseSwitch,
}
await model_async_setup_entry(
hass,
config_entry,
async_add_entities,
SENSOR_SERVICES,
SENSOR_TYPES,
dispatcher,
)
await async_add_entities(hass, config_entry, dispatcher)
# ---------------------------