add start stop and restart services

docker_monitor/__init__.py

@@ -56,6 +56,7 @@
 
 UTILISATION_MONITOR_VERSION = 'utilization_version'
 
+CONTAINER_NAME = 'container_name'
 CONTAINER_MONITOR_STATUS = 'container_status'
 CONTAINER_MONITOR_UPTIME = 'container_uptime'
 CONTAINER_MONITOR_IMAGE = 'container_image'
@@ -105,6 +106,12 @@
     })
 }, extra=vol.ALLOW_EXTRA)
 
+SERVICE_START = 'start'
+SERVICE_STOP = 'stop'
+SERVICE_RESTART = 'restart'
+SERVICE_SCHEMA = vol.Schema({
+    vol.Required(CONTAINER_NAME):  cv.string
+})
 
 def setup(hass, config):
     _LOGGER.info("Settings: {}".format(config[DOMAIN]))
@@ -148,6 +155,29 @@ def event_listener(message):
         if config[DOMAIN][CONF_EVENTS]:
             api.events(event_listener)
 
+
+        def handle_start(call):
+            container_name = call.data[CONTAINER_NAME]
+            container = api.get_container(container_name)
+            if container:
+                container.start()
+
+        def handle_stop(call):
+            container_name = call.data[CONTAINER_NAME]
+            container = api.get_container(container_name)
+            if container:
+                container.stop()
+
+        def handle_restart(call):
+            container_name = call.data[CONTAINER_NAME]
+            container = api.get_container(container_name)
+            if container:
+                container.restart()
+
+        hass.services.register(DOMAIN, SERVICE_START, handle_start, SERVICE_SCHEMA)
+        hass.services.register(DOMAIN, SERVICE_STOP, handle_stop, SERVICE_SCHEMA)
+        hass.services.register(DOMAIN, SERVICE_RESTART, handle_restart, SERVICE_SCHEMA)
+
         return True
 
 
@@ -297,6 +327,10 @@ def stop(self, timeout=10):
         _LOGGER.info("Stop container {}".format(self._name))
         self._container.stop(timeout=timeout)
 
+    def restart(self):
+        _LOGGER.info("Restart container {}".format(self._name))
+        self._container.restart()
+
     def _notify(self, message):
         _LOGGER.debug("Send notify for container {}".format(self._name))
         for callback in self._subscribers:

docker_monitor/services.yaml

@@ -0,0 +1,19 @@
+start:
+  description: start a docker container
+  fields:
+    container_name:
+      description: name of the container to start.
+      example: docker_nginx_1
+stop:
+  description: stop a docker container
+  fields:
+    container_name:
+      description: name of the container to stop.
+      example: docker_nginx_1
+restart:
+  description: restart a docker container
+  fields:
+    container_name:
+      description: name of the container to restart.
+      example: docker_nginx_1
+

docker_monitor/switch.py

@@ -9,7 +9,7 @@
 from homeassistant.components.switch import (
     ENTITY_ID_FORMAT,
     PLATFORM_SCHEMA,
-    SwitchDevice
+    SwitchEntity
 )
 from homeassistant.const import (
     ATTR_ATTRIBUTION,
@@ -49,7 +49,7 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
         return False
 
 
-class ContainerSwitch(SwitchDevice):
+class ContainerSwitch(SwitchEntity):
     def __init__(self, api, clientname, container_name):
         self._api = api
         self._clientname = clientname