jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-01 23:19:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Validate password before restoring backup (#133647)

Browse Source 
* Validate password before restoring backup

* Raise specific error when password is incorrect
This commit is contained in:
Erik Montnemery
2024-12-20 15:43:46 +01:00
committed by GitHub
parent 1c0135880d
commit 5afb9a5053
7 changed files with 220 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
homeassistant/components/backup/websocket.py
View File

@@ -9,7 +9,7 @@ from homeassistant.core import HomeAssistant, callback
from .config import ScheduleState
from .const import DATA_MANAGER, LOGGER
from .manager import ManagerStateEvent
from .manager import IncorrectPasswordError, ManagerStateEvent
from .models import Folder
@@ -131,16 +131,20 @@ async def handle_restore(
msg: dict[str, Any],
) -> None:
"""Restore a backup."""
await hass.data[DATA_MANAGER].async_restore_backup(
msg["backup_id"],
agent_id=msg["agent_id"],
password=msg.get("password"),
restore_addons=msg.get("restore_addons"),
restore_database=msg["restore_database"],
restore_folders=msg.get("restore_folders"),
restore_homeassistant=msg["restore_homeassistant"],
)
connection.send_result(msg["id"])
try:
await hass.data[DATA_MANAGER].async_restore_backup(
msg["backup_id"],
agent_id=msg["agent_id"],
password=msg.get("password"),
restore_addons=msg.get("restore_addons"),
restore_database=msg["restore_database"],
restore_folders=msg.get("restore_folders"),
restore_homeassistant=msg["restore_homeassistant"],
)
except IncorrectPasswordError:
connection.send_error(msg["id"], "password_incorrect", "Incorrect password")
else:
connection.send_result(msg["id"])
@websocket_api.require_admin