feat: add usage metrics to the client organization query (#346)

Author: yolanfery

openhexa/graphql/graphql_client/__init__.py

@@ -391,6 +391,7 @@
 from .organizations import (
     Organizations,
     OrganizationsOrganizations,
+    OrganizationsOrganizationsUsage,
     OrganizationsOrganizationsWorkspaces,
     OrganizationsOrganizationsWorkspacesItems,
 )
@@ -716,6 +717,7 @@
     "OrganizationOrganizationWorkspacesItemsCountries",
     "Organizations",
     "OrganizationsOrganizations",
+    "OrganizationsOrganizationsUsage",
     "OrganizationsOrganizationsWorkspaces",
     "OrganizationsOrganizationsWorkspacesItems",
     "ParameterInput",

openhexa/graphql/graphql_client/client.py

@@ -919,6 +919,11 @@ def organizations(self, **kwargs: Any) -> List[OrganizationsOrganizations]:
                     name
                   }
                 }
+                usage {
+                  pipelineRuns
+                  users
+                  workspaces
+                }
               }
             }
             """

openhexa/graphql/graphql_client/organizations.py

@@ -3,6 +3,8 @@
 
 from typing import Any, List
 
+from pydantic import Field
+
 from .base_model import BaseModel
 
 
@@ -14,6 +16,7 @@ class OrganizationsOrganizations(BaseModel):
     id: Any
     name: str
     workspaces: "OrganizationsOrganizationsWorkspaces"
+    usage: "OrganizationsOrganizationsUsage"
 
 
 class OrganizationsOrganizationsWorkspaces(BaseModel):
@@ -25,6 +28,12 @@ class OrganizationsOrganizationsWorkspacesItems(BaseModel):
     name: str
 
 
+class OrganizationsOrganizationsUsage(BaseModel):
+    pipeline_runs: int = Field(alias="pipelineRuns")
+    users: int
+    workspaces: int
+
+
 Organizations.model_rebuild()
 OrganizationsOrganizations.model_rebuild()
 OrganizationsOrganizationsWorkspaces.model_rebuild()

openhexa/graphql/queries.graphql

@@ -364,6 +364,12 @@ query Organizations {
         name
       }
     }
+      usage {
+          pipelineRuns
+          users
+          workspaces
+
+      }
   }
 }