Black

homeassistant/components/ffmpeg/camera.py

@@ -4,8 +4,7 @@ import logging
 
 import voluptuous as vol
 
-from homeassistant.components.camera import (
-    PLATFORM_SCHEMA, Camera, SUPPORT_STREAM)
+from homeassistant.components.camera import PLATFORM_SCHEMA, Camera, SUPPORT_STREAM
 from homeassistant.const import CONF_NAME
 from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
 import homeassistant.helpers.config_validation as cv
@@ -14,18 +13,19 @@ from . import CONF_EXTRA_ARGUMENTS, CONF_INPUT, DATA_FFMPEG
 
 _LOGGER = logging.getLogger(__name__)
 
-DEFAULT_NAME = 'FFmpeg'
+DEFAULT_NAME = "FFmpeg"
 DEFAULT_ARGUMENTS = "-pred 1"
 
-PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
-    vol.Required(CONF_INPUT): cv.string,
-    vol.Optional(CONF_EXTRA_ARGUMENTS, default=DEFAULT_ARGUMENTS): cv.string,
-    vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
-})
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
+    {
+        vol.Required(CONF_INPUT): cv.string,
+        vol.Optional(CONF_EXTRA_ARGUMENTS, default=DEFAULT_ARGUMENTS): cv.string,
+        vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
+    }
+)
 
 
-async def async_setup_platform(hass, config, async_add_entities,
-                               discovery_info=None):
+async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
     """Set up a FFmpeg camera."""
     async_add_entities([FFmpegCamera(hass, config)])
 
@@ -49,16 +49,19 @@ class FFmpegCamera(Camera):
 
     async def stream_source(self):
         """Return the stream source."""
-        return self._input.split(' ')[-1]
+        return self._input.split(" ")[-1]
 
     async def async_camera_image(self):
         """Return a still image response from the camera."""
         from haffmpeg.tools import ImageFrame, IMAGE_JPEG
+
         ffmpeg = ImageFrame(self._manager.binary, loop=self.hass.loop)
 
-        image = await asyncio.shield(ffmpeg.get_image(
-            self._input, output_format=IMAGE_JPEG,
-            extra_cmd=self._extra_arguments))
+        image = await asyncio.shield(
+            ffmpeg.get_image(
+                self._input, output_format=IMAGE_JPEG, extra_cmd=self._extra_arguments
+            )
+        )
         return image
 
     async def handle_async_mjpeg_stream(self, request):
@@ -66,14 +69,16 @@ class FFmpegCamera(Camera):
         from haffmpeg.camera import CameraMjpeg
 
         stream = CameraMjpeg(self._manager.binary, loop=self.hass.loop)
-        await stream.open_camera(
-            self._input, extra_cmd=self._extra_arguments)
+        await stream.open_camera(self._input, extra_cmd=self._extra_arguments)
 
         try:
             stream_reader = await stream.get_reader()
             return await async_aiohttp_proxy_stream(
-                self.hass, request, stream_reader,
-                self._manager.ffmpeg_stream_content_type)
+                self.hass,
+                request,
+                stream_reader,
+                self._manager.ffmpeg_stream_content_type,
+            )
         finally:
             await stream.close()