primitive 0.1.6__tar.gz → 0.1.8__tar.gz

{primitive-0.1.6 → primitive-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.6
+Version: 0.1.8
 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.1.6 → primitive-0.1.8}/src/primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@steins.studio>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.6"
+__version__ = "0.1.8"

primitive-0.1.8/src/primitive/agent/actions.py

@@ -0,0 +1,71 @@
+import sys
+from time import sleep
+from primitive.utils.actions import BaseAction
+from loguru import logger
+from primitive.__about__ import __version__
+
+
+class Agent(BaseAction):
+    def execute(
+        self,
+    ):
+        logger.enable("primitive")
+        logger.info(" [*] primitive")
+        logger.info(f" [*] Version: {__version__}")
+
+        # self.primitive.hardware.update_hardware_system_info()
+        self.primitive.hardware.check_in_http()
+
+        try:
+            while True:
+                hardware = self.primitive.hardware.get_own_hardware_details()
+
+                active_reservation_id = None
+                if hardware.get("activeReservation"):
+                    active_reservation_id = hardware["activeReservation"]["id"]
+                if not active_reservation_id:
+                    logger.debug("No active reservation found")
+                    sleep(5)
+                    continue
+
+                job_runs_data = self.primitive.projects.get_job_runs(
+                    status="pending", first=1, reservation_id=active_reservation_id
+                )
+
+                pending_job_runs = [
+                    edge["node"] for edge in job_runs_data["jobRuns"]["edges"]
+                ]
+
+                for job_run in pending_job_runs:
+                    logger.debug("Found pending Job Run")
+                    logger.debug(f"Job Run ID: {job_run['id']}")
+                    logger.debug(f"Job Name: {job_run['job']['name']}")
+
+                    if job_run["job"]["slug"] == "lint":
+                        logger.debug("Executing Lint Job")
+
+                        self.primitive.projects.job_run_update(
+                            job_run["id"], status="request_completed"
+                        )
+
+                        result, message = self.primitive.lint.execute()
+                        if result:
+                            self.primitive.projects.job_run_update(
+                                job_run["id"],
+                                status="request_completed",
+                                conclusion="success",
+                                stdout=message,
+                            )
+                        else:
+                            self.primitive.projects.job_run_update(
+                                job_run["id"],
+                                status="request_completed",
+                                conclusion="failure",
+                                stdout=message,
+                            )
+
+                sleep(5)
+        except KeyboardInterrupt:
+            logger.info(" [*] Stopping primitive...")
+            self.primitive.hardware.check_in_http(is_available=False, is_online=False)
+            sys.exit()

primitive-0.1.6/src/primitive/agent/actions.py

@@ -1,59 +0,0 @@
-from time import sleep
-from primitive.utils.actions import BaseAction
-from loguru import logger
-
-
-class Agent(BaseAction):
-    def execute(
-        self,
-    ):
-        logger.enable("primitive")
-        logger.debug("Starting Primitive Agent")
-        while True:
-            hardware = self.primitive.hardware.get_own_hardware_details()
-
-            active_reservation_id = None
-            if hardware.get("activeReservation"):
-                active_reservation_id = hardware["activeReservation"]["id"]
-            if not active_reservation_id:
-                logger.debug("No active reservation found")
-                sleep(5)
-                continue
-
-            job_runs_data = self.primitive.projects.get_job_runs(
-                status="pending", first=1, reservation_id=active_reservation_id
-            )
-
-            pending_job_runs = [
-                edge["node"] for edge in job_runs_data["jobRuns"]["edges"]
-            ]
-
-            for job_run in pending_job_runs:
-                logger.debug("Found pending Job Run")
-                logger.debug(f"Job Run ID: {job_run['id']}")
-                logger.debug(f"Job Name: {job_run['job']['name']}")
-
-                if job_run["job"]["slug"] == "lint":
-                    logger.debug("Executing Lint Job")
-
-                    self.primitive.projects.job_run_update(
-                        job_run["id"], status="request_completed"
-                    )
-
-                    result, message = self.primitive.lint.execute()
-                    if result:
-                        self.primitive.projects.job_run_update(
-                            job_run["id"],
-                            status="request_completed",
-                            conclusion="success",
-                            stdout=message,
-                        )
-                    else:
-                        self.primitive.projects.job_run_update(
-                            job_run["id"],
-                            status="request_completed",
-                            conclusion="failure",
-                            stdout=message,
-                        )
-
-            sleep(5)