orkun
/
ha_samsung_soundbar
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

fix woofer number box

This commit is contained in:
samuelspagl 2023-09-07 13:02:47 +02:00
parent e010f0af50
commit 8a547a3caf
1 changed files with 9 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
custom_components/samsung_soundbar/number.py
View File

@ -19,39 +19,27 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
device = device_config.device device = device_config.device
if device.device_id == config_entry.data.get(CONF_ENTRY_DEVICE_ID): if device.device_id == config_entry.data.get(CONF_ENTRY_DEVICE_ID):
entities.append( entities.append(
SoundbarNumberEntity( SoundbarWooferNumberEntity(
device, device,
"woofer_level", "woofer_level",
lambda: device.woofer_level,
device.set_woofer,
(-6, 12),
unit="dB",
mode=NumberMode.BOX
) )
) )
async_add_entities(entities) async_add_entities(entities)
return True return True
class SoundbarNumberEntity(NumberEntity): class SoundbarWooferNumberEntity(NumberEntity):
def __init__( def __init__(
self, self,
device: SoundbarDevice, device: SoundbarDevice,
append_unique_id: str, append_unique_id: str,
state_function,
on_function,
min_max: tuple,
*,
unit: str = "%",
step_size: float = 1,
mode: NumberMode = NumberMode.SLIDER
): ):
self.entity_id = f"number.{device.device_name}_{append_unique_id}" self.entity_id = f"number.{device.device_name}_{append_unique_id}"
self.entity_description = NumberEntityDescription(native_max_value=min_max[1], self.entity_description = NumberEntityDescription(native_max_value=6,
native_min_value=min_max[0], native_min_value=-10,
mode=mode, mode=NumberMode.BOX,
native_step=step_size, native_step=1,
native_unit_of_measurement=unit, native_unit_of_measurement="dB",
key=append_unique_id, key=append_unique_id,
) )
self.__device = device self.__device = device
@ -65,11 +53,6 @@ class SoundbarNumberEntity(NumberEntity):
) )
self.__append_unique_id = append_unique_id self.__append_unique_id = append_unique_id
self.__current_value_function = state_function
self.__set_value_function = on_function
self.__min_value = min_max[0]
self.__max_value = min_max[1]
# ---------- GENERAL --------------- # ---------- GENERAL ---------------
@property @property
@ -80,8 +63,7 @@ class SoundbarNumberEntity(NumberEntity):
@property @property
def native_value(self) -> float | None: def native_value(self) -> float | None:
_LOGGER.info(f"[{DOMAIN}] Soundbar Woofer number value {self.__current_value_function()}") return self.__device.woofer_level
return self.__current_value_function()
async def async_set_native_value(self, value: float): async def async_set_native_value(self, value: float):
await self.__set_value_function(value) await self.__device.set_woofer(int(value))