Refactor unifiprotect media_source to remove type ignores (#119516)

This commit is contained in:
J. Nick Koston 2024-06-12 12:02:53 -05:00 committed by GitHub
parent 7f7128adbf
commit 8f7686082c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -543,21 +543,20 @@ class ProtectMediaSource(MediaSource):
return source
now = dt_util.now()
args = {
"data": data,
"start": now - timedelta(days=days),
"end": now,
"reserve": True,
"event_types": get_ufp_event(event_type),
}
camera: Camera | None = None
event_camera_id: str | None = None
if camera_id != "all":
camera = data.api.bootstrap.cameras.get(camera_id)
args["camera_id"] = camera_id
event_camera_id = camera_id
events = await self._build_events(**args) # type: ignore[arg-type]
events = await self._build_events(
data=data,
start=now - timedelta(days=days),
end=now,
camera_id=event_camera_id,
event_types=get_ufp_event(event_type),
reserve=True,
)
source.children = events
source.title = self._breadcrumb(
data,
@ -674,21 +673,21 @@ class ProtectMediaSource(MediaSource):
else:
end_dt = start_dt + timedelta(hours=24)
args = {
"data": data,
"start": start_dt,
"end": end_dt,
"reserve": False,
"event_types": get_ufp_event(event_type),
}
camera: Camera | None = None
event_camera_id: str | None = None
if camera_id != "all":
camera = data.api.bootstrap.cameras.get(camera_id)
args["camera_id"] = camera_id
event_camera_id = camera_id
title = f"{start.strftime('%B %Y')} > {title}"
events = await self._build_events(**args) # type: ignore[arg-type]
events = await self._build_events(
data=data,
start=start_dt,
end=end_dt,
camera_id=event_camera_id,
reserve=False,
event_types=get_ufp_event(event_type),
)
source.children = events
source.title = self._breadcrumb(
data,