Make add_feature private

This commit is contained in:
Teemu Rytilahti 2024-02-14 20:00:37 +01:00
parent fc29d2456e
commit 231a3e4dc5
4 changed files with 12 additions and 12 deletions

View File

@ -302,7 +302,7 @@ class Device(ABC):
"""Return the list of supported features."""
return self._features
def add_feature(self, feature: Feature):
def _add_feature(self, feature: Feature):
"""Add a new feature to the device."""
desc_name = feature.name.lower().replace(" ", "_")
if desc_name in self._features:

View File

@ -308,18 +308,18 @@ class IotDevice(Device):
self._set_sys_info(self._last_update["system"]["get_sysinfo"])
async def _initialize_features(self):
self.add_feature(
self._add_feature(
Feature(
device=self, name="RSSI", attribute_getter="rssi", icon="mdi:signal"
)
)
self.add_feature(
self._add_feature(
Feature(
device=self, name="Time", attribute_getter="time", show_in_hass=False
)
)
if "on_time" in self._sys_info:
self.add_feature(
self._add_feature(
Feature(
device=self,
name="On since",
@ -344,7 +344,7 @@ class IotDevice(Device):
if module.is_supported:
supported[module._module] = module
for module_feat in module._module_features.values():
self.add_feature(module_feat)
self._add_feature(module_feat)
self._supported_modules = supported

View File

@ -57,7 +57,7 @@ class IotPlug(IotDevice):
self.add_module("time", Time(self, "time"))
self.add_module("cloud", Cloud(self, "cnCloud"))
self.add_feature(
self._add_feature(
Feature(
device=self,
name="LED",

View File

@ -133,7 +133,7 @@ class SmartDevice(Device):
async def _initialize_features(self):
"""Initialize device features."""
self.add_feature(
self._add_feature(
Feature(
self,
"Signal Level",
@ -141,7 +141,7 @@ class SmartDevice(Device):
icon="mdi:signal",
)
)
self.add_feature(
self._add_feature(
Feature(
self,
"RSSI",
@ -149,17 +149,17 @@ class SmartDevice(Device):
icon="mdi:signal",
)
)
self.add_feature(
self._add_feature(
Feature(
device=self, name="Time", attribute_getter="time", show_in_hass=False
)
)
self.add_feature(
self._add_feature(
Feature(device=self, name="SSID", attribute_getter="ssid", icon="mdi:wifi")
)
if "overheated" in self._info:
self.add_feature(
self._add_feature(
Feature(
self,
"Overheated",
@ -172,7 +172,7 @@ class SmartDevice(Device):
# We check for the key available, and not for the property truthiness,
# as the value is falsy when the device is off.
if "on_time" in self._info:
self.add_feature(
self._add_feature(
Feature(
device=self,
name="On since",