Add missing schema to dominos order service (#63016)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2022-01-02 20:49:41 +01:00 committed by GitHub
parent d364186571
commit f14e162610
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -7,7 +7,7 @@ from pizzapi.address import StoreException
import voluptuous as vol import voluptuous as vol
from homeassistant.components import http from homeassistant.components import http
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, ServiceCall, callback
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_component import EntityComponent from homeassistant.helpers.entity_component import EntityComponent
@ -72,7 +72,16 @@ def setup(hass: HomeAssistant, config: ConfigType) -> bool:
entities = [] entities = []
conf = config[DOMAIN] conf = config[DOMAIN]
hass.services.register(DOMAIN, "order", dominos.handle_order) hass.services.register(
DOMAIN,
"order",
dominos.handle_order,
vol.Schema(
{
vol.Required(ATTR_ORDER_ENTITY): cv.entity_ids,
}
),
)
if conf.get(ATTR_SHOW_MENU): if conf.get(ATTR_SHOW_MENU):
hass.http.register_view(DominosProductListView(dominos)) hass.http.register_view(DominosProductListView(dominos))
@ -112,9 +121,9 @@ class Dominos:
except StoreException: except StoreException:
self.closest_store = None self.closest_store = None
def handle_order(self, call): def handle_order(self, call: ServiceCall) -> None:
"""Handle ordering pizza.""" """Handle ordering pizza."""
entity_ids = call.data.get(ATTR_ORDER_ENTITY) entity_ids = call.data[ATTR_ORDER_ENTITY]
target_orders = [ target_orders = [
order order