Protect SupervisorIssues.update method (#113425)

This commit is contained in:
Mike Degatano 2024-03-15 13:06:14 -04:00 committed by GitHub
parent 38164c705e
commit 6ede1c543f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -300,13 +300,13 @@ class SupervisorIssues:
async def setup(self) -> None: async def setup(self) -> None:
"""Create supervisor events listener.""" """Create supervisor events listener."""
await self.update() await self._update()
async_dispatcher_connect( async_dispatcher_connect(
self._hass, EVENT_SUPERVISOR_EVENT, self._supervisor_events_to_issues self._hass, EVENT_SUPERVISOR_EVENT, self._supervisor_events_to_issues
) )
async def update(self, _: datetime | None = None) -> None: async def _update(self, _: datetime | None = None) -> None:
"""Update issues from Supervisor resolution center.""" """Update issues from Supervisor resolution center."""
try: try:
data = await self._client.get_resolution_info() data = await self._client.get_resolution_info()
@ -315,7 +315,7 @@ class SupervisorIssues:
async_call_later( async_call_later(
self._hass, self._hass,
REQUEST_REFRESH_DELAY, REQUEST_REFRESH_DELAY,
HassJob(self.update, cancel_on_shutdown=True), HassJob(self._update, cancel_on_shutdown=True),
) )
return return
self.unhealthy_reasons = set(data[ATTR_UNHEALTHY]) self.unhealthy_reasons = set(data[ATTR_UNHEALTHY])
@ -342,7 +342,7 @@ class SupervisorIssues:
event[ATTR_WS_EVENT] == EVENT_SUPERVISOR_UPDATE event[ATTR_WS_EVENT] == EVENT_SUPERVISOR_UPDATE
and event.get(ATTR_UPDATE_KEY) == UPDATE_KEY_SUPERVISOR and event.get(ATTR_UPDATE_KEY) == UPDATE_KEY_SUPERVISOR
): ):
self._hass.async_create_task(self.update()) self._hass.async_create_task(self._update())
elif event[ATTR_WS_EVENT] == EVENT_HEALTH_CHANGED: elif event[ATTR_WS_EVENT] == EVENT_HEALTH_CHANGED:
self.unhealthy_reasons = ( self.unhealthy_reasons = (