jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-26 02:07:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reolink only fetch data for enabled entities (#104157)

Browse Source
This commit is contained in:
starkillerOG 2023-12-02 14:10:44 +01:00 committed by GitHub
parent a3dd2b8ea9
commit 381036e46a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 63 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/reolink/entity.py
View File

@ -24,6 +24,7 @@ _T = TypeVar("_T")
class ReolinkChannelEntityDescription(EntityDescription):
"""A class that describes entities for a camera channel."""
cmd_key: str | None = None
supported: Callable[[Host, int], bool] = lambda api, ch: True
@ -31,6 +32,7 @@ class ReolinkChannelEntityDescription(EntityDescription):
class ReolinkHostEntityDescription(EntityDescription):
"""A class that describes host entities."""
cmd_key: str | None = None
supported: Callable[[Host], bool] = lambda api: True
@ -84,6 +86,15 @@ class ReolinkHostCoordinatorEntity(ReolinkBaseCoordinatorEntity[None]):
self._attr_unique_id = f"{self._host.unique_id}_{self.entity_description.key}"
async def async_added_to_hass(self) -> None:
"""Entity created."""
await super().async_added_to_hass()
if (
self.entity_description.cmd_key is not None
and self.entity_description.cmd_key not in self._host.update_cmd_list
):
self._host.update_cmd_list.append(self.entity_description.cmd_key)
class ReolinkChannelCoordinatorEntity(ReolinkHostCoordinatorEntity):
"""Parent class for Reolink hardware camera entities connected to a channel of the NVR."""

4
homeassistant/components/reolink/host.py
View File

@ -60,6 +60,8 @@ class ReolinkHost:
timeout=DEFAULT_TIMEOUT,
)
self.update_cmd_list: list[str] = []
self.webhook_id: str | None = None
self._onvif_push_supported: bool = True
self._onvif_long_poll_supported: bool = True
@ -311,7 +313,7 @@ class ReolinkHost:
async def update_states(self) -> None:
"""Call the API of the camera device to update the internal states."""
await self._api.get_states()
await self._api.get_states(cmd_list=self.update_cmd_list)
async def disconnect(self) -> None:
"""Disconnect from the API, so the connection will be released."""

3
homeassistant/components/reolink/light.py
View File

