9704057959
* This change will enable the functionality for Arlo Q cameras. When we added the code to enable/disable motion detection, we assumed that base station will be present for all arlo type of cameras. But found recently that Arlo Q cameras does not have base station. So, removed the base_station dependency in the init code. Also added code in enable/disable motion detection code to first check if base station is detected by library. If base station is detected then it will use it to enable the motion detection. If not detected, even if service was called, it will not do anything. Enabling/disabling the motion detection for Arlo Q cameras have to added by someone who has that camera. I don't have the Arlo Q cameras. * Fixed a typo in the code.
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
"""
|
|
This component provides basic support for Netgear Arlo IP cameras.
|
|
|
|
For more details about this platform, please refer to the documentation at
|
|
https://home-assistant.io/components/camera.arlo/
|
|
"""
|
|
import asyncio
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
|
|
from homeassistant.components.arlo import DEFAULT_BRAND, DATA_ARLO
|
|
from homeassistant.components.camera import Camera, PLATFORM_SCHEMA
|
|
from homeassistant.components.ffmpeg import DATA_FFMPEG
|
|
|
|
DEPENDENCIES = ['arlo', 'ffmpeg']
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_FFMPEG_ARGUMENTS = 'ffmpeg_arguments'
|
|
ARLO_MODE_ARMED = 'armed'
|
|
ARLO_MODE_DISARMED = 'disarmed'
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Optional(CONF_FFMPEG_ARGUMENTS): cv.string,
|
|
})
|
|
|
|
|
|
@asyncio.coroutine
|
|
def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
|
|
"""Set up an Arlo IP Camera."""
|
|
arlo = hass.data.get(DATA_ARLO)
|
|
if not arlo:
|
|
return False
|
|
|
|
cameras = []
|
|
for camera in arlo.cameras:
|
|
cameras.append(ArloCam(hass, camera, config))
|
|
|
|
async_add_devices(cameras, True)
|
|
|
|
|
|
class ArloCam(Camera):
|
|
"""An implementation of a Netgear Arlo IP camera."""
|
|
|
|
def __init__(self, hass, camera, device_info):
|
|
"""Initialize an Arlo camera."""
|
|
super().__init__()
|
|
self._camera = camera
|
|
self._name = self._camera.name
|
|
self._motion_status = False
|
|
self._ffmpeg = hass.data[DATA_FFMPEG]
|
|
self._ffmpeg_arguments = device_info.get(CONF_FFMPEG_ARGUMENTS)
|
|
|
|
def camera_image(self):
|
|
"""Return a still image response from the camera."""
|
|
return self._camera.last_image
|
|
|
|
@asyncio.coroutine
|
|
def handle_async_mjpeg_stream(self, request):
|
|
"""Generate an HTTP MJPEG stream from the camera."""
|
|
from haffmpeg import CameraMjpeg
|
|
video = self._camera.last_video
|
|
if not video:
|
|
return
|
|
|
|
stream = CameraMjpeg(self._ffmpeg.binary, loop=self.hass.loop)
|
|
yield from stream.open_camera(
|
|
video.video_url, extra_cmd=self._ffmpeg_arguments)
|
|
|
|
yield from async_aiohttp_proxy_stream(
|
|
self.hass, request, stream,
|
|
'multipart/x-mixed-replace;boundary=ffserver')
|
|
yield from stream.close()
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of this camera."""
|
|
return self._name
|
|
|
|
@property
|
|
def model(self):
|
|
"""Camera model."""
|
|
return self._camera.model_id
|
|
|
|
@property
|
|
def brand(self):
|
|
"""Camera brand."""
|
|
return DEFAULT_BRAND
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Camera should poll periodically."""
|
|
return True
|
|
|
|
@property
|
|
def motion_detection_enabled(self):
|
|
"""Camera Motion Detection Status."""
|
|
return self._motion_status
|
|
|
|
def set_base_station_mode(self, mode):
|
|
"""Set the mode in the base station."""
|
|
# Get the list of base stations identified by library
|
|
base_stations = self.hass.data[DATA_ARLO].base_stations
|
|
|
|
# Some Arlo cameras does not have basestation
|
|
# So check if there is base station detected first
|
|
# if yes, then choose the primary base station
|
|
# Set the mode on the chosen base station
|
|
if base_stations:
|
|
primary_base_station = base_stations[0]
|
|
primary_base_station.mode = mode
|
|
|
|
def enable_motion_detection(self):
|
|
"""Enable the Motion detection in base station (Arm)."""
|
|
self._motion_status = True
|
|
self.set_base_station_mode(ARLO_MODE_ARMED)
|
|
|
|
def disable_motion_detection(self):
|
|
"""Disable the motion detection in base station (Disarm)."""
|
|
self._motion_status = False
|
|
self.set_base_station_mode(ARLO_MODE_DISARMED)
|