Fix synology_dsm coordinator typing (#73301)

This commit is contained in:
J. Nick Koston 2022-06-09 18:24:39 -10:00 committed by GitHub
parent 0505c596a5
commit 0f4080bca3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,6 +3,7 @@ from __future__ import annotations
from datetime import timedelta from datetime import timedelta
import logging import logging
from typing import Any
import async_timeout import async_timeout
from synology_dsm.api.surveillance_station.camera import SynoCamera from synology_dsm.api.surveillance_station.camera import SynoCamera
@ -65,7 +66,7 @@ class SynologyDSMSwitchUpdateCoordinator(SynologyDSMUpdateCoordinator):
) )
self.version = info["data"]["CMSMinVersion"] self.version = info["data"]["CMSMinVersion"]
async def _async_update_data(self) -> dict[str, dict[str, SynoCamera]] | None: async def _async_update_data(self) -> dict[str, dict[str, Any]]:
"""Fetch all data from api.""" """Fetch all data from api."""
surveillance_station = self.api.surveillance_station surveillance_station = self.api.surveillance_station
return { return {
@ -96,7 +97,7 @@ class SynologyDSMCentralUpdateCoordinator(SynologyDSMUpdateCoordinator):
), ),
) )
async def _async_update_data(self) -> dict[str, dict[str, SynoCamera]] | None: async def _async_update_data(self) -> None:
"""Fetch all data from api.""" """Fetch all data from api."""
try: try:
await self.api.async_update() await self.api.async_update()
@ -117,7 +118,7 @@ class SynologyDSMCameraUpdateCoordinator(SynologyDSMUpdateCoordinator):
"""Initialize DataUpdateCoordinator for cameras.""" """Initialize DataUpdateCoordinator for cameras."""
super().__init__(hass, entry, api, timedelta(seconds=30)) super().__init__(hass, entry, api, timedelta(seconds=30))
async def _async_update_data(self) -> dict[str, dict[str, SynoCamera]] | None: async def _async_update_data(self) -> dict[str, dict[str, SynoCamera]]:
"""Fetch all camera data from api.""" """Fetch all camera data from api."""
surveillance_station = self.api.surveillance_station surveillance_station = self.api.surveillance_station
current_data: dict[str, SynoCamera] = { current_data: dict[str, SynoCamera] = {