feat(deployment.py): add spec_annotations to deployments and carry forward

Add the spec_annotations dict and carry it forward to all the functions that control the manifest for deployments.

Author: Cryptophobia

rootfs/scheduler/__init__.py

@@ -232,6 +232,7 @@ def deploy(self, namespace, name, image, entrypoint, command, **kwargs):  # noqa
         """Deploy Deployment depending on what's requested"""
         app_type = kwargs.get('app_type')
         version = kwargs.get('version')
+        spec_annotations = {}
 
         # If an RC already exists then stop processing of the deploy
         try:
@@ -255,19 +256,24 @@ def deploy(self, namespace, name, image, entrypoint, command, **kwargs):  # noqa
             }
             # this depends on the deployment object having the latest information
             deployment = self.deployment.get(namespace, name).json()
+            # a hack to persist the spec annotations on the deployment object to next release
+            # instantiate spec_annotations and set to blank to avoid errors
+            if 'annotations' in deployment['spec']['template']['metadata'].keys():
+                old_spec_annotations = deployment['spec']['template']['metadata']['annotations']
+                spec_annotations = old_spec_annotations
             if deployment['spec']['template']['metadata']['labels'] == labels:
                 self.log(namespace, 'Deployment {} with release {} already exists. Stopping deploy'.format(name, version))  # noqa
                 return
         except KubeException:
             # create the initial deployment object (and the first revision)
             self.deployment.create(
-                namespace, name, image, entrypoint, command, **kwargs
+                namespace, name, image, entrypoint, command, spec_annotations, **kwargs
             )
         else:
             try:
                 # kick off a new revision of the deployment
                 self.deployment.update(
-                    namespace, name, image, entrypoint, command, **kwargs
+                    namespace, name, image, entrypoint, command, spec_annotations, **kwargs
                 )
             except KubeException as e:
                 raise KubeException(
@@ -283,7 +289,10 @@ def scale(self, namespace, name, image, entrypoint, command, **kwargs):
             if e.response.status_code == 404:
                 # create missing deployment - deleted if it fails
                 try:
-                    self.deployment.create(namespace, name, image, entrypoint, command, **kwargs)
+                    spec_annotations = {}
+                    self.deployment.create(
+                        namespace, name, image, entrypoint, command, spec_annotations, **kwargs
+                    )
                 except KubeException:
                     # see if the deployment got created
                     try:

rootfs/scheduler/resources/deployment.py

@@ -30,7 +30,7 @@ def get(self, namespace, name=None, **kwargs):
 
         return response
 
-    def manifest(self, namespace, name, image, entrypoint, command, **kwargs):
+    def manifest(self, namespace, name, image, entrypoint, command, spec_annotations, **kwargs):
         replicas = kwargs.get('replicas', 0)
         batches = kwargs.get('deploy_batches', None)
         tags = kwargs.get('tags', {})
@@ -104,11 +104,14 @@ def manifest(self, namespace, name, image, entrypoint, command, **kwargs):
         # pod manifest spec
         manifest['spec']['template'] = self.pod.manifest(namespace, name, image, **kwargs)
 
+        # set the old deployment spec annotations on this deployment
+        manifest['spec']['template']['metadata']['annotations'] = spec_annotations
+
         return manifest
 
-    def create(self, namespace, name, image, entrypoint, command, **kwargs):
+    def create(self, namespace, name, image, entrypoint, command, spec_annotations, **kwargs):
         manifest = self.manifest(namespace, name, image,
-                                 entrypoint, command, **kwargs)
+                                 entrypoint, command, spec_annotations, **kwargs)
 
         url = self.api("/namespaces/{}/deployments", namespace)
         response = self.http_post(url, json=manifest)
@@ -124,9 +127,9 @@ def create(self, namespace, name, image, entrypoint, command, **kwargs):
 
         return response
 
-    def update(self, namespace, name, image, entrypoint, command, **kwargs):
+    def update(self, namespace, name, image, entrypoint, command, spec_annotations, **kwargs):
         manifest = self.manifest(namespace, name, image,
-                                 entrypoint, command, **kwargs)
+                                 entrypoint, command, spec_annotations, **kwargs)
 
         url = self.api("/namespaces/{}/deployments/{}", namespace, name)
         response = self.http_put(url, json=manifest)