diff --git a/homeassistant/components/duke_energy/__init__.py b/homeassistant/components/duke_energy/__init__.py index 6eacc15880f..bfa89d81c69 100644 --- a/homeassistant/components/duke_energy/__init__.py +++ b/homeassistant/components/duke_energy/__init__.py @@ -10,7 +10,7 @@ from .coordinator import DukeEnergyConfigEntry, DukeEnergyCoordinator async def async_setup_entry(hass: HomeAssistant, entry: DukeEnergyConfigEntry) -> bool: """Set up Duke Energy from a config entry.""" - coordinator = DukeEnergyCoordinator(hass, entry.data) + coordinator = DukeEnergyCoordinator(hass, entry) await coordinator.async_config_entry_first_refresh() entry.runtime_data = coordinator diff --git a/homeassistant/components/duke_energy/coordinator.py b/homeassistant/components/duke_energy/coordinator.py index 2b0ae46b405..12a2f5fd6ae 100644 --- a/homeassistant/components/duke_energy/coordinator.py +++ b/homeassistant/components/duke_energy/coordinator.py @@ -2,7 +2,6 @@ from datetime import datetime, timedelta import logging -from types import MappingProxyType from typing import Any, cast from aiodukeenergy import DukeEnergy @@ -37,22 +36,21 @@ class DukeEnergyCoordinator(DataUpdateCoordinator[None]): config_entry: DukeEnergyConfigEntry def __init__( - self, - hass: HomeAssistant, - entry_data: MappingProxyType[str, Any], + self, hass: HomeAssistant, config_entry: DukeEnergyConfigEntry ) -> None: """Initialize the data handler.""" super().__init__( hass, _LOGGER, + config_entry=config_entry, name="Duke Energy", # Data is updated daily on Duke Energy. # Refresh every 12h to be at most 12h behind. update_interval=timedelta(hours=12), ) self.api = DukeEnergy( - entry_data[CONF_USERNAME], - entry_data[CONF_PASSWORD], + config_entry.data[CONF_USERNAME], + config_entry.data[CONF_PASSWORD], async_get_clientsession(hass), ) self._statistic_ids: set = set()