@ -41,6 +41,7 @@ class ReolinkLightEntityDescription(
LIGHT_ENTITIES = (
ReolinkLightEntityDescription(
key="floodlight",
cmd_key="GetWhiteLed",
translation_key="floodlight",
icon="mdi:spotlight-beam",
supported=lambda api, ch: api.supported(ch, "floodLight"),
@ -51,6 +52,7 @@ LIGHT_ENTITIES = (
),
ReolinkLightEntityDescription(
key="ir_lights",
cmd_key="GetIrLights",
translation_key="ir_lights",
icon="mdi:led-off",
entity_category=EntityCategory.CONFIG,
@ -60,6 +62,7 @@ LIGHT_ENTITIES = (
),
ReolinkLightEntityDescription(
key="status_led",
cmd_key="GetPowerLed",
translation_key="status_led",
icon="mdi:lightning-bolt-circle",
entity_category=EntityCategory.CONFIG,

22
homeassistant/components/reolink/number.py
View File

@ -41,6 +41,7 @@ class ReolinkNumberEntityDescription(
NUMBER_ENTITIES = (
ReolinkNumberEntityDescription(
key="zoom",
cmd_key="GetZoomFocus",
translation_key="zoom",
icon="mdi:magnify",
mode=NumberMode.SLIDER,
@ -53,6 +54,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="focus",
cmd_key="GetZoomFocus",
translation_key="focus",
icon="mdi:focus-field",
mode=NumberMode.SLIDER,
@ -68,6 +70,7 @@ NUMBER_ENTITIES = (
# or when using the "light.floodlight" entity.
ReolinkNumberEntityDescription(
key="floodlight_brightness",
cmd_key="GetWhiteLed",
translation_key="floodlight_brightness",
icon="mdi:spotlight-beam",
entity_category=EntityCategory.CONFIG,
@ -80,6 +83,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="volume",
cmd_key="GetAudioCfg",
translation_key="volume",
icon="mdi:volume-high",
entity_category=EntityCategory.CONFIG,
@ -92,6 +96,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="guard_return_time",
cmd_key="GetPtzGuard",
translation_key="guard_return_time",
icon="mdi:crosshairs-gps",
entity_category=EntityCategory.CONFIG,
@ -105,6 +110,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="motion_sensitivity",
cmd_key="GetMdAlarm",
translation_key="motion_sensitivity",
icon="mdi:motion-sensor",
entity_category=EntityCategory.CONFIG,
@ -117,6 +123,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_face_sensititvity",
cmd_key="GetAiAlarm",
translation_key="ai_face_sensititvity",
icon="mdi:face-recognition",
entity_category=EntityCategory.CONFIG,
@ -131,6 +138,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_person_sensititvity",
cmd_key="GetAiAlarm",
translation_key="ai_person_sensititvity",
icon="mdi:account",
entity_category=EntityCategory.CONFIG,
@ -145,6 +153,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_vehicle_sensititvity",
cmd_key="GetAiAlarm",
translation_key="ai_vehicle_sensititvity",
icon="mdi:car",
entity_category=EntityCategory.CONFIG,
@ -159,6 +168,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_pet_sensititvity",
cmd_key="GetAiAlarm",
translation_key="ai_pet_sensititvity",
icon="mdi:dog-side",
entity_category=EntityCategory.CONFIG,
@ -175,6 +185,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_pet_sensititvity",
cmd_key="GetAiAlarm",
translation_key="ai_animal_sensititvity",
icon="mdi:paw",
entity_category=EntityCategory.CONFIG,
@ -189,6 +200,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_face_delay",
cmd_key="GetAiAlarm",
translation_key="ai_face_delay",
icon="mdi:face-recognition",
entity_category=EntityCategory.CONFIG,
@ -205,6 +217,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_person_delay",
cmd_key="GetAiAlarm",
translation_key="ai_person_delay",
icon="mdi:account",
entity_category=EntityCategory.CONFIG,
@ -221,6 +234,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_vehicle_delay",
cmd_key="GetAiAlarm",
translation_key="ai_vehicle_delay",
icon="mdi:car",
entity_category=EntityCategory.CONFIG,
@ -237,6 +251,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_pet_delay",
cmd_key="GetAiAlarm",
translation_key="ai_pet_delay",
icon="mdi:dog-side",
entity_category=EntityCategory.CONFIG,
@ -255,6 +270,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="ai_pet_delay",
cmd_key="GetAiAlarm",
translation_key="ai_animal_delay",
icon="mdi:paw",
entity_category=EntityCategory.CONFIG,
@ -271,6 +287,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="auto_quick_reply_time",
cmd_key="GetAutoReply",
translation_key="auto_quick_reply_time",
icon="mdi:message-reply-text-outline",
entity_category=EntityCategory.CONFIG,
@ -284,6 +301,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="auto_track_limit_left",
cmd_key="GetPtzTraceSection",
translation_key="auto_track_limit_left",
icon="mdi:angle-acute",
mode=NumberMode.SLIDER,
@ -297,6 +315,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="auto_track_limit_right",
cmd_key="GetPtzTraceSection",
translation_key="auto_track_limit_right",
icon="mdi:angle-acute",
mode=NumberMode.SLIDER,
@ -310,6 +329,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="auto_track_disappear_time",
cmd_key="GetAiCfg",
translation_key="auto_track_disappear_time",
icon="mdi:target-account",
entity_category=EntityCategory.CONFIG,
@ -325,6 +345,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="auto_track_stop_time",
cmd_key="GetAiCfg",
translation_key="auto_track_stop_time",
icon="mdi:target-account",
entity_category=EntityCategory.CONFIG,
@ -338,6 +359,7 @@ NUMBER_ENTITIES = (
),
ReolinkNumberEntityDescription(
key="day_night_switch_threshold",
cmd_key="GetIsp",
translation_key="day_night_switch_threshold",
icon="mdi:theme-light-dark",
entity_category=EntityCategory.CONFIG,

5
homeassistant/components/reolink/select.py
View File

@ -44,6 +44,7 @@ class ReolinkSelectEntityDescription(
SELECT_ENTITIES = (
ReolinkSelectEntityDescription(
key="floodlight_mode",
cmd_key="GetWhiteLed",
translation_key="floodlight_mode",
icon="mdi:spotlight-beam",
entity_category=EntityCategory.CONFIG,
@ -54,6 +55,7 @@ SELECT_ENTITIES = (
),
ReolinkSelectEntityDescription(
key="day_night_mode",
cmd_key="GetIsp",
translation_key="day_night_mode",
icon="mdi:theme-light-dark",
entity_category=EntityCategory.CONFIG,
@ -72,6 +74,7 @@ SELECT_ENTITIES = (
),
ReolinkSelectEntityDescription(
key="auto_quick_reply_message",
cmd_key="GetAutoReply",
translation_key="auto_quick_reply_message",
icon="mdi:message-reply-text-outline",
entity_category=EntityCategory.CONFIG,
@ -84,6 +87,7 @@ SELECT_ENTITIES = (
),
ReolinkSelectEntityDescription(
key="auto_track_method",
cmd_key="GetAiCfg",
translation_key="auto_track_method",
icon="mdi:target-account",
entity_category=EntityCategory.CONFIG,
@ -94,6 +98,7 @@ SELECT_ENTITIES = (
),
ReolinkSelectEntityDescription(
key="status_led",
cmd_key="GetPowerLed",
translation_key="status_led",
icon="mdi:lightning-bolt-circle",
entity_category=EntityCategory.CONFIG,

2
homeassistant/components/reolink/sensor.py
View File

@ -52,6 +52,7 @@ class ReolinkHostSensorEntityDescription(
SENSORS = (
ReolinkSensorEntityDescription(
key="ptz_pan_position",
cmd_key="GetPtzCurPos",
translation_key="ptz_pan_position",
icon="mdi:pan",
state_class=SensorStateClass.MEASUREMENT,
@ -64,6 +65,7 @@ SENSORS = (
HOST_SENSORS = (
ReolinkHostSensorEntityDescription(
key="wifi_signal",
cmd_key="GetWifiSignal",
translation_key="wifi_signal",
icon="mdi:wifi",
state_class=SensorStateClass.MEASUREMENT,

17
homeassistant/components/reolink/switch.py
View File

@ -50,6 +50,7 @@ class ReolinkNVRSwitchEntityDescription(
SWITCH_ENTITIES = (
ReolinkSwitchEntityDescription(
key="record_audio",
cmd_key="GetEnc",
translation_key="record_audio",
icon="mdi:microphone",
entity_category=EntityCategory.CONFIG,
@ -59,6 +60,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="siren_on_event",
cmd_key="GetAudioAlarm",
translation_key="siren_on_event",
icon="mdi:alarm-light",
entity_category=EntityCategory.CONFIG,
@ -68,6 +70,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="auto_tracking",
cmd_key="GetAiCfg",
translation_key="auto_tracking",
icon="mdi:target-account",
entity_category=EntityCategory.CONFIG,
@ -77,6 +80,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="auto_focus",
cmd_key="GetAutoFocus",
translation_key="auto_focus",
icon="mdi:focus-field",
entity_category=EntityCategory.CONFIG,
@ -86,6 +90,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="gaurd_return",
cmd_key="GetPtzGuard",
translation_key="gaurd_return",
icon="mdi:crosshairs-gps",
entity_category=EntityCategory.CONFIG,
@ -95,6 +100,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="email",
cmd_key="GetEmail",
translation_key="email",
icon="mdi:email",
entity_category=EntityCategory.CONFIG,
@ -104,6 +110,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="ftp_upload",
cmd_key="GetFtp",
translation_key="ftp_upload",
icon="mdi:swap-horizontal",
entity_category=EntityCategory.CONFIG,
@ -113,6 +120,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="push_notifications",
cmd_key="GetPush",
translation_key="push_notifications",
icon="mdi:message-badge",
entity_category=EntityCategory.CONFIG,
@ -122,6 +130,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="record",
cmd_key="GetRec",
translation_key="record",
icon="mdi:record-rec",
entity_category=EntityCategory.CONFIG,
@ -131,6 +140,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="buzzer",
cmd_key="GetBuzzerAlarmV20",
translation_key="buzzer",
icon="mdi:room-service",
entity_category=EntityCategory.CONFIG,
@ -140,6 +150,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="doorbell_button_sound",
cmd_key="GetAudioCfg",
translation_key="doorbell_button_sound",
icon="mdi:volume-high",
entity_category=EntityCategory.CONFIG,
@ -149,6 +160,7 @@ SWITCH_ENTITIES = (
),
ReolinkSwitchEntityDescription(
key="hdr",
cmd_key="GetIsp",
translation_key="hdr",
icon="mdi:hdr",
entity_category=EntityCategory.CONFIG,
@ -162,6 +174,7 @@ SWITCH_ENTITIES = (
NVR_SWITCH_ENTITIES = (
ReolinkNVRSwitchEntityDescription(
key="email",
cmd_key="GetEmail",
translation_key="email",
icon="mdi:email",
entity_category=EntityCategory.CONFIG,
@ -171,6 +184,7 @@ NVR_SWITCH_ENTITIES = (
),
ReolinkNVRSwitchEntityDescription(
key="ftp_upload",
cmd_key="GetFtp",
translation_key="ftp_upload",
icon="mdi:swap-horizontal",
entity_category=EntityCategory.CONFIG,
@ -180,6 +194,7 @@ NVR_SWITCH_ENTITIES = (
),
ReolinkNVRSwitchEntityDescription(
key="push_notifications",
cmd_key="GetPush",
translation_key="push_notifications",
icon="mdi:message-badge",
entity_category=EntityCategory.CONFIG,
@ -189,6 +204,7 @@ NVR_SWITCH_ENTITIES = (
),
ReolinkNVRSwitchEntityDescription(
key="record",
cmd_key="GetRec",
translation_key="record",
icon="mdi:record-rec",
entity_category=EntityCategory.CONFIG,
@ -198,6 +214,7 @@ NVR_SWITCH_ENTITIES = (
),
ReolinkNVRSwitchEntityDescription(
key="buzzer",
cmd_key="GetBuzzerAlarmV20",
translation_key="buzzer",
icon="mdi:room-service",
entity_category=EntityCategory.CONFIG,