ee7ce47860
* Initial working commit * Create const file. Load camera from component. * Handle failed authentication. Bump library version. * Remove line break * Camera attributes and recording services * Add services, manifest, constant update, and exclude_channels. Prefix channel name. Update service argument. * Update codeowners * Update coveragerc * Remove codeowners line * Update codeowners again from python3 -m script.hassfest * Update homeassistant/components/qvrpro/__init__.py Co-Authored-By: springstan <46536646+springstan@users.noreply.github.com> * Requested changes * Fix typo * Update to use exception. Bump library version. * Support stream component * Update module header * Missing property wrapper * Partial requested changes * Update coveragerc and codeowners * Move constants to const file. Add SHORT_NAME * Add conf variable * Use camera domain * More requested changes * Requested changes * Requested changes * Update prefix * Handle error condition when camera is not configured to support live streaming * Move method to camera setup. Disable stream component support. * Move auth string to library to prevent private member access Co-authored-by: springstan <46536646+springstan@users.noreply.github.com>
103 lines
2.4 KiB
Python
103 lines
2.4 KiB
Python
"""Support for QVR Pro streams."""
|
|
|
|
import logging
|
|
|
|
from pyqvrpro.client import QVRResponseError
|
|
|
|
from homeassistant.components.camera import Camera
|
|
|
|
from .const import DOMAIN, SHORT_NAME
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the QVR Pro camera platform."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
client = hass.data[DOMAIN]["client"]
|
|
|
|
entities = []
|
|
|
|
for channel in hass.data[DOMAIN]["channels"]:
|
|
|
|
stream_source = get_stream_source(channel["guid"], client)
|
|
entities.append(
|
|
QVRProCamera(**channel, stream_source=stream_source, client=client)
|
|
)
|
|
|
|
add_entities(entities)
|
|
|
|
|
|
def get_stream_source(guid, client):
|
|
"""Get channel stream source."""
|
|
try:
|
|
resp = client.get_channel_live_stream(guid, protocol="rtsp")
|
|
|
|
full_url = resp["resourceUris"]
|
|
|
|
protocol = full_url[:7]
|
|
auth = f"{client.get_auth_string()}@"
|
|
url = full_url[7:]
|
|
|
|
return f"{protocol}{auth}{url}"
|
|
|
|
except QVRResponseError as ex:
|
|
_LOGGER.error(ex)
|
|
return None
|
|
|
|
|
|
class QVRProCamera(Camera):
|
|
"""Representation of a QVR Pro camera."""
|
|
|
|
def __init__(self, name, model, brand, channel_index, guid, stream_source, client):
|
|
"""Init QVR Pro camera."""
|
|
|
|
self._name = f"{SHORT_NAME} {name}"
|
|
self._model = model
|
|
self._brand = brand
|
|
self.index = channel_index
|
|
self.guid = guid
|
|
self._client = client
|
|
self._stream_source = stream_source
|
|
|
|
self._supported_features = 0
|
|
|
|
super().__init__()
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the entity."""
|
|
return self._name
|
|
|
|
@property
|
|
def model(self):
|
|
"""Return the model of the entity."""
|
|
return self._model
|
|
|
|
@property
|
|
def brand(self):
|
|
"""Return the brand of the entity."""
|
|
return self._brand
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Get the state attributes."""
|
|
attrs = {"qvr_guid": self.guid}
|
|
|
|
return attrs
|
|
|
|
def camera_image(self):
|
|
"""Get image bytes from camera."""
|
|
return self._client.get_snapshot(self.guid)
|
|
|
|
async def stream_source(self):
|
|
"""Get stream source."""
|
|
return self._stream_source
|
|
|
|
@property
|
|
def supported_features(self):
|
|
"""Get supported features."""
|
|
return self._supported_features
|