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

Handle UPS disconnects in NUT (#35758)

Browse source
This commit is contained in:
J. Nick Koston 2020-05-18 09:23:05 -05:00 • committed by GitHub
parent affd11b372
commit 50105eed74
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
homeassistant/components/nut/__init__.py
View file

@ -18,7 +18,7 @@ from homeassistant.const import (
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import ( from .const import (
COORDINATOR, COORDINATOR,
@ -61,7 +61,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
async def async_update_data(): async def async_update_data():
"""Fetch data from NUT.""" """Fetch data from NUT."""
async with async_timeout.timeout(10): async with async_timeout.timeout(10):
return await hass.async_add_executor_job(data.update) await hass.async_add_executor_job(data.update)
if not data.status:
raise UpdateFailed("Error fetching UPS state")
coordinator = DataUpdateCoordinator( coordinator = DataUpdateCoordinator(
hass, hass,

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

@ -189,6 +189,8 @@ class NUTSensor(Entity):
@property @property
def state(self): def state(self):
"""Return entity state from ups.""" """Return entity state from ups."""
if not self._data.status:
return None
if self._type == KEY_STATUS_DISPLAY: if self._type == KEY_STATUS_DISPLAY:
return _format_display_state(self._data.status) return _format_display_state(self._data.status)
return self._data.status.get(self._type) return self._data.status.get(self._type)