Use enums in pluugwise (#62067)

This commit is contained in:
Robert Hillis 2021-12-16 09:40:32 -05:00 committed by GitHub
parent a5cf783e6a
commit e1a7f6d1b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,17 +3,9 @@
import logging import logging
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
DEVICE_CLASS_BATTERY, SensorDeviceClass,
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_GAS,
DEVICE_CLASS_ILLUMINANCE,
DEVICE_CLASS_POWER,
DEVICE_CLASS_PRESSURE,
DEVICE_CLASS_TEMPERATURE,
STATE_CLASS_MEASUREMENT,
STATE_CLASS_TOTAL,
STATE_CLASS_TOTAL_INCREASING,
SensorEntity, SensorEntity,
SensorStateClass,
) )
from homeassistant.const import ( from homeassistant.const import (
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
@ -46,26 +38,26 @@ _LOGGER = logging.getLogger(__name__)
ATTR_TEMPERATURE = [ ATTR_TEMPERATURE = [
"Temperature", "Temperature",
TEMP_CELSIUS, TEMP_CELSIUS,
DEVICE_CLASS_TEMPERATURE, SensorDeviceClass.TEMPERATURE,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
] ]
ATTR_BATTERY_LEVEL = [ ATTR_BATTERY_LEVEL = [
"Charge", "Charge",
PERCENTAGE, PERCENTAGE,
DEVICE_CLASS_BATTERY, SensorDeviceClass.BATTERY,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
] ]
ATTR_ILLUMINANCE = [ ATTR_ILLUMINANCE = [
"Illuminance", "Illuminance",
UNIT_LUMEN, UNIT_LUMEN,
DEVICE_CLASS_ILLUMINANCE, SensorDeviceClass.ILLUMINANCE,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
] ]
ATTR_PRESSURE = [ ATTR_PRESSURE = [
"Pressure", "Pressure",
PRESSURE_BAR, PRESSURE_BAR,
DEVICE_CLASS_PRESSURE, SensorDeviceClass.PRESSURE,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
] ]
TEMP_SENSOR_MAP = { TEMP_SENSOR_MAP = {
@ -82,122 +74,122 @@ ENERGY_SENSOR_MAP = {
"electricity_consumed": [ "electricity_consumed": [
"Current Consumed Power", "Current Consumed Power",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"electricity_produced": [ "electricity_produced": [
"Current Produced Power", "Current Produced Power",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"electricity_consumed_interval": [ "electricity_consumed_interval": [
"Consumed Power Interval", "Consumed Power Interval",
ENERGY_WATT_HOUR, ENERGY_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"electricity_consumed_peak_interval": [ "electricity_consumed_peak_interval": [
"Consumed Power Interval", "Consumed Power Interval",
ENERGY_WATT_HOUR, ENERGY_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"electricity_consumed_off_peak_interval": [ "electricity_consumed_off_peak_interval": [
"Consumed Power Interval (off peak)", "Consumed Power Interval (off peak)",
ENERGY_WATT_HOUR, ENERGY_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"electricity_produced_interval": [ "electricity_produced_interval": [
"Produced Power Interval", "Produced Power Interval",
ENERGY_WATT_HOUR, ENERGY_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"electricity_produced_peak_interval": [ "electricity_produced_peak_interval": [
"Produced Power Interval", "Produced Power Interval",
ENERGY_WATT_HOUR, ENERGY_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"electricity_produced_off_peak_interval": [ "electricity_produced_off_peak_interval": [
"Produced Power Interval (off peak)", "Produced Power Interval (off peak)",
ENERGY_WATT_HOUR, ENERGY_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"electricity_consumed_off_peak_point": [ "electricity_consumed_off_peak_point": [
"Current Consumed Power (off peak)", "Current Consumed Power (off peak)",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"electricity_consumed_peak_point": [ "electricity_consumed_peak_point": [
"Current Consumed Power", "Current Consumed Power",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"electricity_consumed_off_peak_cumulative": [ "electricity_consumed_off_peak_cumulative": [
"Cumulative Consumed Power (off peak)", "Cumulative Consumed Power (off peak)",
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL_INCREASING, SensorStateClass.TOTAL_INCREASING,
], ],
"electricity_consumed_peak_cumulative": [ "electricity_consumed_peak_cumulative": [
"Cumulative Consumed Power", "Cumulative Consumed Power",
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL_INCREASING, SensorStateClass.TOTAL_INCREASING,
], ],
"electricity_produced_off_peak_point": [ "electricity_produced_off_peak_point": [
"Current Produced Power (off peak)", "Current Produced Power (off peak)",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"electricity_produced_peak_point": [ "electricity_produced_peak_point": [
"Current Produced Power", "Current Produced Power",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"electricity_produced_off_peak_cumulative": [ "electricity_produced_off_peak_cumulative": [
"Cumulative Produced Power (off peak)", "Cumulative Produced Power (off peak)",
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL_INCREASING, SensorStateClass.TOTAL_INCREASING,
], ],
"electricity_produced_peak_cumulative": [ "electricity_produced_peak_cumulative": [
"Cumulative Produced Power", "Cumulative Produced Power",
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL_INCREASING, SensorStateClass.TOTAL_INCREASING,
], ],
"gas_consumed_interval": [ "gas_consumed_interval": [
"Current Consumed Gas Interval", "Current Consumed Gas Interval",
VOLUME_CUBIC_METERS, VOLUME_CUBIC_METERS,
DEVICE_CLASS_GAS, SensorDeviceClass.GAS,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
"gas_consumed_cumulative": [ "gas_consumed_cumulative": [
"Consumed Gas", "Consumed Gas",
VOLUME_CUBIC_METERS, VOLUME_CUBIC_METERS,
DEVICE_CLASS_GAS, SensorDeviceClass.GAS,
STATE_CLASS_TOTAL_INCREASING, SensorStateClass.TOTAL_INCREASING,
], ],
"net_electricity_point": [ "net_electricity_point": [
"Current net Power", "Current net Power",
POWER_WATT, POWER_WATT,
DEVICE_CLASS_POWER, SensorDeviceClass.POWER,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
], ],
"net_electricity_cumulative": [ "net_electricity_cumulative": [
"Cumulative net Power", "Cumulative net Power",
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
DEVICE_CLASS_ENERGY, SensorDeviceClass.ENERGY,
STATE_CLASS_TOTAL, SensorStateClass.TOTAL,
], ],
} }
@ -208,9 +200,14 @@ MISC_SENSOR_MAP = {
"Heater Modulation Level", "Heater Modulation Level",
PERCENTAGE, PERCENTAGE,
None, None,
STATE_CLASS_MEASUREMENT, SensorStateClass.MEASUREMENT,
],
"valve_position": [
"Valve Position",
PERCENTAGE,
None,
SensorStateClass.MEASUREMENT,
], ],
"valve_position": ["Valve Position", PERCENTAGE, None, STATE_CLASS_MEASUREMENT],
"water_pressure": ATTR_PRESSURE, "water_pressure": ATTR_PRESSURE,
} }