diff --git a/homeassistant/components/input_datetime/__init__.py b/homeassistant/components/input_datetime/__init__.py index c25680b7180..fe034a8edb5 100644 --- a/homeassistant/components/input_datetime/__init__.py +++ b/homeassistant/components/input_datetime/__init__.py @@ -19,6 +19,9 @@ from homeassistant.core import HomeAssistant, ServiceCall, callback from homeassistant.helpers import collection import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity_component import EntityComponent +from homeassistant.helpers.integration_platform import ( + async_process_integration_platform_for_component, +) from homeassistant.helpers.restore_state import RestoreEntity import homeassistant.helpers.service from homeassistant.helpers.storage import Store @@ -125,6 +128,11 @@ RELOAD_SERVICE_SCHEMA = vol.Schema({}) async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: """Set up an input datetime.""" component = EntityComponent(_LOGGER, DOMAIN, hass) + + # Process integration platforms right away since + # we will create entities before firing EVENT_COMPONENT_LOADED + await async_process_integration_platform_for_component(hass, DOMAIN) + id_manager = collection.IDManager() yaml_collection = collection.YamlCollection( diff --git a/homeassistant/components/input_datetime/recorder.py b/homeassistant/components/input_datetime/recorder.py new file mode 100644 index 00000000000..91c33ee0811 --- /dev/null +++ b/homeassistant/components/input_datetime/recorder.py @@ -0,0 +1,13 @@ +"""Integration platform for recorder.""" +from __future__ import annotations + +from homeassistant.const import ATTR_EDITABLE +from homeassistant.core import HomeAssistant, callback + +from . import CONF_HAS_DATE, CONF_HAS_TIME + + +@callback +def exclude_attributes(hass: HomeAssistant) -> set[str]: + """Exclude some attributes from being recorded in the database.""" + return {ATTR_EDITABLE, CONF_HAS_DATE, CONF_HAS_TIME} diff --git a/tests/components/input_datetime/test_recorder.py b/tests/components/input_datetime/test_recorder.py new file mode 100644 index 00000000000..ee42c4042b2 --- /dev/null +++ b/tests/components/input_datetime/test_recorder.py @@ -0,0 +1,51 @@ +"""The tests for recorder platform.""" +from __future__ import annotations + +from datetime import timedelta + +from homeassistant.components.input_datetime import CONF_HAS_DATE, CONF_HAS_TIME, DOMAIN +from homeassistant.components.recorder.models import StateAttributes, States +from homeassistant.components.recorder.util import session_scope +from homeassistant.const import ATTR_EDITABLE +from homeassistant.core import HomeAssistant, State +from homeassistant.setup import async_setup_component +from homeassistant.util import dt as dt_util + +from tests.common import async_fire_time_changed, async_init_recorder_component +from tests.components.recorder.common import async_wait_recording_done_without_instance + + +async def test_exclude_attributes( + hass: HomeAssistant, enable_custom_integrations: None +): + """Test attributes to be excluded.""" + await async_init_recorder_component(hass) + assert await async_setup_component( + hass, DOMAIN, {DOMAIN: {"test": {CONF_HAS_TIME: True}}} + ) + + state = hass.states.get("input_datetime.test") + assert state + assert state.attributes[ATTR_EDITABLE] is False + assert state.attributes[CONF_HAS_DATE] is False + assert state.attributes[CONF_HAS_TIME] is True + + await hass.async_block_till_done() + async_fire_time_changed(hass, dt_util.utcnow() + timedelta(minutes=5)) + await hass.async_block_till_done() + await async_wait_recording_done_without_instance(hass) + + def _fetch_states() -> list[State]: + with session_scope(hass=hass) as session: + native_states = [] + for db_state, db_state_attributes in session.query(States, StateAttributes): + state = db_state.to_native() + state.attributes = db_state_attributes.to_native() + native_states.append(state) + return native_states + + states: list[State] = await hass.async_add_executor_job(_fetch_states) + assert len(states) == 1 + assert ATTR_EDITABLE not in states[0].attributes + assert CONF_HAS_DATE not in states[0].attributes + assert CONF_HAS_TIME not in states[0].attributes