605 lines
18 KiB
Python
605 lines
18 KiB
Python
"""Handle core shared data."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from typing import TYPE_CHECKING, Any, Callable, Coroutine, Optional, TypeVar
|
|
|
|
import aiohttp
|
|
import sentry_sdk
|
|
|
|
from .config import CoreConfig
|
|
from .const import UpdateChannel
|
|
from .docker import DockerAPI
|
|
from .misc.hardware import Hardware
|
|
|
|
if TYPE_CHECKING:
|
|
from .addons import AddonManager
|
|
from .api import RestAPI
|
|
from .arch import CpuArch
|
|
from .auth import Auth
|
|
from .core import Core
|
|
from .dbus import DBusManager
|
|
from .discovery import Discovery
|
|
from .hassos import HassOS
|
|
from .misc.scheduler import Scheduler
|
|
from .misc.hwmon import HwMonitor
|
|
from .misc.tasks import Tasks
|
|
from .homeassistant import HomeAssistant
|
|
from .host import HostManager
|
|
from .ingress import Ingress
|
|
from .services import ServiceManager
|
|
from .snapshots import SnapshotManager
|
|
from .supervisor import Supervisor
|
|
from .store import StoreManager
|
|
from .updater import Updater
|
|
from .plugins import PluginManager
|
|
from .resolution import ResolutionManager
|
|
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
class CoreSys:
|
|
"""Class that handle all shared data."""
|
|
|
|
def __init__(self):
|
|
"""Initialize coresys."""
|
|
# Static attributes protected
|
|
self._machine_id: Optional[str] = None
|
|
self._machine: Optional[str] = None
|
|
|
|
# External objects
|
|
self._loop: asyncio.BaseEventLoop = asyncio.get_running_loop()
|
|
self._websession: aiohttp.ClientSession = aiohttp.ClientSession()
|
|
self._websession_ssl: aiohttp.ClientSession = aiohttp.ClientSession(
|
|
connector=aiohttp.TCPConnector(ssl=False)
|
|
)
|
|
|
|
# Global objects
|
|
self._config: CoreConfig = CoreConfig()
|
|
self._hardware: Hardware = Hardware()
|
|
self._docker: DockerAPI = DockerAPI()
|
|
|
|
# Internal objects pointers
|
|
self._core: Optional[Core] = None
|
|
self._arch: Optional[CpuArch] = None
|
|
self._auth: Optional[Auth] = None
|
|
self._homeassistant: Optional[HomeAssistant] = None
|
|
self._supervisor: Optional[Supervisor] = None
|
|
self._addons: Optional[AddonManager] = None
|
|
self._api: Optional[RestAPI] = None
|
|
self._updater: Optional[Updater] = None
|
|
self._snapshots: Optional[SnapshotManager] = None
|
|
self._tasks: Optional[Tasks] = None
|
|
self._host: Optional[HostManager] = None
|
|
self._ingress: Optional[Ingress] = None
|
|
self._dbus: Optional[DBusManager] = None
|
|
self._hassos: Optional[HassOS] = None
|
|
self._services: Optional[ServiceManager] = None
|
|
self._scheduler: Optional[Scheduler] = None
|
|
self._store: Optional[StoreManager] = None
|
|
self._discovery: Optional[Discovery] = None
|
|
self._hwmonitor: Optional[HwMonitor] = None
|
|
self._plugins: Optional[PluginManager] = None
|
|
self._resolution: Optional[ResolutionManager] = None
|
|
|
|
@property
|
|
def dev(self) -> bool:
|
|
"""Return True if we run dev mode."""
|
|
if self._updater is None:
|
|
return False
|
|
return self._updater.channel == UpdateChannel.DEV
|
|
|
|
@property
|
|
def loop(self) -> asyncio.BaseEventLoop:
|
|
"""Return loop object."""
|
|
return self._loop
|
|
|
|
@property
|
|
def websession(self) -> aiohttp.ClientSession:
|
|
"""Return websession object."""
|
|
return self._websession
|
|
|
|
@property
|
|
def websession_ssl(self) -> aiohttp.ClientSession:
|
|
"""Return websession object with disabled SSL."""
|
|
return self._websession_ssl
|
|
|
|
@property
|
|
def config(self) -> CoreConfig:
|
|
"""Return CoreConfig object."""
|
|
return self._config
|
|
|
|
@property
|
|
def hardware(self) -> Hardware:
|
|
"""Return Hardware object."""
|
|
return self._hardware
|
|
|
|
@property
|
|
def docker(self) -> DockerAPI:
|
|
"""Return DockerAPI object."""
|
|
return self._docker
|
|
|
|
@property
|
|
def scheduler(self) -> Scheduler:
|
|
"""Return Scheduler object."""
|
|
if self._scheduler is None:
|
|
raise RuntimeError("Scheduler not set!")
|
|
return self._scheduler
|
|
|
|
@scheduler.setter
|
|
def scheduler(self, value: Scheduler) -> None:
|
|
"""Set a Scheduler object."""
|
|
if self._scheduler:
|
|
raise RuntimeError("Scheduler already set!")
|
|
self._scheduler = value
|
|
|
|
@property
|
|
def core(self) -> Core:
|
|
"""Return core object."""
|
|
if self._core is None:
|
|
raise RuntimeError("Core not set!")
|
|
return self._core
|
|
|
|
@core.setter
|
|
def core(self, value: Core) -> None:
|
|
"""Set a Core object."""
|
|
if self._core:
|
|
raise RuntimeError("Core already set!")
|
|
self._core = value
|
|
|
|
@property
|
|
def plugins(self) -> PluginManager:
|
|
"""Return PluginManager object."""
|
|
if self._plugins is None:
|
|
raise RuntimeError("PluginManager not set!")
|
|
return self._plugins
|
|
|
|
@plugins.setter
|
|
def plugins(self, value: PluginManager) -> None:
|
|
"""Set a PluginManager object."""
|
|
if self._plugins:
|
|
raise RuntimeError("PluginManager already set!")
|
|
self._plugins = value
|
|
|
|
@property
|
|
def arch(self) -> CpuArch:
|
|
"""Return CpuArch object."""
|
|
if self._arch is None:
|
|
raise RuntimeError("CpuArch not set!")
|
|
return self._arch
|
|
|
|
@arch.setter
|
|
def arch(self, value: CpuArch) -> None:
|
|
"""Set a CpuArch object."""
|
|
if self._arch:
|
|
raise RuntimeError("CpuArch already set!")
|
|
self._arch = value
|
|
|
|
@property
|
|
def auth(self) -> Auth:
|
|
"""Return Auth object."""
|
|
if self._auth is None:
|
|
raise RuntimeError("Auth not set!")
|
|
return self._auth
|
|
|
|
@auth.setter
|
|
def auth(self, value: Auth) -> None:
|
|
"""Set a Auth object."""
|
|
if self._auth:
|
|
raise RuntimeError("Auth already set!")
|
|
self._auth = value
|
|
|
|
@property
|
|
def homeassistant(self) -> HomeAssistant:
|
|
"""Return Home Assistant object."""
|
|
if self._homeassistant is None:
|
|
raise RuntimeError("Home Assistant not set!")
|
|
return self._homeassistant
|
|
|
|
@homeassistant.setter
|
|
def homeassistant(self, value: HomeAssistant) -> None:
|
|
"""Set a HomeAssistant object."""
|
|
if self._homeassistant:
|
|
raise RuntimeError("Home Assistant already set!")
|
|
self._homeassistant = value
|
|
|
|
@property
|
|
def supervisor(self) -> Supervisor:
|
|
"""Return Supervisor object."""
|
|
if self._supervisor is None:
|
|
raise RuntimeError("Supervisor not set!")
|
|
return self._supervisor
|
|
|
|
@supervisor.setter
|
|
def supervisor(self, value: Supervisor) -> None:
|
|
"""Set a Supervisor object."""
|
|
if self._supervisor:
|
|
raise RuntimeError("Supervisor already set!")
|
|
self._supervisor = value
|
|
|
|
@property
|
|
def api(self) -> RestAPI:
|
|
"""Return API object."""
|
|
if self._api is None:
|
|
raise RuntimeError("API not set!")
|
|
return self._api
|
|
|
|
@api.setter
|
|
def api(self, value: RestAPI) -> None:
|
|
"""Set an API object."""
|
|
if self._api:
|
|
raise RuntimeError("API already set!")
|
|
self._api = value
|
|
|
|
@property
|
|
def updater(self) -> Updater:
|
|
"""Return Updater object."""
|
|
if self._updater is None:
|
|
raise RuntimeError("Updater not set!")
|
|
return self._updater
|
|
|
|
@updater.setter
|
|
def updater(self, value: Updater) -> None:
|
|
"""Set a Updater object."""
|
|
if self._updater:
|
|
raise RuntimeError("Updater already set!")
|
|
self._updater = value
|
|
|
|
@property
|
|
def addons(self) -> AddonManager:
|
|
"""Return AddonManager object."""
|
|
if self._addons is None:
|
|
raise RuntimeError("AddonManager not set!")
|
|
return self._addons
|
|
|
|
@addons.setter
|
|
def addons(self, value: AddonManager) -> None:
|
|
"""Set a AddonManager object."""
|
|
if self._addons:
|
|
raise RuntimeError("AddonManager already set!")
|
|
self._addons = value
|
|
|
|
@property
|
|
def store(self) -> StoreManager:
|
|
"""Return StoreManager object."""
|
|
if self._store is None:
|
|
raise RuntimeError("StoreManager not set!")
|
|
return self._store
|
|
|
|
@store.setter
|
|
def store(self, value: StoreManager) -> None:
|
|
"""Set a StoreManager object."""
|
|
if self._store:
|
|
raise RuntimeError("StoreManager already set!")
|
|
self._store = value
|
|
|
|
@property
|
|
def snapshots(self) -> SnapshotManager:
|
|
"""Return SnapshotManager object."""
|
|
if self._snapshots is None:
|
|
raise RuntimeError("SnapshotManager not set!")
|
|
return self._snapshots
|
|
|
|
@snapshots.setter
|
|
def snapshots(self, value: SnapshotManager) -> None:
|
|
"""Set a SnapshotManager object."""
|
|
if self._snapshots:
|
|
raise RuntimeError("SnapshotsManager already set!")
|
|
self._snapshots = value
|
|
|
|
@property
|
|
def tasks(self) -> Tasks:
|
|
"""Return Tasks object."""
|
|
if self._tasks is None:
|
|
raise RuntimeError("Tasks not set!")
|
|
return self._tasks
|
|
|
|
@tasks.setter
|
|
def tasks(self, value: Tasks) -> None:
|
|
"""Set a Tasks object."""
|
|
if self._tasks:
|
|
raise RuntimeError("Tasks already set!")
|
|
self._tasks = value
|
|
|
|
@property
|
|
def services(self) -> ServiceManager:
|
|
"""Return ServiceManager object."""
|
|
if self._services is None:
|
|
raise RuntimeError("Services not set!")
|
|
return self._services
|
|
|
|
@services.setter
|
|
def services(self, value: ServiceManager) -> None:
|
|
"""Set a ServiceManager object."""
|
|
if self._services:
|
|
raise RuntimeError("Services already set!")
|
|
self._services = value
|
|
|
|
@property
|
|
def discovery(self) -> Discovery:
|
|
"""Return ServiceManager object."""
|
|
if self._discovery is None:
|
|
raise RuntimeError("Discovery not set!")
|
|
return self._discovery
|
|
|
|
@discovery.setter
|
|
def discovery(self, value: Discovery) -> None:
|
|
"""Set a Discovery object."""
|
|
if self._discovery:
|
|
raise RuntimeError("Discovery already set!")
|
|
self._discovery = value
|
|
|
|
@property
|
|
def dbus(self) -> DBusManager:
|
|
"""Return DBusManager object."""
|
|
if self._dbus is None:
|
|
raise RuntimeError("DBusManager not set!")
|
|
return self._dbus
|
|
|
|
@dbus.setter
|
|
def dbus(self, value: DBusManager) -> None:
|
|
"""Set a DBusManager object."""
|
|
if self._dbus:
|
|
raise RuntimeError("DBusManager already set!")
|
|
self._dbus = value
|
|
|
|
@property
|
|
def host(self) -> HostManager:
|
|
"""Return HostManager object."""
|
|
if self._host is None:
|
|
raise RuntimeError("HostManager not set!")
|
|
return self._host
|
|
|
|
@host.setter
|
|
def host(self, value: HostManager) -> None:
|
|
"""Set a HostManager object."""
|
|
if self._host:
|
|
raise RuntimeError("HostManager already set!")
|
|
self._host = value
|
|
|
|
@property
|
|
def hwmonitor(self) -> HwMonitor:
|
|
"""Return HwMonitor object."""
|
|
if self._hwmonitor is None:
|
|
raise RuntimeError("HwMonitor not set!")
|
|
return self._hwmonitor
|
|
|
|
@hwmonitor.setter
|
|
def hwmonitor(self, value: HwMonitor) -> None:
|
|
"""Set a HwMonitor object."""
|
|
if self._hwmonitor:
|
|
raise RuntimeError("HwMonitor already set!")
|
|
self._hwmonitor = value
|
|
|
|
@property
|
|
def ingress(self) -> Ingress:
|
|
"""Return Ingress object."""
|
|
if self._ingress is None:
|
|
raise RuntimeError("Ingress not set!")
|
|
return self._ingress
|
|
|
|
@ingress.setter
|
|
def ingress(self, value: Ingress) -> None:
|
|
"""Set a Ingress object."""
|
|
if self._ingress:
|
|
raise RuntimeError("Ingress already set!")
|
|
self._ingress = value
|
|
|
|
@property
|
|
def hassos(self) -> HassOS:
|
|
"""Return HassOS object."""
|
|
if self._hassos is None:
|
|
raise RuntimeError("HassOS not set!")
|
|
return self._hassos
|
|
|
|
@hassos.setter
|
|
def hassos(self, value: HassOS) -> None:
|
|
"""Set a HassOS object."""
|
|
if self._hassos:
|
|
raise RuntimeError("HassOS already set!")
|
|
self._hassos = value
|
|
|
|
@property
|
|
def resolution(self) -> ResolutionManager:
|
|
"""Return resolution manager object."""
|
|
if self._resolution is None:
|
|
raise RuntimeError("resolution manager not set!")
|
|
return self._resolution
|
|
|
|
@resolution.setter
|
|
def resolution(self, value: ResolutionManager) -> None:
|
|
"""Set a resolution manager object."""
|
|
if self._resolution:
|
|
raise RuntimeError("resolution manager already set!")
|
|
self._resolution = value
|
|
|
|
@property
|
|
def machine(self) -> Optional[str]:
|
|
"""Return machine type string."""
|
|
return self._machine
|
|
|
|
@machine.setter
|
|
def machine(self, value: str) -> None:
|
|
"""Set a machine type string."""
|
|
if self._machine:
|
|
raise RuntimeError("Machine type already set!")
|
|
self._machine = value
|
|
|
|
@property
|
|
def machine_id(self) -> Optional[str]:
|
|
"""Return machine-id type string."""
|
|
return self._machine_id
|
|
|
|
@machine_id.setter
|
|
def machine_id(self, value: str) -> None:
|
|
"""Set a machine-id type string."""
|
|
if self._machine_id:
|
|
raise RuntimeError("Machine-ID type already set!")
|
|
self._machine_id = value
|
|
|
|
|
|
class CoreSysAttributes:
|
|
"""Inherit basic CoreSysAttributes."""
|
|
|
|
coresys: CoreSys
|
|
|
|
@property
|
|
def sys_machine(self) -> Optional[str]:
|
|
"""Return running machine type of the Supervisor system."""
|
|
return self.coresys.machine
|
|
|
|
@property
|
|
def sys_dev(self) -> bool:
|
|
"""Return True if we run dev mode."""
|
|
return self.coresys.dev
|
|
|
|
@property
|
|
def sys_loop(self) -> asyncio.BaseEventLoop:
|
|
"""Return loop object."""
|
|
return self.coresys.loop
|
|
|
|
@property
|
|
def sys_websession(self) -> aiohttp.ClientSession:
|
|
"""Return websession object."""
|
|
return self.coresys.websession
|
|
|
|
@property
|
|
def sys_websession_ssl(self) -> aiohttp.ClientSession:
|
|
"""Return websession object with disabled SSL."""
|
|
return self.coresys.websession_ssl
|
|
|
|
@property
|
|
def sys_config(self) -> CoreConfig:
|
|
"""Return CoreConfig object."""
|
|
return self.coresys.config
|
|
|
|
@property
|
|
def sys_hardware(self) -> Hardware:
|
|
"""Return Hardware object."""
|
|
return self.coresys.hardware
|
|
|
|
@property
|
|
def sys_docker(self) -> DockerAPI:
|
|
"""Return DockerAPI object."""
|
|
return self.coresys.docker
|
|
|
|
@property
|
|
def sys_scheduler(self) -> Scheduler:
|
|
"""Return Scheduler object."""
|
|
return self.coresys.scheduler
|
|
|
|
@property
|
|
def sys_core(self) -> Core:
|
|
"""Return core object."""
|
|
return self.coresys.core
|
|
|
|
@property
|
|
def sys_plugins(self) -> PluginManager:
|
|
"""Return PluginManager object."""
|
|
return self.coresys.plugins
|
|
|
|
@property
|
|
def sys_arch(self) -> CpuArch:
|
|
"""Return CpuArch object."""
|
|
return self.coresys.arch
|
|
|
|
@property
|
|
def sys_auth(self) -> Auth:
|
|
"""Return Auth object."""
|
|
return self.coresys.auth
|
|
|
|
@property
|
|
def sys_homeassistant(self) -> HomeAssistant:
|
|
"""Return Home Assistant object."""
|
|
return self.coresys.homeassistant
|
|
|
|
@property
|
|
def sys_supervisor(self) -> Supervisor:
|
|
"""Return Supervisor object."""
|
|
return self.coresys.supervisor
|
|
|
|
@property
|
|
def sys_api(self) -> RestAPI:
|
|
"""Return API object."""
|
|
return self.coresys.api
|
|
|
|
@property
|
|
def sys_updater(self) -> Updater:
|
|
"""Return Updater object."""
|
|
return self.coresys.updater
|
|
|
|
@property
|
|
def sys_addons(self) -> AddonManager:
|
|
"""Return AddonManager object."""
|
|
return self.coresys.addons
|
|
|
|
@property
|
|
def sys_store(self) -> StoreManager:
|
|
"""Return StoreManager object."""
|
|
return self.coresys.store
|
|
|
|
@property
|
|
def sys_snapshots(self) -> SnapshotManager:
|
|
"""Return SnapshotManager object."""
|
|
return self.coresys.snapshots
|
|
|
|
@property
|
|
def sys_tasks(self) -> Tasks:
|
|
"""Return Tasks object."""
|
|
return self.coresys.tasks
|
|
|
|
@property
|
|
def sys_services(self) -> ServiceManager:
|
|
"""Return ServiceManager object."""
|
|
return self.coresys.services
|
|
|
|
@property
|
|
def sys_discovery(self) -> Discovery:
|
|
"""Return ServiceManager object."""
|
|
return self.coresys.discovery
|
|
|
|
@property
|
|
def sys_dbus(self) -> DBusManager:
|
|
"""Return DBusManager object."""
|
|
return self.coresys.dbus
|
|
|
|
@property
|
|
def sys_host(self) -> HostManager:
|
|
"""Return HostManager object."""
|
|
return self.coresys.host
|
|
|
|
@property
|
|
def sys_hwmonitor(self) -> HwMonitor:
|
|
"""Return HwMonitor object."""
|
|
return self.coresys.hwmonitor
|
|
|
|
@property
|
|
def sys_ingress(self) -> Ingress:
|
|
"""Return Ingress object."""
|
|
return self.coresys.ingress
|
|
|
|
@property
|
|
def sys_hassos(self) -> HassOS:
|
|
"""Return HassOS object."""
|
|
return self.coresys.hassos
|
|
|
|
@property
|
|
def sys_resolution(self) -> ResolutionManager:
|
|
"""Return Resolution manager object."""
|
|
return self.coresys.resolution
|
|
|
|
def sys_run_in_executor(
|
|
self, funct: Callable[..., T], *args: Any
|
|
) -> Coroutine[Any, Any, T]:
|
|
"""Add an job to the executor pool."""
|
|
return self.sys_loop.run_in_executor(None, funct, *args)
|
|
|
|
def sys_create_task(self, coroutine: Coroutine) -> asyncio.Task:
|
|
"""Create an async task."""
|
|
return self.sys_loop.create_task(coroutine)
|
|
|
|
def sys_capture_exception(self, err: Exception) -> None:
|
|
"""Capture a exception."""
|
|
sentry_sdk.capture_exception(err)
|