todo.py

Go to the documentation of this file.

"""Definition of Picnic shopping cart."""
 
from __future__ import annotations
 
import logging
from typing import cast
 
from homeassistant.components.todo import (
    TodoItem,
    TodoItemStatus,
    TodoListEntity,
    TodoListEntityFeature,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import CONF_COORDINATOR, DOMAIN
from .coordinator import PicnicUpdateCoordinator
from .services import product_search
 
_LOGGER = logging.getLogger(__name__)
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the Picnic shopping cart todo platform config entry."""
    picnic_coordinator = hass.data[DOMAIN][config_entry.entry_id][CONF_COORDINATOR]
 
    async_add_entities([PicnicCart(picnic_coordinator, config_entry)])
 
 
class PicnicCart(TodoListEntity, CoordinatorEntity[PicnicUpdateCoordinator]):
    """A Picnic Shopping Cart TodoListEntity."""
 
    _attr_has_entity_name = True
    _attr_supported_features = TodoListEntityFeature.CREATE_TODO_ITEM
    _attr_translation_key = "shopping_cart"
 
    def __init__(
        self,
        coordinator: PicnicUpdateCoordinator,
        config_entry: ConfigEntry,
    ) -> None:
        """Initialize PicnicCart."""
        super().__init__(coordinator)
        self._attr_device_info_attr_device_info = DeviceInfo(
            entry_type=DeviceEntryType.SERVICE,
            identifiers={(DOMAIN, cast(str, config_entry.unique_id))},
            manufacturer="Picnic",
            model=config_entry.unique_id,
        )
        self._attr_unique_id_attr_unique_id = f"{config_entry.unique_id}-cart"
 
    @property
    def todo_items(self) -> list[TodoItem] | None:
        """Get the current set of items in cart items."""
        if self.coordinator.data is None:
            return None
 
        _LOGGER.debug(self.coordinator.data["cart_data"]["items"])
 
        return [
            TodoItem(
                summary=f"{article['name']} ({article['unit_quantity']})",
                uid=f"{item['id']}-{article['id']}",
                status=TodoItemStatus.NEEDS_ACTION,  # We set 'NEEDS_ACTION' so they count as state
            )
            for item in self.coordinator.data["cart_data"]["items"]
            for article in item["items"]
        ]
 
    async def async_create_todo_item(self, item: TodoItem) -> None:
        """Add item to shopping cart."""
        product_id = await self.hasshasshass.async_add_executor_job(
            product_search, self.coordinator.picnic_api_client, item.summary
        )
 
        if not product_id:
            raise ServiceValidationError("No product found or no product ID given")
 
        await self.hasshasshass.async_add_executor_job(
            self.coordinator.picnic_api_client.add_product, product_id, 1
        )
 
        await self.coordinator.async_refresh()