vincent/koolnova-BMS-Integration
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

correct switch bug and rename async functions

Browse Source
This commit is contained in:
BENOIT Vincent
2024-02-27 14:52:29 +01:00
parent cee71535bf
commit 8eae65de57
8 changed files with 86 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_components/__init__.py
+1 -2
View File
@@ -29,7 +29,6 @@ async def async_setup_entry(hass: HomeAssistant,
stopbits: int = entry.data['Stopbits'] stopbits: int = entry.data['Stopbits']
timeout: int = entry.data['Timeout'] timeout: int = entry.data['Timeout']
#_LOGGER.debug("Appel de async_setup_entry - entry: entry_id={}, data={}".format(entry.entry_id, entry.data))
try: try:
device = Koolnova(name, port, addr, baudrate, parity, bytesize, stopbits, timeout) device = Koolnova(name, port, addr, baudrate, parity, bytesize, stopbits, timeout)
# connect to modbus client # connect to modbus client
@@ -37,7 +36,7 @@ async def async_setup_entry(hass: HomeAssistant,
# update attributes # update attributes
await device.update() await device.update()
# record each area in device # record each area in device
#_LOGGER.debug("Koolnova areas: {}".format(entry.data['areas'])) _LOGGER.debug("Koolnova areas: {}".format(entry.data['areas']))
for area in entry.data['areas']: for area in entry.data['areas']:
await device.add_manual_registered_zone(name=area['Name'], await device.add_manual_registered_zone(name=area['Name'],
id_zone=area['Area_id']) id_zone=area['Area_id'])
custom_components/climate.py
+1 -1
View File
@@ -164,7 +164,7 @@ class AreaClimateEntity(CoordinatorEntity, ClimateEntity):
""" Handle updated data from the coordinator """ """ Handle updated data from the coordinator """
for _cur_area in self.coordinator.data['areas']: for _cur_area in self.coordinator.data['areas']:
if _cur_area.id_zone == self._area.id_zone: if _cur_area.id_zone == self._area.id_zone:
_LOGGER.debug("[Climate {}] temp:{} - target:{} - state: {} - hvac:{} - fan:{}".format(_cur_area.id_zone, _LOGGER.debug("[UPDATE] [Climate {}] temp:{} - target:{} - state: {} - hvac:{} - fan:{}".format(_cur_area.id_zone,
_cur_area.real_temp, _cur_area.real_temp,
_cur_area.order_temp, _cur_area.order_temp,
_cur_area.state, _cur_area.state,
custom_components/const.py
+3 -3
View File
@@ -22,8 +22,8 @@ from .koolnova.const import (
DOMAIN = "testVBE_4" DOMAIN = "testVBE_4"
PLATFORMS: list[Platform] = [Platform.SENSOR, PLATFORMS: list[Platform] = [Platform.SENSOR,
Platform.SELECT, Platform.SELECT,
Platform.SWITCH, Platform.SWITCH,
Platform.CLIMATE] Platform.CLIMATE]
CONF_NAME = "koolnova_test_HA" CONF_NAME = "koolnova_test_HA"
@@ -31,7 +31,7 @@ CONF_DEVICE_ID = "device_id"
DEVICE_MANUFACTURER = "koolnova" DEVICE_MANUFACTURER = "koolnova"
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=60) #MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=60)
GLOBAL_MODE_POS_1 = "cold" GLOBAL_MODE_POS_1 = "cold"
GLOBAL_MODE_POS_2 = "heat" GLOBAL_MODE_POS_2 = "heat"
custom_components/koolnova/device.py
+1
View File
@@ -500,6 +500,7 @@ class Koolnova:
''' Set System State ''' ''' Set System State '''
if not isinstance(val, const.SysState): if not isinstance(val, const.SysState):
raise AssertionError('Input variable must be Enum SysState') raise AssertionError('Input variable must be Enum SysState')
_LOGGER.debug("set system state : {}".format(val))
ret = await self._client.set_system_status(val) ret = await self._client.set_system_status(val)
if not ret: if not ret:
_LOGGER.error("[SYS_STATE] Error writing {} to modbus".format(val)) _LOGGER.error("[SYS_STATE] Error writing {} to modbus".format(val))
custom_components/koolnova/operations.py
+28 -28
View File
@@ -62,13 +62,13 @@ class Operations:
if debug: if debug:
pymodbus_apply_logging_config("DEBUG") pymodbus_apply_logging_config("DEBUG")
async def __read_register(self, reg:int) -> (int, bool): async def __async_read_register(self, reg:int) -> (int, bool):
''' Read one holding register (code 0x03) ''' ''' Read one holding register (code 0x03) '''
rr = None rr = None
if not self._client.connected: if not self._client.connected:
raise ModbusConnexionError('Client Modbus not connected') raise ModbusConnexionError('Client Modbus not connected')
try: try:
#_LOGGER.debug("reading holding register: {} - Addr: {}".format(hex(reg), self._addr)) _LOGGER.debug("reading holding register: {} - Addr: {}".format(hex(reg), self._addr))
rr = await self._client.read_holding_registers(address=reg, count=1, slave=self._addr) rr = await self._client.read_holding_registers(address=reg, count=1, slave=self._addr)
if rr.isError(): if rr.isError():
_LOGGER.error("reading holding register error") _LOGGER.error("reading holding register error")
@@ -85,7 +85,7 @@ class Operations:
return None, False return None, False
return rr.registers[0], True return rr.registers[0], True
async def __read_registers(self, start_reg:int, count:int) -> (int, bool): async def __async_read_registers(self, start_reg:int, count:int) -> (int, bool):
''' Read holding registers (code 0x03) ''' ''' Read holding registers (code 0x03) '''
rr = None rr = None
if not self._client.connected: if not self._client.connected:
@@ -107,14 +107,14 @@ class Operations:
return None, False return None, False
return rr.registers, True return rr.registers, True
async def __write_register(self, reg:int, val:int) -> bool: async def __async_write_register(self, reg:int, val:int) -> bool:
''' Write one register (code 0x06) ''' ''' Write one register (code 0x06) '''
rq = None rq = None
ret = True ret = True
if not self._client.connected: if not self._client.connected:
raise ModbusConnexionError('Client Modbus not connected') raise ModbusConnexionError('Client Modbus not connected')
try: try:
#_LOGGER.debug("writing single register: {} - Addr: {} - Val: {}".format(hex(reg), self._addr, hex(val))) _LOGGER.debug("writing single register: {} - Addr: {} - Val: {}".format(hex(reg), self._addr, hex(val)))
rq = await self._client.write_register(address=reg, value=val, slave=self._addr) rq = await self._client.write_register(address=reg, value=val, slave=self._addr)
if rq.isError(): if rq.isError():
_LOGGER.error("writing register error") _LOGGER.error("writing register error")
@@ -142,8 +142,8 @@ class Operations:
self._client.close() self._client.close()
async def discover_registered_zones(self) -> list: async def discover_registered_zones(self) -> list:
''' Discover all zones registered to the system ''' ''' Discover all areas registered to the system '''
regs, ret = await self.__read_registers(start_reg=const.REG_START_ZONE, regs, ret = await self.__async_read_registers(start_reg=const.REG_START_ZONE,
count=const.NB_ZONE_MAX * const.NUM_REG_PER_ZONE) count=const.NB_ZONE_MAX * const.NUM_REG_PER_ZONE)
if not ret: if not ret:
raise ReadRegistersError("Read holding regsiter error") raise ReadRegistersError("Read holding regsiter error")
@@ -179,7 +179,7 @@ class Operations:
if zone_id > const.NB_ZONE_MAX or zone_id == 0: if zone_id > const.NB_ZONE_MAX or zone_id == 0:
raise ZoneIdError('Zone Id must be between 1 to {}'.format(const.NB_ZONE_MAX)) raise ZoneIdError('Zone Id must be between 1 to {}'.format(const.NB_ZONE_MAX))
zone_dict = {} zone_dict = {}
regs, ret = await self.__read_registers(start_reg = const.REG_START_ZONE + (4 * (zone_id - 1)), regs, ret = await self.__async_read_registers(start_reg = const.REG_START_ZONE + (4 * (zone_id - 1)),
count = const.NUM_REG_PER_ZONE) count = const.NUM_REG_PER_ZONE)
if not ret: if not ret:
raise ReadRegistersError("Error reading holding register") raise ReadRegistersError("Error reading holding register")
@@ -199,7 +199,7 @@ class Operations:
""" Get all areas values """ """ Get all areas values """
_areas_dict:dict = {} _areas_dict:dict = {}
# retreive all areas (registered and unregistered) # retreive all areas (registered and unregistered)
regs, ret = await self.__read_registers(start_reg = const.REG_START_ZONE, regs, ret = await self.__async_read_registers(start_reg = const.REG_START_ZONE,
count = const.NUM_REG_PER_ZONE * const.NB_ZONE_MAX) count = const.NUM_REG_PER_ZONE * const.NB_ZONE_MAX)
if not ret: if not ret:
raise ReadRegistersError("Error reading holding register") raise ReadRegistersError("Error reading holding register")
@@ -221,7 +221,7 @@ class Operations:
async def system_status(self) -> (bool, const.SysState): async def system_status(self) -> (bool, const.SysState):
''' Read system status register ''' ''' Read system status register '''
reg, ret = await self.__read_register(const.REG_SYS_STATE) reg, ret = await self.__async_read_register(const.REG_SYS_STATE)
if not ret: if not ret:
_LOGGER.error('Error retreive system status') _LOGGER.error('Error retreive system status')
reg = 0 reg = 0
@@ -231,14 +231,14 @@ class Operations:
opt:const.SysState, opt:const.SysState,
) -> bool: ) -> bool:
''' Write system status ''' ''' Write system status '''
ret = await self.__write_register(reg = const.REG_SYS_STATE, val = int(opt)) ret = await self.__async_write_register(reg = const.REG_SYS_STATE, val = int(opt))
if not ret: if not ret:
_LOGGER.error('Error writing system status') _LOGGER.error('Error writing system status')
return ret return ret
async def global_mode(self) -> (bool, const.GlobalMode): async def global_mode(self) -> (bool, const.GlobalMode):
''' Read global mode ''' ''' Read global mode '''
reg, ret = await self.__read_register(const.REG_GLOBAL_MODE) reg, ret = await self.__async_read_register(const.REG_GLOBAL_MODE)
if not ret: if not ret:
_LOGGER.error('Error retreive global mode') _LOGGER.error('Error retreive global mode')
reg = 0 reg = 0
@@ -248,14 +248,14 @@ class Operations:
opt:const.GlobalMode, opt:const.GlobalMode,
) -> bool: ) -> bool:
''' Write global mode ''' ''' Write global mode '''
ret = await self.__write_register(reg = const.REG_GLOBAL_MODE, val = int(opt)) ret = await self.__async_write_register(reg = const.REG_GLOBAL_MODE, val = int(opt))
if not ret: if not ret:
_LOGGER.error('Error writing global mode') _LOGGER.error('Error writing global mode')
return ret return ret
async def efficiency(self) -> (bool, const.Efficiency): async def efficiency(self) -> (bool, const.Efficiency):
''' read efficiency/speed ''' ''' read efficiency/speed '''
reg, ret = await self.__read_register(const.REG_EFFICIENCY) reg, ret = await self.__async_read_register(const.REG_EFFICIENCY)
if not ret: if not ret:
_LOGGER.error('Error retreive efficiency') _LOGGER.error('Error retreive efficiency')
reg = 0 reg = 0
@@ -265,7 +265,7 @@ class Operations:
opt:const.GlobalMode, opt:const.GlobalMode,
) -> bool: ) -> bool:
''' Write efficiency ''' ''' Write efficiency '''
ret = await self.__write_register(reg = const.REG_EFFICIENCY, val = int(opt)) ret = await self.__async_write_register(reg = const.REG_EFFICIENCY, val = int(opt))
if not ret: if not ret:
_LOGGER.error('Error writing efficiency') _LOGGER.error('Error writing efficiency')
return ret return ret
@@ -273,7 +273,7 @@ class Operations:
async def engines_throughput(self) -> (bool, list): async def engines_throughput(self) -> (bool, list):
''' read engines throughput AC1, AC2, AC3, AC4 ''' ''' read engines throughput AC1, AC2, AC3, AC4 '''
engines_lst = [] engines_lst = []
regs, ret = await self.__read_registers(const.REG_START_FLOW_ENGINE, regs, ret = await self.__async_read_registers(const.REG_START_FLOW_ENGINE,
const.NUM_OF_ENGINES) const.NUM_OF_ENGINES)
if ret: if ret:
for idx, reg in enumerate(regs): for idx, reg in enumerate(regs):
@@ -288,7 +288,7 @@ class Operations:
''' read engine throughput specified by id ''' ''' read engine throughput specified by id '''
if engine_id < 1 or engine_id > 4: if engine_id < 1 or engine_id > 4:
raise UnitIdError("engine Id must be between 1 and 4") raise UnitIdError("engine Id must be between 1 and 4")
reg, ret = await self.__read_register(const.REG_START_FLOW_ENGINE + (engine_id - 1)) reg, ret = await self.__async_read_register(const.REG_START_FLOW_ENGINE + (engine_id - 1))
if not ret: if not ret:
_LOGGER.error('Error retreive engine throughput for id:{}'.format(engine_id)) _LOGGER.error('Error retreive engine throughput for id:{}'.format(engine_id))
reg = 0 reg = 0
@@ -299,7 +299,7 @@ class Operations:
) -> (bool, const.FlowEngine): ) -> (bool, const.FlowEngine):
if engine_id < 1 or engine_id > 4: if engine_id < 1 or engine_id > 4:
raise UnitIdError("Engine id must be between 1 and 4") raise UnitIdError("Engine id must be between 1 and 4")
reg, ret = await self.__read_register(const.REG_START_FLOW_STATE_ENGINE + (engine_id - 1)) reg, ret = await self.__async_read_register(const.REG_START_FLOW_STATE_ENGINE + (engine_id - 1))
if not ret: if not ret:
_LOGGER.error('Error retreive engine state for id:{}'.format(engine_id)) _LOGGER.error('Error retreive engine state for id:{}'.format(engine_id))
reg = 0 reg = 0
@@ -310,7 +310,7 @@ class Operations:
) -> (bool, float): ) -> (bool, float):
if engine_id < 1 or engine_id > 4: if engine_id < 1 or engine_id > 4:
raise UnitIdError("Engine id must be between 1 and 4") raise UnitIdError("Engine id must be between 1 and 4")
reg, ret = await self.__read_register(const.REG_START_ORDER_TEMP + (engine_id - 1)) reg, ret = await self.__async_read_register(const.REG_START_ORDER_TEMP + (engine_id - 1))
if not ret: if not ret:
_LOGGER.error('Error retreive engine order temp for id:{}'.format(engine_id)) _LOGGER.error('Error retreive engine order temp for id:{}'.format(engine_id))
reg = 0 reg = 0
@@ -319,7 +319,7 @@ class Operations:
async def engine_orders_temp(self) -> (bool, list): async def engine_orders_temp(self) -> (bool, list):
''' read orders temperature for engines : AC1, AC2, AC3, AC4 ''' ''' read orders temperature for engines : AC1, AC2, AC3, AC4 '''
engines_lst = [] engines_lst = []
regs, ret = await self.__read_registers(const.REG_START_ORDER_TEMP, const.NUM_OF_ENGINES) regs, ret = await self.__async_read_registers(const.REG_START_ORDER_TEMP, const.NUM_OF_ENGINES)
if ret: if ret:
for idx, reg in enumerate(regs): for idx, reg in enumerate(regs):
engines_lst.append(reg/2) engines_lst.append(reg/2)
@@ -337,7 +337,7 @@ class Operations:
if val > const.MAX_TEMP_ORDER or val < const.MIN_TEMP_ORDER: if val > const.MAX_TEMP_ORDER or val < const.MIN_TEMP_ORDER:
_LOGGER.error('Order Temperature must be between {} and {}'.format(const.MIN_TEMP_ORDER, const.MAX_TEMP_ORDER)) _LOGGER.error('Order Temperature must be between {} and {}'.format(const.MIN_TEMP_ORDER, const.MAX_TEMP_ORDER))
return False return False
ret = await self.__write_register(reg = const.REG_START_ZONE + (4 * (zone_id - 1)) + const.REG_TEMP_ORDER, val = int(val * 2)) ret = await self.__async_write_register(reg = const.REG_START_ZONE + (4 * (zone_id - 1)) + const.REG_TEMP_ORDER, val = int(val * 2))
if not ret: if not ret:
_LOGGER.error('Error writing area order temperature') _LOGGER.error('Error writing area order temperature')
@@ -347,7 +347,7 @@ class Operations:
id_zone:int = 0, id_zone:int = 0,
) -> (bool, float): ) -> (bool, float):
""" get temperature of specific area id """ """ get temperature of specific area id """
reg, ret = await self.__read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_TEMP_REAL) reg, ret = await self.__async_read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_TEMP_REAL)
if not ret: if not ret:
_LOGGER.error('Error retreive area real temp') _LOGGER.error('Error retreive area real temp')
reg = 0 reg = 0
@@ -357,7 +357,7 @@ class Operations:
id_zone:int = 0, id_zone:int = 0,
) -> (bool, float): ) -> (bool, float):
""" get target temperature of specific area id """ """ get target temperature of specific area id """
reg, ret = await self.__read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_TEMP_ORDER) reg, ret = await self.__async_read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_TEMP_ORDER)
if not ret: if not ret:
_LOGGER.error('Error retreive area target temp') _LOGGER.error('Error retreive area target temp')
reg = 0 reg = 0
@@ -367,7 +367,7 @@ class Operations:
id_zone:int = 0, id_zone:int = 0,
) -> (bool, const.ZoneFanMode, const.ZoneClimMode): ) -> (bool, const.ZoneFanMode, const.ZoneClimMode):
""" get climate and fan mode of specific area id """ """ get climate and fan mode of specific area id """
reg, ret = await self.__read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_STATE_AND_FLOW) reg, ret = await self.__async_read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_STATE_AND_FLOW)
if not ret: if not ret:
_LOGGER.error('Error retreive area fan and climate values') _LOGGER.error('Error retreive area fan and climate values')
reg = 0 reg = 0
@@ -377,7 +377,7 @@ class Operations:
id_zone:int = 0, id_zone:int = 0,
) -> (bool, const.ZoneRegister, const.ZoneState): ) -> (bool, const.ZoneRegister, const.ZoneState):
""" get area state and register """ """ get area state and register """
reg, ret = await self.__read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_LOCK_ZONE) reg, ret = await self.__async_read_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_LOCK_ZONE)
if not ret: if not ret:
_LOGGER.error('Error retreive area register value') _LOGGER.error('Error retreive area register value')
reg = 0 reg = 0
@@ -397,7 +397,7 @@ class Operations:
_LOGGER.error("Error reading state and register mode") _LOGGER.error("Error reading state and register mode")
return ret return ret
#_LOGGER.debug("register & state: {}".format(hex((int(register) << 1) | (int(val) & 0b01)))) #_LOGGER.debug("register & state: {}".format(hex((int(register) << 1) | (int(val) & 0b01))))
ret = await self.__write_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_LOCK_ZONE, ret = await self.__async_write_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_LOCK_ZONE,
val = int(int(register) << 1) | (int(val) & 0b01)) val = int(int(register) << 1) | (int(val) & 0b01))
if not ret: if not ret:
_LOGGER.error('Error writing area state value') _LOGGER.error('Error writing area state value')
@@ -418,7 +418,7 @@ class Operations:
_LOGGER.error("Error reading fan and clim mode") _LOGGER.error("Error reading fan and clim mode")
return ret return ret
#_LOGGER.debug("Fan & Clim: {}".format(hex((int(fan) << 4) | (int(val) & 0x0F)))) #_LOGGER.debug("Fan & Clim: {}".format(hex((int(fan) << 4) | (int(val) & 0x0F))))
ret = await self.__write_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_STATE_AND_FLOW, ret = await self.__async_write_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_STATE_AND_FLOW,
val = int(int(fan) << 4) | (int(val) & 0x0F)) val = int(int(fan) << 4) | (int(val) & 0x0F))
if not ret: if not ret:
_LOGGER.error('Error writing area climate mode') _LOGGER.error('Error writing area climate mode')
@@ -439,7 +439,7 @@ class Operations:
_LOGGER.error("Error reading fan and clim mode") _LOGGER.error("Error reading fan and clim mode")
return ret return ret
#_LOGGER.debug("Fan & Clim: {}".format(hex((int(val) << 4) | (int(clim) & 0x0F)))) #_LOGGER.debug("Fan & Clim: {}".format(hex((int(val) << 4) | (int(clim) & 0x0F))))
ret = await self.__write_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_STATE_AND_FLOW, ret = await self.__async_write_register(reg = const.REG_START_ZONE + (4 * (id_zone - 1)) + const.REG_STATE_AND_FLOW,
val = int(int(val) << 4) | (int(clim) & 0x0F)) val = int(int(val) << 4) | (int(clim) & 0x0F))
if not ret: if not ret:
_LOGGER.error('Error writing area fan mode') _LOGGER.error('Error writing area fan mode')
custom_components/select.py
+10 -2
View File
@@ -6,6 +6,7 @@ import logging
from homeassistant.core import HomeAssistant, callback, Event, State from homeassistant.core import HomeAssistant, callback, Event, State
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.entity import EntityCategory
from homeassistant.components.select import SelectEntity from homeassistant.components.select import SelectEntity
from homeassistant.util import Throttle from homeassistant.util import Throttle
from homeassistant.const import UnitOfTime from homeassistant.const import UnitOfTime
@@ -15,7 +16,6 @@ from homeassistant.helpers.update_coordinator import (
from .const import ( from .const import (
DOMAIN, DOMAIN,
MIN_TIME_BETWEEN_UPDATES,
GLOBAL_MODES, GLOBAL_MODES,
GLOBAL_MODE_TRANSLATION, GLOBAL_MODE_TRANSLATION,
EFF_MODES, EFF_MODES,
@@ -50,6 +50,8 @@ async def async_setup_entry(hass: HomeAssistant,
class GlobalModeSelect(CoordinatorEntity, SelectEntity): class GlobalModeSelect(CoordinatorEntity, SelectEntity):
""" Select component to set global HVAC mode """ """ Select component to set global HVAC mode """
_attr_entity_category: EntityCategory = EntityCategory.CONFIG
def __init__(self, def __init__(self,
coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument
device: Koolnova, # pylint: disable=unused-argument, device: Koolnova, # pylint: disable=unused-argument,
@@ -83,13 +85,17 @@ class GlobalModeSelect(CoordinatorEntity, SelectEntity):
def _handle_coordinator_update(self) -> None: def _handle_coordinator_update(self) -> None:
""" Handle updated data from the coordinator """ Handle updated data from the coordinator
Retrieve latest state of global mode """ Retrieve latest state of global mode """
_LOGGER.debug("[UPDATE] Global Mode: {}".format(self.coordinator.data['glob']))
self.select_option( self.select_option(
GLOBAL_MODE_TRANSLATION[int(self.coordinator.data['glob'])] GLOBAL_MODE_TRANSLATION[int(self.coordinator.data['glob'])]
) )
self.async_write_ha_state()
class EfficiencySelect(CoordinatorEntity, SelectEntity): class EfficiencySelect(CoordinatorEntity, SelectEntity):
"""Select component to set global efficiency """ """Select component to set global efficiency """
_attr_entity_category: EntityCategory = EntityCategory.CONFIG
def __init__(self, def __init__(self,
coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument
device: Koolnova, # pylint: disable=unused-argument, device: Koolnova, # pylint: disable=unused-argument,
@@ -128,6 +134,8 @@ class EfficiencySelect(CoordinatorEntity, SelectEntity):
def _handle_coordinator_update(self) -> None: def _handle_coordinator_update(self) -> None:
""" Handle updated data from the coordinator """ Handle updated data from the coordinator
Retrieve latest state of global efficiency """ Retrieve latest state of global efficiency """
_LOGGER.debug("[UPDATE] Efficiency: {}".format(self.coordinator.data['eff']))
self.select_option( self.select_option(
EFF_TRANSLATION[int(self.coordinator.data['eff'])] EFF_TRANSLATION[int(self.coordinator.data['eff'])]
) )
self.async_write_ha_state()
custom_components/sensor.py
+2
View File
@@ -148,6 +148,7 @@ class DiagEngineThroughputSensor(CoordinatorEntity, SensorEntity):
""" Handle updated data from the coordinator """ """ Handle updated data from the coordinator """
for _cur_engine in self.coordinator.data['engines']: for _cur_engine in self.coordinator.data['engines']:
if self._engine.engine_id == _cur_engine.engine_id: if self._engine.engine_id == _cur_engine.engine_id:
_LOGGER.debug("[UPDATE] [ENGINE AC{}] Troughput: {}".format(_cur_engine.engine_id, _cur_engine.throughput))
self._attr_native_value = "{}".format(_cur_engine.throughput) self._attr_native_value = "{}".format(_cur_engine.throughput)
self.async_write_ha_state() self.async_write_ha_state()
@@ -182,5 +183,6 @@ class DiagEngineTempOrderSensor(CoordinatorEntity, SensorEntity):
""" Handle updated data from the coordinator """ """ Handle updated data from the coordinator """
for _cur_engine in self.coordinator.data['engines']: for _cur_engine in self.coordinator.data['engines']:
if self._engine.engine_id == _cur_engine.engine_id: if self._engine.engine_id == _cur_engine.engine_id:
_LOGGER.debug("[UPDATE] [ENGINE AC{}] Order temp: {}".format(_cur_engine.engine_id, _cur_engine.order_temp))
self._attr_native_value = "{}".format(_cur_engine.order_temp) self._attr_native_value = "{}".format(_cur_engine.order_temp)
self.async_write_ha_state() self.async_write_ha_state()
custom_components/switch.py
+40 -14
View File
@@ -5,21 +5,29 @@ import logging
from homeassistant.core import HomeAssistant, callback, Event, State from homeassistant.core import HomeAssistant, callback, Event, State
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.entity import EntityCategory
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.components.switch import SwitchEntity from homeassistant.components.switch import (
SwitchEntity,
SwitchDeviceClass
)
from homeassistant.util import Throttle from homeassistant.util import Throttle
from homeassistant.helpers.update_coordinator import ( from homeassistant.helpers.update_coordinator import (
CoordinatorEntity, CoordinatorEntity,
) )
from .const import ( from .const import (
DOMAIN, DOMAIN
MIN_TIME_BETWEEN_UPDATES,
) )
from .coordinator import KoolnovaCoordinator from .coordinator import KoolnovaCoordinator
from homeassistant.const import UnitOfTime from homeassistant.const import (
STATE_OFF,
STATE_ON
)
from .koolnova.device import Koolnova from .koolnova.device import Koolnova
from .koolnova.const import ( from .koolnova.const import (
SysState, SysState,
@@ -43,40 +51,58 @@ async def async_setup_entry(hass: HomeAssistant,
class SystemStateSwitch(CoordinatorEntity, SwitchEntity): class SystemStateSwitch(CoordinatorEntity, SwitchEntity):
"""Select component to set system state """ """Select component to set system state """
_attr_has_entity_name = True _attr_has_entity_name: bool = True
_attr_device_class: SwitchDeviceClass = SwitchDeviceClass.SWITCH
_attr_entity_category: EntityCategory = EntityCategory.CONFIG
def __init__(self, def __init__(self,
coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument
device: Koolnova, # pylint: disable=unused-argument, device: Koolnova, # pylint: disable=unused-argument
) -> None: ) -> None:
super().__init__(coordinator) super().__init__(coordinator)
self._device = device self._device = device
self._attr_name = f"{device.name} Global HVAC State" self._attr_name = f"{device.name} Global HVAC State"
self._attr_device_info = device.device_info self._attr_device_info = device.device_info
self._attr_unique_id = f"{DOMAIN}-Global-HVACState-switch" self._attr_unique_id = f"{DOMAIN}-Global-HVAC-State-switch"
self._is_on = bool(int(self._device.sys_state)) self._attr_is_on = bool(int(self._device.sys_state))
if bool(int(self._device.sys_state)):
self._attr_state = STATE_ON
else:
self._attr_state = STATE_OFF
async def async_turn_on(self, **kwargs): async def async_turn_on(self, **kwargs):
""" Turn the entity on. """ """ Turn the entity on. """
_LOGGER.debug("Turn on system") _LOGGER.debug("Turn on system")
await self._device.set_sys_state(SysState.SYS_STATE_ON) await self._device.set_sys_state(SysState.SYS_STATE_ON)
self._is_on = True self._attr_is_on = True
self._attr_state = STATE_ON
self.async_write_ha_state()
async def async_turn_off(self, **kwargs): async def async_turn_off(self, **kwargs):
""" Turn the entity off. """ """ Turn the entity off. """
_LOGGER.debug("Turn off system") _LOGGER.debug("Turn off system")
await self._device.set_sys_state(SysState.SYS_STATE_OFF) await self._device.set_sys_state(SysState.SYS_STATE_OFF)
self._is_on = False self._attr_is_on = False
self._attr_state = STATE_OFF
self.async_write_ha_state()
@callback @callback
def _handle_coordinator_update(self) -> None: def _handle_coordinator_update(self) -> None:
""" Handle updated data from the coordinator """ """ Handle updated data from the coordinator """
self._is_on = bool(int(self.coordinator.data['sys'])) self._attr_is_on = bool(int(self.coordinator.data['sys']))
_LOGGER.debug("[UPDATE] Switch State: {} - is_on ? {} - state ? {}".format(bool(int(self.coordinator.data['sys'])),
self._attr_is_on,
self._attr_state))
if bool(int(self.coordinator.data['sys'])):
self._attr_state = STATE_ON
else:
self._attr_state = STATE_OFF
self.async_write_ha_state()
@property @property
def is_on(self): def is_on(self) -> bool | None:
""" If the switch is currently on or off. """ """Return True if entity is on."""
return self._is_on return bool(int(self._device.sys_state))
@property @property
def icon(self) -> str | None: def icon(self) -> str | None: