contree-cli 0.7.0__tar.gz → 0.7.1__tar.gz

{contree_cli-0.7.0 → contree_cli-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: contree-cli
-Version: 0.7.0
+Version: 0.7.1
 Summary: CLI client for the ConTree sandbox platform
 Keywords: sandbox,cli,vm,ai-agent,contree
 Author: Dmitry Orlov

{contree_cli-0.7.0 → contree_cli-0.7.1}/contree_cli/cli/file.py

@@ -328,37 +328,35 @@ def cmd_file_ls(args: FileListArgs) -> int | None:
     if args.until is not None:
         params["until"] = isoformat_datetime(args.until)
 
-    fetcher = PaginatedFetcher(
+    emitted = 0
+    hit_limit = False
+    with PaginatedFetcher(
         client,
         "/v1/files",
         params,
         lambda body: json.loads(body).get("files", []),
         limit=args.limit,
         concurrency=CONTREE_CONCURRENCY,
-    )
-
-    emitted = 0
-    hit_limit = False
-    for page in fetcher:
-        for entry in page:
-            if emitted >= args.limit:
-                hit_limit = True
+    ) as fetcher:
+        for page in fetcher:
+            for entry in page:
+                if emitted >= args.limit:
+                    hit_limit = True
+                    break
+                uuid_str = entry.get("uuid")
+                source = sources.get(uuid_str, "") if isinstance(uuid_str, str) else ""
+                if args.quiet:
+                    formatter(
+                        uuid=uuid_str,
+                        sha256=entry.get("sha256", ""),
+                        source=source,
+                    )
+                else:
+                    formatter(**{**entry, "source": source})
+                emitted += 1
+            formatter.flush()
+            if hit_limit:
                 break
-            uuid_str = entry.get("uuid")
-            source = sources.get(uuid_str, "") if isinstance(uuid_str, str) else ""
-            if args.quiet:
-                formatter(
-                    uuid=uuid_str,
-                    sha256=entry.get("sha256", ""),
-                    source=source,
-                )
-            else:
-                formatter(**{**entry, "source": source})
-            emitted += 1
-        formatter.flush()
-        if hit_limit:
-            fetcher.stop()
-            break
 
     if hit_limit:
         logger.warning(

{contree_cli-0.7.0 → contree_cli-0.7.1}/contree_cli/cli/images.py

@@ -272,28 +272,26 @@ def cmd_images(args: ImagesArgs) -> None:
     if args.until is not None:
         base_params["until"] = isoformat_datetime(args.until)
 
-    fetcher = PaginatedFetcher(
+    emitted = 0
+    hit_limit = False
+    with PaginatedFetcher(
         client,
         "/v1/images",
         base_params,
         lambda body: json.loads(body)["images"],
         limit=args.limit,
         concurrency=CONTREE_CONCURRENCY,
-    )
-
-    emitted = 0
-    hit_limit = False
-    for page in fetcher:
-        for image in page:
-            if emitted >= args.limit:
-                hit_limit = True
+    ) as fetcher:
+        for page in fetcher:
+            for image in page:
+                if emitted >= args.limit:
+                    hit_limit = True
+                    break
+                formatter(**image)
+                emitted += 1
+            formatter.flush()
+            if hit_limit:
                 break
-            formatter(**image)
-            emitted += 1
-        formatter.flush()
-        if hit_limit:
-            fetcher.stop()
-            break
 
     if hit_limit:
         logger.warning(

{contree_cli-0.7.0 → contree_cli-0.7.1}/contree_cli/cli/operation.py

@@ -393,31 +393,29 @@ def cmd_list(args: ListArgs) -> None:
         base_params["until"] = isoformat_datetime(args.until)
 
     limit = args.show_max
-    fetcher = PaginatedFetcher(
+    emitted = 0
+    hit_limit = False
+    with PaginatedFetcher(
         client,
         "/v1/operations",
         base_params,
         json.loads,
         limit=limit,
         concurrency=CONTREE_CONCURRENCY,
-    )
-
-    emitted = 0
-    hit_limit = False
-    for page in fetcher:
-        for op in page:
-            if limit is not None and emitted >= limit:
-                hit_limit = True
+    ) as fetcher:
+        for page in fetcher:
+            for op in page:
+                if limit is not None and emitted >= limit:
+                    hit_limit = True
+                    break
+                if args.quiet:
+                    print(op["uuid"])
+                else:
+                    formatter(**op)
+                emitted += 1
+            formatter.flush()
+            if hit_limit:
                 break
-            if args.quiet:
-                print(op["uuid"])
-            else:
-                formatter(**op)
-            emitted += 1
-        formatter.flush()
-        if hit_limit:
-            fetcher.stop()
-            break
 
     if hit_limit:
         logger.warning(

{contree_cli-0.7.0 → contree_cli-0.7.1}/contree_cli/client.py

@@ -619,6 +619,17 @@ class PaginatedFetcher:
         """Signal that the caller has seen enough; skip pending fetches."""
         self._stop.set()
 
+    def __enter__(self) -> PaginatedFetcher:
+        return self
+
+    def __exit__(self, *_: object) -> None:
+        # Setting the stop event short-circuits any worker that hasn't
+        # started yet and prevents the iterator's refill from enqueueing
+        # more fetches. Callers wrap iteration in `with PaginatedFetcher(...)`
+        # so they don't have to remember an explicit `stop()` after
+        # breaking out of a paged loop.
+        self.stop()
+
     def _fetch(self, offset: int) -> list[dict[str, Any]]:
         if self._stop.is_set():
             return []

{contree_cli-0.7.0 → contree_cli-0.7.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "contree-cli"
-version = "0.7.0"
+version = "0.7.1"
 description = "CLI client for the ConTree sandbox platform"
 readme = "README.md"
 requires-python = ">=3.10"