primitive 0.1.13__tar.gz → 0.1.14__tar.gz

{primitive-0.1.13 → primitive-0.1.14}/PKG-INFO

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

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

{primitive-0.1.13 → primitive-0.1.14}/src/primitive/agent/actions.py

@@ -32,7 +32,7 @@ class Agent(BaseAction):
                     sleep(5)
                     continue
 
-                job_runs_data = self.primitive.projects.get_job_runs(
+                job_runs_data = self.primitive.jobs.get_job_runs(
                     status="pending", first=1, reservation_id=active_reservation_id
                 )
 
@@ -52,7 +52,7 @@ class Agent(BaseAction):
                     )
 
                     github_access_token = (
-                        self.primitive.projects.github_access_token_for_job_run(
+                        self.primitive.jobs.github_access_token_for_job_run(
                             job_run["id"]
                         )
                     )
@@ -68,7 +68,7 @@ class Agent(BaseAction):
                     if job_run["job"]["slug"] == "lint":
                         logger.debug("Executing Lint Job")
 
-                        self.primitive.projects.job_run_update(
+                        self.primitive.jobs.job_run_update(
                             job_run["id"], status="request_in_progress"
                         )
 
@@ -79,7 +79,7 @@ class Agent(BaseAction):
                             conclusion = "success"
                         else:
                             conclusion = "failure"
-                        self.primitive.projects.job_run_update(
+                        self.primitive.jobs.job_run_update(
                             job_run["id"],
                             status="request_completed",
                             conclusion=conclusion,

{primitive-0.1.13 → primitive-0.1.14}/src/primitive/cli.py

@@ -10,6 +10,9 @@ from .lint.commands import cli as lint_commands
 from .agent.commands import cli as agent_commands
 from .git.commands import cli as git_commands
 from .daemons.commands import cli as daemons_commands
+from .jobs.commands import cli as jobs_commands
+from .organizations.commands import cli as organizations_commands
+from .projects.commands import cli as projects_commands
 
 
 @click.group()
@@ -61,6 +64,9 @@ cli.add_command(lint_commands, "lint")
 cli.add_command(agent_commands, "agent")
 cli.add_command(git_commands, "git")
 cli.add_command(daemons_commands, "daemons")
+cli.add_command(jobs_commands, "jobs")
+cli.add_command(organizations_commands, "organizations")
+cli.add_command(projects_commands, "projects")
 
 if __name__ == "__main__":
     cli(obj={})

{primitive-0.1.13 → primitive-0.1.14}/src/primitive/client.py

@@ -10,6 +10,8 @@ from .lint.actions import Lint
 from .agent.actions import Agent
 from .git.actions import Git
 from .daemons.actions import Daemons
+from .jobs.actions import Jobs
+from .organizations.actions import Organizations
 
 from loguru import logger
 
@@ -55,7 +57,9 @@ class Primitive:
         )
 
         self.auth: Auth = Auth(self)
+        self.organizations: Organizations = Organizations(self)
         self.projects: Projects = Projects(self)
+        self.jobs: Jobs = Jobs(self)
         self.files: Files = Files(self)
         self.simulations: Simulations = Simulations(self)
         self.hardware: Hardware = Hardware(self)

{primitive-0.1.13/src/primitive/projects → primitive-0.1.14/src/primitive/jobs}/actions.py

@@ -5,7 +5,88 @@ from gql import gql
 from primitive.utils.actions import BaseAction
 
 
-class Projects(BaseAction):
+class Jobs(BaseAction):
+    def get_jobs(
+        self,
+        organization_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        job_id: Optional[str] = None,
+        slug: Optional[str] = None,
+        first: Optional[int] = 1,
+        last: Optional[int] = None,
+    ):
+        query = gql(
+            """ 
+            fragment PageInfoFragment on PageInfo {
+                hasNextPage
+                hasPreviousPage
+                startCursor
+                endCursor
+            }
+
+            fragment JobFragment on Job {
+                id
+                pk
+                slug
+                name
+                createdAt
+                updatedAt
+            }
+
+            query jobs(
+                $before: String
+                $after: String
+                $first: Int
+                $last: Int
+                $filters: JobFilters
+            ) {
+                jobs(
+                    before: $before
+                    after: $after
+                    first: $first
+                    last: $last
+                    filters: $filters
+                ) {
+                    totalCount
+                    pageInfo {
+                        ...PageInfoFragment
+                    }
+                    edges {
+                        cursor
+                        node {
+                            ...JobFragment
+                        }
+                    }
+                }
+            }
+            """
+        )
+
+        filters = {}
+        if organization_id:
+            filters["organization"] = {"id": organization_id}
+        if project_id:
+            filters["project"] = {"id": project_id}
+        if job_id:
+            filters["id"] = job_id
+        if slug:
+            filters["slug"] = {"exact": slug}
+
+        variables = {
+            "first": first,
+            "last": last,
+            "filters": filters,
+            "order": {
+                "createdAt": "DESC",
+            },
+        }
+
+        result = self.primitive.session.execute(
+            query, variable_values=variables, get_execution_result=True
+        )
+        jobs = [edge["node"] for edge in result.data["jobs"]["edges"]]
+        return jobs
+
     def get_job_runs(
         self,
         organization_id: Optional[str] = None,
@@ -114,12 +195,34 @@ query jobRuns(
     def get_job_run(self, id: str):
         query = gql(
             """
+            fragment JobRunFragment on JobRun {
+                id
+                pk
+                createdAt
+                updatedAt
+                completedAt
+                startedAt
+                status
+                conclusion
+                stdout
+                job {
+                    id
+                    pk
+                    slug
+                    name
+                    createdAt
+                    updatedAt
+                }
+                gitCommit {
+                    sha
+                    branch
+                    repoFullName
+                }
+            }
+
             query jobRun($id: GlobalID!) {
                 jobRun(id: $id) {
-                    id
-                    organization {
-                        id
-                    }
+                    ...JobRunFragment
                 }
             }
             """
@@ -173,3 +276,28 @@ query ghAppTokenForJobRun($jobRunId: GlobalID!) {
         variables = {"jobRunId": job_run_id}
         result = self.primitive.session.execute(query, variable_values=variables)
         return result["ghAppTokenForJobRun"]
+
+    def get_latest_job_run_for_job(
+        self, job_slug: Optional[str] = None, job_id: Optional[str] = None
+    ):
+        if not job_slug and not job_id:
+            raise ValueError("job_slug or job_id is required")
+        jobs_results = self.get_jobs(slug=job_slug)
+        jobs = [edge["node"] for edge in jobs_results.data["jobs"]["edges"]]
+
+        job_id = jobs.id
+        job_run_results = self.get_job_runs(job_id=job_id, first=1)
+        job_run = [edge["node"] for edge in job_run_results.data["job_runs"]["edges"]][
+            0
+        ]
+        return job_run
+
+    def job_run_start(
+        self,
+        job_slug: str,
+        job_id: str,
+    ):
+        if not job_slug and not job_id:
+            raise ValueError("job_slug or job_id is required")
+
+        self.get_jobs(slug=job_slug)

primitive-0.1.14/src/primitive/jobs/commands.py

@@ -0,0 +1,41 @@
+import click
+
+from ..utils.printer import print_result
+
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.group()
+@click.pass_context
+def cli(context):
+    """Jobs Commands"""
+    pass
+
+
+@cli.command("list")
+@click.pass_context
+def list(
+    context,
+    organization_slug: str = None,
+    organization_id: str = None,
+):
+    """List Job"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    message = primitive.jobs.get_jobs()
+    print_result(message=message, context=context)
+
+
+@cli.command("details")
+@click.argument("job_slug")
+@click.pass_context
+def details(
+    context,
+    job_slug: str = None,
+):
+    """List Job"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    message = primitive.jobs.get_jobs(slug=job_slug)
+    print_result(message=message, context=context)

primitive-0.1.14/src/primitive/organizations/actions.py

@@ -0,0 +1,84 @@
+from typing import Optional
+from gql import gql
+
+
+from primitive.utils.actions import BaseAction
+
+
+class Organizations(BaseAction):
+    def get_organizations(
+        self,
+        organization_id: Optional[str] = None,
+        slug: Optional[str] = None,
+        first: Optional[int] = 1,
+        last: Optional[int] = None,
+    ):
+        query = gql(
+            """ 
+            fragment PageInfoFragment on PageInfo {
+                hasNextPage
+                hasPreviousPage
+                startCursor
+                endCursor
+            }
+
+            fragment OrganizationFragment on Organization {
+                id
+                pk
+                slug
+                name
+                createdAt
+                updatedAt
+            }
+
+            query organizations(
+                $before: String
+                $after: String
+                $first: Int
+                $last: Int
+                $filters: OrganizationFilters
+                $order: OrganizationOrder
+            ) {
+                organizations(
+                    before: $before
+                    after: $after
+                    first: $first
+                    last: $last
+                    filters: $filters
+                    order: $order
+                ) {
+                    totalCount
+                    pageInfo {
+                        ...PageInfoFragment
+                    }
+                    edges {
+                        cursor
+                        node {
+                            ...OrganizationFragment
+                        }
+                    }
+                }
+            }
+            """
+        )
+
+        filters = {}
+        if organization_id:
+            filters["organization"] = {"id": organization_id}
+        if slug:
+            filters["slug"] = {"exact": slug}
+
+        variables = {
+            "first": first,
+            "last": last,
+            "filters": filters,
+            "order": {
+                "createdAt": "DESC",
+            },
+        }
+
+        result = self.primitive.session.execute(
+            query, variable_values=variables, get_execution_result=True
+        )
+        organizations = [edge["node"] for edge in result.data["organizations"]["edges"]]
+        return organizations

primitive-0.1.14/src/primitive/organizations/commands.py

@@ -0,0 +1,24 @@
+import click
+
+from ..utils.printer import print_result
+
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.group()
+@click.pass_context
+def cli(context):
+    """Organizations Commands"""
+    pass
+
+
+@cli.command("list")
+@click.pass_context
+def details(context):
+    """List Organizations"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    message = primitive.organizations.get_organizations()
+    print_result(message=message, context=context)

primitive-0.1.14/src/primitive/projects/actions.py

@@ -0,0 +1,79 @@
+from gql import gql
+
+from typing import Optional
+from primitive.utils.actions import BaseAction
+
+
+class Projects(BaseAction):
+
+    def get_projects(
+        self,
+        organization_id: Optional[str] = None,
+        slug: Optional[str] = None,
+        first: Optional[int] = 1,
+        last: Optional[int] = None,
+    ):
+        query = gql(
+            """ 
+            fragment PageInfoFragment on PageInfo {
+                hasNextPage
+                hasPreviousPage
+                startCursor
+                endCursor
+            }
+
+            fragment ProjectFragment on Project {
+                id
+                pk
+                slug
+                name
+                createdAt
+                updatedAt
+            }
+
+            query projects(
+                $before: String
+                $after: String
+                $first: Int
+                $last: Int
+                $filters: ProjectFilters
+            ) {
+                projects(
+                    before: $before
+                    after: $after
+                    first: $first
+                    last: $last
+                    filters: $filters
+                ) {
+                    totalCount
+                    pageInfo {
+                        ...PageInfoFragment
+                    }
+                    edges {
+                        cursor
+                        node {
+                            ...ProjectFragment
+                        }
+                    }
+                }
+            }
+            """
+        )
+
+        filters = {}
+        if organization_id:
+            filters["organization"] = {"id": organization_id}
+        if slug:
+            filters["slug"] = {"exact": slug}
+
+        variables = {
+            "first": first,
+            "last": last,
+            "filters": filters,
+        }
+
+        result = self.primitive.session.execute(
+            query, variable_values=variables, get_execution_result=True
+        )
+        projects = [edge["node"] for edge in result.data["projects"]["edges"]]
+        return projects

primitive-0.1.14/src/primitive/projects/commands.py

@@ -0,0 +1,24 @@
+import click
+
+from ..utils.printer import print_result
+
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.group()
+@click.pass_context
+def cli(context):
+    """Projects Commands"""
+    pass
+
+
+@cli.command("list")
+@click.pass_context
+def list(context):
+    """List Projects"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    message = primitive.projects.get_projects()
+    print_result(message=message, context=context)