primitive 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

primitive/__about__.py

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

primitive/agent/actions.py

@@ -0,0 +1,59 @@
+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)

primitive/agent/commands.py

@@ -0,0 +1,13 @@
+import click
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.command("agent")
+@click.pass_context
+def cli(context):
+    """agent"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    primitive.agent.execute()

primitive/cli.py

@@ -7,6 +7,7 @@ from .auth.commands import config_command, whoami_command
 from .files.commands import cli as file_commands
 from .hardware.commands import cli as hardware_commands
 from .lint.commands import cli as lint_commands
+from .agent.commands import cli as agent_commands
 
 
 @click.group()
@@ -55,6 +56,7 @@ cli.add_command(whoami_command, "whoami")
 cli.add_command(file_commands, "files")
 cli.add_command(hardware_commands, "hardware")
 cli.add_command(lint_commands, "lint")
+cli.add_command(agent_commands, "agent")
 
 if __name__ == "__main__":
     cli(obj={})

primitive/client.py

@@ -7,6 +7,7 @@ from .files.actions import Files
 from .simulations.actions import Simulations
 from .hardware.actions import Hardware
 from .lint.actions import Lint
+from .agent.actions import Agent
 
 from loguru import logger
 
@@ -57,6 +58,7 @@ class Primitive:
         self.simulations: Simulations = Simulations(self)
         self.hardware: Hardware = Hardware(self)
         self.lint: Lint = Lint(self)
+        self.agent: Agent = Agent(self)
 
     def get_host_config(self):
         self.full_config = read_config_file()

primitive/hardware/actions.py

@@ -4,7 +4,7 @@ import json
 import platform
 from shutil import which
 import subprocess
-from typing import Dict, List
+from typing import Dict, List, Optional
 import click
 from loguru import logger
 from primitive.utils.memory_size import MemorySize
@@ -445,3 +445,67 @@ class Hardware(BaseAction):
             message = " [*] Failed to check in! "
             logger.error(message)
             raise exception
+
+    def get_hardware_list(self, fingerprint: Optional[str] = None):
+        query = gql(
+            """
+fragment HardwareFragment on Hardware {
+  id
+  pk
+  name
+  slug
+  createdAt
+  updatedAt
+  capabilities {
+    id
+    pk
+  }
+  activeReservation {
+    id
+    pk
+  }
+}
+
+query hardwareList(
+  $before: String
+  $after: String
+  $first: Int
+  $last: Int
+  $filters: HardwareFilters
+) {
+  hardwareList(
+    before: $before
+    after: $after
+    first: $first
+    last: $last
+    filters: $filters
+  ) {
+    totalCount
+    edges {
+      cursor
+      node {
+        ...HardwareFragment
+      }
+    }
+  }
+}
+"""
+        )
+
+        filters = {}
+        if fingerprint:
+            filters["fingerprint"] = {"exact": fingerprint}
+
+        variables = {
+            "first": 1,
+            "filters": filters,
+        }
+        result = self.primitive.session.execute(query, variable_values=variables)
+        return result
+
+    def get_own_hardware_details(self):
+        hardware_list_data = self.get_hardware_list(
+            fingerprint=self.primitive.host_config.get("fingerprint")
+        )
+        hardware = hardware_list_data.get("hardwareList").get("edges")[0].get("node")
+        return hardware

primitive/projects/actions.py

@@ -6,6 +6,106 @@ from primitive.utils.actions import BaseAction
 
 
 class Projects(BaseAction):
+    def get_job_runs(
+        self,
+        organization_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        job_id: Optional[str] = None,
+        reservation_id: Optional[str] = None,
+        git_commit_id: Optional[str] = None,
+        status: Optional[str] = None,
+        conclusion: Optional[str] = None,
+        first: Optional[int] = 1,
+        last: Optional[int] = None,
+    ):
+        query = gql(
+            """
+fragment PageInfoFragment on PageInfo {
+  hasNextPage
+  hasPreviousPage
+  startCursor
+  endCursor
+}
+
+fragment JobRunFragment on JobRun {
+  id
+  pk
+  createdAt
+  updatedAt
+  completedAt
+  startedAt
+  status
+  conclusion
+  stdout
+  job {
+    id
+    pk
+    slug
+    name
+    createdAt
+    updatedAt
+  }
+}
+
+query jobRuns(
+  $before: String
+  $after: String
+  $first: Int
+  $last: Int
+  $filters: JobRunFilters
+  $order: JobRunOrder
+) {
+  jobRuns(
+    before: $before
+    after: $after
+    first: $first
+    last: $last
+    filters: $filters
+    order: $order
+  ) {
+    totalCount
+    pageInfo {
+      ...PageInfoFragment
+    }
+    edges {
+      cursor
+      node {
+        ...JobRunFragment
+      }
+    }
+  }
+}
+"""
+        )
+
+        filters = {}
+        if organization_id:
+            filters["organization"] = {"id": organization_id}
+        if project_id:
+            filters["project"] = {"id": project_id}
+        if job_id:
+            filters["job"] = {"id": job_id}
+        if reservation_id:
+            filters["reservation"] = {"id": reservation_id}
+        if git_commit_id:
+            filters["gitCommit"] = {"id": git_commit_id}
+        if status:
+            filters["status"] = {"exact": status}
+        if conclusion:
+            filters["conclusion"] = {"exact": status}
+
+        variables = {
+            "first": first,
+            "last": last,
+            "filters": filters,
+            "order": {
+                "createdAt": "DESC",
+            },
+        }
+
+        result = self.primitive.session.execute(query, variable_values=variables)
+        return result
+
     def get_job_run(self, id: str):
         query = gql(
             """
