Address jellyfin sensor feedback (#80222)

This commit is contained in:
Chris Talkington
2022-10-19 20:37:18 -05:00
committed by GitHub
parent 265e2741e7
commit ce35812937

View File

@@ -52,7 +52,6 @@ class JellyfinDataUpdateCoordinator(DataUpdateCoordinator[JellyfinDataT]):
@abstractmethod
async def _fetch_data(self) -> JellyfinDataT:
"""Fetch the actual data."""
raise NotImplementedError
class SessionsDataUpdateCoordinator(
@@ -60,10 +59,14 @@ class SessionsDataUpdateCoordinator(
):
"""Sessions update coordinator for Jellyfin."""
async def _fetch_data(self) -> dict:
async def _fetch_data(self) -> dict[str, dict[str, Any]]:
"""Fetch the data."""
sessions = await self.hass.async_add_executor_job(
self.api_client.jellyfin.sessions
)
return {session["Id"]: session for session in sessions}
sessions_by_id: dict[str, Any] = {
session["Id"]: session for session in sessions
}
return sessions_by_id