evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Some minor cleanups in WLED (#117453)

Browse source
This commit is contained in:
Franck Nijhof 2024-05-14 21:02:31 +02:00 • committed by GitHub
parent 0df9006bf7
commit faff5f4738
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 13 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

5
homeassistant/components/wled/binary_sensor.py
View file

@ -12,7 +12,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
from .models import WLEDEntity
from .entity import WLEDEntity
async def async_setup_entry(
@ -21,10 +21,9 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up a WLED binary sensor based on a config entry."""
coordinator = entry.runtime_data
async_add_entities(
[
WLEDUpdateBinarySensor(coordinator),
WLEDUpdateBinarySensor(entry.runtime_data),
]
)

5
homeassistant/components/wled/button.py
View file

@ -9,8 +9,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
async def async_setup_entry(
@ -19,8 +19,7 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up WLED button based on a config entry."""
coordinator = entry.runtime_data
async_add_entities([WLEDRestartButton(coordinator)])
async_add_entities([WLEDRestartButton(entry.runtime_data)])
class WLEDRestartButton(WLEDEntity, ButtonEntity):

2
homeassistant/components/wled/models.py → homeassistant/components/wled/entity.py
View file

@ -1,4 +1,4 @@
"""Models for WLED."""
"""Base entity for WLED."""
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity

2
homeassistant/components/wled/helpers.py
View file

@ -9,7 +9,7 @@ from wled import WLEDConnectionError, WLEDError
from homeassistant.exceptions import HomeAssistantError
from .models import WLEDEntity
from .entity import WLEDEntity
_WLEDEntityT = TypeVar("_WLEDEntityT", bound=WLEDEntity)
_P = ParamSpec("_P")

2
homeassistant/components/wled/light.py
View file

@ -21,8 +21,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .const import ATTR_COLOR_PRIMARY, ATTR_ON, ATTR_SEGMENT_ID
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
PARALLEL_UPDATES = 1

2
homeassistant/components/wled/number.py
View file

@ -16,8 +16,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .const import ATTR_INTENSITY, ATTR_SPEED
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
PARALLEL_UPDATES = 1

2
homeassistant/components/wled/select.py
View file

@ -13,8 +13,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
PARALLEL_UPDATES = 1

2
homeassistant/components/wled/sensor.py
View file

@ -28,7 +28,7 @@ from homeassistant.util.dt import utcnow
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
from .models import WLEDEntity
from .entity import WLEDEntity
@dataclass(frozen=True, kw_only=True)

2
homeassistant/components/wled/switch.py
View file

@ -13,8 +13,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .const import ATTR_DURATION, ATTR_FADE, ATTR_TARGET_BRIGHTNESS, ATTR_UDP_PORT
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
PARALLEL_UPDATES = 1

5
homeassistant/components/wled/update.py
View file

@ -14,8 +14,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
async def async_setup_entry(
@ -24,8 +24,7 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up WLED update based on a config entry."""
coordinator = entry.runtime_data
async_add_entities([WLEDUpdateEntity(coordinator)])
async_add_entities([WLEDUpdateEntity(entry.runtime_data)])
class WLEDUpdateEntity(WLEDEntity, UpdateEntity):