python-3.x - 将日志添加到 Airflow 日志

Question

如何将我自己的日志添加到自动生成的 Apache Airflow 日志中？任何打印语句都不会在那里登录，所以我想知道如何添加我的日志以便它也显示在 UI 上？

Answer 1

我认为您可以通过使用日志记录模块并将配置信任给 Airflow 来解决此问题。

就像是：

import ...

dag = ...

def print_params_fn(**kwargs):
    import logging
    logging.info(kwargs)
    return None

print_params = PythonOperator(task_id="print_params",
                              python_callable=print_params_fn,
                              provide_context=True,
                              dag=dag)

Answer 2

在 Python 的可调用 PythonOperator 中，您可以使用：

import logging

LOGGER = logging.getLogger("airflow.task")
LOGGER.info("airflow.task >>> 2 - INFO logger test")

这将产生正确的输出，如：

[2019-12-26 09:42:55,813] {operations.py:86} INFO - airflow.task >>> 2 - INFO logger test

对于您的自定义记录器：

LOGGER = logging.getLogger(__name__)
LOGGER.info("__name__ >>> 2 - INFO logger test")

你会得到重复的格式：

[2019-12-26 09:42:55,813] {logging_mixin.py:112} INFO - [2019-12-26 09:42:55,813] {operations.py:79} INFO - __name__ >>> 2 - INFO logger test

Answer 3

如果您查看 PythonOperator：https ://github.com/apache/incubator-airflow/blob/master/airflow/operators/python_operator.py#L80-L81 ，看起来没有办法从python 可调用到气流日志中。

但是，如果您查看 BashOperator：https://github.com/apache/incubator-airflow/blob/master/airflow/operators/bash_operator.py#L79-L94，来自那里的 STDOUT/STDERR 与气流日志。因此，如果日志对您很重要，我建议将 python 代码添加到单独的文件中并使用 BashOperator 调用它。