ddadminandClaude Opus 4.6
e28232ef50
- Зачем: - обработка нескольких файлов за один вызов с загрузкой модели один раз. - хранение дефолтов (модель, язык, устройство) в .transcriber.toml. - Что: - добавлен config.py: поиск .transcriber.toml (CWD → ~/.config), парсинг, валидация, приоритет CLI > конфиг > хардкод. - рефакторинг transcriber.py: выделены load_model() и _transcribe_file() с TranscribeFileResult для переиспользования модели в батче. - добавлены expand_globs() с дедупликацией и has_existing_transcript() в utils.py. - CLI: files: list[Path], --force/-f, prescan-first батч с итоговой статистикой и временем, Status-спиннер для прогресса. - README: секции батч-режим, конфигурационный файл, --force в таблице опций. - ADR-002: зафиксированы архитектурные решения (prescan-first, TranscribeFileResult, конфиг без мержа). - Проверка: - uv run pytest -q — 94 passed, 1 skipped. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
129 lines
3.6 KiB
Python
129 lines
3.6 KiB
Python
import subprocess
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from local_transcriber.utils import (
|
|
build_output_path,
|
|
detect_device,
|
|
expand_globs,
|
|
get_gpu_name,
|
|
has_existing_transcript,
|
|
validate_input_file,
|
|
)
|
|
|
|
|
|
def test_validate_input_file_not_found(tmp_path):
|
|
missing = tmp_path / "no_such_file.mp3"
|
|
with pytest.raises(FileNotFoundError):
|
|
validate_input_file(missing)
|
|
|
|
|
|
def test_validate_input_file_empty(tmp_path):
|
|
empty = tmp_path / "empty.mp3"
|
|
empty.touch()
|
|
with pytest.raises(ValueError, match="пустой"):
|
|
validate_input_file(empty)
|
|
|
|
|
|
def test_validate_input_file_unknown_ext(tmp_path):
|
|
f = tmp_path / "notes.txt"
|
|
f.write_text("hello")
|
|
with pytest.warns(UserWarning, match="не входит в список"):
|
|
result = validate_input_file(f)
|
|
assert result == f.resolve()
|
|
|
|
|
|
def test_validate_input_file_ok(tmp_path):
|
|
f = tmp_path / "audio.mp3"
|
|
f.write_bytes(b"\x00" * 16)
|
|
result = validate_input_file(f)
|
|
assert result == f.resolve()
|
|
|
|
|
|
def test_build_output_path_default():
|
|
inp = Path("/some/dir/meeting-2026-03-17.mp4")
|
|
result = build_output_path(inp)
|
|
assert result == Path("/some/dir/meeting-2026-03-17-transcript.md")
|
|
|
|
|
|
def test_build_output_path_custom():
|
|
inp = Path("/some/dir/audio.mp3")
|
|
custom = Path("/out/result.md")
|
|
result = build_output_path(inp, output=custom)
|
|
assert result == custom
|
|
|
|
|
|
def test_detect_device_explicit():
|
|
assert detect_device("cpu") == "cpu"
|
|
assert detect_device("cuda") == "cuda"
|
|
|
|
|
|
def test_get_gpu_name_no_nvidia_smi():
|
|
with patch("shutil.which", return_value=None):
|
|
with patch("subprocess.run", side_effect=FileNotFoundError):
|
|
result = get_gpu_name()
|
|
assert result is None
|
|
|
|
|
|
def test_get_gpu_name_empty_stdout():
|
|
"""nvidia-smi returns 0 but stdout is empty — should return None, not crash."""
|
|
mock_result = subprocess.CompletedProcess(
|
|
args=[], returncode=0, stdout="", stderr=""
|
|
)
|
|
with patch("subprocess.run", return_value=mock_result):
|
|
result = get_gpu_name()
|
|
assert result is None
|
|
|
|
|
|
def test_get_gpu_name_success():
|
|
mock_result = subprocess.CompletedProcess(
|
|
args=[], returncode=0, stdout="NVIDIA GeForce RTX 3060\n", stderr=""
|
|
)
|
|
with patch("subprocess.run", return_value=mock_result):
|
|
result = get_gpu_name()
|
|
assert result == "NVIDIA GeForce RTX 3060"
|
|
|
|
|
|
def test_expand_globs_no_patterns(tmp_path):
|
|
a = tmp_path / "a.mp3"
|
|
b = tmp_path / "b.mp3"
|
|
result = expand_globs([a, b])
|
|
assert result == [a, b]
|
|
|
|
|
|
def test_expand_globs_with_star(tmp_path):
|
|
(tmp_path / "x.mp3").write_bytes(b"fake")
|
|
(tmp_path / "y.mp3").write_bytes(b"fake")
|
|
(tmp_path / "z.txt").write_bytes(b"fake")
|
|
result = expand_globs([Path(str(tmp_path / "*.mp3"))])
|
|
assert len(result) == 2
|
|
assert all(p.suffix == ".mp3" for p in result)
|
|
|
|
|
|
def test_expand_globs_no_match(tmp_path):
|
|
result = expand_globs([Path(str(tmp_path / "*.wav"))])
|
|
assert result == []
|
|
|
|
|
|
def test_expand_globs_deduplicates(tmp_path):
|
|
f = tmp_path / "a.mp3"
|
|
f.write_bytes(b"fake")
|
|
result = expand_globs([f, Path(str(tmp_path / "*.mp3"))])
|
|
assert len(result) == 1
|
|
|
|
|
|
def test_has_existing_transcript_true(tmp_path):
|
|
audio = tmp_path / "meeting.mp3"
|
|
audio.write_bytes(b"fake")
|
|
transcript = tmp_path / "meeting-transcript.md"
|
|
transcript.write_text("content")
|
|
assert has_existing_transcript(audio) is True
|
|
|
|
|
|
def test_has_existing_transcript_false(tmp_path):
|
|
audio = tmp_path / "meeting.mp3"
|
|
audio.write_bytes(b"fake")
|
|
assert has_existing_transcript(audio) is False
|