@@ -28,6 +128,7 @@ class Projects(BaseAction):
         id: str,
         status: str = None,
         conclusion: str = None,
+        stdout: str = None,
         file_ids: Optional[List[str]] = [],
     ):
         mutation = gql(
@@ -50,6 +151,8 @@ class Projects(BaseAction):
             input["conclusion"] = conclusion
         if file_ids and len(file_ids) > 0:
             input["files"] = file_ids
+        if stdout:
+            input["stdout"] = stdout
         variables = {"input": input}
         result = self.primitive.session.execute(mutation, variable_values=variables)
         return result

{primitive-0.1.5.dist-info → primitive-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.5
+Version: 0.1.6
 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.5.dist-info → primitive-0.1.6.dist-info}/RECORD

@@ -1,7 +1,9 @@
-primitive/__about__.py,sha256=eiRjGM6WtdYyj1eyR-jQ63lTpMSkzch2LxAzdXLwqzU,128
+primitive/__about__.py,sha256=P5wASxX77-g_YRSjb4MnBDu4CWjITURFbU8iN8R3rlM,128
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
-primitive/cli.py,sha256=QC4jbu8s0uhNxFmaTO2B-X1o1gpyFX3xJVHU-j5CvQE,1705
-primitive/client.py,sha256=cZiJjzZ4HkCtYViDNOIoNuf7mcp772eSCSztsqh381E,2010
+primitive/cli.py,sha256=LXSpPtqQLxkOXsnsbs4UN3n7G_pHCz8AkBFCrk7Is4g,1796
+primitive/client.py,sha256=3RMRNY4ZiGT2OKnZtquWyrXjRSgDe5HlZvtFNKZErLQ,2083
+primitive/agent/actions.py,sha256=Fq6GEJqIwv-hWABFOl3YC8l0gBNME3byycNYqICpZG8,2144
+primitive/agent/commands.py,sha256=-dVDilELfkGfbZB7qfEPs77Dm1oT62qJj4tsIk4KoxI,254
 primitive/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/auth/actions.py,sha256=N2bGcwXNsB89pzs66gF9A5_WzUScY5fhfOyWixqo2y8,1054
 primitive/auth/commands.py,sha256=JahUq0E2e7Xa-FX1WEUv7TgM6ieDvNH4VwRRtxAW7HE,2340
@@ -9,12 +11,12 @@ primitive/files/actions.py,sha256=f4JN3QFB2WXw-0JpnE-4-movnqtvXIpCrGd_9pdkeW4,26
 primitive/files/commands.py,sha256=DDizo3xJnU3KLUBTMeeM72viVpnJinLwxs84tmqKhqo,810
 primitive/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/graphql/sdk.py,sha256=BhCGmDtc4sNnH8CxbQSJyFwOZ-ZSqMtjsxMB3JRBhPw,1456
-primitive/hardware/actions.py,sha256=BbFz17NKVDiI-9SiV1R1GPqsXRZwUm9oNj-Dc_bBFJ8,17133
+primitive/hardware/actions.py,sha256=UL2Wyh_XFXKVNEvlVCe1ltCZHpNaRSXPNwSR8hGJyR0,18343
 primitive/hardware/commands.py,sha256=QE7LLeFdfOqlvz3JwdwJJRZAY3fHI1zB9kYmmDajpq0,1477
 primitive/lint/actions.py,sha256=FVigksFrADPDD3zHS0g0riyVrge7eEtwWk5PG2aZv9A,2352
 primitive/lint/commands.py,sha256=3CZvkOEMpJspJWmaQzA5bpPKx0_VCijQIXA9l-eTnZE,487
 primitive/projects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/projects/actions.py,sha256=Zi0uFCUG5KU7w7HNE8eGlhUPdsJyjVrakcdU1PR8-MQ,1481
+primitive/projects/actions.py,sha256=h1rgt3tWUPye0gnFvIZzZz7ibTbxyBS4q9RryjkVLzY,3600
 primitive/simulations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/simulations/actions.py,sha256=YR0oxxd7_kuUIH77BWZLUs9rLRiSJztPPhpgDJU2PbY,1267
 primitive/utils/actions.py,sha256=HOFrmM3-0A_A3NS84MqrZ6JmQEiiPSoDqEeuu6b_qfQ,196
@@ -25,8 +27,8 @@ primitive/utils/memory_size.py,sha256=4xfha21kW82nFvOTtDFx9Jk2ZQoEhkfXii-PGNTpIU
 primitive/utils/printer.py,sha256=f1XUpqi5dkTL3GWvYRUGlSwtj2IxU1q745T4Fxo7Tn4,370
 primitive/utils/shell.py,sha256=-7UjQaBqSGHzEEyX8pNjeYFFP0P3lVnDV0OkgPz1qHU,1050
 primitive/utils/verible.py,sha256=QYczN1IvxODfj4jeq0nqjFuF0Oi0Zdx-Q32ySOJgcw8,2205
-primitive-0.1.5.dist-info/METADATA,sha256=30UnO3-byVnVwOjKRro-Agt53Q8Nae137DUfcZF5A1Y,1817
-primitive-0.1.5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-primitive-0.1.5.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.1.5.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.1.5.dist-info/RECORD,,
+primitive-0.1.6.dist-info/METADATA,sha256=k2nIZBpBfzNRizaA-5a9BW7qtOGmdQTeNF2lOdvocPU,1817
+primitive-0.1.6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+primitive-0.1.6.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.1.6.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.1.6.dist-info/RECORD,,