Align kitchen_sink with changes in BackupAgent (#139768)

This commit is contained in:
Erik Montnemery 2025-03-04 16:38:03 +01:00 committed by GitHub
parent e3a90831bf
commit 0ebdb1c2a8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -7,7 +7,13 @@ from collections.abc import AsyncIterator, Callable, Coroutine
import logging import logging
from typing import Any from typing import Any
from homeassistant.components.backup import AddonInfo, AgentBackup, BackupAgent, Folder from homeassistant.components.backup import (
AddonInfo,
AgentBackup,
BackupAgent,
BackupNotFound,
Folder,
)
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from . import DATA_BACKUP_AGENT_LISTENERS, DOMAIN from . import DATA_BACKUP_AGENT_LISTENERS, DOMAIN
@ -110,9 +116,9 @@ class KitchenSinkBackupAgent(BackupAgent):
self, self,
backup_id: str, backup_id: str,
**kwargs: Any, **kwargs: Any,
) -> AgentBackup | None: ) -> AgentBackup:
"""Return a backup.""" """Return a backup."""
for backup in self._uploads: for backup in self._uploads:
if backup.backup_id == backup_id: if backup.backup_id == backup_id:
return backup return backup
return None raise BackupNotFound(f"Backup {backup_id} not found")