Add QVR Pro integration

.coveragerc

@@ -558,6 +558,7 @@ omit =
     homeassistant/components/qnap/sensor.py
     homeassistant/components/qrcode/image_processing.py
     homeassistant/components/quantum_gateway/device_tracker.py
+    homeassistant/components/qvr_pro/*
     homeassistant/components/qwikswitch/*
     homeassistant/components/rachio/*
     homeassistant/components/radarr/sensor.py

CODEOWNERS

@@ -272,6 +272,7 @@ homeassistant/components/pvoutput/* @fabaff
 homeassistant/components/qld_bushfire/* @exxamalte
 homeassistant/components/qnap/* @colinodell
 homeassistant/components/quantum_gateway/* @cisasteelersfan
+homeassistant/components/qvr_pro/* @oblogic7
 homeassistant/components/qwikswitch/* @kellerza
 homeassistant/components/rainbird/* @konikvranik
 homeassistant/components/raincloud/* @vanstinator

homeassistant/components/qvr_pro/__init__.py

@@ -0,0 +1,91 @@
+"""Support for QVR Pro NVR software by QNAP."""
+
+import logging
+
+from pyqvrpro import Client
+from pyqvrpro.client import AuthenticationError, InsufficientPermissionsError
+import voluptuous as vol
+
+from homeassistant.components.camera import DOMAIN as CAMERA_DOMAIN
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
+import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.discovery import load_platform
+
+from .const import CONF_EXCLUDE_CHANNELS, DOMAIN, SHORT_NAME, START_RECORD, STOP_RECORD
+
+_LOGGER = logging.getLogger(__name__)
+
+CONFIG_SCHEMA = vol.Schema(
+    {
+        DOMAIN: vol.Schema(
+            {
+                vol.Required(CONF_HOST): cv.string,
+                vol.Required(CONF_USERNAME): cv.string,
+                vol.Required(CONF_PASSWORD): cv.string,
+                vol.Optional(CONF_EXCLUDE_CHANNELS, default=[]): vol.All(
+                    cv.ensure_list_csv, [cv.positive_int]
+                ),
+            }
+        )
+    },
+    extra=vol.ALLOW_EXTRA,
+)
+
+
+def setup(hass, config):
+    """Set up the QVR Pro component."""
+    conf = config[DOMAIN]
+    user = conf[CONF_USERNAME]
+    password = conf[CONF_PASSWORD]
+    host = conf[CONF_HOST]
+    excluded_channels = conf[CONF_EXCLUDE_CHANNELS]
+
+    try:
+        qvrpro = Client(user, password, host)
+
+        channel_resp = qvrpro.get_channel_list()
+
+    except InsufficientPermissionsError:
+        _LOGGER.error("User must have Surveillance Management permission")
+        return False
+    except AuthenticationError:
+        _LOGGER.error("Authentication failed")
+        return False
+
+    channels = []
+
+    for channel in channel_resp["channels"]:
+        if channel["channel_index"] + 1 in excluded_channels:
+            continue
+
+        channels.append(QVRChannel(**channel))
+
+    hass.data[DOMAIN] = {"channels": channels, "client": qvrpro}
+
+    load_platform(hass, CAMERA_DOMAIN, DOMAIN, {}, config)
+
+    # Register services
+    def handle_start_record(call):
+        guid = call.data.get("guid")
+        qvrpro.start_recording(guid)
+
+    def handle_stop_record(call):
+        guid = call.data.get("guid")
+        qvrpro.stop_recording(guid)
+
+    hass.services.register(DOMAIN, START_RECORD, handle_start_record)
+    hass.services.register(DOMAIN, STOP_RECORD, handle_stop_record)
+
+    return True
+
+
+class QVRChannel:
+    """Representation of a QVR channel."""
+
+    def __init__(self, name, model, brand, channel_index, guid):
+        """Initialize QVRChannel."""
+        self.name = f"{SHORT_NAME} {name}"
+        self.model = model
+        self.brand = brand
+        self.index = channel_index
+        self.guid = guid

homeassistant/components/qvr_pro/camera.py

@@ -0,0 +1,66 @@
+"""Support for QVR Pro streams."""
+
+from homeassistant.components.camera import SUPPORT_STREAM, Camera
+
+from .const import DOMAIN
+
+
+def setup_platform(hass, config, add_entities, discovery_info=None):
+    """Set up the QVR Pro camera platform."""
+    client = hass.data[DOMAIN]["client"]
+
+    entities = []
+
+    for channel in hass.data[DOMAIN]["channels"]:
+        entities.append(QVRProCamera(channel, client))
+
+    add_entities(entities)
+
+
+class QVRProCamera(Camera):
+    """Representation of a QVR Pro camera."""
+
+    def __init__(self, channel, client):
+        """Init QVR Pro camera."""
+
+        self._channel = channel
+        self._client = client
+
+        super().__init__()
+
+    @property
+    def name(self):
+        """Get the name of the camera."""
+        return self._channel.name
+
+    @property
+    def model(self):
+        """Get the model of the camera."""
+        return self._channel.model
+
+    @property
+    def brand(self):
+        """Get the brand of the camera."""
+        return self._channel.brand
+
+    @property
+    def state_attributes(self):
+        """Get the state attributes."""
+        attrs = super().state_attributes
+
+        attrs["qvr_guid"] = self._channel.guid
+
+        return attrs
+
+    def camera_image(self):
+        """Get image bytes from camera."""
+        return self._client.get_snapshot(self._channel.guid)
+
+    def stream_source(self):
+        """Get stream source."""
+        return self._client.get_channel_live_stream(self._channel.guid)
+
+    @property
+    def supported_features(self):
+        """Get supported features."""
+        return SUPPORT_STREAM

homeassistant/components/qvr_pro/const.py

@@ -0,0 +1,9 @@
+"""Constants for QVR Pro component."""
+
+DOMAIN = "qvr_pro"
+SHORT_NAME = "QVR"
+
+CONF_EXCLUDE_CHANNELS = "exclude_channels"
+
+STOP_RECORD = "stop_record"
+START_RECORD = "start_record"

homeassistant/components/qvr_pro/manifest.json

@@ -0,0 +1,8 @@
+{
+  "domain": "qvr_pro",
+  "name": "QVR Pro",
+  "documentation": "https://www.home-assistant.io/integrations/qvr_pro",
+  "requirements": ["pyqvrpro==0.49"],
+  "dependencies": [],
+  "codeowners": ["@oblogic7"]
+}

homeassistant/components/qvr_pro/services.yaml

@@ -0,0 +1,13 @@
+start_record:
+  description: Start QVR Pro recording on specified channel.
+  fields:
+    guid:
+      description: GUID of the channel to start recording.
+      example: '245EBE933C0A597EBE865C0A245E0002'
+
+stop_record:
+  description: Stop QVR Pro recording on specified channel.
+  fields:
+    guid:
+      description: GUID of the channel to stop recording.
+      example: '245EBE933C0A597EBE865C0A245E0002'

requirements_all.txt

@@ -1454,6 +1454,9 @@ pypoint==1.1.2
 # homeassistant.components.ps4
 pyps4-2ndscreen==1.0.6
 
+# homeassistant.components.qvr_pro
+pyqvrpro==0.49
+
 # homeassistant.components.qwikswitch
 pyqwikswitch==0.93