def get_context_from_env_var(): """ Extract context from env variable, e.g. dag_id, task_id and execution_date, so that they can be used inside BashOperator and PythonOperator. :return: The context of interest. """ return {format_map['default']: os.environ.get(format_map['env_var_format'], '') for format_map in AIRFLOW_VAR_NAME_FORMAT_MAPPING.values()}
def get_context_from_env_var(): """ Extract context from env variable, e.g. dag_id, task_id and execution_date, so that they can be used inside BashOperator and PythonOperator. :return: The context of interest. """ return {format_map['default']: os.environ.get(format_map['env_var_format'], '') for format_map in AIRFLOW_VAR_NAME_FORMAT_MAPPING.values()}
def generate_trino_client_info() -> str: """Return json string with dag_id, task_id, execution_date and try_number""" context_var = { format_map['default'].replace(DEFAULT_FORMAT_PREFIX, ''): os.environ.get(format_map['env_var_format'], '') for format_map in AIRFLOW_VAR_NAME_FORMAT_MAPPING.values() } task_info = { 'dag_id': context_var['dag_id'], 'task_id': context_var['task_id'], 'execution_date': context_var['execution_date'], 'try_number': context_var['try_number'], 'dag_run_id': context_var['dag_run_id'], 'dag_owner': context_var['dag_owner'], } return json.dumps(task_info, sort_keys=True)
def generate_trino_client_info() -> str: """Return json string with dag_id, task_id, execution_date and try_number""" context_var = { format_map['default'].replace(DEFAULT_FORMAT_PREFIX, ''): os.environ.get(format_map['env_var_format'], '') for format_map in AIRFLOW_VAR_NAME_FORMAT_MAPPING.values() } # try_number isn't available in context for airflow < 2.2.5 # https://github.com/apache/airflow/issues/23059 try_number = context_var.get('try_number', '') task_info = { 'dag_id': context_var['dag_id'], 'task_id': context_var['task_id'], 'execution_date': context_var['execution_date'], 'try_number': try_number, 'dag_run_id': context_var['dag_run_id'], 'dag_owner': context_var['dag_owner'], } return json.dumps(task_info, sort_keys=True)