Merge pull request #33 from stax-labs/feat/examples-tasks

feat(tasks) add read tasks to examples

Author: ZXYmania

examples/tasks/read_tasks.py

@@ -0,0 +1,19 @@
+import os
+
+from staxapp.config import Config
+from staxapp.openapi import StaxClient
+
+Config.access_key = os.getenv("ACCESS_KEY")
+Config.secret_key = os.getenv("SECRET_KEY")
+
+# Read all tasks
+task_client = StaxClient("tasks")
+all_tasks = task_client.ReadTasks()
+print(all_tasks)
+
+# Task Id in the response of a different StaxClient call
+task_id = <Task Id>
+
+# Get a specific task
+task = task_client.ReadTask(task_id=task_id)
+print(task)

examples/workloads/create_catalogue_item.py

@@ -26,10 +26,12 @@
 
 #Upload the cfn to the deployment bucket
 s3 = boto3.resource('s3')
-workload_client = StaxClient("workloads")
 cfn_name = f'{catalogue_version}-{catalogue_name}.yaml'
 s3.Bucket(bucket_name["Parameter"]["Value"]).upload_file(cloudformation_manifest_path, cfn_name)
 
+## Invoke the Stax SDK
+workload_client = StaxClient("workloads")
+
 # Make the cfn into a workload catalogue item
 manifest_body = f"""Resources:
   - WorkloadSSM:

examples/workloads/create_catalogue_version.py

@@ -1,7 +1,7 @@
 import boto3
 from staxapp.config import Config
 from staxapp.openapi import StaxClient
-
+     
 # Requirements
 # - Logged into the deployment bucket account
 #	- Logged into the SDK
@@ -26,10 +26,12 @@
 
 #Upload the cfn to the deployment bucket
 s3 = boto3.resource('s3')
-workload_client = StaxClient("workloads")
 cfn_name = f'{catalogue_version}-{catalogue_name}.yaml'
 s3.Bucket(bucket_name["Parameter"]["Value"]).upload_file(cloudformation_manifest_path, cfn_name)
 
+# # Invoke the Stax SDK
+workload_client = StaxClient("workloads")
+
 # Make the cfn into a workload catalogue item
 manifest_body = f"""Resources:
   - WorkloadSSM:

examples/workloads/create_workload.py

@@ -2,7 +2,6 @@
 from staxapp.openapi import StaxClient
 
 # Requirements
-# - Logged into the deployment bucket account
 #	- Logged into the SDK
 
 Config.access_key = <Access Key>
@@ -12,7 +11,7 @@
 catalogue_id = <Catalogue Id>
 workload_name = <Workload Name> 
 account_id = <Account Id>
-parameter_dict = <Dictionary of Parameters
+parameter_dict = <Dictionary of Parameters>
 tags_dict = <Dictionary of Tags>
 
 workload_client = StaxClient('workloads')

examples/workloads/update_workload.py

@@ -0,0 +1,15 @@
+from staxapp.config import Config
+from staxapp.openapi import StaxClient
+
+Config.access_key = <Access Key>
+Config.secret_key = <Secret Key>
+
+# The workload to update
+workload_id = <Workload Id> 
+# The catalogue version id to update the worklaod too
+catalogue_version_id = <Catalogue Version Id>
+
+workload_client = StaxClient('workloads')
+
+response = workload_client.UpdateWorkload(workload_id=workload_id, catalogue_version_id=catalogue_version_id)
+print(response)