fix unsaved

This commit is contained in:
J. Nick Koston
2023-09-07 08:39:51 -05:00
parent 7a13849221
commit ac67bde0c1
2 changed files with 6 additions and 4 deletions

View File

@@ -12,6 +12,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import ElkAttachedEntity, ElkEntity, create_elk_entities from . import ElkAttachedEntity, ElkEntity, create_elk_entities
from .const import DOMAIN from .const import DOMAIN
from .models import ELKM1Data
async def async_setup_entry( async def async_setup_entry(
@@ -20,9 +21,9 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Create the Elk-M1 scene platform.""" """Create the Elk-M1 scene platform."""
elk_data = hass.data[DOMAIN][config_entry.entry_id] elk_data: ELKM1Data = hass.data[DOMAIN][config_entry.entry_id]
elk = elk_data.elk
entities: list[ElkEntity] = [] entities: list[ElkEntity] = []
elk = elk_data["elk"]
create_elk_entities(elk_data, elk.tasks, "task", ElkTask, entities) create_elk_entities(elk_data, elk.tasks, "task", ElkTask, entities)
async_add_entities(entities) async_add_entities(entities)

View File

@@ -23,6 +23,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import ElkAttachedEntity, ElkEntity, create_elk_entities from . import ElkAttachedEntity, ElkEntity, create_elk_entities
from .const import ATTR_VALUE, DOMAIN, ELK_USER_CODE_SERVICE_SCHEMA from .const import ATTR_VALUE, DOMAIN, ELK_USER_CODE_SERVICE_SCHEMA
from .models import ELKM1Data
SERVICE_SENSOR_COUNTER_REFRESH = "sensor_counter_refresh" SERVICE_SENSOR_COUNTER_REFRESH = "sensor_counter_refresh"
SERVICE_SENSOR_COUNTER_SET = "sensor_counter_set" SERVICE_SENSOR_COUNTER_SET = "sensor_counter_set"
@@ -41,9 +42,9 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Create the Elk-M1 sensor platform.""" """Create the Elk-M1 sensor platform."""
elk_data = hass.data[DOMAIN][config_entry.entry_id] elk_data: ELKM1Data = hass.data[DOMAIN][config_entry.entry_id]
elk = elk_data.elk
entities: list[ElkEntity] = [] entities: list[ElkEntity] = []
elk = elk_data["elk"]
create_elk_entities(elk_data, elk.counters, "counter", ElkCounter, entities) create_elk_entities(elk_data, elk.counters, "counter", ElkCounter, entities)
create_elk_entities(elk_data, elk.keypads, "keypad", ElkKeypad, entities) create_elk_entities(elk_data, elk.keypads, "keypad", ElkKeypad, entities)
create_elk_entities(elk_data, [elk.panel], "panel", ElkPanel, entities) create_elk_entities(elk_data, [elk.panel], "panel", ElkPanel, entities)