/
/
/
1"""Setup flow for the AmpliPi 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.amplipi.constants import CONF_HOST
13
14if TYPE_CHECKING:
15 from music_assistant.models.setup_flow import SetupSession
16
17_ENTRIES = (
18 ConfigEntry(
19 key=CONF_HOST,
20 type=ConfigEntryType.STRING,
21 required=True,
22 ),
23)
24
25
26async def run_setup(session: SetupSession) -> None:
27 """Run the setup flow: collect the connection details and create the provider."""
28 errors: dict[str, str] | None = None
29 setup_data = dict(session.context.setup_data)
30 while True:
31 entries = [
32 replace(entry, value=setup_data.get(entry.key, entry.value)) for entry in _ENTRIES
33 ]
34 submitted = await session.form(entries, step_id="user", errors=errors, last_step=True)
35 setup_data.update(submitted)
36 try:
37 await session.finish(setup_data)
38 return
39 except SetupFlowError as err:
40 errors = {"base": err.translation_key or str(err)}
41