142 lines
4.9 KiB
Python
142 lines
4.9 KiB
Python
"""
|
|
Camera platform that has a Raspberry Pi camera.
|
|
|
|
For more details about this platform, please refer to the documentation at
|
|
https://home-assistant.io/components/camera.rpi_camera/
|
|
"""
|
|
import os
|
|
import subprocess
|
|
import logging
|
|
import shutil
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.camera import (Camera, PLATFORM_SCHEMA)
|
|
from homeassistant.const import (CONF_NAME, CONF_FILE_PATH,
|
|
EVENT_HOMEASSISTANT_STOP)
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_HORIZONTAL_FLIP = 'horizontal_flip'
|
|
CONF_IMAGE_HEIGHT = 'image_height'
|
|
CONF_IMAGE_QUALITY = 'image_quality'
|
|
CONF_IMAGE_ROTATION = 'image_rotation'
|
|
CONF_IMAGE_WIDTH = 'image_width'
|
|
CONF_TIMELAPSE = 'timelapse'
|
|
CONF_VERTICAL_FLIP = 'vertical_flip'
|
|
|
|
DEFAULT_HORIZONTAL_FLIP = 0
|
|
DEFAULT_IMAGE_HEIGHT = 480
|
|
DEFAULT_IMAGE_QUALITIY = 7
|
|
DEFAULT_IMAGE_ROTATION = 0
|
|
DEFAULT_IMAGE_WIDTH = 640
|
|
DEFAULT_NAME = 'Raspberry Pi Camera'
|
|
DEFAULT_TIMELAPSE = 1000
|
|
DEFAULT_VERTICAL_FLIP = 0
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Optional(CONF_FILE_PATH): cv.string,
|
|
vol.Optional(CONF_HORIZONTAL_FLIP, default=DEFAULT_HORIZONTAL_FLIP):
|
|
vol.All(vol.Coerce(int), vol.Range(min=0, max=1)),
|
|
vol.Optional(CONF_IMAGE_HEIGHT, default=DEFAULT_IMAGE_HEIGHT):
|
|
vol.Coerce(int),
|
|
vol.Optional(CONF_IMAGE_QUALITY, default=DEFAULT_IMAGE_QUALITIY):
|
|
vol.All(vol.Coerce(int), vol.Range(min=0, max=100)),
|
|
vol.Optional(CONF_IMAGE_ROTATION, default=DEFAULT_IMAGE_ROTATION):
|
|
vol.All(vol.Coerce(int), vol.Range(min=0, max=359)),
|
|
vol.Optional(CONF_IMAGE_WIDTH, default=DEFAULT_IMAGE_WIDTH):
|
|
vol.Coerce(int),
|
|
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
|
|
vol.Optional(CONF_TIMELAPSE, default=1000): vol.Coerce(int),
|
|
vol.Optional(CONF_VERTICAL_FLIP, default=DEFAULT_VERTICAL_FLIP):
|
|
vol.All(vol.Coerce(int), vol.Range(min=0, max=1)),
|
|
})
|
|
|
|
|
|
def kill_raspistill(*args):
|
|
"""Kill any previously running raspistill process.."""
|
|
subprocess.Popen(['killall', 'raspistill'],
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.STDOUT)
|
|
|
|
|
|
def setup_platform(hass, config, add_devices, discovery_info=None):
|
|
"""Setup the Raspberry Camera."""
|
|
if shutil.which("raspistill") is None:
|
|
_LOGGER.error("'raspistill' was not found")
|
|
return False
|
|
|
|
setup_config = (
|
|
{
|
|
CONF_NAME: config.get(CONF_NAME),
|
|
CONF_IMAGE_WIDTH: config.get(CONF_IMAGE_WIDTH),
|
|
CONF_IMAGE_HEIGHT: config.get(CONF_IMAGE_HEIGHT),
|
|
CONF_IMAGE_QUALITY: config.get(CONF_IMAGE_QUALITY),
|
|
CONF_IMAGE_ROTATION: config.get(CONF_IMAGE_ROTATION),
|
|
CONF_TIMELAPSE: config.get(CONF_TIMELAPSE),
|
|
CONF_HORIZONTAL_FLIP: config.get(CONF_HORIZONTAL_FLIP),
|
|
CONF_VERTICAL_FLIP: config.get(CONF_VERTICAL_FLIP),
|
|
CONF_FILE_PATH: config.get(CONF_FILE_PATH,
|
|
os.path.join(os.path.dirname(__file__),
|
|
'image.jpg'))
|
|
}
|
|
)
|
|
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, kill_raspistill)
|
|
|
|
try:
|
|
# Try to create an empty file (or open existing) to ensure we have
|
|
# proper permissions.
|
|
open(setup_config[CONF_FILE_PATH], 'a').close()
|
|
|
|
add_devices([RaspberryCamera(setup_config)])
|
|
except PermissionError:
|
|
_LOGGER.error("File path is not writable")
|
|
return False
|
|
except FileNotFoundError:
|
|
_LOGGER.error("Could not create output file (missing directory?)")
|
|
return False
|
|
|
|
|
|
class RaspberryCamera(Camera):
|
|
"""Representation of a Raspberry Pi camera."""
|
|
|
|
def __init__(self, device_info):
|
|
"""Initialize Raspberry Pi camera component."""
|
|
super().__init__()
|
|
|
|
self._name = device_info[CONF_NAME]
|
|
self._config = device_info
|
|
|
|
# Kill if there's raspistill instance
|
|
kill_raspistill()
|
|
|
|
cmd_args = [
|
|
'raspistill', '--nopreview', '-o', device_info[CONF_FILE_PATH],
|
|
'-t', '0', '-w', str(device_info[CONF_IMAGE_WIDTH]),
|
|
'-h', str(device_info[CONF_IMAGE_HEIGHT]),
|
|
'-tl', str(device_info[CONF_TIMELAPSE]),
|
|
'-q', str(device_info[CONF_IMAGE_QUALITY]),
|
|
'-rot', str(device_info[CONF_IMAGE_ROTATION])
|
|
]
|
|
if device_info[CONF_HORIZONTAL_FLIP]:
|
|
cmd_args.append("-hf")
|
|
|
|
if device_info[CONF_VERTICAL_FLIP]:
|
|
cmd_args.append("-vf")
|
|
|
|
subprocess.Popen(cmd_args,
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.STDOUT)
|
|
|
|
def camera_image(self):
|
|
"""Return raspstill image response."""
|
|
with open(self._config[CONF_FILE_PATH], 'rb') as file:
|
|
return file.read()
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of this camera."""
|
|
return self._name
|