jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate openweathermap weather to CoordinatorEntity (#98799)

Browse Source
This commit is contained in:
Erik Montnemery 2023-08-22 10:27:34 +02:00 committed by GitHub
parent 00b75ce58d
commit 52b1e34af0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
homeassistant/components/openweathermap/weather.py
View File

@ -29,6 +29,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ( from .const import (
ATTR_API_CLOUDS, ATTR_API_CLOUDS,
@ -95,7 +96,7 @@ async def async_setup_entry(
async_add_entities([owm_weather], False) async_add_entities([owm_weather], False)
class OpenWeatherMapWeather(WeatherEntity): class OpenWeatherMapWeather(CoordinatorEntity[WeatherUpdateCoordinator], WeatherEntity):
"""Implementation of an OpenWeatherMap sensor.""" """Implementation of an OpenWeatherMap sensor."""
_attr_attribution = ATTRIBUTION _attr_attribution = ATTRIBUTION
@ -113,6 +114,7 @@ class OpenWeatherMapWeather(WeatherEntity):
weather_coordinator: WeatherUpdateCoordinator, weather_coordinator: WeatherUpdateCoordinator,
) -> None: ) -> None:
"""Initialize the sensor.""" """Initialize the sensor."""
super().__init__(weather_coordinator)
self._attr_name = name self._attr_name = name
self._attr_unique_id = unique_id self._attr_unique_id = unique_id
self._attr_device_info = DeviceInfo( self._attr_device_info = DeviceInfo(
@ -121,62 +123,61 @@ class OpenWeatherMapWeather(WeatherEntity):
manufacturer=MANUFACTURER, manufacturer=MANUFACTURER,
name=DEFAULT_NAME, name=DEFAULT_NAME,
) )
self._weather_coordinator = weather_coordinator
@property @property
def condition(self) -> str | None: def condition(self) -> str | None:
"""Return the current condition.""" """Return the current condition."""
return self._weather_coordinator.data[ATTR_API_CONDITION] return self.coordinator.data[ATTR_API_CONDITION]
@property @property
def cloud_coverage(self) -> float | None: def cloud_coverage(self) -> float | None:
"""Return the Cloud coverage in %.""" """Return the Cloud coverage in %."""
return self._weather_coordinator.data[ATTR_API_CLOUDS] return self.coordinator.data[ATTR_API_CLOUDS]
@property @property
def native_apparent_temperature(self) -> float | None: def native_apparent_temperature(self) -> float | None:
"""Return the apparent temperature.""" """Return the apparent temperature."""
return self._weather_coordinator.data[ATTR_API_FEELS_LIKE_TEMPERATURE] return self.coordinator.data[ATTR_API_FEELS_LIKE_TEMPERATURE]
@property @property
def native_temperature(self) -> float | None: def native_temperature(self) -> float | None:
"""Return the temperature.""" """Return the temperature."""
return self._weather_coordinator.data[ATTR_API_TEMPERATURE] return self.coordinator.data[ATTR_API_TEMPERATURE]
@property @property
def native_pressure(self) -> float | None: def native_pressure(self) -> float | None:
"""Return the pressure.""" """Return the pressure."""
return self._weather_coordinator.data[ATTR_API_PRESSURE] return self.coordinator.data[ATTR_API_PRESSURE]
@property @property
def humidity(self) -> float | None: def humidity(self) -> float | None:
"""Return the humidity.""" """Return the humidity."""
return self._weather_coordinator.data[ATTR_API_HUMIDITY] return self.coordinator.data[ATTR_API_HUMIDITY]
@property @property
def native_dew_point(self) -> float | None: def native_dew_point(self) -> float | None:
"""Return the dew point.""" """Return the dew point."""
return self._weather_coordinator.data[ATTR_API_DEW_POINT] return self.coordinator.data[ATTR_API_DEW_POINT]
@property @property
def native_wind_gust_speed(self) -> float | None: def native_wind_gust_speed(self) -> float | None:
"""Return the wind gust speed.""" """Return the wind gust speed."""
return self._weather_coordinator.data[ATTR_API_WIND_GUST] return self.coordinator.data[ATTR_API_WIND_GUST]
@property @property
def native_wind_speed(self) -> float | None: def native_wind_speed(self) -> float | None:
"""Return the wind speed.""" """Return the wind speed."""
return self._weather_coordinator.data[ATTR_API_WIND_SPEED] return self.coordinator.data[ATTR_API_WIND_SPEED]
@property @property
def wind_bearing(self) -> float | str | None: def wind_bearing(self) -> float | str | None:
"""Return the wind bearing.""" """Return the wind bearing."""
return self._weather_coordinator.data[ATTR_API_WIND_BEARING] return self.coordinator.data[ATTR_API_WIND_BEARING]
@property @property
def forecast(self) -> list[Forecast] | None: def forecast(self) -> list[Forecast] | None:
"""Return the forecast array.""" """Return the forecast array."""
api_forecasts = self._weather_coordinator.data[ATTR_API_FORECAST] api_forecasts = self.coordinator.data[ATTR_API_FORECAST]
forecasts = [ forecasts = [
{ {
ha_key: forecast[api_key] ha_key: forecast[api_key]
@ -186,18 +187,3 @@ class OpenWeatherMapWeather(WeatherEntity):
for forecast in api_forecasts for forecast in api_forecasts
] ]
return cast(list[Forecast], forecasts) return cast(list[Forecast], forecasts)
@property
def available(self) -> bool:
"""Return True if entity is available."""
return self._weather_coordinator.last_update_success
async def async_added_to_hass(self) -> None:
"""Connect to dispatcher listening for entity data notifications."""
self.async_on_remove(
self._weather_coordinator.async_add_listener(self.async_write_ha_state)
)
async def async_update(self) -> None:
"""Get the latest data from OWM and updates the states."""
await self._weather_coordinator.async_request_refresh()