Description

We presently have a number of XCOM_PULL_TEMPLATE.format(...) calls as part of our DAG definitions. Since this is a frequently used templated string, it would be nice to have a shorthand utility function for it. E.g.:

def xcom_pull(operator: BaseOperator, key: str = "return_value"):
    return XCOM_PULL_TEMPLATE.format(
        operator.task_id,
        key
    )

Defaulting to return_value would allow us to make very simple calls for simple XComs.

We could then go through and replace all instances of XCOM_PULL_TEMPLATE formatting with this function instead.

Additional context

Discussed on the Make WordPress Slack here.

Implementation

• 🙋 I would be interested in implementing this feature.