evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Switch Reolink from hass.data to runtime_data (#126002)

Browse source 
Switch from hass.data to runtime_data
This commit is contained in:
starkillerOG 2024-09-15 21:05:59 +02:00 • committed by GitHub
parent d9812f0d48
commit e768bea298
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
18 changed files with 75 additions and 104 deletions
Download patch file Download diff file Expand all files Collapse all files

24
homeassistant/components/reolink/media_source.py
View file

@ -22,8 +22,8 @@ from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
from . import ReolinkData
from .const import DOMAIN
from .host import ReolinkHost
_LOGGER = logging.getLogger(__name__)
@ -46,6 +46,13 @@ def res_name(stream: str) -> str:
return "Low res."
def get_host(hass: HomeAssistant, config_entry_id: str) -> ReolinkHost:
"""Return the Reolink host from the config entry id."""
config_entry = hass.config_entries.async_get_entry(config_entry_id)
assert config_entry is not None
return config_entry.runtime_data.host
class ReolinkVODMediaSource(MediaSource):
"""Provide Reolink camera VODs as media sources."""
@ -65,8 +72,7 @@ class ReolinkVODMediaSource(MediaSource):
_, config_entry_id, channel_str, stream_res, filename = identifier
channel = int(channel_str)
data: dict[str, ReolinkData] = self.hass.data[DOMAIN]
host = data[config_entry_id].host
host = get_host(self.hass, config_entry_id)
def get_vod_type() -> VodRequestType:
if filename.endswith(".mp4"):
@ -151,8 +157,7 @@ class ReolinkVODMediaSource(MediaSource):
if config_entry.state != ConfigEntryState.LOADED:
continue
channels: list[str] = []
data: dict[str, ReolinkData] = self.hass.data[DOMAIN]
host = data[config_entry.entry_id].host
host = config_entry.runtime_data.host
entities = er.async_entries_for_config_entry(
entity_reg, config_entry.entry_id
)
@ -213,8 +218,7 @@ class ReolinkVODMediaSource(MediaSource):
self, config_entry_id: str, channel: int
) -> BrowseMediaSource:
"""Allow the user to select the high or low playback resolution, (low loads faster)."""
data: dict[str, ReolinkData] = self.hass.data[DOMAIN]
host = data[config_entry_id].host
host = get_host(self.hass, config_entry_id)
main_enc = await host.api.get_encoding(channel, "main")
if main_enc == "h265":
@ -297,8 +301,7 @@ class ReolinkVODMediaSource(MediaSource):
self, config_entry_id: str, channel: int, stream: str
) -> BrowseMediaSource:
"""Return all days on which recordings are available for a reolink camera."""
data: dict[str, ReolinkData] = self.hass.data[DOMAIN]
host = data[config_entry_id].host
host = get_host(self.hass, config_entry_id)
# We want today of the camera, not necessarily today of the server
now = host.api.time() or await host.api.async_get_time()
@ -354,8 +357,7 @@ class ReolinkVODMediaSource(MediaSource):
day: int,
) -> BrowseMediaSource:
"""Return all recording files on a specific day of a Reolink camera."""
data: dict[str, ReolinkData] = self.hass.data[DOMAIN]
host = data[config_entry_id].host
host = get_host(self.hass, config_entry_id)
start = dt.datetime(year, month, day, hour=0, minute=0, second=0)
end = dt.datetime(year, month, day, hour=23, minute=59, second=59)