From a55fbd2be79fdf08c690252a94512708ae8ee857 Mon Sep 17 00:00:00 2001
From: Max Muth <kontakt@maxi-muth.de>
Date: Sat, 26 May 2018 13:53:48 +0200
Subject: [PATCH] Add services for adding and removing items to shopping list
 (#14574)

---
 homeassistant/components/services.yaml    | 14 ++++++++
 homeassistant/components/shopping_list.py | 41 ++++++++++++++++++++++-
 2 files changed, 54 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/services.yaml b/homeassistant/components/services.yaml
index 746c3c7f483..c0279ef1d0f 100644
--- a/homeassistant/components/services.yaml
+++ b/homeassistant/components/services.yaml
@@ -556,3 +556,17 @@ xiaomi_aqara:
       device_id:
         description: Hardware address of the device to remove.
         example: 158d0000000000
+
+shopping_list:
+  add_item:
+    description: Adds an item to the shopping list.
+    fields:
+      name:
+        description: The name of the item to add.
+        example: Beer
+  complete_item:
+    description: Marks an item as completed in the shopping list. It does not remove the item.
+    fields:
+      name:
+        description: The name of the item to mark as completed.
+        example: Beer
diff --git a/homeassistant/components/shopping_list.py b/homeassistant/components/shopping_list.py
index 0ca0fef6e06..f113561429a 100644
--- a/homeassistant/components/shopping_list.py
+++ b/homeassistant/components/shopping_list.py
@@ -14,6 +14,8 @@ from homeassistant.helpers import intent
 import homeassistant.helpers.config_validation as cv
 from homeassistant.util.json import load_json, save_json
 
+ATTR_NAME = 'name'
+
 DOMAIN = 'shopping_list'
 DEPENDENCIES = ['http']
 _LOGGER = logging.getLogger(__name__)
@@ -23,20 +25,57 @@ INTENT_ADD_ITEM = 'HassShoppingListAddItem'
 INTENT_LAST_ITEMS = 'HassShoppingListLastItems'
 ITEM_UPDATE_SCHEMA = vol.Schema({
     'complete': bool,
-    'name': str,
+    ATTR_NAME: str,
 })
 PERSISTENCE = '.shopping_list.json'
 
+SERVICE_ADD_ITEM = 'add_item'
+SERVICE_COMPLETE_ITEM = 'complete_item'
+
+SERVICE_ITEM_SCHEMA = vol.Schema({
+    vol.Required(ATTR_NAME): vol.Any(None, cv.string)
+})
+
 
 @asyncio.coroutine
 def async_setup(hass, config):
     """Initialize the shopping list."""
+    @asyncio.coroutine
+    def add_item_service(call):
+        """Add an item with `name`."""
+        data = hass.data[DOMAIN]
+        name = call.data.get(ATTR_NAME)
+        if name is not None:
+            data.async_add(name)
+
+    @asyncio.coroutine
+    def complete_item_service(call):
+        """Mark the item provided via `name` as completed."""
+        data = hass.data[DOMAIN]
+        name = call.data.get(ATTR_NAME)
+        if name is None:
+            return
+        try:
+            item = [item for item in data.items if item['name'] == name][0]
+        except IndexError:
+            _LOGGER.error("Removing of item failed: %s cannot be found", name)
+        else:
+            data.async_update(item['id'], {'name': name, 'complete': True})
+
     data = hass.data[DOMAIN] = ShoppingData(hass)
     yield from data.async_load()
 
     intent.async_register(hass, AddItemIntent())
     intent.async_register(hass, ListTopItemsIntent())
 
+    hass.services.async_register(
+        DOMAIN, SERVICE_ADD_ITEM, add_item_service, schema=SERVICE_ITEM_SCHEMA
+    )
+    hass.services.async_register(
+        DOMAIN, SERVICE_COMPLETE_ITEM, complete_item_service,
+        schema=SERVICE_ITEM_SCHEMA
+    )
+
     hass.http.register_view(ShoppingListView)
     hass.http.register_view(CreateShoppingListItemView)
     hass.http.register_view(UpdateShoppingListItemView)