[new] component rest_command (#5055)

* New component rest_command

* add unittests

* change handling like other command

* change unittest

* address @balloob comments

homeassistant/components/rest_command.py

@@ -0,0 +1,115 @@
+"""
+Exposes regular rest commands as services.
+
+For more details about this platform, please refer to the documentation at
+https://home-assistant.io/components/rest_command/
+"""
+import asyncio
+import logging
+
+import aiohttp
+import async_timeout
+import voluptuous as vol
+
+from homeassistant.const import (
+    CONF_TIMEOUT, CONF_USERNAME, CONF_PASSWORD, CONF_URL, CONF_PAYLOAD,
+    CONF_METHOD)
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+import homeassistant.helpers.config_validation as cv
+
+DOMAIN = 'rest_command'
+
+_LOGGER = logging.getLogger(__name__)
+
+DEFAULT_TIMEOUT = 10
+DEFAULT_METHOD = 'get'
+
+SUPPORT_REST_METHODS = [
+    'get',
+    'post',
+    'put',
+    'delete',
+]
+
+COMMAND_SCHEMA = vol.Schema({
+    vol.Required(CONF_URL): cv.template,
+    vol.Optional(CONF_METHOD, default=DEFAULT_METHOD):
+        vol.All(vol.Lower, vol.In(SUPPORT_REST_METHODS)),
+    vol.Inclusive(CONF_USERNAME, 'authentication'): cv.string,
+    vol.Inclusive(CONF_PASSWORD, 'authentication'): cv.string,
+    vol.Optional(CONF_PAYLOAD): cv.template,
+    vol.Optional(CONF_TIMEOUT, default=DEFAULT_TIMEOUT): vol.Coerce(int),
+})
+
+CONFIG_SCHEMA = vol.Schema({
+    DOMAIN: vol.Schema({
+        cv.slug: COMMAND_SCHEMA,
+    }),
+}, extra=vol.ALLOW_EXTRA)
+
+
+@asyncio.coroutine
+def async_setup(hass, config):
+    """Setup the rest_command component."""
+    websession = async_get_clientsession(hass)
+
+    def async_register_rest_command(name, command_config):
+        """Create service for rest command."""
+        timeout = command_config[CONF_TIMEOUT]
+        method = command_config[CONF_METHOD]
+
+        template_url = command_config[CONF_URL]
+        template_url.hass = hass
+
+        auth = None
+        if CONF_USERNAME in command_config:
+            username = command_config[CONF_USERNAME]
+            password = command_config.get(CONF_PASSWORD, '')
+            auth = aiohttp.BasicAuth(username, password=password)
+
+        template_payload = None
+        if CONF_PAYLOAD in command_config:
+            template_payload = command_config[CONF_PAYLOAD]
+            template_payload.hass = hass
+
+        @asyncio.coroutine
+        def async_service_handler(service):
+            """Execute a shell command service."""
+            payload = None
+            if template_payload:
+                payload = bytes(
+                    template_payload.async_render(variables=service.data),
+                    'utf-8')
+
+            request = None
+            try:
+                with async_timeout.timeout(timeout, loop=hass.loop):
+                    request = yield from getattr(websession, method)(
+                        template_url.async_render(variables=service.data),
+                        data=payload,
+                        auth=auth
+                    )
+
+                    if request.status == 200:
+                        _LOGGER.info("Success call %s.", request.url)
+                        return
+
+                    _LOGGER.warning(
+                        "Error %d on call %s.", request.status, request.url)
+            except asyncio.TimeoutError:
+                _LOGGER.warning("Timeout call %s.", request.url)
+
+            except aiohttp.errors.ClientError:
+                _LOGGER.error("Client error %s.", request.url)
+
+            finally:
+                if request is not None:
+                    yield from request.release()
+
+        # register services
+        hass.services.async_register(DOMAIN, name, async_service_handler)
+
+    for command, command_config in config[DOMAIN].items():
+        async_register_rest_command(command, command_config)
+
+    return True