Explicitly pass in the config_entry in qbittorrent coordinator (#138029)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 17:32:22 +01:00 committed by GitHub
parent 685e882847
commit 474d8bbd65
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 2 deletions

View File

@ -124,7 +124,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
except APIConnectionError as exc: except APIConnectionError as exc:
raise ConfigEntryNotReady("Fail to connect to qBittorrent") from exc raise ConfigEntryNotReady("Fail to connect to qBittorrent") from exc
coordinator = QBittorrentDataCoordinator(hass, client) coordinator = QBittorrentDataCoordinator(hass, config_entry, client)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = coordinator hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = coordinator

View File

@ -15,6 +15,7 @@ from qbittorrentapi import (
) )
from qbittorrentapi.torrents import TorrentStatusesT from qbittorrentapi.torrents import TorrentStatusesT
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@ -27,7 +28,11 @@ _LOGGER = logging.getLogger(__name__)
class QBittorrentDataCoordinator(DataUpdateCoordinator[SyncMainDataDictionary]): class QBittorrentDataCoordinator(DataUpdateCoordinator[SyncMainDataDictionary]):
"""Coordinator for updating QBittorrent data.""" """Coordinator for updating QBittorrent data."""
def __init__(self, hass: HomeAssistant, client: Client) -> None: config_entry: ConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: ConfigEntry, client: Client
) -> None:
"""Initialize coordinator.""" """Initialize coordinator."""
self.client = client self.client = client
self._is_alternative_mode_enabled = False self._is_alternative_mode_enabled = False
@ -42,6 +47,7 @@ class QBittorrentDataCoordinator(DataUpdateCoordinator[SyncMainDataDictionary]):
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
config_entry=config_entry,
name=DOMAIN, name=DOMAIN,
update_interval=timedelta(seconds=30), update_interval=timedelta(seconds=30),
) )