mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-08-20 13:17:35 +00:00
f6faa18409
It seems that the codebase is not formatted with the latest ruff version. This PR reformats the codebase with ruff 0.5.7.
145 lines
4.1 KiB
Python
145 lines
4.1 KiB
Python
"""Validate some things around restore."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from awesomeversion import AwesomeVersion
|
|
import voluptuous as vol
|
|
|
|
from ..backups.const import BackupType
|
|
from ..const import (
|
|
ATTR_ADDONS,
|
|
ATTR_COMPRESSED,
|
|
ATTR_CRYPTO,
|
|
ATTR_DATE,
|
|
ATTR_DAYS_UNTIL_STALE,
|
|
ATTR_DOCKER,
|
|
ATTR_EXCLUDE_DATABASE,
|
|
ATTR_FOLDERS,
|
|
ATTR_HOMEASSISTANT,
|
|
ATTR_NAME,
|
|
ATTR_PROTECTED,
|
|
ATTR_REPOSITORIES,
|
|
ATTR_SIZE,
|
|
ATTR_SLUG,
|
|
ATTR_SUPERVISOR_VERSION,
|
|
ATTR_TYPE,
|
|
ATTR_VERSION,
|
|
CRYPTO_AES128,
|
|
FOLDER_ADDONS,
|
|
FOLDER_HOMEASSISTANT,
|
|
FOLDER_MEDIA,
|
|
FOLDER_SHARE,
|
|
FOLDER_SSL,
|
|
)
|
|
from ..store.validate import repositories
|
|
from ..validate import SCHEMA_DOCKER_CONFIG, version_tag
|
|
|
|
ALL_FOLDERS = [
|
|
FOLDER_SHARE,
|
|
FOLDER_ADDONS,
|
|
FOLDER_SSL,
|
|
FOLDER_MEDIA,
|
|
]
|
|
|
|
|
|
def unique_addons(addons_list):
|
|
"""Validate that an add-on is unique."""
|
|
single = {addon[ATTR_SLUG] for addon in addons_list}
|
|
|
|
if len(single) != len(addons_list):
|
|
raise vol.Invalid("Invalid addon list in backup!") from None
|
|
return addons_list
|
|
|
|
|
|
def v1_homeassistant(
|
|
homeassistant_data: dict[str, Any] | None,
|
|
) -> dict[str, Any] | None:
|
|
"""Cleanup homeassistant artefacts from v1."""
|
|
if not homeassistant_data:
|
|
return None
|
|
|
|
if homeassistant_data.get(ATTR_VERSION) is None:
|
|
return None
|
|
|
|
return homeassistant_data
|
|
|
|
|
|
def v1_folderlist(folder_data: list[str]) -> list[str]:
|
|
"""Cleanup folder artefacts from v1."""
|
|
if FOLDER_HOMEASSISTANT in folder_data:
|
|
folder_data.remove(FOLDER_HOMEASSISTANT)
|
|
return folder_data
|
|
|
|
|
|
def v1_protected(protected: bool | str) -> bool:
|
|
"""Cleanup old protected handling."""
|
|
if isinstance(protected, bool):
|
|
return protected
|
|
return True
|
|
|
|
|
|
# pylint: disable=no-value-for-parameter
|
|
days_until_stale = vol.All(vol.Coerce(int), vol.Range(min=1))
|
|
|
|
SCHEMA_BACKUP = vol.Schema(
|
|
{
|
|
vol.Optional(ATTR_VERSION, default=1): vol.All(vol.Coerce(int), vol.In((1, 2))),
|
|
vol.Optional(
|
|
ATTR_SUPERVISOR_VERSION, default=AwesomeVersion("2022.08.3")
|
|
): version_tag,
|
|
vol.Required(ATTR_SLUG): str,
|
|
vol.Required(ATTR_TYPE): vol.Coerce(BackupType),
|
|
vol.Required(ATTR_NAME): str,
|
|
vol.Required(ATTR_DATE): str,
|
|
vol.Optional(ATTR_COMPRESSED, default=True): vol.Boolean(),
|
|
vol.Optional(ATTR_PROTECTED, default=False): vol.All(
|
|
v1_protected, vol.Boolean()
|
|
),
|
|
vol.Optional(ATTR_CRYPTO, default=None): vol.Maybe(CRYPTO_AES128),
|
|
vol.Optional(ATTR_HOMEASSISTANT, default=None): vol.All(
|
|
v1_homeassistant,
|
|
vol.Maybe(
|
|
vol.Schema(
|
|
{
|
|
vol.Required(ATTR_VERSION): version_tag,
|
|
vol.Optional(ATTR_SIZE, default=0): vol.Coerce(float),
|
|
vol.Optional(
|
|
ATTR_EXCLUDE_DATABASE, default=False
|
|
): vol.Boolean(),
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|
|
),
|
|
),
|
|
vol.Optional(ATTR_DOCKER, default=dict): SCHEMA_DOCKER_CONFIG,
|
|
vol.Optional(ATTR_FOLDERS, default=list): vol.All(
|
|
v1_folderlist, [vol.In(ALL_FOLDERS)], vol.Unique()
|
|
),
|
|
vol.Optional(ATTR_ADDONS, default=list): vol.All(
|
|
[
|
|
vol.Schema(
|
|
{
|
|
vol.Required(ATTR_SLUG): str,
|
|
vol.Required(ATTR_NAME): str,
|
|
vol.Required(ATTR_VERSION): version_tag,
|
|
vol.Optional(ATTR_SIZE, default=0): vol.Coerce(float),
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|
|
],
|
|
unique_addons,
|
|
),
|
|
vol.Optional(ATTR_REPOSITORIES, default=list): repositories,
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
SCHEMA_BACKUPS_CONFIG = vol.Schema(
|
|
{
|
|
vol.Optional(ATTR_DAYS_UNTIL_STALE, default=30): days_until_stale,
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|