mirror of
https://github.com/home-assistant/core.git
synced 2025-07-15 01:07:10 +00:00
Make sun solar_rising
a binary_sensor (#140956)
* Make sun solar_rising a binary_sensor. * Add a state translation * code review * fix test * move PLATFORMS * Update strings.json
This commit is contained in:
parent
ad493e077e
commit
33fc700952
@ -16,7 +16,10 @@ from homeassistant.helpers.typing import ConfigType
|
|||||||
# as we will always load it and we do not want to have
|
# as we will always load it and we do not want to have
|
||||||
# to wait for the import executor when its busy later
|
# to wait for the import executor when its busy later
|
||||||
# in the startup process.
|
# in the startup process.
|
||||||
from . import sensor as sensor_pre_import # noqa: F401
|
from . import (
|
||||||
|
binary_sensor as binary_sensor_pre_import, # noqa: F401
|
||||||
|
sensor as sensor_pre_import, # noqa: F401
|
||||||
|
)
|
||||||
from .const import ( # noqa: F401 # noqa: F401
|
from .const import ( # noqa: F401 # noqa: F401
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
STATE_ABOVE_HORIZON,
|
STATE_ABOVE_HORIZON,
|
||||||
@ -24,6 +27,8 @@ from .const import ( # noqa: F401 # noqa: F401
|
|||||||
)
|
)
|
||||||
from .entity import Sun, SunConfigEntry
|
from .entity import Sun, SunConfigEntry
|
||||||
|
|
||||||
|
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR]
|
||||||
|
|
||||||
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
|
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
|
||||||
|
|
||||||
_LOGGER = logging.getLogger(__name__)
|
_LOGGER = logging.getLogger(__name__)
|
||||||
@ -52,14 +57,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: SunConfigEntry) -> bool:
|
|||||||
await component.async_add_entities([sun])
|
await component.async_add_entities([sun])
|
||||||
entry.runtime_data = sun
|
entry.runtime_data = sun
|
||||||
entry.async_on_unload(sun.remove_listeners)
|
entry.async_on_unload(sun.remove_listeners)
|
||||||
await hass.config_entries.async_forward_entry_setups(entry, [Platform.SENSOR])
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
async def async_unload_entry(hass: HomeAssistant, entry: SunConfigEntry) -> bool:
|
async def async_unload_entry(hass: HomeAssistant, entry: SunConfigEntry) -> bool:
|
||||||
"""Unload a config entry."""
|
"""Unload a config entry."""
|
||||||
if unload_ok := await hass.config_entries.async_unload_platforms(
|
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
|
||||||
entry, [Platform.SENSOR]
|
|
||||||
):
|
|
||||||
await entry.runtime_data.async_remove()
|
await entry.runtime_data.async_remove()
|
||||||
return unload_ok
|
return unload_ok
|
||||||
|
100
homeassistant/components/sun/binary_sensor.py
Normal file
100
homeassistant/components/sun/binary_sensor.py
Normal file
@ -0,0 +1,100 @@
|
|||||||
|
"""Binary Sensor platform for Sun integration."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from collections.abc import Callable
|
||||||
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
from homeassistant.components.binary_sensor import (
|
||||||
|
DOMAIN as BINARY_SENSOR_DOMAIN,
|
||||||
|
BinarySensorEntity,
|
||||||
|
BinarySensorEntityDescription,
|
||||||
|
)
|
||||||
|
from homeassistant.const import EntityCategory
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
|
||||||
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
||||||
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
||||||
|
|
||||||
|
from .const import DOMAIN, SIGNAL_EVENTS_CHANGED
|
||||||
|
from .entity import Sun, SunConfigEntry
|
||||||
|
|
||||||
|
ENTITY_ID_BINARY_SENSOR_FORMAT = BINARY_SENSOR_DOMAIN + ".sun_{}"
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(kw_only=True, frozen=True)
|
||||||
|
class SunBinarySensorEntityDescription(BinarySensorEntityDescription):
|
||||||
|
"""Describes a Sun binary sensor entity."""
|
||||||
|
|
||||||
|
value_fn: Callable[[Sun], bool | None]
|
||||||
|
signal: str
|
||||||
|
|
||||||
|
|
||||||
|
BINARY_SENSOR_TYPES: tuple[SunBinarySensorEntityDescription, ...] = (
|
||||||
|
SunBinarySensorEntityDescription(
|
||||||
|
key="solar_rising",
|
||||||
|
translation_key="solar_rising",
|
||||||
|
value_fn=lambda data: data.rising,
|
||||||
|
entity_registry_enabled_default=False,
|
||||||
|
signal=SIGNAL_EVENTS_CHANGED,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def async_setup_entry(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
entry: SunConfigEntry,
|
||||||
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
||||||
|
) -> None:
|
||||||
|
"""Set up Sun binary sensor platform."""
|
||||||
|
|
||||||
|
sun = entry.runtime_data
|
||||||
|
|
||||||
|
async_add_entities(
|
||||||
|
[
|
||||||
|
SunBinarySensor(sun, description, entry.entry_id)
|
||||||
|
for description in BINARY_SENSOR_TYPES
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class SunBinarySensor(BinarySensorEntity):
|
||||||
|
"""Representation of a Sun binary sensor."""
|
||||||
|
|
||||||
|
_attr_has_entity_name = True
|
||||||
|
_attr_should_poll = False
|
||||||
|
_attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||||
|
entity_description: SunBinarySensorEntityDescription
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
sun: Sun,
|
||||||
|
entity_description: SunBinarySensorEntityDescription,
|
||||||
|
entry_id: str,
|
||||||
|
) -> None:
|
||||||
|
"""Initiate Sun Binary Sensor."""
|
||||||
|
self.entity_description = entity_description
|
||||||
|
self.entity_id = ENTITY_ID_BINARY_SENSOR_FORMAT.format(entity_description.key)
|
||||||
|
self._attr_unique_id = f"{entry_id}-{entity_description.key}"
|
||||||
|
self.sun = sun
|
||||||
|
self._attr_device_info = DeviceInfo(
|
||||||
|
name="Sun",
|
||||||
|
identifiers={(DOMAIN, entry_id)},
|
||||||
|
entry_type=DeviceEntryType.SERVICE,
|
||||||
|
)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_on(self) -> bool | None:
|
||||||
|
"""Return value of binary sensor."""
|
||||||
|
return self.entity_description.value_fn(self.sun)
|
||||||
|
|
||||||
|
async def async_added_to_hass(self) -> None:
|
||||||
|
"""Register signal listener when added to hass."""
|
||||||
|
await super().async_added_to_hass()
|
||||||
|
self.async_on_remove(
|
||||||
|
async_dispatcher_connect(
|
||||||
|
self.hass,
|
||||||
|
self.entity_description.signal,
|
||||||
|
self.async_write_ha_state,
|
||||||
|
)
|
||||||
|
)
|
@ -28,6 +28,15 @@
|
|||||||
"solar_rising": {
|
"solar_rising": {
|
||||||
"default": "mdi:sun-clock"
|
"default": "mdi:sun-clock"
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"binary_sensor": {
|
||||||
|
"solar_rising": {
|
||||||
|
"default": "mdi:weather-sunny-off",
|
||||||
|
"state": {
|
||||||
|
"on": "mdi:weather-sunset-up",
|
||||||
|
"off": "mdi:weather-sunset-down"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -27,6 +27,15 @@
|
|||||||
"solar_azimuth": { "name": "Solar azimuth" },
|
"solar_azimuth": { "name": "Solar azimuth" },
|
||||||
"solar_elevation": { "name": "Solar elevation" },
|
"solar_elevation": { "name": "Solar elevation" },
|
||||||
"solar_rising": { "name": "Solar rising" }
|
"solar_rising": { "name": "Solar rising" }
|
||||||
|
},
|
||||||
|
"binary_sensor": {
|
||||||
|
"solar_rising": {
|
||||||
|
"name": "Solar rising",
|
||||||
|
"state": {
|
||||||
|
"on": "Rising",
|
||||||
|
"off": "Setting"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
44
tests/components/sun/test_binary_sensor.py
Normal file
44
tests/components/sun/test_binary_sensor.py
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
"""The tests for the Sun binary_sensor platform."""
|
||||||
|
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
from freezegun.api import FrozenDateTimeFactory
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from homeassistant.components import sun
|
||||||
|
from homeassistant.const import EntityCategory
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.helpers import entity_registry as er
|
||||||
|
from homeassistant.setup import async_setup_component
|
||||||
|
from homeassistant.util import dt as dt_util
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
||||||
|
async def test_setting_rising(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
entity_registry: er.EntityRegistry,
|
||||||
|
freezer: FrozenDateTimeFactory,
|
||||||
|
) -> None:
|
||||||
|
"""Test retrieving sun setting and rising."""
|
||||||
|
utc_now = datetime(2016, 11, 1, 8, 0, 0, tzinfo=dt_util.UTC)
|
||||||
|
freezer.move_to(utc_now)
|
||||||
|
await async_setup_component(hass, sun.DOMAIN, {sun.DOMAIN: {}})
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert hass.states.get("binary_sensor.sun_solar_rising").state == "on"
|
||||||
|
|
||||||
|
entry_ids = hass.config_entries.async_entries("sun")
|
||||||
|
|
||||||
|
freezer.tick(timedelta(hours=12))
|
||||||
|
# Block once for Sun to update
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
# Block another time for the sensors to update
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
# Make sure all the signals work
|
||||||
|
assert hass.states.get("binary_sensor.sun_solar_rising").state == "off"
|
||||||
|
|
||||||
|
entity = entity_registry.async_get("binary_sensor.sun_solar_rising")
|
||||||
|
assert entity
|
||||||
|
assert entity.entity_category is EntityCategory.DIAGNOSTIC
|
||||||
|
assert entity.unique_id == f"{entry_ids[0].entry_id}-solar_rising"
|
Loading…
x
Reference in New Issue
Block a user