mirror of
https://github.com/elisspace/core.git
synced 2026-09-30 06:19:55 +00:00
Exclude sun attributes from being recorded in the database (#68887)
This commit is contained in:
@@ -5,11 +5,12 @@ import logging
|
||||
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
|
||||
from homeassistant.const import (
|
||||
CONF_ELEVATION,
|
||||
EVENT_COMPONENT_LOADED,
|
||||
EVENT_CORE_CONFIG_UPDATE,
|
||||
SUN_EVENT_SUNRISE,
|
||||
SUN_EVENT_SUNSET,
|
||||
)
|
||||
from homeassistant.core import HomeAssistant, callback
|
||||
from homeassistant.core import Event, HomeAssistant, callback
|
||||
from homeassistant.helpers import event
|
||||
from homeassistant.helpers.entity import Entity
|
||||
from homeassistant.helpers.sun import (
|
||||
@@ -17,6 +18,7 @@ from homeassistant.helpers.sun import (
|
||||
get_location_astral_event_next,
|
||||
)
|
||||
from homeassistant.helpers.typing import ConfigType
|
||||
from homeassistant.setup import ATTR_COMPONENT
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from .const import DOMAIN
|
||||
@@ -124,26 +126,43 @@ class Sun(Entity):
|
||||
self._config_listener = None
|
||||
self._update_events_listener = None
|
||||
self._update_sun_position_listener = None
|
||||
|
||||
@callback
|
||||
def update_location(_event):
|
||||
location, elevation = get_astral_location(self.hass)
|
||||
if location == self.location:
|
||||
return
|
||||
self.location = location
|
||||
self.elevation = elevation
|
||||
if self._update_events_listener:
|
||||
self._update_events_listener()
|
||||
self.update_events()
|
||||
|
||||
update_location(None)
|
||||
self._loaded_listener = None
|
||||
self._config_listener = self.hass.bus.async_listen(
|
||||
EVENT_CORE_CONFIG_UPDATE, update_location
|
||||
EVENT_CORE_CONFIG_UPDATE, self.update_location
|
||||
)
|
||||
self._loaded_listener = self.hass.bus.async_listen(
|
||||
EVENT_COMPONENT_LOADED, self.loading_complete
|
||||
)
|
||||
|
||||
@callback
|
||||
def loading_complete(self, event_: Event) -> None:
|
||||
"""Update location when loading is complete."""
|
||||
if event_.data[ATTR_COMPONENT] == DOMAIN:
|
||||
self.update_location()
|
||||
self._remove_loaded_listener()
|
||||
|
||||
@callback
|
||||
def update_location(self, *_):
|
||||
"""Update location."""
|
||||
location, elevation = get_astral_location(self.hass)
|
||||
if location == self.location:
|
||||
return
|
||||
self.location = location
|
||||
self.elevation = elevation
|
||||
if self._update_events_listener:
|
||||
self._update_events_listener()
|
||||
self.update_events()
|
||||
|
||||
@callback
|
||||
def _remove_loaded_listener(self):
|
||||
"""Remove the loaded listener."""
|
||||
if self._loaded_listener:
|
||||
self._loaded_listener()
|
||||
|
||||
@callback
|
||||
def remove_listeners(self):
|
||||
"""Remove listeners."""
|
||||
self._remove_loaded_listener()
|
||||
if self._config_listener:
|
||||
self._config_listener()
|
||||
if self._update_events_listener:
|
||||
|
||||
@@ -0,0 +1,32 @@
|
||||
"""Integration platform for recorder."""
|
||||
from __future__ import annotations
|
||||
|
||||
from homeassistant.core import HomeAssistant, callback
|
||||
|
||||
from . import (
|
||||
STATE_ATTR_AZIMUTH,
|
||||
STATE_ATTR_ELEVATION,
|
||||
STATE_ATTR_NEXT_DAWN,
|
||||
STATE_ATTR_NEXT_DUSK,
|
||||
STATE_ATTR_NEXT_MIDNIGHT,
|
||||
STATE_ATTR_NEXT_NOON,
|
||||
STATE_ATTR_NEXT_RISING,
|
||||
STATE_ATTR_NEXT_SETTING,
|
||||
STATE_ATTR_RISING,
|
||||
)
|
||||
|
||||
|
||||
@callback
|
||||
def exclude_attributes(hass: HomeAssistant) -> set[str]:
|
||||
"""Exclude sun attributes from being recorded in the database."""
|
||||
return {
|
||||
STATE_ATTR_AZIMUTH,
|
||||
STATE_ATTR_ELEVATION,
|
||||
STATE_ATTR_RISING,
|
||||
STATE_ATTR_NEXT_DAWN,
|
||||
STATE_ATTR_NEXT_DUSK,
|
||||
STATE_ATTR_NEXT_MIDNIGHT,
|
||||
STATE_ATTR_NEXT_NOON,
|
||||
STATE_ATTR_NEXT_RISING,
|
||||
STATE_ATTR_NEXT_SETTING,
|
||||
}
|
||||
Reference in New Issue
Block a user