/
/
/
1"""Setup flow for the Storytel provider."""
2
3from __future__ import annotations
4
5from dataclasses import replace
6from typing import TYPE_CHECKING
7
8from music_assistant_models.config_entries import ConfigEntry
9from music_assistant_models.enums import ConfigEntryType
10
11from music_assistant.models.setup_flow import SetupFlowError
12from music_assistant.providers.storytel.constants import (
13 CONF_PASSWORD,
14 CONF_USERNAME,
15)
16
17if TYPE_CHECKING:
18 from music_assistant.models.setup_flow import SetupSession
19
20
21_ENTRIES = (
22 ConfigEntry(key=CONF_USERNAME, type=ConfigEntryType.STRING, required=True),
23 ConfigEntry(key=CONF_PASSWORD, type=ConfigEntryType.SECURE_STRING, required=True),
24)
25
26
27async def run_setup(session: SetupSession) -> None:
28 """Run the setup flow: collect credentials and create the provider."""
29 errors: dict[str, str] | None = None
30 setup_data = dict(session.context.setup_data)
31 while True:
32 entries = [
33 replace(entry, value=setup_data.get(entry.key, entry.value)) for entry in _ENTRIES
34 ]
35 submitted = await session.form(entries, step_id="user", errors=errors, last_step=True)
36 setup_data.update(submitted)
37 try:
38 await session.finish(setup_data)
39 return
40 except SetupFlowError as err:
41 errors = {"base": err.translation_key or str(err)}
42