fix prompt

Author: p141592

avito/cli/accounts.py

@@ -8,6 +8,7 @@
 
 import click
 
+from avito.cli.click_params import RequiredPromptArgument, RequiredPromptOption
 from avito.cli.config import (
     AccountsDocument,
     AccountStore,
@@ -33,9 +34,32 @@ def account_group() -> None:
     """Управлять локальными учетными записями."""
 
 
+class _AuthRequiredPromptOption(RequiredPromptOption):
+    """Обязательный option с учетом глобального запрета интерактивного ввода."""
+
+    def _missing_value_error(self) -> click.ClickException:
+        """Создать auth-ошибку отсутствующего значения для non-interactive режима."""
+
+        return CliAuthRequiredError(
+            f"{self.human_readable_name} не передан, а интерактивный ввод отключен.",
+        )
+
+
 @account_group.command("add")
-@click.argument("account_name", metavar="ACCOUNT-NAME")
-@click.option("--client-id", required=True, metavar="CLIENT-ID", help="Client ID учетной записи.")
+@click.argument(
+    "account_name",
+    cls=RequiredPromptArgument,
+    prompt="Имя учетной записи",
+    metavar="ACCOUNT-NAME",
+)
+@click.option(
+    "--client-id",
+    cls=_AuthRequiredPromptOption,
+    required=True,
+    prompt="Client ID",
+    metavar="CLIENT-ID",
+    help="Client ID учетной записи.",
+)
 @click.option(
     "--client-secret",
     metavar="CLIENT-SECRET",
@@ -144,7 +168,12 @@ def list_accounts(ctx: CliContext) -> None:
 
 
 @account_group.command("use")
-@click.argument("account_name", metavar="ACCOUNT-NAME")
+@click.argument(
+    "account_name",
+    cls=RequiredPromptArgument,
+    prompt="Имя учетной записи",
+    metavar="ACCOUNT-NAME",
+)
 @click.pass_obj
 def use_account(ctx: CliContext, account_name: str) -> None:
     """Сделать учетную запись активной."""
@@ -203,7 +232,12 @@ def current_account(ctx: CliContext) -> None:
 
 
 @account_group.command("delete")
-@click.argument("account_name", metavar="ACCOUNT-NAME")
+@click.argument(
+    "account_name",
+    cls=RequiredPromptArgument,
+    prompt="Имя учетной записи",
+    metavar="ACCOUNT-NAME",
+)
 @click.option("--yes", is_flag=True, help="Удалить без интерактивного подтверждения.")
 @click.option("--confirm", metavar="ACCOUNT-NAME", help="Подтвердить имя удаляемой учетной записи.")
 @click.pass_obj

avito/cli/app.py

@@ -10,6 +10,7 @@
 
 from avito.cli import commands as api_commands
 from avito.cli.accounts import account_group
+from avito.cli.click_params import RequiredPromptOption
 from avito.cli.context import CliContext
 from avito.cli.errors import CliUsageError, InvalidFlagCombinationError
 from avito.cli.help import render_registry_help
@@ -274,10 +275,11 @@ def _parameter_click_options(
     """Преобразовать registry parameter records в Click options."""
 
     return [
-        click.Option(
+        RequiredPromptOption(
             param_decls=(parameter.flag,),
             multiple=parameter.multiple,
-            required=False,
+            required=parameter.required,
+            prompt=_prompt_label(parameter.name) if parameter.required else False,
             metavar="VALUE",
             help=f"Параметр SDK `{parameter.name}`.",
         )
@@ -336,6 +338,12 @@ def _optional_string(value: object) -> str | None:
     return str(value)
 
 
+def _prompt_label(name: str) -> str:
+    """Вернуть человекочитаемую подпись prompt для CLI-параметра."""
+
+    return name.replace("_", " ")
+
+
 def _raw_values_from_click(raw_options: dict[str, object]) -> dict[str, tuple[str, ...]]:
     """Преобразовать Click kwargs в raw CLI values для coercion."""
 

avito/cli/click_params.py

@@ -0,0 +1,64 @@
+"""Click-параметры с единым интерактивным поведением."""
+
+from __future__ import annotations
+
+import collections.abc as cabc
+from typing import Any
+
+import click
+
+from avito.cli.context import CliContext
+from avito.cli.errors import CliUsageError, CliValidationError
+
+
+class RequiredPromptOption(click.Option):
+    """Обязательный option, который спрашивает значение до callback."""
+
+    def prompt_for_value(self, ctx: click.Context) -> object:
+        """Запросить значение через Click prompt, если интерактивный ввод разрешен."""
+
+        if _no_input(ctx):
+            raise self._missing_value_error()
+        if self.multiple:
+            prompt = self.prompt or self.name or "value"
+            return (click.prompt(prompt, type=str),)
+        return super().prompt_for_value(ctx)
+
+    def _missing_value_error(self) -> click.ClickException:
+        """Создать ошибку отсутствующего значения для non-interactive режима."""
+
+        flag = self.opts[0] if self.opts else self.human_readable_name
+        return CliValidationError(
+            f"Не указан обязательный параметр {flag}. Интерактивный ввод отключен.",
+            details={"parameter": self.name},
+        )
+
+
+class RequiredPromptArgument(click.Argument):
+    """Обязательный argument, который спрашивает значение до callback."""
+
+    def __init__(self, param_decls: cabc.Sequence[str], **attrs: Any) -> None:
+        """Сохранить prompt label и инициализировать Click argument."""
+
+        self.prompt = str(attrs.pop("prompt"))
+        super().__init__(param_decls, **attrs)
+
+    def process_value(self, ctx: click.Context, value: object) -> object:
+        """Подставить интерактивно введенное значение, если argument не передан."""
+
+        if self.required and (value == () or self.value_is_missing(value)):
+            if _no_input(ctx):
+                raise CliUsageError(
+                    f"Не указан обязательный аргумент {self.human_readable_name}. "
+                    "Интерактивный ввод отключен.",
+                    details={"parameter": self.name},
+                )
+            value = click.prompt(self.prompt, type=str)
+        return super().process_value(ctx, value)
+
+
+def _no_input(ctx: click.Context) -> bool:
+    """Вернуть глобальный запрет интерактивного ввода из Click context."""
+
+    cli_context = ctx.find_object(CliContext)
+    return cli_context is not None and cli_context.no_input

avito/cli/local.py

@@ -10,6 +10,7 @@
 
 import click
 
+from avito.cli.click_params import RequiredPromptArgument
 from avito.cli.config import (
     AccountStore,
     CliConfigDocument,
@@ -34,7 +35,7 @@ def config_group() -> None:
 
 
 @config_group.command("get")
-@click.argument("key", metavar="KEY")
+@click.argument("key", cls=RequiredPromptArgument, prompt="Ключ", metavar="KEY")
 @click.option("--show-source", is_flag=True, help="Показать источник значения.")
 @click.pass_obj
 def config_get(ctx: CliContext, key: str, show_source: bool) -> None:
@@ -52,8 +53,8 @@ def config_get(ctx: CliContext, key: str, show_source: bool) -> None:
 
 
 @config_group.command("set")
-@click.argument("key", metavar="KEY")
-@click.argument("value", metavar="VALUE")
+@click.argument("key", cls=RequiredPromptArgument, prompt="Ключ", metavar="KEY")
+@click.argument("value", cls=RequiredPromptArgument, prompt="Значение", metavar="VALUE")
 @click.pass_obj
 def config_set(ctx: CliContext, key: str, value: str) -> None:
     """Сохранить значение локальной конфигурации."""
@@ -73,7 +74,7 @@ def config_set(ctx: CliContext, key: str, value: str) -> None:
 
 
 @config_group.command("unset")
-@click.argument("key", metavar="KEY")
+@click.argument("key", cls=RequiredPromptArgument, prompt="Ключ", metavar="KEY")
 @click.pass_obj
 def config_unset(ctx: CliContext, key: str) -> None:
     """Удалить значение локальной конфигурации."""

tests/cli/test_accounts.py

@@ -114,6 +114,52 @@ def test_account_add_no_input_without_secret_fails_without_prompt(tmp_path: Path
     assert "AUTH_REQUIRED" in result.stderr
 
 
+def test_account_add_prompts_for_missing_client_id(tmp_path: Path) -> None:
+    result = _runner(tmp_path).invoke(
+        app,
+        ["account", "add", "main", "--client-secret", "client-secret"],
+        input="prompt-client-id\n",
+    )
+
+    assert result.exit_code == 0
+    assert "Client ID" in result.output
+    assert "prompt-client-id" in result.output
+
+
+def test_account_add_prompts_for_missing_account_name(tmp_path: Path) -> None:
+    result = _runner(tmp_path).invoke(
+        app,
+        ["account", "add", "--client-id", "client-id", "--client-secret", "client-secret"],
+        input="main\n",
+    )
+
+    assert result.exit_code == 0
+    assert "Имя учетной записи" in result.output
+    assert "Учетная запись добавлена: main" in result.output
+
+
+def test_account_add_no_input_without_client_id_fails_without_prompt(tmp_path: Path) -> None:
+    result = _runner(tmp_path).invoke(
+        app,
+        ["--no-input", "account", "add", "main", "--client-secret", "client-secret"],
+    )
+
+    assert result.exit_code == 4
+    assert "AUTH_REQUIRED" in result.stderr
+    assert "интерактивный ввод отключен" in result.stderr
+
+
+def test_account_add_no_input_without_account_name_fails_without_prompt(tmp_path: Path) -> None:
+    result = _runner(tmp_path).invoke(
+        app,
+        ["--no-input", "account", "add", "--client-id", "client-id", "--client-secret", "client-secret"],
+    )
+
+    assert result.exit_code == 2
+    assert "CLI_USAGE_ERROR" in result.stderr
+    assert "Интерактивный ввод отключен" in result.stderr
+
+
 def test_account_add_accepts_ticket_aliases_api_key_and_endpoint(tmp_path: Path) -> None:
     runner = _runner(tmp_path)
 

tests/cli/test_config_commands.py

@@ -52,5 +52,25 @@ def test_config_rejects_unknown_key(tmp_path: Path) -> None:
     assert "не поддерживается" in result.stderr
 
 
+def test_config_set_prompts_for_missing_required_arguments(tmp_path: Path) -> None:
+    runner = _runner(tmp_path)
+
+    set_result = runner.invoke(app, ["config", "set"], input="active-profile\nmain\n")
+    get_result = runner.invoke(app, ["config", "get", "active-profile"])
+
+    assert set_result.exit_code == 0
+    assert "Ключ" in set_result.output
+    assert "Значение" in set_result.output
+    assert get_result.stdout.strip() == "main"
+
+
+def test_config_get_no_input_without_key_fails_without_prompt(tmp_path: Path) -> None:
+    result = _runner(tmp_path).invoke(app, ["--no-input", "config", "get"])
+
+    assert result.exit_code == 2
+    assert "CLI_USAGE_ERROR" in result.stderr
+    assert "Интерактивный ввод отключен" in result.stderr
+
+
 def _runner(tmp_path: Path) -> CliRunner:
     return CliRunner(env={"AVITO_PY_HOME": str(tmp_path / "home")})

tests/cli/test_domain_smoke_commands.py

@@ -83,6 +83,53 @@ def test_every_read_factory_has_at_least_one_smoke_command() -> None:
     assert smoked_factories == read_factories
 
 
+def test_api_command_prompts_for_missing_required_option(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    command = _api_command("account.get-operations-history")
+    fake = SwaggerFakeTransport(registry=load_swagger_registry())
+    fake.add_success_operation(command.operation_key)
+    _install_fake_client(monkeypatch, fake)
+    _write_account(tmp_path)
+    args = [
+        "--profile",
+        "main",
+        command.resource,
+        command.action,
+        *_cli_args_except(command, "date_from"),
+    ]
+
+    result = CliRunner(env={"AVITO_PY_HOME": str(tmp_path)}).invoke(
+        app,
+        args,
+        input="2026-05-01T00:00:00+00:00\n",
+    )
+
+    assert result.exit_code == 0, result.output
+    assert "date from" in result.output
+    assert fake.count() >= 1
+
+
+def test_api_command_no_input_without_required_option_fails_without_prompt(tmp_path: Path) -> None:
+    command = _api_command("account.get-operations-history")
+    _write_account(tmp_path)
+    args = [
+        "--profile",
+        "main",
+        "--no-input",
+        command.resource,
+        command.action,
+        *_cli_args_except(command, "date_from"),
+    ]
+
+    result = CliRunner(env={"AVITO_PY_HOME": str(tmp_path)}).invoke(app, args)
+
+    assert result.exit_code == 7
+    assert "VALIDATION_FAILED" in result.stderr
+    assert "Интерактивный ввод отключен" in result.stderr
+
+
 @pytest.mark.parametrize("command", _WRITE_COMMANDS, ids=lambda command: command.command_id)
 def test_write_api_command_is_registered_and_renders_help(
     command: ApiCommandRecord,
@@ -172,6 +219,22 @@ def _cli_args(command: ApiCommandRecord) -> tuple[str, ...]:
     return tuple(args)
 
 
+def _cli_args_except(command: ApiCommandRecord, skipped_parameter: str) -> tuple[str, ...]:
+    args: list[str] = []
+    for parameter in command.parameters:
+        if parameter.name != skipped_parameter:
+            args.extend((parameter.flag, _value_for_parameter(parameter)))
+    return tuple(args)
+
+
+def _api_command(command_id: str) -> ApiCommandRecord:
+    matches = tuple(
+        command for command in build_cli_registry().api_commands if command.command_id == command_id
+    )
+    assert len(matches) == 1
+    return matches[0]
+
+
 def _value_for_parameter(parameter: CliParameterSchema) -> str:
     if parameter.value_kind == "list":
         item_kind = parameter.item_value_kind or "string"