savetest-cli 0.3.0rc1__tar.gz → 0.3.0rc2__tar.gz

{savetest_cli-0.3.0rc1/savetest_cli.egg-info → savetest_cli-0.3.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: savetest-cli
-Version: 0.3.0rc1
+Version: 0.3.0rc2
 Summary: CLI for SaveTest: upload Allure results as a ZIP, as raw files, or by watching a directory
 License-Expression: MIT
 Requires-Python: >=3.8
@@ -220,7 +220,7 @@ savetest watch ./allure-results \
 - Если тесты и загрузка — разные джобы, отдайте каталог `allure-results` артефактом тоже с `when: always`.
 - `--wait` оставляйте включённым: джоба должна завершиться только когда импорт дошёл до конца.
 - Вложения включайте только если они реально нужны. По умолчанию CLI их не шлёт.
-- `0.3.0rc1` — предварительная версия для SaveTest 1.5.0: `pip install savetest-cli==0.3.0rc1`. Обычный `pip install savetest-cli` остаётся на последней стабильной версии.
+- `0.3.0rc2` — предварительная версия для SaveTest 1.5.0: `pip install savetest-cli==0.3.0rc2`. Обычный `pip install savetest-cli` остаётся на последней стабильной версии.
 
 ### Результаты по мере прогона
 

{savetest_cli-0.3.0rc1 → savetest_cli-0.3.0rc2}/README.md

@@ -210,7 +210,7 @@ savetest watch ./allure-results \
 - Если тесты и загрузка — разные джобы, отдайте каталог `allure-results` артефактом тоже с `when: always`.
 - `--wait` оставляйте включённым: джоба должна завершиться только когда импорт дошёл до конца.
 - Вложения включайте только если они реально нужны. По умолчанию CLI их не шлёт.
-- `0.3.0rc1` — предварительная версия для SaveTest 1.5.0: `pip install savetest-cli==0.3.0rc1`. Обычный `pip install savetest-cli` остаётся на последней стабильной версии.
+- `0.3.0rc2` — предварительная версия для SaveTest 1.5.0: `pip install savetest-cli==0.3.0rc2`. Обычный `pip install savetest-cli` остаётся на последней стабильной версии.
 
 ### Результаты по мере прогона
 

{savetest_cli-0.3.0rc1 → savetest_cli-0.3.0rc2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "savetest-cli"
-version = "0.3.0rc1"
+version = "0.3.0rc2"
 description = "CLI for SaveTest: upload Allure results as a ZIP, as raw files, or by watching a directory"
 readme = "README.md"
 requires-python = ">=3.8"

savetest_cli-0.3.0rc2/savetest_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.0rc2"

{savetest_cli-0.3.0rc1 → savetest_cli-0.3.0rc2}/savetest_cli/cli.py

@@ -22,6 +22,7 @@ DEFAULT_POLL_INTERVAL_SEC = 1.5
 DEFAULT_WAIT_TIMEOUT_SEC = 3600
 DEFAULT_FILE_BATCH_COUNT = 20
 DEFAULT_FILE_BATCH_BYTES = 32 * 1024 * 1024
+DEFAULT_WATCH_BATCH_COUNT = 200
 DEFAULT_WATCH_INTERVAL_SEC = 0.5
 DEFAULT_SETTLE_SEC = 1.0
 DEFAULT_ATTACHMENT_WAIT_SEC = 5.0
@@ -868,14 +869,28 @@ def _snapshot_raw_files(root: Path, include_attachments: bool) -> Dict[str, Tupl
     return found
 
 
+def _index_snapshot_names(
+    snapshot: Dict[str, Tuple[Path, int, int]],
+) -> Dict[str, List[str]]:
+    """Имя файла → пути. Сопоставление вложений не должно перебирать весь каталог."""
+    index: Dict[str, List[str]] = {}
+    for rel in snapshot:
+        index.setdefault(Path(rel).name, []).append(rel)
+    return index
+
+
 def _resolve_named_file(
     snapshot: Dict[str, Tuple[Path, int, int]],
     result_rel: str,
     source_name: str,
+    name_index: Optional[Dict[str, List[str]]] = None,
 ) -> Optional[str]:
     name = Path(source_name).name
     result_dir = str(Path(result_rel).parent)
-    matches = [rel for rel in snapshot if Path(rel).name == name]
+    if name_index is None:
+        matches = [rel for rel in snapshot if Path(rel).name == name]
+    else:
+        matches = name_index.get(name) or []
     if not matches:
         return None
     for rel in matches:
@@ -921,6 +936,7 @@ def _pick_watch_batch(
             continue
         ready[rel] = path
 
+    name_index = _index_snapshot_names(snapshot)
     held = set()
     if include_attachments:
         for rel, path in list(ready.items()):
@@ -929,7 +945,7 @@ def _pick_watch_batch(
             document = _load_json_if_complete(path) or {}
             missing = False
             for source in _attachment_sources(document):
-                attachment_rel = _resolve_named_file(snapshot, rel, source)
+                attachment_rel = _resolve_named_file(snapshot, rel, source, name_index)
                 if attachment_rel is None or attachment_rel not in ready:
                     missing = True
                     break
@@ -952,7 +968,7 @@ def _pick_watch_batch(
     for rel, path in send_results:
         document = _load_json_if_complete(path) or {}
         for source in _attachment_sources(document):
-            attachment_rel = _resolve_named_file(snapshot, rel, source)
+            attachment_rel = _resolve_named_file(snapshot, rel, source, name_index)
             if attachment_rel and attachment_rel in ready:
                 needed.add(attachment_rel)
 
@@ -1005,6 +1021,9 @@ def _chunk_watch_selection(
     """Режет готовые файлы на пачки. Результат не отрывается от своих вложений."""
     results = [(rel, path) for rel, path in selected if _is_result_file(path)]
     extras = {rel: (rel, path) for rel, path in selected if not _is_result_file(path)}
+    extras_by_name: Dict[str, List[str]] = {}
+    for extra_rel in extras:
+        extras_by_name.setdefault(Path(extra_rel).name, []).append(extra_rel)
     used = set()
     groups: List[List[Tuple[str, Path]]] = []
     for rel, path in results:
@@ -1012,10 +1031,10 @@ def _chunk_watch_selection(
         document = _load_json_if_complete(path) or {}
         for source in _attachment_sources(document):
             name = Path(source).name
-            for extra_rel, extra in extras.items():
-                if extra_rel in used or Path(extra_rel).name != name:
+            for extra_rel in extras_by_name.get(name) or []:
+                if extra_rel in used:
                     continue
-                group.append(extra)
+                group.append(extras[extra_rel])
                 used.add(extra_rel)
                 break
         groups.append(group)
@@ -1193,8 +1212,8 @@ def watch(
 
     Готовые файлы не уходят по одному. Если они появляются подряд, watch держит их
     --batch-window секунд после последнего нового и отправляет вместе. Дольше
-    --batch-max-wait секунд пачка не копится. Предел одной отправки — 20 файлов
-    или 32 МБ, как у upload_results.
+    --batch-max-wait секунд пачка не копится. Предел одной отправки — 200 файлов
+    или 32 МБ.
     """
     command = [arg for arg in ctx.args if arg != "--"]
     host = host.rstrip("/")
@@ -1209,6 +1228,7 @@ def watch(
     run_create_failed = False
     sent_batches = 0
     drain_until = 0.0
+    announced_flush = False
     hold_opened = 0.0
     hold_quiet = 0.0
     hold_key: Optional[Tuple[str, ...]] = None
@@ -1267,6 +1287,10 @@ def watch(
             flushing = stop["requested"] or timed_out
 
             snapshot = _snapshot_raw_files(root, include_attachments) if root.is_dir() else {}
+            if flushing and not announced_flush:
+                pending = sum(1 for rel in snapshot if rel not in uploaded)
+                click.echo(f"После остановки осталось файлов: {pending}.")
+                announced_flush = True
             if any(rel not in uploaded for rel in snapshot):
                 last_activity = now
 
@@ -1300,7 +1324,7 @@ def watch(
                     hold_key = ready_key
 
             batch_full = bool(batch) and (
-                len(batch) >= DEFAULT_FILE_BATCH_COUNT
+                len(batch) >= DEFAULT_WATCH_BATCH_COUNT
                 or _selection_bytes(batch) >= DEFAULT_FILE_BATCH_BYTES
             )
             send_now = bool(batch) and _watch_batch_due(
@@ -1313,7 +1337,10 @@ def watch(
                 flushing=flushing,
             )
             if send_now and not run_create_failed and (known_run_id or not inflight):
-                chunks = _chunk_watch_selection(batch)
+                chunks = _chunk_watch_selection(
+                    batch,
+                    max_count=DEFAULT_WATCH_BATCH_COUNT,
+                )
                 if not known_run_id:
                     chunks = chunks[:1]
                 for chunk in chunks:

{savetest_cli-0.3.0rc1 → savetest_cli-0.3.0rc2/savetest_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: savetest-cli
-Version: 0.3.0rc1
+Version: 0.3.0rc2
 Summary: CLI for SaveTest: upload Allure results as a ZIP, as raw files, or by watching a directory
 License-Expression: MIT
 Requires-Python: >=3.8
@@ -220,7 +220,7 @@ savetest watch ./allure-results \
 - Если тесты и загрузка — разные джобы, отдайте каталог `allure-results` артефактом тоже с `when: always`.
 - `--wait` оставляйте включённым: джоба должна завершиться только когда импорт дошёл до конца.
 - Вложения включайте только если они реально нужны. По умолчанию CLI их не шлёт.
-- `0.3.0rc1` — предварительная версия для SaveTest 1.5.0: `pip install savetest-cli==0.3.0rc1`. Обычный `pip install savetest-cli` остаётся на последней стабильной версии.
+- `0.3.0rc2` — предварительная версия для SaveTest 1.5.0: `pip install savetest-cli==0.3.0rc2`. Обычный `pip install savetest-cli` остаётся на последней стабильной версии.
 
 ### Результаты по мере прогона
 

savetest_cli-0.3.0rc1/savetest_cli/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.3.0rc1"