fix(cli): добавлен пользовательский вывод предупреждений

- Зачем:
  - предупреждения из задачи #7 выглядели как фрагменты Python traceback.
- Что:
  - на время запуска CLI установлен однострочный Rich-рендер предупреждений.
  - сохранены буквальный текст сообщений и восстановление глобального hook.
  - single- и batch-сценарии покрыты автоматическим тестом.
- Проверка:
  - uv run pytest tests/test_cli.py -q.
This commit is contained in:
Dmitry Dementiev committed 2026-08-29 01:20:24 +03:00
1 parent f966d4554b
commit b412b73fc0
2 files changed
+71

No files matched your search

+27
View File
@@ -2,6 +2,9 @@
import sys
import time
import warnings
from collections.abc import Callable
from functools import wraps
from pathlib import Path
import typer
@@ -56,6 +59,29 @@ app = typer.Typer()
console = Console(stderr=True)
def _show_cli_warning(
message: Warning | str,
*_args: object,
**_kwargs: object,
) -> None:
"""Печатает предупреждение в пользовательском формате CLI."""
console.print(f"Внимание: {message}", style="yellow", soft_wrap=True, markup=False)
def _with_cli_warning_renderer(
command: Callable[..., None],
) -> Callable[..., None]:
"""Устанавливает CLI-рендер предупреждений на время одного запуска."""
@wraps(command)
def wrapped(*args: object, **kwargs: object) -> None:
with warnings.catch_warnings():
warnings.showwarning = _show_cli_warning
command(*args, **kwargs)
return wrapped
def _format_device_info(device_used: str) -> str:
"""Формирует строку устройства для шапки транскрипта."""
if device_used == "cuda":
@@ -193,6 +219,7 @@ def _print_diarization_report(
@app.command()
@_with_cli_warning_renderer
def main(
files: list[Path] | None = typer.Argument(None, help="Пути к аудио/видеофайлам"),
model: str | None = typer.Option(
+44
View File
@@ -1,3 +1,4 @@
import warnings
from pathlib import Path
from unittest.mock import MagicMock, patch
@@ -118,6 +119,49 @@ def test_cli_happy_path_exit_code_zero(tmp_path):
assert out.exit_code == 0
@pytest.mark.parametrize("file_count", [1, 2])
def test_cli_renders_runtime_warning_without_python_details(tmp_path, file_count):
files = [tmp_path / f"test-{index}.mp3" for index in range(file_count)]
for file in files:
file.write_bytes(b"fake")
result = _make_result()
model = _make_model()
backend = _make_backend()
tfr = _make_tfr(result=result, model=model, backend=backend)
warning_message = (
"Тестовое [bold]предупреждение[/bold] с длинным текстом, который "
"должен остаться одной логической строкой без служебных подробностей Python"
)
def transcribe_with_warning(**_kwargs):
warnings.warn(warning_message, stacklevel=2)
return tfr
original_showwarning = warnings.showwarning
with (
patch("local_transcriber.cli.load_config", return_value={}),
patch("local_transcriber.cli.validate_input_file", side_effect=lambda p: p),
patch("local_transcriber.cli.detect_device", return_value="cpu"),
patch(
"local_transcriber.cli.load_model",
return_value=(model, "cpu", backend, "/models/medium"),
),
patch(
"local_transcriber.cli._transcribe_file",
side_effect=transcribe_with_warning,
),
patch("local_transcriber.cli.write_transcript"),
):
out = runner.invoke(app, [str(file) for file in files])
warning_lines = [line for line in out.output.splitlines() if "Тестовое" in line]
assert warning_lines == [f"Внимание: {warning_message}"]
assert "UserWarning" not in out.output
assert "warnings.warn" not in out.output
assert warnings.showwarning is original_showwarning
def test_cli_default_options_passed_to_transcribe(tmp_path):
audio = tmp_path / "test.mp3"
audio.write_bytes(b"fake")