Better upgrade support

src/examples/app.py

@@ -93,6 +93,15 @@ def workload_safe(self, cluster, project, id):
         workload = api.get_workload_safe(project, id)
         return workload
 
+    @appier.route("/clusters/<str:cluster>/projects/<str:project>/workloads/<str:id>/upgrade", "GET")
+    def upgrade_workload(self, cluster, project, id):
+        api = self.get_api()
+        cluster = api.get_cluster_safe(cluster)["id"]
+        project = api.get_project_safe(cluster, project)["id"]
+        id = api.get_workload_safe(project, id)["id"]
+        workload = api.upgrade_workload(project, id)
+        return workload
+
     @appier.route("/stacks", "GET")
     def stacks(self):
         api = self.get_api()

src/rancher/workload.py

@@ -37,6 +37,8 @@
 __license__ = "Apache License, Version 2.0"
 """ The license for the module """
 
+import datetime
+
 class WorkloadAPI(object):
     """
     The workload API endpoints used by the Rancher 2.x
@@ -55,7 +57,7 @@ def list_workloads_name(self, project, name):
         data = contents["data"]
         return data
 
-    def get_workload(self, cluster, project, id):
+    def get_workload(self, project, id):
         url = self.base_url + "projects/%s/workloads/%s" % (project, id)
         contents = self.get(url)
         return contents
@@ -64,3 +66,15 @@ def get_workload_safe(self, project, id):
         contents = self.list_workloads_name(project, id)
         if contents: return contents[0]
         return self.get_workload(project, id)
+
+    def update_workload(self, project, id, payload = {}):
+        url = self.base_url + "projects/%s/workloads/%s" % (project, id)
+        contents = self.put(url, data_j = payload)
+        return contents
+
+    def upgrade_workload(self, project, id):
+        workload = self.get_workload(project, id)
+        current_date = datetime.datetime.utcnow().strftime("%Y-%m-%dT%H-%M-%SZ")
+        workload["annotations"]["cattle.io/timestamp"] = current_date
+        self.update_workload(project, id, payload = workload)
+        return workload