Add api key integration to deployments D-53 (#377)

Author: paperspace-philip

gradient/api_sdk/graphql.py

@@ -4,7 +4,9 @@
 
 from .config import config
 
-def graphql_client(api_key=config.PAPERSPACE_API_KEY):
+def graphql_client(api_key=None):
+	if api_key is None:
+		api_key = config.PAPERSPACE_API_KEY
 	headers = {
 		'Authorization': f'Bearer {api_key}',
 	}

gradient/api_sdk/repositories/gradient_deployments.py

@@ -2,8 +2,8 @@
 from ..graphql import graphql_client
 
 
-def create_deployment(name, project_id, spec, cluster_id=None):
-    client = graphql_client()
+def create_deployment(name, project_id, spec, cluster_id=None, api_key=None):
+    client = graphql_client(api_key)
     query = gql(
     """
         mutation createDeployment($input: CreateDeploymentInput!) {
@@ -26,8 +26,8 @@ def create_deployment(name, project_id, spec, cluster_id=None):
     return client.execute(query, variable_values=params)['createDeployment']['deployment']
 
 
-def update_deployment(id, name=None, project_id=None, spec=None, cluster_id=None):
-    client = graphql_client()
+def update_deployment(id, name=None, project_id=None, spec=None, cluster_id=None, api_key=None):
+    client = graphql_client(api_key)
     query = gql(
     """
         mutation updateDeployment($input: UpdateDeploymentInput!) {
@@ -58,8 +58,8 @@ def update_deployment(id, name=None, project_id=None, spec=None, cluster_id=None
     return client.execute(query, variable_values=params)['updateDeployment']['deployment']
 
 
-def get_deployment(id, first=100):
-    client = graphql_client()
+def get_deployment(id, first=100, api_key=None):
+    client = graphql_client(api_key)
     query = gql(
     """
         query getDeployment($id: UUID!, $first: Int!) {
@@ -130,8 +130,8 @@ def get_deployment(id, first=100):
     return client.execute(query, variable_values=params)
 
 
-def list_deployments(first=100):
-    client = graphql_client()
+def list_deployments(first=100, api_key=None):
+    client = graphql_client(api_key)
     query = gql(
     """
         query getDeployments($first: Int!) {
@@ -184,8 +184,8 @@ def list_deployments(first=100):
     return client.execute(query, variable_values=params)['deployments']['nodes']
 
 
-def delete_deployment(id):
-    client = graphql_client()
+def delete_deployment(id, api_key=None):
+    client = graphql_client(api_key)
     query = gql(
     """
         mutation deleteDeployment($input: DeleteDeploymentInput!) {

gradient/cli/gradient_deployments.py

@@ -70,7 +70,7 @@ def create_deployment_command(ctx, api_key, name, project_id, cluster_id, spec_p
         return
 
     try:
-        deployment = create_deployment(name, project_id, spec, cluster_id)
+        deployment = create_deployment(name, project_id, spec, cluster_id, api_key=api_key)
         print(f'Created deployment: {deployment["id"]}')
     except TransportQueryError as error:
         logger.error(error.errors[0]['message'])
@@ -126,7 +126,7 @@ def update_deployment_command(ctx, api_key, id, name, project_id, spec_path, clu
             return
 
     try:
-        deployment = update_deployment(id, name, project_id, spec, cluster_id)
+        deployment = update_deployment(id, name, project_id, spec, cluster_id, api_key=api_key)
         print(f'Updated deployment: {deployment["id"]}')
     except TransportQueryError as error:
         logger.error(error.errors[0]['message'])
@@ -139,7 +139,7 @@ def update_deployment_command(ctx, api_key, id, name, project_id, spec_path, clu
 @click.pass_context
 def list_deployments_command(ctx, api_key):
     try:
-        deployments = list_deployments()
+        deployments = list_deployments(api_key=api_key)
         if len(deployments) == 0:
             print('No deployments found')
             return
@@ -167,7 +167,7 @@ def list_deployments_command(ctx, api_key):
 @click.pass_context
 def get_deployment_command(ctx, api_key, id):
     try:
-        deployment = get_deployment(id)
+        deployment = get_deployment(id, api_key=api_key)
         if deployment['deployment'] is None:
             print('Deployment not found')
         else:
@@ -191,7 +191,7 @@ def get_deployment_command(ctx, api_key, id):
 @click.pass_context
 def delete_deployment_command(ctx, api_key, id):
     try:
-        deployment = delete_deployment(id)
+        deployment = delete_deployment(id, api_key=api_key)
 
         if deployment is None:
             print('Deployment not found')