Breaking News: Grepper is joining You.com. Read the official announcement!

airflow get execution context dictionary kubernetes pod name

DreamCoder answered on July 18, 2022 Popularity 2/10 Helpfulness 1/10

answer airflow get execution context dictionary kubernetes pod name

airflow get execution context dictionary kubernetes pod name

Comment

import airflow.configuration as config
from airflow.contrib.operators.kubernetes_pod_operator import KubernetesPodOperator as AirflowKubernetesPodOperator

class KubernetesPodOperator(AirflowKubernetesPodOperator):

    def execute(self, context):

        environment = config.conf.get('webserver', 'web_server_name')

        ti = context['ti']
        dag_id = ti.dag_id
        task_id = ti.task_id
        run_id = context['run_id']
        try_number = str(ti._try_number)

        labels = { 
            'ENVIRONMENT' : environment,
            'DAG_ID'      : dag_id, 
            'TASK_ID'     : task_id, 
            'RUN_ID'      : run_id,
            'TRY_NUMBER'  : try_number,
        }

        self.env_vars.update(labels)
        super().execute(context)

Popularity 2/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: airflow dictionary execution get kubernetes-pod python

Link to this answer
Share Copy Link