Add tools to octoprint when the printer comes back online (#59666)

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
Ryan Fleming 2022-02-23 05:59:28 -05:00 committed by GitHub
parent 4fecd5d8af
commit 9db2d8768b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 37 additions and 15 deletions

View File

@ -1,4 +1,6 @@
"""Support for monitoring OctoPrint 3D printers."""
from __future__ import annotations
from datetime import timedelta
import logging
from typing import cast
@ -175,7 +177,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await coordinator.async_config_entry_first_refresh()
hass.data[DOMAIN][entry.entry_id] = {"coordinator": coordinator, "client": client}
hass.data[DOMAIN][entry.entry_id] = {
"coordinator": coordinator,
"client": client,
}
hass.config_entries.async_setup_platforms(entry, PLATFORMS)

View File

@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, TEMP_CELSIUS
from homeassistant.core import HomeAssistant
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@ -47,13 +47,23 @@ async def async_setup_entry(
assert device_id is not None
entities: list[SensorEntity] = []
if coordinator.data["printer"]:
printer_info = coordinator.data["printer"]
types = ["actual", "target"]
for tool in printer_info.temperatures:
for temp_type in types:
entities.append(
known_tools = set()
@callback
def async_add_tool_sensors() -> None:
if not coordinator.data["printer"]:
return
new_tools = []
for tool in [
tool
for tool in coordinator.data["printer"].temperatures
if tool.name not in known_tools
]:
assert device_id is not None
known_tools.add(tool.name)
for temp_type in ("actual", "target"):
new_tools.append(
OctoPrintTemperatureSensor(
coordinator,
tool.name,
@ -61,13 +71,20 @@ async def async_setup_entry(
device_id,
)
)
else:
_LOGGER.debug("Printer appears to be offline, skipping temperature sensors")
if new_tools:
async_add_entities(new_tools)
entities.append(OctoPrintStatusSensor(coordinator, device_id))
entities.append(OctoPrintJobPercentageSensor(coordinator, device_id))
entities.append(OctoPrintEstimatedFinishTimeSensor(coordinator, device_id))
entities.append(OctoPrintStartTimeSensor(coordinator, device_id))
config_entry.async_on_unload(coordinator.async_add_listener(async_add_tool_sensors))
if coordinator.data["printer"]:
async_add_tool_sensors()
entities: list[SensorEntity] = [
OctoPrintStatusSensor(coordinator, device_id),
OctoPrintJobPercentageSensor(coordinator, device_id),
OctoPrintEstimatedFinishTimeSensor(coordinator, device_id),
OctoPrintStartTimeSensor(coordinator, device_id),
]
async_add_entities(entities)