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 the list of supported features."""
return self._features return self._features
def add_feature(self, feature: Feature): def _add_feature(self, feature: Feature):
"""Add a new feature to the device.""" """Add a new feature to the device."""
desc_name = feature.name.lower().replace(" ", "_") desc_name = feature.name.lower().replace(" ", "_")
if desc_name in self._features: 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"]) self._set_sys_info(self._last_update["system"]["get_sysinfo"])
async def _initialize_features(self): async def _initialize_features(self):
self.add_feature( self._add_feature(
Feature( Feature(
device=self, name="RSSI", attribute_getter="rssi", icon="mdi:signal" device=self, name="RSSI", attribute_getter="rssi", icon="mdi:signal"
) )
) )
self.add_feature( self._add_feature(
Feature( Feature(
device=self, name="Time", attribute_getter="time", show_in_hass=False device=self, name="Time", attribute_getter="time", show_in_hass=False
) )
) )
if "on_time" in self._sys_info: if "on_time" in self._sys_info:
self.add_feature( self._add_feature(
Feature( Feature(
device=self, device=self,
name="On since", name="On since",
@ -344,7 +344,7 @@ class IotDevice(Device):
if module.is_supported: if module.is_supported:
supported[module._module] = module supported[module._module] = module
for module_feat in module._module_features.values(): for module_feat in module._module_features.values():
self.add_feature(module_feat) self._add_feature(module_feat)
self._supported_modules = supported self._supported_modules = supported

View File

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

View File

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