Update __init__.py

This commit is contained in:
thekiwismarthome
2026-02-12 12:02:22 +13:00
committed by GitHub
parent 78829aefb0
commit b6941a1499
@@ -6,6 +6,8 @@ import logging
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.components import websocket_api as ha_websocket from homeassistant.components import websocket_api as ha_websocket
from .websocket_api import websocket_create_list
from .const import DOMAIN from .const import DOMAIN
from .manager import ShoppingListManager from .manager import ShoppingListManager
@@ -34,6 +36,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
hass.data[DOMAIN]["manager"] = manager hass.data[DOMAIN]["manager"] = manager
# Register WebSocket commands using Home Assistant's websocket_api # Register WebSocket commands using Home Assistant's websocket_api
ha_websocket.async_register_command(hass, websocket_create_list)
ha_websocket.async_register_command(hass, websocket_add_product) ha_websocket.async_register_command(hass, websocket_add_product)
ha_websocket.async_register_command(hass, websocket_set_qty) ha_websocket.async_register_command(hass, websocket_set_qty)
ha_websocket.async_register_command(hass, websocket_get_products) ha_websocket.async_register_command(hass, websocket_get_products)
@@ -50,4 +53,4 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload Shopping List Manager.""" """Unload Shopping List Manager."""
hass.data[DOMAIN].pop("manager", None) hass.data[DOMAIN].pop("manager", None)
return True return True