[TWTR][EWT-472] Add lifecycle support while launching worker pods (twitter-forks#59)

Author: msumit

airflow/contrib/kubernetes/kubernetes_request_factory/kubernetes_request_factory.py

@@ -262,3 +262,8 @@ def _apply_env_from(pod, req):
     def extract_priority_class(pod, req):
         if pod.priority_class:
             req['spec']['priorityClassName'] = pod.priority_class
+
+    @staticmethod
+    def extract_lifecycle(pod, req):
+        if pod.lifecycle:
+            req['spec']['containers'][0]['lifecycle'] = pod.lifecycle

airflow/contrib/kubernetes/kubernetes_request_factory/pod_request_factory.py

@@ -66,6 +66,7 @@ def create(self, pod):
         self.extract_security_context(pod, req)
         self.extract_dnspolicy(pod, req)
         self.extract_priority_class(pod, req)
+        self.extract_lifecycle(pod, req)
         return req
 
 
@@ -137,4 +138,5 @@ def create(self, pod):
         self.extract_security_context(pod, req)
         self.extract_dnspolicy(pod, req)
         self.extract_priority_class(pod, req)
+        self.extract_lifecycle(pod, req)
         return req

airflow/contrib/kubernetes/pod.py

@@ -114,7 +114,8 @@ def __init__(
             configmaps=None,
             pod_runtime_info_envs=None,
             dnspolicy=None,
-            priority_class=None
+            priority_class=None,
+            lifecycle=None
     ):
         self.image = image
         self.envs = envs or {}
@@ -143,3 +144,4 @@ def __init__(
         self.pod_runtime_info_envs = pod_runtime_info_envs or []
         self.dnspolicy = dnspolicy
         self.priority_class = priority_class
+        self.lifecycle = lifecycle or {}

airflow/version.py

@@ -18,5 +18,5 @@
 # under the License.
 #
 
-version = '1.10.4+twtr18'
+version = '1.10.4+twtr19'