primitive 0.2.8__tar.gz → 0.2.10__tar.gz

{primitive-0.2.8 → primitive-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: primitive
-Version: 0.2.8
+Version: 0.2.10
 Project-URL: Documentation, https://github.com//primitivecorp/primitive-cli#readme
 Project-URL: Issues, https://github.com//primitivecorp/primitive-cli/issues
 Project-URL: Source, https://github.com//primitivecorp/primitive-cli

{primitive-0.2.8 → primitive-0.2.10}/src/primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@primitive.tech>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.2.8"
+__version__ = "0.2.10"

{primitive-0.2.8 → primitive-0.2.10}/src/primitive/agent/runner.py

@@ -1,16 +1,18 @@
+import asyncio
 import os
 import re
-import typing
 import shutil
-from typing import Dict, TypedDict, List
+import time
+import typing
 from abc import abstractmethod
-from enum import IntEnum, Enum
+from enum import Enum, IntEnum
 from pathlib import Path, PurePath
-from loguru import logger
+from typing import Dict, List, TypedDict
+
 import yaml
-import asyncio
-import time
-from ..utils.cache import get_artifacts_cache, get_sources_cache, get_logs_cache
+from loguru import logger
+
+from ..utils.cache import get_artifacts_cache, get_logs_cache, get_sources_cache
 from ..utils.shell import env_to_dict
 
 try:
@@ -168,10 +170,61 @@ class Runner:
         else:
             logger.success(f"Completed {self.job['slug']} job")
 
+        number_of_files_produced = self.get_number_of_files_produced()
+        logger.info(
+            f"Produced {number_of_files_produced} files for {self.job['slug']} job"
+        )
         self.primitive.jobs.job_run_update(
-            self.job_run["id"], status="request_completed", conclusion=conclusion
+            self.job_run["id"],
+            status="request_completed",
+            conclusion=conclusion,
+            number_of_files_produced=number_of_files_produced,
         )
 
+    def get_number_of_files_produced(self) -> int:
+        """Returns the number of files produced by the job."""
+        number_of_files_produced = 0
+
+        # Logs can be produced even if no artifact stores are created for the job run.
+        job_run_logs_cache = get_logs_cache(self.job_run["id"])
+        has_walk = getattr(job_run_logs_cache, "walk", None)
+        if has_walk:
+            log_files = [
+                file
+                for _, _, current_path_files in job_run_logs_cache.walk()
+                for file in current_path_files
+            ]
+        else:
+            log_files = [
+                file
+                for _, _, current_path_files in os.walk(job_run_logs_cache)
+                for file in current_path_files
+            ]
+
+        number_of_files_produced += len(log_files)
+
+        if "stores" not in self.config:
+            return number_of_files_produced
+
+        job_run_artifacts_cache = get_artifacts_cache(self.job_run["id"])
+        has_walk = getattr(job_run_artifacts_cache, "walk", None)
+        if has_walk:
+            artifact_files = [
+                file
+                for _, _, current_path_files in job_run_artifacts_cache.walk()
+                for file in current_path_files
+            ]
+        else:
+            artifact_files = [
+                file
+                for _, _, current_path_files in os.walk(job_run_artifacts_cache)
+                for file in current_path_files
+            ]
+
+        number_of_files_produced += len(artifact_files)
+
+        return number_of_files_produced
+
     async def run_task(self, task: Task) -> bool:
         for cmd in task["cmd"].strip().split("\n"):
             # Adding an additional echo and utilizing stdbuf to force line buffering

{primitive-0.2.8 → primitive-0.2.10}/src/primitive/jobs/actions.py

@@ -113,6 +113,7 @@ class Jobs(BaseAction):
         status: str = None,
         conclusion: str = None,
         file_ids: Optional[List[str]] = [],
+        number_of_files_produced: Optional[int] = None,
     ):
         mutation = gql(job_run_update_mutation)
         input = {"id": id}
@@ -122,6 +123,8 @@ class Jobs(BaseAction):
             input["conclusion"] = conclusion
         if file_ids and len(file_ids) > 0:
             input["files"] = file_ids
+        if number_of_files_produced is not None:
+            input["numberOfFilesProduced"] = number_of_files_produced
         variables = {"input": input}
         result = self.primitive.session.execute(
             mutation, variable_values=variables, get_execution_result=True

{primitive-0.2.8 → primitive-0.2.10}/uv.lock

@@ -283,7 +283,7 @@ name = "click"
 version = "8.1.8"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "colorama", marker = "platform_system == 'Windows'" },
+    { name = "colorama", marker = "sys_platform == 'win32'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/b9/2e/0090cbf739cee7d23781ad4b89a9894a41538e4fcf4c31dcdd705b78eb8b/click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a", size = 226593 }
 wheels = [
@@ -710,7 +710,6 @@ wheels = [
 
 [[package]]
 name = "primitive"
-version = "0.1.89"
 source = { editable = "." }
 dependencies = [
     { name = "click" },