Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use typed config entry in SABnzbd coordinator (#132098)

pull/132180/head
Jan-Philipp Benecke 2024-12-03 15:48:56 +01:00 committed by GitHub
parent 6fc4f45def
commit 7ae80b542a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/components/sabnzbd/__init__.py
View File

@ -8,7 +8,7 @@ from typing import Any
import voluptuous as vol import voluptuous as vol
from homeassistant.config_entries import ConfigEntry, ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import Platform from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, ServiceCall, callback from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
@ -24,7 +24,7 @@ from .const import (
SERVICE_RESUME, SERVICE_RESUME,
SERVICE_SET_SPEED, SERVICE_SET_SPEED,
) )
from .coordinator import SabnzbdUpdateCoordinator from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
from .helpers import get_client from .helpers import get_client
PLATFORMS = [Platform.BINARY_SENSOR, Platform.BUTTON, Platform.NUMBER, Platform.SENSOR] PLATFORMS = [Platform.BINARY_SENSOR, Platform.BUTTON, Platform.NUMBER, Platform.SENSOR]
@ -48,8 +48,6 @@ SERVICE_SPEED_SCHEMA = SERVICE_BASE_SCHEMA.extend(
} }
) )
type SabnzbdConfigEntry = ConfigEntry[SabnzbdUpdateCoordinator]
@callback @callback
def async_get_entry_for_service_call( def async_get_entry_for_service_call(

2
homeassistant/components/sabnzbd/binary_sensor.py
View File

@ -15,7 +15,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import SabnzbdConfigEntry from .coordinator import SabnzbdConfigEntry
from .entity import SabnzbdEntity from .entity import SabnzbdEntity

3
homeassistant/components/sabnzbd/button.py
View File

@ -11,9 +11,8 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import SabnzbdConfigEntry
from .const import DOMAIN from .const import DOMAIN
from .coordinator import SabnzbdUpdateCoordinator from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
from .entity import SabnzbdEntity from .entity import SabnzbdEntity

6
homeassistant/components/sabnzbd/coordinator.py
View File

@ -12,16 +12,18 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
type SabnzbdConfigEntry = ConfigEntry[SabnzbdUpdateCoordinator]
class SabnzbdUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]): class SabnzbdUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""The SABnzbd update coordinator.""" """The SABnzbd update coordinator."""
config_entry: ConfigEntry config_entry: SabnzbdConfigEntry
def __init__( def __init__(
self, self,
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: SabnzbdConfigEntry,
sab_api: SabnzbdApi, sab_api: SabnzbdApi,
) -> None: ) -> None:
"""Initialize the SABnzbd update coordinator.""" """Initialize the SABnzbd update coordinator."""

3
homeassistant/components/sabnzbd/number.py
View File

@ -17,9 +17,8 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import SabnzbdConfigEntry
from .const import DOMAIN from .const import DOMAIN
from .coordinator import SabnzbdUpdateCoordinator from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
from .entity import SabnzbdEntity from .entity import SabnzbdEntity

2
homeassistant/components/sabnzbd/sensor.py
View File

@ -15,7 +15,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from . import SabnzbdConfigEntry from .coordinator import SabnzbdConfigEntry
from .entity import SabnzbdEntity from .entity import SabnzbdEntity