mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-10-18 16:19:38 +00:00
31193abb7b
* FileConfiguration uses executor for I/O * Fix credentials tests * Remove migrate_system_env as its very deprecated
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
"""Init file for Supervisor add-on data."""
|
|
|
|
from copy import deepcopy
|
|
from typing import Any
|
|
|
|
from ..const import (
|
|
ATTR_IMAGE,
|
|
ATTR_OPTIONS,
|
|
ATTR_SYSTEM,
|
|
ATTR_USER,
|
|
ATTR_VERSION,
|
|
FILE_HASSIO_ADDONS,
|
|
)
|
|
from ..coresys import CoreSys, CoreSysAttributes
|
|
from ..store.addon import AddonStore
|
|
from ..utils.common import FileConfiguration
|
|
from .addon import Addon
|
|
from .validate import SCHEMA_ADDONS_FILE
|
|
|
|
Config = dict[str, Any]
|
|
|
|
|
|
class AddonsData(FileConfiguration, CoreSysAttributes):
|
|
"""Hold data for installed Add-ons inside Supervisor."""
|
|
|
|
def __init__(self, coresys: CoreSys):
|
|
"""Initialize data holder."""
|
|
super().__init__(FILE_HASSIO_ADDONS, SCHEMA_ADDONS_FILE)
|
|
self.coresys: CoreSys = coresys
|
|
|
|
@property
|
|
def user(self):
|
|
"""Return local add-on user data."""
|
|
return self._data[ATTR_USER]
|
|
|
|
@property
|
|
def system(self):
|
|
"""Return local add-on data."""
|
|
return self._data[ATTR_SYSTEM]
|
|
|
|
async def install(self, addon: AddonStore) -> None:
|
|
"""Set addon as installed."""
|
|
self.system[addon.slug] = deepcopy(addon.data)
|
|
self.user[addon.slug] = {
|
|
ATTR_OPTIONS: {},
|
|
ATTR_VERSION: addon.version,
|
|
ATTR_IMAGE: addon.image,
|
|
}
|
|
await self.save_data()
|
|
|
|
async def uninstall(self, addon: Addon) -> None:
|
|
"""Set add-on as uninstalled."""
|
|
self.system.pop(addon.slug, None)
|
|
self.user.pop(addon.slug, None)
|
|
await self.save_data()
|
|
|
|
async def update(self, addon: AddonStore) -> None:
|
|
"""Update version of add-on."""
|
|
self.system[addon.slug] = deepcopy(addon.data)
|
|
self.user[addon.slug].update(
|
|
{ATTR_VERSION: addon.version, ATTR_IMAGE: addon.image}
|
|
)
|
|
await self.save_data()
|
|
|
|
async def restore(
|
|
self, slug: str, user: Config, system: Config, image: str
|
|
) -> None:
|
|
"""Restore data to add-on."""
|
|
self.user[slug] = deepcopy(user)
|
|
self.system[slug] = deepcopy(system)
|
|
|
|
self.user[slug][ATTR_IMAGE] = image
|
|
await self.save_data()
|