diff --git a/custom_components/senec/const.py b/custom_components/senec/const.py index db1aea5..c0aaac1 100644 --- a/custom_components/senec/const.py +++ b/custom_components/senec/const.py @@ -16,15 +16,14 @@ NumberMode ) from homeassistant.const import ( - ENERGY_KILO_WATT_HOUR, PERCENTAGE, - POWER_WATT, - POWER_KILO_WATT, - TEMP_CELSIUS, UnitOfElectricPotential, UnitOfElectricCurrent, UnitOfFrequency, UnitOfTime, + UnitOfTemperature, + UnitOfEnergy, + UnitOfPower ) from homeassistant.helpers.entity import EntityCategory @@ -177,7 +176,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="consumption_total", name="House consumed", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:home-import-outline", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL, @@ -186,7 +185,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="powergenerated_total", name="Solar generated", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:solar-power", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL, @@ -195,7 +194,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="accuimport_total", name="Battery discharged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:home-battery-outline", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL, @@ -204,7 +203,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="accuexport_total", name="Battery charged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:home-battery", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL, @@ -213,7 +212,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="gridimport_total", name="Grid Imported", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:transmission-tower-export", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL, @@ -222,7 +221,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="gridexport_total", name="Grid Exported", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:transmission-tower-import", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL, @@ -239,7 +238,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="powergenerated_now", name="Solar Generated Power", - native_unit_of_measurement=POWER_KILO_WATT, + native_unit_of_measurement=UnitOfPower.KILO_WATT, icon="mdi:solar-power", device_class=SensorDeviceClass.POWER, suggested_display_precision=3, @@ -248,7 +247,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="consumption_now", name="House Power", - native_unit_of_measurement=POWER_KILO_WATT, + native_unit_of_measurement=UnitOfPower.KILO_WATT, icon="mdi:home-import-outline", device_class=SensorDeviceClass.POWER, suggested_display_precision=3, @@ -257,7 +256,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="accuimport_now", name="Battery Discharge Power", - native_unit_of_measurement=POWER_KILO_WATT, + native_unit_of_measurement=UnitOfPower.KILO_WATT, icon="mdi:home-battery", device_class=SensorDeviceClass.POWER, suggested_display_precision=3, @@ -266,7 +265,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="accuexport_now", name="Battery Charge Power", - native_unit_of_measurement=POWER_KILO_WATT, + native_unit_of_measurement=UnitOfPower.KILO_WATT, icon="mdi:home-battery-outline", device_class=SensorDeviceClass.POWER, suggested_display_precision=3, @@ -283,7 +282,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="gridimport_now", name="Grid Imported Power", - native_unit_of_measurement=POWER_KILO_WATT, + native_unit_of_measurement=UnitOfPower.KILO_WATT, icon="mdi:transmission-tower-export", device_class=SensorDeviceClass.POWER, suggested_display_precision=3, @@ -292,7 +291,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="gridexport_now", name="Grid Exported Power", - native_unit_of_measurement=POWER_KILO_WATT, + native_unit_of_measurement=UnitOfPower.KILO_WATT, icon="mdi:transmission-tower-import", device_class=SensorDeviceClass.POWER, suggested_display_precision=3, @@ -358,7 +357,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="ac_power", name="AC Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:solar-power", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -367,7 +366,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="ac_power_fast", name="AC Power (fast)", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:solar-power", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -406,7 +405,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): controls=("bdc_only"), key="bdc_bat_power", name="BDC Battery Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:battery-charging-100", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -436,7 +435,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): controls=("bdc_only"), key="bdc_link_power", name="BDC Link Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:power-plug-outline", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -480,7 +479,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="gridpower", name="Grid Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:transmission-tower", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -489,7 +488,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="gridconsumedpower", name="Grid consumed Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:transmission-tower-import", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -498,7 +497,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="gridinjectedpower", name="Grid injected Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:transmission-tower-export", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -507,7 +506,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="ownconsumedpower", name="Own consumed Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:home-import-outline", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -932,7 +931,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): native_max_value=65535, native_min_value=0, native_step=1, - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, ), ExtNumberEntityDescription( senec_lala_section=SENEC_SECTION_SOCKETS, @@ -947,7 +946,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): native_max_value=65535, native_min_value=0, native_step=1, - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, ), ExtNumberEntityDescription( senec_lala_section=SENEC_SECTION_SOCKETS, @@ -1018,7 +1017,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): native_max_value=65535, native_min_value=0, native_step=1, - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, ), ExtNumberEntityDescription( senec_lala_section=SENEC_SECTION_SOCKETS, @@ -1033,7 +1032,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): native_max_value=65535, native_min_value=0, native_step=1, - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, ), ExtNumberEntityDescription( senec_lala_section=SENEC_SECTION_SOCKETS, @@ -1148,28 +1147,28 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="battery_temp", name="Battery Temperature", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, icon="mdi:thermometer", entity_category=EntityCategory.DIAGNOSTIC, ), ExtSensorEntityDescription( key="case_temp", name="Case Temperature", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, icon="mdi:thermometer", entity_category=EntityCategory.DIAGNOSTIC, ), ExtSensorEntityDescription( key="mcu_temp", name="Controller Temperature", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, icon="mdi:thermometer", entity_category=EntityCategory.DIAGNOSTIC, ), ExtSensorEntityDescription( key="solar_generated_power", name="Solar Generated Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:solar-power", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1177,7 +1176,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="house_power", name="House Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:home-import-outline", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1185,7 +1184,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="battery_state_power", name="Battery State Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:home-battery", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1209,7 +1208,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="battery_charge_power", name="Battery Charge Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:home-battery", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1217,7 +1216,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="battery_discharge_power", name="Battery Discharge Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:home-battery-outline", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1233,7 +1232,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="grid_state_power", name="Grid State Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:transmission-tower", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1241,7 +1240,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="grid_imported_power", name="Grid Imported Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:transmission-tower-export", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1249,7 +1248,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="grid_exported_power", name="Grid Exported Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:transmission-tower-import", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -1258,7 +1257,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="house_total_consumption", controls=("require_stats_fields"), name="House consumed", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:home-import-outline", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -1267,7 +1266,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="solar_total_generated", controls=("require_stats_fields"), name="Solar generated", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:solar-power", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -1276,7 +1275,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="battery_total_charged", controls=("require_stats_fields"), name="Battery charged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:home-battery", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -1285,7 +1284,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="battery_total_discharged", controls=("require_stats_fields"), name="Battery discharged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:home-battery-outline", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -1294,7 +1293,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="grid_total_import", controls=("require_stats_fields"), name="Grid Imported", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:transmission-tower-export", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -1303,7 +1302,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="grid_total_export", controls=("require_stats_fields"), name="Grid Exported", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:transmission-tower-import", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -1327,7 +1326,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="solar_mpp1_power", name="MPP1 Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1351,7 +1350,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="solar_mpp2_power", name="MPP2 Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1375,7 +1374,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="solar_mpp3_power", name="MPP3 Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1391,7 +1390,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="enfluri_net_power_total", name="Enfluri Net Total Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1447,7 +1446,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="enfluri_net_power_p1", name="Enfluri Net Power Phase 1", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1455,7 +1454,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="enfluri_net_power_p2", name="Enfluri Net Power Phase 2", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1463,7 +1462,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): ExtSensorEntityDescription( key="enfluri_net_power_p3", name="Enfluri Net Power Phase 3", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1481,7 +1480,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="enfluri_usage_power_total", name="Enfluri Usage Total Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1544,7 +1543,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="enfluri_usage_power_p1", name="Enfluri Usage Power Phase 1", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1553,7 +1552,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="enfluri_usage_power_p2", name="Enfluri Usage Power Phase 2", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1562,7 +1561,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="enfluri_usage_power_p3", name="Enfluri Usage Power Phase 3", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:meter-electric", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.MEASUREMENT, @@ -1573,7 +1572,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_a1", name="Module A: Cell Temperature A1", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1584,7 +1583,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_a2", name="Module A: Cell Temperature A2", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1595,7 +1594,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_a3", name="Module A: Cell Temperature A3", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1606,7 +1605,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_a4", name="Module A: Cell Temperature A4", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1617,7 +1616,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_a5", name="Module A: Cell Temperature A5", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1628,7 +1627,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_a6", name="Module A: Cell Temperature A6", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1639,7 +1638,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_b1", name="Module B: Cell Temperature B1", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1650,7 +1649,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_b2", name="Module B: Cell Temperature B2", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1661,7 +1660,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_b3", name="Module B: Cell Temperature B3", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1672,7 +1671,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_b4", name="Module B: Cell Temperature B4", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1683,7 +1682,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_b5", name="Module B: Cell Temperature B5", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1694,7 +1693,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_b6", name="Module B: Cell Temperature B6", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1705,7 +1704,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_c1", name="Module C: Cell Temperature C1", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1716,7 +1715,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_c2", name="Module C: Cell Temperature C2", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1727,7 +1726,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_c3", name="Module C: Cell Temperature C3", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1738,7 +1737,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_c4", name="Module C: Cell Temperature C4", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1749,7 +1748,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_c5", name="Module C: Cell Temperature C5", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1760,7 +1759,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_c6", name="Module C: Cell Temperature C6", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1771,7 +1770,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_d1", name="Module D: Cell Temperature D1", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1782,7 +1781,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_d2", name="Module D: Cell Temperature D2", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1793,7 +1792,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_d3", name="Module D: Cell Temperature D3", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1804,7 +1803,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_d4", name="Module D: Cell Temperature D4", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1815,7 +1814,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_d5", name="Module D: Cell Temperature D5", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -1826,7 +1825,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): key="bms_cell_temp_d6", name="Module D: Cell Temperature D6", icon="mdi:thermometer", - native_unit_of_measurement=TEMP_CELSIUS, + native_unit_of_measurement=UnitOfTemperature.CELSIUS, device_class=SensorDeviceClass.TEMPERATURE, state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, @@ -2587,7 +2586,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="wallbox_1_power", name="Wallbox I Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:car-arrow-left", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -2605,7 +2604,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): controls=("require_stats_fields"), key="wallbox_1_energy", name="Wallbox I charged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:ev-station", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -2663,7 +2662,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="wallbox_2_power", name="Wallbox II Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:car-arrow-left", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -2681,7 +2680,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): controls=("require_stats_fields"), key="wallbox_2_energy", name="Wallbox II charged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:ev-station", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -2739,7 +2738,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="wallbox_3_power", name="Wallbox III Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:car-arrow-left", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -2757,7 +2756,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): controls=("require_stats_fields"), key="wallbox_3_energy", name="Wallbox III charged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:ev-station", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING, @@ -2815,7 +2814,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): entity_registry_enabled_default=False, key="wallbox_4_power", name="Wallbox IV Power", - native_unit_of_measurement=POWER_WATT, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:car-arrow-left", device_class=SensorDeviceClass.POWER, state_class=SensorStateClass.MEASUREMENT, @@ -2833,7 +2832,7 @@ class ExtNumberEntityDescription(NumberEntityDescription): controls=("require_stats_fields"), key="wallbox_4_energy", name="Wallbox IV charged", - native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, icon="mdi:ev-station", device_class=SensorDeviceClass.ENERGY, state_class=SensorStateClass.TOTAL_INCREASING,