84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
"""Support to help onboard new users."""
|
|
from typing import TYPE_CHECKING
|
|
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.storage import Store
|
|
from homeassistant.helpers.typing import ConfigType
|
|
from homeassistant.loader import bind_hass
|
|
|
|
from . import views
|
|
from .const import (
|
|
DOMAIN,
|
|
STEP_ANALYTICS,
|
|
STEP_CORE_CONFIG,
|
|
STEP_INTEGRATION,
|
|
STEP_USER,
|
|
STEPS,
|
|
)
|
|
|
|
STORAGE_KEY = DOMAIN
|
|
STORAGE_VERSION = 4
|
|
|
|
|
|
class OnboadingStorage(Store):
|
|
"""Store onboarding data."""
|
|
|
|
async def _async_migrate_func(self, old_major_version, old_minor_version, old_data):
|
|
"""Migrate to the new version."""
|
|
# From version 1 -> 2, we automatically mark the integration step done
|
|
if old_major_version < 2:
|
|
old_data["done"].append(STEP_INTEGRATION)
|
|
if old_major_version < 3:
|
|
old_data["done"].append(STEP_CORE_CONFIG)
|
|
if old_major_version < 4:
|
|
old_data["done"].append(STEP_ANALYTICS)
|
|
return old_data
|
|
|
|
|
|
@bind_hass
|
|
@callback
|
|
def async_is_onboarded(hass: HomeAssistant) -> bool:
|
|
"""Return if Home Assistant has been onboarded."""
|
|
data = hass.data.get(DOMAIN)
|
|
return data is None or data is True
|
|
|
|
|
|
@bind_hass
|
|
@callback
|
|
def async_is_user_onboarded(hass: HomeAssistant) -> bool:
|
|
"""Return if a user has been created as part of onboarding."""
|
|
return async_is_onboarded(hass) or STEP_USER in hass.data[DOMAIN]["done"]
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
|
|
"""Set up the onboarding component."""
|
|
store = OnboadingStorage(hass, STORAGE_VERSION, STORAGE_KEY, private=True)
|
|
if (data := await store.async_load()) is None:
|
|
data = {"done": []}
|
|
|
|
if TYPE_CHECKING:
|
|
assert isinstance(data, dict)
|
|
|
|
if STEP_USER not in data["done"]:
|
|
# Users can already have created an owner account via the command line
|
|
# If so, mark the user step as done.
|
|
has_owner = False
|
|
|
|
for user in await hass.auth.async_get_users():
|
|
if user.is_owner:
|
|
has_owner = True
|
|
break
|
|
|
|
if has_owner:
|
|
data["done"].append(STEP_USER)
|
|
await store.async_save(data)
|
|
|
|
if set(data["done"]) == set(STEPS):
|
|
return True
|
|
|
|
hass.data[DOMAIN] = data
|
|
|
|
await views.async_setup(hass, data, store)
|
|
|
|
return True
|