VictronMPPT-ESPHOME/components/victron/sensor.py
2022-01-24 18:03:53 +01:00

135 lines
4.3 KiB
Python

import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import sensor
from esphome.const import (
CONF_BATTERY_VOLTAGE,
DEVICE_CLASS_CURRENT,
DEVICE_CLASS_EMPTY,
DEVICE_CLASS_POWER,
DEVICE_CLASS_VOLTAGE,
ICON_CURRENT_AC,
ICON_EMPTY,
ICON_FLASH,
ICON_POWER,
UNIT_AMPERE,
UNIT_EMPTY,
UNIT_VOLT,
UNIT_WATT,
UNIT_WATT_HOURS,
)
from . import CONF_VICTRON_ID, VictronComponent
DEPENDENCIES = ["victron"]
CODEOWNERS = ["@KinDR007"]
CONF_MAX_POWER_YESTERDAY = "max_power_yesterday"
CONF_MAX_POWER_TODAY = "max_power_today"
CONF_YIELD_TOTAL = "yield_total"
CONF_YIELD_YESTERDAY = "yield_yesterday"
CONF_YIELD_TODAY = "yield_today"
CONF_PANEL_VOLTAGE = "panel_voltage"
CONF_PANEL_POWER = "panel_power"
CONF_BATTERY_CURRENT = "battery_current"
CONF_DAY_NUMBER = "day_number"
CONF_CHARGING_MODE_ID = "charging_mode_id"
CONF_ERROR_CODE = "error_code"
CONF_WARNING_CODE = "warning_code"
CONF_TRACKING_MODE_ID = "tracking_mode_id"
CONF_DEVICE_MODE_ID = "device_mode_id"
CONF_LOAD_CURRENT = "load_current"
CONF_AC_OUT_VOLTAGE = "ac_out_voltage"
CONF_AC_OUT_CURRENT = "ac_out_current"
SENSORS = [
CONF_BATTERY_VOLTAGE,
CONF_AC_OUT_VOLTAGE,
CONF_MAX_POWER_YESTERDAY,
CONF_MAX_POWER_TODAY,
CONF_YIELD_TOTAL,
CONF_YIELD_YESTERDAY,
CONF_YIELD_TODAY,
CONF_PANEL_VOLTAGE,
CONF_PANEL_POWER,
CONF_BATTERY_CURRENT,
CONF_AC_OUT_CURRENT,
CONF_DAY_NUMBER,
CONF_CHARGING_MODE_ID,
CONF_ERROR_CODE,
CONF_WARNING_CODE,
CONF_TRACKING_MODE_ID,
CONF_DEVICE_MODE_ID,
CONF_LOAD_CURRENT,
]
CONFIG_SCHEMA = cv.Schema(
{
cv.GenerateID(CONF_VICTRON_ID): cv.use_id(VictronComponent),
cv.Optional(CONF_MAX_POWER_YESTERDAY): sensor.sensor_schema(
UNIT_WATT, ICON_POWER, 0, DEVICE_CLASS_POWER
),
cv.Optional(CONF_MAX_POWER_TODAY): sensor.sensor_schema(
UNIT_WATT, ICON_POWER, 0, DEVICE_CLASS_POWER
),
cv.Optional(CONF_YIELD_TOTAL): sensor.sensor_schema(
UNIT_WATT_HOURS, ICON_POWER, 0, DEVICE_CLASS_POWER
),
cv.Optional(CONF_YIELD_YESTERDAY): sensor.sensor_schema(
UNIT_WATT_HOURS, ICON_POWER, 0, DEVICE_CLASS_POWER
),
cv.Optional(CONF_YIELD_TODAY): sensor.sensor_schema(
UNIT_WATT_HOURS, ICON_POWER, 0, DEVICE_CLASS_POWER
),
cv.Optional(CONF_PANEL_VOLTAGE): sensor.sensor_schema(
UNIT_VOLT, ICON_FLASH, 3, DEVICE_CLASS_VOLTAGE
),
cv.Optional(CONF_PANEL_POWER): sensor.sensor_schema(
UNIT_WATT, ICON_POWER, 0, DEVICE_CLASS_POWER
),
cv.Optional(CONF_BATTERY_VOLTAGE): sensor.sensor_schema(
UNIT_VOLT, ICON_FLASH, 3, DEVICE_CLASS_VOLTAGE
),
cv.Optional(CONF_AC_OUT_VOLTAGE): sensor.sensor_schema(
UNIT_VOLT, ICON_FLASH, 3, DEVICE_CLASS_VOLTAGE
),
cv.Optional(CONF_BATTERY_CURRENT): sensor.sensor_schema(
UNIT_AMPERE, ICON_CURRENT_AC, 3, DEVICE_CLASS_CURRENT
),
cv.Optional(CONF_AC_OUT_CURRENT): sensor.sensor_schema(
UNIT_AMPERE, ICON_CURRENT_AC, 3, DEVICE_CLASS_CURRENT
),
cv.Optional(CONF_DAY_NUMBER): sensor.sensor_schema(
UNIT_EMPTY, ICON_EMPTY, 0, DEVICE_CLASS_EMPTY
),
cv.Optional(CONF_CHARGING_MODE_ID): sensor.sensor_schema(
UNIT_EMPTY, ICON_EMPTY, 0, DEVICE_CLASS_EMPTY
),
cv.Optional(CONF_ERROR_CODE): sensor.sensor_schema(
UNIT_EMPTY, ICON_EMPTY, 0, DEVICE_CLASS_EMPTY
),
cv.Optional(CONF_WARNING_CODE): sensor.sensor_schema(
UNIT_EMPTY, ICON_EMPTY, 0, DEVICE_CLASS_EMPTY
),
cv.Optional(CONF_TRACKING_MODE_ID): sensor.sensor_schema(
UNIT_EMPTY, ICON_EMPTY, 0, DEVICE_CLASS_EMPTY
),
cv.Optional(CONF_DEVICE_MODE_ID): sensor.sensor_schema(
UNIT_EMPTY, ICON_EMPTY, 0, DEVICE_CLASS_EMPTY
),
cv.Optional(CONF_LOAD_CURRENT): sensor.sensor_schema(
UNIT_AMPERE, ICON_CURRENT_AC, 3, DEVICE_CLASS_CURRENT
),
}
)
def to_code(config):
hub = yield cg.get_variable(config[CONF_VICTRON_ID])
for key in SENSORS:
if key in config:
conf = config[key]
sens = yield sensor.new_sensor(conf)
cg.add(getattr(hub, f"set_{key}_sensor")(sens))