Use typed config entry in SABnzbd coordinator (#132098)
parent
6fc4f45def
commit
7ae80b542a
|
@ -8,7 +8,7 @@ from typing import Any
|
|||
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.config_entries import ConfigEntry, ConfigEntryState
|
||||
from homeassistant.config_entries import ConfigEntryState
|
||||
from homeassistant.const import Platform
|
||||
from homeassistant.core import HomeAssistant, ServiceCall, callback
|
||||
from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
|
||||
|
@ -24,7 +24,7 @@ from .const import (
|
|||
SERVICE_RESUME,
|
||||
SERVICE_SET_SPEED,
|
||||
)
|
||||
from .coordinator import SabnzbdUpdateCoordinator
|
||||
from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
|
||||
from .helpers import get_client
|
||||
|
||||
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
|
||||
def async_get_entry_for_service_call(
|
||||
|
|
|
@ -15,7 +15,7 @@ from homeassistant.const import EntityCategory
|
|||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from . import SabnzbdConfigEntry
|
||||
from .coordinator import SabnzbdConfigEntry
|
||||
from .entity import SabnzbdEntity
|
||||
|
||||
|
||||
|
|
|
@ -11,9 +11,8 @@ from homeassistant.core import HomeAssistant
|
|||
from homeassistant.exceptions import HomeAssistantError
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from . import SabnzbdConfigEntry
|
||||
from .const import DOMAIN
|
||||
from .coordinator import SabnzbdUpdateCoordinator
|
||||
from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
|
||||
from .entity import SabnzbdEntity
|
||||
|
||||
|
||||
|
|
|
@ -12,16 +12,18 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
|
|||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
type SabnzbdConfigEntry = ConfigEntry[SabnzbdUpdateCoordinator]
|
||||
|
||||
|
||||
class SabnzbdUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
|
||||
"""The SABnzbd update coordinator."""
|
||||
|
||||
config_entry: ConfigEntry
|
||||
config_entry: SabnzbdConfigEntry
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
config_entry: SabnzbdConfigEntry,
|
||||
sab_api: SabnzbdApi,
|
||||
) -> None:
|
||||
"""Initialize the SABnzbd update coordinator."""
|
||||
|
|
|
@ -17,9 +17,8 @@ from homeassistant.core import HomeAssistant
|
|||
from homeassistant.exceptions import HomeAssistantError
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from . import SabnzbdConfigEntry
|
||||
from .const import DOMAIN
|
||||
from .coordinator import SabnzbdUpdateCoordinator
|
||||
from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
|
||||
from .entity import SabnzbdEntity
|
||||
|
||||
|
||||
|
|
|
@ -15,7 +15,7 @@ from homeassistant.core import HomeAssistant
|
|||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
from homeassistant.helpers.typing import StateType
|
||||
|
||||
from . import SabnzbdConfigEntry
|
||||
from .coordinator import SabnzbdConfigEntry
|
||||
from .entity import SabnzbdEntity
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue