Add dataclass to store AdGuard data (#115668)
* Add dataclass to store AdGuard data * Unify version call
This commit is contained in:
parent
11ff00f637
commit
6a7a44c998
5 changed files with 43 additions and 45 deletions
|
@ -2,13 +2,14 @@
|
|||
|
||||
from __future__ import annotations
|
||||
|
||||
from adguardhome import AdGuardHome, AdGuardHomeError
|
||||
from adguardhome import AdGuardHomeError
|
||||
|
||||
from homeassistant.config_entries import SOURCE_HASSIO, ConfigEntry
|
||||
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
|
||||
from homeassistant.helpers.entity import Entity
|
||||
|
||||
from .const import DATA_ADGUARD_VERSION, DOMAIN, LOGGER
|
||||
from . import AdGuardData
|
||||
from .const import DOMAIN, LOGGER
|
||||
|
||||
|
||||
class AdGuardHomeEntity(Entity):
|
||||
|
@ -19,12 +20,13 @@ class AdGuardHomeEntity(Entity):
|
|||
|
||||
def __init__(
|
||||
self,
|
||||
adguard: AdGuardHome,
|
||||
data: AdGuardData,
|
||||
entry: ConfigEntry,
|
||||
) -> None:
|
||||
"""Initialize the AdGuard Home entity."""
|
||||
self._entry = entry
|
||||
self.adguard = adguard
|
||||
self.data = data
|
||||
self.adguard = data.client
|
||||
|
||||
async def async_update(self) -> None:
|
||||
"""Update AdGuard Home entity."""
|
||||
|
@ -68,8 +70,6 @@ class AdGuardHomeEntity(Entity):
|
|||
},
|
||||
manufacturer="AdGuard Team",
|
||||
name="AdGuard Home",
|
||||
sw_version=self.hass.data[DOMAIN][self._entry.entry_id].get(
|
||||
DATA_ADGUARD_VERSION
|
||||
),
|
||||
sw_version=self.data.version,
|
||||
configuration_url=config_url,
|
||||
)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue