Use EntityFeature enum in components (u**) (#69462)

This commit is contained in:
epenet 2022-04-07 09:35:15 +02:00 committed by GitHub
parent 95fb4695e4
commit 75eec850c1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 66 additions and 98 deletions

View file

@ -8,7 +8,7 @@ from pyunifiprotect.api import ProtectApiClient
from pyunifiprotect.data import Camera as UFPCamera, StateType
from pyunifiprotect.data.devices import CameraChannel
from homeassistant.components.camera import SUPPORT_STREAM, Camera
from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -134,7 +134,7 @@ class ProtectCamera(ProtectDeviceEntity, Camera):
None if disable_stream else rtsp_url
)
self._attr_supported_features: int = (
SUPPORT_STREAM if self._stream_source else 0
CameraEntityFeature.STREAM if self._stream_source else 0
)
@callback