primitive 0.2.51__py3-none-any.whl → 0.2.53__py3-none-any.whl

primitive/__about__.py

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

primitive/agent/actions.py

@@ -89,71 +89,84 @@ class Agent(BaseAction):
                 logger.debug(f"Job Run ID: {job_run_data.get('id')}")
                 logger.debug(f"Job Name: {job_run_data.get('job').get('name')}")
 
-                logger.info(
-                    f"Setting JobRun {job_run_data.get('job').get('name')} to request_in_progress"
-                )
+                job_run_status = job_run_data.get("status", None)
 
-                job_run_status = job_run_data.get("jobRun", {}).get("status", None)
-
-                while job_run_status != "in_progress":
-                    if job_run_status == "request_completed":
-                        logger.error(
-                            f"JobRun {job_run_data.get('name')} is already completed. Exiting."
-                        )
-                        if RUNNING_IN_CONTAINER:
-                            logger.info("Running in container, exiting after job run")
-                            return
-                    elif job_run_status == "pending":
+                while True:
+                    if job_run_status == "pending":
                         # we are setting to request_in_progress here which puts a started_at time on the JobRun in the API's database
                         # any time spent pulling Git repositories, setting up, etc, counts as compute time
-                        self.primitive.jobs.job_run_update(
+                        logger.info(
+                            f"Setting JobRun {job_run_data.get('id')} to request_in_progress."
+                        )
+                        # get the status back from the job run update, this should be 'in progress'
+                        job_run_update_result = self.primitive.jobs.job_run_update(
                             id=job_run_id, status="request_in_progress"
                         )
-                    logger.info(
-                        f"Waiting for JobRun {job_run_data.get('name')} to be in_progress. Current status: {job_run_status}"
-                    )
-                    sleep(1)
+                        job_run_update_data = job_run_update_result.data
+                        if job_run_update_data is not None:
+                            # if the job_run_status is not in "in_progress", short circuit
+                            job_run_status = job_run_update_data.get(
+                                "jobRunUpdate", {}
+                            ).get("status", None)
+                    if job_run_status == "in_progress":
+                        logger.info(f"JobRun {job_run_data.get('id')} in progress.")
+                        break
+                    if (
+                        job_run_status == "request_completed"
+                        or job_run_status == "completed"
+                    ):
+                        logger.error(
+                            f"JobRun {job_run_data.get('name')} is already completed with conclusion {job_run_data.get('jobRun', {}).get('conclusion', 'unknown')}. Exiting."
+                        )
+                        break
+
                     job_run_result = self.primitive.jobs.get_job_run(id=job_run_id)
                     if job_run_result.data is not None:
                         job_run_data = job_run_result.data.get("jobRun", {})
                         job_run_status = job_run_data.get("status", None)
-
-                try:
-                    runner = Runner(
-                        primitive=self.primitive,
-                        job_run=job_run_data,
-                    )
-                    runner.setup()
-                except Exception as exception:
-                    logger.exception(
-                        f"Exception while initializing runner: {exception}"
-                    )
-                    self.primitive.jobs.job_run_update(
-                        id=job_run_id,
-                        status="request_completed",
-                        conclusion="failure",
+                    logger.info(
+                        f"Waiting for JobRun {job_run_data.get('id')} to be in_progress. Current status: {job_run_status}"
                     )
-                    continue
+                    sleep(1)
 
-                try:
-                    runner.execute_job_run()
-                except Exception as exception:
-                    logger.exception(f"Exception while executing job: {exception}")
-                    self.primitive.jobs.job_run_update(
-                        id=job_run_id,
-                        status="request_completed",
-                        conclusion="failure",
-                    )
-                finally:
-                    runner.cleanup()
+                # the backend has said it's okay for the agent to run this job
+                if job_run_status == "in_progress":
+                    try:
+                        runner = Runner(
+                            primitive=self.primitive,
+                            job_run=job_run_data,
+                        )
+                        runner.setup()
+                    except Exception as exception:
+                        logger.exception(
+                            f"Exception while initializing runner: {exception}"
+                        )
+                        self.primitive.jobs.job_run_update(
+                            id=job_run_id,
+                            status="request_completed",
+                            conclusion="failure",
+                        )
+                        continue
 
-                    # NOTE: also run scan here to force upload of artifacts
-                    # This should probably eventually be another daemon?
-                    uploader.scan()
+                    try:
+                        runner.execute_job_run()
+                    except Exception as exception:
+                        logger.exception(f"Exception while executing job: {exception}")
+                        self.primitive.jobs.job_run_update(
+                            id=job_run_id,
+                            status="request_completed",
+                            conclusion="failure",
+                        )
+                    finally:
+                        runner.cleanup()
 
-                if RUNNING_IN_CONTAINER:
-                    logger.info("Running in container, exiting after job run")
-                    break
+                        # NOTE: also run scan here to force upload of artifacts
+                        # This should probably eventually be another daemon?
+                        uploader.scan()
+
+                    if RUNNING_IN_CONTAINER:
+                        logger.info("Running in container, exiting after job run")
+                        break
 
                 sleep(5)
         except KeyboardInterrupt:

primitive/agent/uploader.py

@@ -46,22 +46,24 @@ class Uploader:
 
         for file in files:
             try:
-                result = self.primitive.files.upload_file_direct(
+                upload_file_result = self.primitive.files.upload_file_direct(
                     path=file,
                     key_prefix=str(PurePath(file).relative_to(cache.parent).parent),
                 )
-            except Exception as exception:
-                if "is empty" in str(exception):
-                    logger.warning(f"{file} is empty, skipping upload")
-                continue
 
-            upload_id = result.data["fileUpdate"]["id"]
+                if upload_file_result and upload_file_result.data is not None:
+                    upload_file_data = upload_file_result.data
+                    upload_id = upload_file_data.get("fileUpdate", {}).get("id")
 
-            if upload_id:
-                file_ids.append(upload_id)
-                continue
+                    if upload_id:
+                        file_ids.append(upload_id)
+                        continue
 
-            logger.error(f"Unable to upload file {file}")
+                logger.error(f"Unable to upload file {file}")
+            except Exception as exception:
+                if "is empty" in str(exception):
+                    logger.warning(f"{file} is empty, skipping upload")
+                    continue
 
         # Clean up job cache
         shutil.rmtree(path=cache)

{primitive-0.2.51.dist-info → primitive-0.2.53.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: primitive
-Version: 0.2.51
+Version: 0.2.53
 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.51.dist-info → primitive-0.2.53.dist-info}/RECORD

@@ -1,12 +1,12 @@
-primitive/__about__.py,sha256=98Py-XClBciFeBwMwwz2608bBaQxMYKd4HlOLCukdAM,130
+primitive/__about__.py,sha256=lSz0mtiMJzyxukJ01ol9sDoiQlVzFFSwYibTsET0ZRo,130
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
 primitive/cli.py,sha256=g7EtHI9MATAB0qQu5w-WzbXtxz_8zu8z5E7sETmMkKU,2509
 primitive/client.py,sha256=RMF46F89oK82gfZH6Bf0WZrhXPUu01pbieSO_Vcuoc4,3624
 primitive/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/agent/actions.py,sha256=Ald56Lq8KOtQ0_eFm2TseKkEIa8RP3IClLNVyv9IL4E,7035
+primitive/agent/actions.py,sha256=-vHuwZZv8tWfq5SSljr18tUJz3BbyodrfdXKZvvrWEM,7993
 primitive/agent/commands.py,sha256=o847pK7v7EWQGG67tky6a33qtwoutX6LZrP2FIS_NOk,388
 primitive/agent/runner.py,sha256=deMLa4l758k-BeK7Up4LVeD9hXfmAf6o3jJh8OmURys,11315
-primitive/agent/uploader.py,sha256=ZzrzsajNBogwEC7mT6Ejy0h2Jd9axMYGzt9pbCvVMlk,3171
+primitive/agent/uploader.py,sha256=DT_Nzt5eOTm_uRcYKm1sjBBaQZzp5iNZ_uN5XktfQ30,3382
 primitive/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/auth/actions.py,sha256=9NIEXJ1BNJutJs6AMMSjMN_ziONUAUhY_xHwojYJCLA,942
 primitive/auth/commands.py,sha256=Krm38ioduDJZw0OIrIcb6eR2X6iECiWX0JPAI-2HNxY,2352
@@ -95,8 +95,8 @@ primitive/utils/printer.py,sha256=f1XUpqi5dkTL3GWvYRUGlSwtj2IxU1q745T4Fxo7Tn4,37
 primitive/utils/psutil.py,sha256=xa7ef435UL37jyjmUPbEqCO2ayQMpCs0HCrxVEvLcuM,763
 primitive/utils/shell.py,sha256=Z4zxmOaSyGCrS0D6I436iQci-ewHLt4UxVg1CD9Serc,2171
 primitive/utils/text.py,sha256=XiESMnlhjQ534xE2hMNf08WehE1SKaYFRNih0MmnK0k,829
-primitive-0.2.51.dist-info/METADATA,sha256=PIcqx1gPY31EuSFxHmGQLOxecKUYIGu4KS3l5ieIvno,3513
-primitive-0.2.51.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-primitive-0.2.51.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.2.51.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.2.51.dist-info/RECORD,,
+primitive-0.2.53.dist-info/METADATA,sha256=hflPnuPJOARIUnf8e9toXIKDpGzCpwNqbFPt-J7BAAw,3513
+primitive-0.2.53.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+primitive-0.2.53.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.2.53.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.2.53.dist-info/RECORD,,