diff --git a/homeassistant/components/whois/sensor.py b/homeassistant/components/whois/sensor.py index 0459651e693..6df920f385c 100644 --- a/homeassistant/components/whois/sensor.py +++ b/homeassistant/components/whois/sensor.py @@ -80,13 +80,6 @@ def _ensure_timezone(timestamp: datetime | None) -> datetime | None: return timestamp -def _fetch_attr_if_exists(domain: Domain, attr: str) -> str | None: - """Fetch an attribute if it exists and is truthy or return None.""" - if hasattr(domain, attr) and (value := getattr(domain, attr)): - return cast(str, value) - return None - - SENSORS: tuple[WhoisSensorEntityDescription, ...] = ( WhoisSensorEntityDescription( key="admin", @@ -94,7 +87,7 @@ SENSORS: tuple[WhoisSensorEntityDescription, ...] = ( icon="mdi:account-star", entity_category=EntityCategory.DIAGNOSTIC, entity_registry_enabled_default=False, - value_fn=lambda domain: _fetch_attr_if_exists(domain, "admin"), + value_fn=lambda domain: getattr(domain, "admin", None), ), WhoisSensorEntityDescription( key="creation_date", @@ -130,7 +123,7 @@ SENSORS: tuple[WhoisSensorEntityDescription, ...] = ( icon="mdi:account", entity_category=EntityCategory.DIAGNOSTIC, entity_registry_enabled_default=False, - value_fn=lambda domain: _fetch_attr_if_exists(domain, "owner"), + value_fn=lambda domain: getattr(domain, "owner", None), ), WhoisSensorEntityDescription( key="registrant", @@ -138,7 +131,7 @@ SENSORS: tuple[WhoisSensorEntityDescription, ...] = ( icon="mdi:account-edit", entity_category=EntityCategory.DIAGNOSTIC, entity_registry_enabled_default=False, - value_fn=lambda domain: _fetch_attr_if_exists(domain, "registrant"), + value_fn=lambda domain: getattr(domain, "registrant", None), ), WhoisSensorEntityDescription( key="registrar", @@ -154,7 +147,7 @@ SENSORS: tuple[WhoisSensorEntityDescription, ...] = ( icon="mdi:store", entity_category=EntityCategory.DIAGNOSTIC, entity_registry_enabled_default=False, - value_fn=lambda domain: _fetch_attr_if_exists(domain, "reseller"), + value_fn=lambda domain: getattr(domain, "reseller", None), ), )