Merge pull request #185 from twrecked/take-snapshot-service

add request snapshot service

changelog

@@ -1,4 +1,5 @@
 aarlo/pyaarlo
+0.6.17: Add request_snapshot webservice
 0.6.16: Fix last_image so lovelace card works
         Arlo Baby: fix effect switching
 0.6.15: Copied services into aarlo domain

custom_components.json

@@ -1,6 +1,6 @@
 {
     "aarlo": {
-        "version": "0.6.16",
+        "version": "0.6.17",
         "local_location": "/custom_components/aarlo/__init__.py",
         "remote_location": "https://raw.githubusercontent.com/twrecked/hass-aarlo/master/custom_components/aarlo/__init__.py",
         "visit_repo": "https://github.com/twrecked/hass-aarlo",
@@ -18,7 +18,7 @@
         ]
     },
     "pyaarlo": {
-        "version": "0.6.16",
+        "version": "0.6.17",
         "local_location": "/custom_components/aarlo/pyaarlo/__init__.py",
         "remote_location": "https://raw.githubusercontent.com/twrecked/hass-aarlo/master/custom_components/aarlo/pyaarlo/__init__.py",
         "visit_repo": "https://github.com/twrecked/hass-aarlo",

custom_components/aarlo/__init__.py

@@ -20,7 +20,7 @@
 from homeassistant.components.camera import DOMAIN as CAMERA_DOMAIN
 from homeassistant.components.alarm_control_panel import DOMAIN as ALARM_DOMAIN
 
-__version__ = '0.6.16'
+__version__ = '0.6.17'
 
 _LOGGER = logging.getLogger(__name__)
 

custom_components/aarlo/camera.py

@@ -104,6 +104,7 @@
 WS_TYPE_LIBRARY = 'aarlo_library'
 WS_TYPE_STREAM_URL = 'aarlo_stream_url'
 WS_TYPE_SNAPSHOT_IMAGE = 'aarlo_snapshot_image'
+WS_TYPE_REQUEST_SNAPSHOT = 'aarlo_request_snapshot'
 WS_TYPE_VIDEO_DATA = 'aarlo_video_data'
 WS_TYPE_STOP_ACTIVITY = 'aarlo_stop_activity'
 WS_TYPE_SIREN_ON = 'aarlo_camera_siren_on'
@@ -126,6 +127,10 @@
     vol.Required('type'): WS_TYPE_SNAPSHOT_IMAGE,
     vol.Required('entity_id'): cv.entity_id
 })
+SCHEMA_WS_REQUEST_SNAPSHOT = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
+    vol.Required('type'): WS_TYPE_REQUEST_SNAPSHOT,
+    vol.Required('entity_id'): cv.entity_id
+})
 SCHEMA_WS_VIDEO_DATA = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
     vol.Required('type'): WS_TYPE_VIDEO_DATA,
     vol.Required('entity_id'): cv.entity_id
@@ -253,6 +258,10 @@ async def async_camera_service(call):
         WS_TYPE_SNAPSHOT_IMAGE, websocket_snapshot_image,
         SCHEMA_WS_SNAPSHOT_IMAGE
     )
+    hass.components.websocket_api.async_register_command(
+        WS_TYPE_REQUEST_SNAPSHOT, websocket_request_snapshot,
+        SCHEMA_WS_REQUEST_SNAPSHOT
+    )
     hass.components.websocket_api.async_register_command(
         WS_TYPE_VIDEO_DATA, websocket_video_data,
         SCHEMA_WS_VIDEO_DATA
@@ -615,6 +624,23 @@ async def websocket_snapshot_image(hass, connection, msg):
         connection.send_message(websocket_api.error_message(
             msg['id'], 'image_fetch_failed', 'Unable to fetch image'))
 
+@websocket_api.async_response
+async def websocket_request_snapshot(hass, connection, msg):
+    camera = get_entity_from_domain(hass, DOMAIN, msg['entity_id'])
+    _LOGGER.debug('request_snapshot_image for ' + str(camera.unique_id))
+
+    try:
+        await camera.async_request_snapshot()
+        connection.send_message(websocket_api.result_message(
+            msg['id'], {
+                'snapshot requested'
+            }
+        ))
+
+    except HomeAssistantError:
+        connection.send_message(websocket_api.error_message(
+            msg['id'], 'image_fetch_failed', 'Unable to fetch image'))
+
 
 @websocket_api.async_response
 async def websocket_video_data(hass, connection, msg):

custom_components/aarlo/pyaarlo/__init__.py

@@ -23,7 +23,7 @@
 
 _LOGGER = logging.getLogger('pyaarlo')
 
-__version__ = '0.6.16'
+__version__ = '0.6.17'
 
 
 class PyArlo(object):