feat(deployments): allow filtering by project and cluster IDs in list methods (#379)

Author: cwetherill-ps

gradient/api_sdk/repositories/gradient_deployments.py

@@ -84,7 +84,7 @@ def get_deployment(id, first=100, api_key=None):
                         }
                         cluster {
                             id
-                        }	
+                        }
                         data {
                             command
                             env {
@@ -130,16 +130,25 @@ def get_deployment(id, first=100, api_key=None):
     return client.execute(query, variable_values=params)
 
 
-def list_deployments(first=100, api_key=None):
+def list_deployments(name=None, project_id=None, cluster_id=None, first=100, api_key=None):
     client = graphql_client(api_key)
     query = gql(
         """
-        query getDeployments($first: Int!) {
-            deployments(first: $first) {
+        query getDeployments(
+            $first: Int!
+            $projectId: String
+            $clusterId: String
+            $name: String
+        ) {
+            deployments(
+                first: $first
+                projectId: $projectId
+                name: $name
+            ) {
                 nodes {
                     id
                     name
-                    deploymentSpecs(first: $first) {
+                    deploymentSpecs(first: $first, clusterId: $clusterId) {
                         nodes {
                             id
                             data {
@@ -172,14 +181,17 @@ def list_deployments(first=100, api_key=None):
                                 }
                             }
                         }
-                            }
+                    }
                 }
             }
         }
     """
     )
     params = {
         "first": first,
+        "projectId": project_id,
+        "clusterId": cluster_id,
+        "name": name
     }
     return client.execute(query, variable_values=params)['deployments']['nodes']
 

gradient/cli/gradient_deployments.py

@@ -135,11 +135,36 @@ def update_deployment_command(ctx, api_key, id, name, project_id, spec_path, clu
 
 
 @deployments.command("list", help="List deployments")
+@click.option(
+    "--name",
+    "name",
+    required=False,
+    help="Name",
+    cls=common.GradientOption,
+)
+@click.option(
+    "--projectId",
+    "project_id",
+    required=False,
+    help="Project ID",
+    cls=common.GradientOption,
+)
+@click.option(
+    "--clusterId",
+    "cluster_id",
+    required=False,
+    help="Cluster ID",
+    cls=common.GradientOption,
+)
 @api_key_option
 @click.pass_context
-def list_deployments_command(ctx, api_key):
+def list_deployments_command(ctx, name, project_id, cluster_id, api_key):
     try:
-        deployments = gradient_deployments.list_deployments(api_key=api_key)
+        deployments = gradient_deployments.list_deployments(
+            name,
+            project_id,
+            cluster_id,
+            api_key=api_key)
         if len(deployments) == 0:
             print('No deployments found')
             return