mirror of
https://github.com/esphome/esphome.git
synced 2024-11-30 10:44:13 +01:00
131 lines
4.4 KiB
Python
131 lines
4.4 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import sensor, esp32_ble_tracker
|
|
from esphome.const import (
|
|
CONF_DISTANCE,
|
|
CONF_MAC_ADDRESS,
|
|
CONF_ID,
|
|
ICON_THERMOMETER,
|
|
ICON_RULER,
|
|
UNIT_PERCENT,
|
|
CONF_LEVEL,
|
|
CONF_TEMPERATURE,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
UNIT_CELSIUS,
|
|
STATE_CLASS_MEASUREMENT,
|
|
CONF_BATTERY_LEVEL,
|
|
DEVICE_CLASS_BATTERY,
|
|
)
|
|
|
|
CONF_TANK_TYPE = "tank_type"
|
|
CONF_CUSTOM_DISTANCE_FULL = "custom_distance_full"
|
|
CONF_CUSTOM_DISTANCE_EMPTY = "custom_distance_empty"
|
|
|
|
ICON_PROPANE_TANK = "mdi:propane-tank"
|
|
|
|
TANK_TYPE_CUSTOM = "CUSTOM"
|
|
|
|
UNIT_MILLIMETER = "mm"
|
|
|
|
|
|
def small_distance(value):
|
|
"""small_distance is stored in mm"""
|
|
meters = cv.distance(value)
|
|
return meters * 1000
|
|
|
|
|
|
#
|
|
# Map of standard tank types to their
|
|
# empty and full distance values.
|
|
# Format is - tank name: (empty distance in mm, full distance in mm)
|
|
#
|
|
CONF_SUPPORTED_TANKS_MAP = {
|
|
TANK_TYPE_CUSTOM: (0, 100),
|
|
"20LB_V": (38, 254), # empty/full readings for 20lb US tank
|
|
"30LB_V": (38, 381),
|
|
"40LB_V": (38, 508),
|
|
}
|
|
|
|
CODEOWNERS = ["@spbrogan"]
|
|
DEPENDENCIES = ["esp32_ble_tracker"]
|
|
|
|
mopeka_pro_check_ns = cg.esphome_ns.namespace("mopeka_pro_check")
|
|
MopekaProCheck = mopeka_pro_check_ns.class_(
|
|
"MopekaProCheck", esp32_ble_tracker.ESPBTDeviceListener, cg.Component
|
|
)
|
|
|
|
CONFIG_SCHEMA = (
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(MopekaProCheck),
|
|
cv.Required(CONF_MAC_ADDRESS): cv.mac_address,
|
|
cv.Optional(CONF_CUSTOM_DISTANCE_FULL): small_distance,
|
|
cv.Optional(CONF_CUSTOM_DISTANCE_EMPTY): small_distance,
|
|
cv.Required(CONF_TANK_TYPE): cv.enum(CONF_SUPPORTED_TANKS_MAP, upper=True),
|
|
cv.Optional(CONF_TEMPERATURE): sensor.sensor_schema(
|
|
unit_of_measurement=UNIT_CELSIUS,
|
|
icon=ICON_THERMOMETER,
|
|
accuracy_decimals=0,
|
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
|
state_class=STATE_CLASS_MEASUREMENT,
|
|
),
|
|
cv.Optional(CONF_LEVEL): sensor.sensor_schema(
|
|
unit_of_measurement=UNIT_PERCENT,
|
|
icon=ICON_PROPANE_TANK,
|
|
accuracy_decimals=0,
|
|
state_class=STATE_CLASS_MEASUREMENT,
|
|
),
|
|
cv.Optional(CONF_DISTANCE): sensor.sensor_schema(
|
|
unit_of_measurement=UNIT_MILLIMETER,
|
|
icon=ICON_RULER,
|
|
accuracy_decimals=0,
|
|
state_class=STATE_CLASS_MEASUREMENT,
|
|
),
|
|
cv.Optional(CONF_BATTERY_LEVEL): sensor.sensor_schema(
|
|
unit_of_measurement=UNIT_PERCENT,
|
|
accuracy_decimals=0,
|
|
device_class=DEVICE_CLASS_BATTERY,
|
|
state_class=STATE_CLASS_MEASUREMENT,
|
|
),
|
|
}
|
|
)
|
|
.extend(esp32_ble_tracker.ESP_BLE_DEVICE_SCHEMA)
|
|
.extend(cv.COMPONENT_SCHEMA)
|
|
)
|
|
|
|
|
|
async def to_code(config):
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
await esp32_ble_tracker.register_ble_device(var, config)
|
|
|
|
cg.add(var.set_address(config[CONF_MAC_ADDRESS].as_hex))
|
|
|
|
if config[CONF_TANK_TYPE] == TANK_TYPE_CUSTOM:
|
|
# Support custom tank min/max
|
|
if CONF_CUSTOM_DISTANCE_EMPTY in config:
|
|
cg.add(var.set_tank_empty(config[CONF_CUSTOM_DISTANCE_EMPTY]))
|
|
else:
|
|
cg.add(var.set_tank_empty(CONF_SUPPORTED_TANKS_MAP[TANK_TYPE_CUSTOM][0]))
|
|
if CONF_CUSTOM_DISTANCE_FULL in config:
|
|
cg.add(var.set_tank_full(config[CONF_CUSTOM_DISTANCE_FULL]))
|
|
else:
|
|
cg.add(var.set_tank_full(CONF_SUPPORTED_TANKS_MAP[TANK_TYPE_CUSTOM][1]))
|
|
else:
|
|
# Set the Tank empty and full based on map - User is requesting standard tank
|
|
t = config[CONF_TANK_TYPE]
|
|
cg.add(var.set_tank_empty(CONF_SUPPORTED_TANKS_MAP[t][0]))
|
|
cg.add(var.set_tank_full(CONF_SUPPORTED_TANKS_MAP[t][1]))
|
|
|
|
if CONF_TEMPERATURE in config:
|
|
sens = await sensor.new_sensor(config[CONF_TEMPERATURE])
|
|
cg.add(var.set_temperature(sens))
|
|
if CONF_LEVEL in config:
|
|
sens = await sensor.new_sensor(config[CONF_LEVEL])
|
|
cg.add(var.set_level(sens))
|
|
if CONF_DISTANCE in config:
|
|
sens = await sensor.new_sensor(config[CONF_DISTANCE])
|
|
cg.add(var.set_distance(sens))
|
|
if CONF_BATTERY_LEVEL in config:
|
|
sens = await sensor.new_sensor(config[CONF_BATTERY_LEVEL])
|
|
cg.add(var.set_battery_level(sens))
|