如何抛出AirflowException而不在日志中输出完整traceback信息？

嘿，这个问题我太有共鸣了——默认抛出AirflowException时一堆栈信息堆在日志里，找有用的提示都费劲。其实有几个实用的办法能解决：

• 手动捕获异常+切断异常链
  在你的任务逻辑里，先用try-except块包裹可能出错的代码段，捕获到问题时先打印友好的提示信息（建议用Airflow的logging模块，比如logging.error("数据加载失败：找不到指定文件")），然后抛出AirflowException时加上from None，这样就能切断异常链，避免日志输出完整的栈跟踪：

  from airflow.exceptions import AirflowException
  import logging
  
  def my_task_function():
      try:
          # 这里写你的任务逻辑，比如读取文件、调用接口等
          with open("missing_file.txt", "r") as f:
              pass
      except FileNotFoundError as e:
          # 打印友好提示
          logging.error("任务失败：无法找到所需的输入文件，请检查文件路径是否正确")
          # 抛出AirflowException并切断异常链
          raise AirflowException("输入文件缺失") from None
  

• 调整Airflow日志配置
  如果你想全局控制这类异常的日志输出，可以修改Airflow的airflow.cfg配置文件，调整logging_level或者针对airflow.task日志器设置更严格的级别。不过这个方法会影响所有任务，建议谨慎使用。

• 使用自定义任务装饰器
  要是你有多个任务都需要这个逻辑，可以写一个自定义装饰器，统一处理异常捕获、友好提示和抛出AirflowException的逻辑，不用每个任务都重复写try-except块：

  from functools import wraps
  from airflow.exceptions import AirflowException
  import logging
  
  def handle_task_errors(friendly_msg):
      def decorator(func):
          @wraps(func)
          def wrapper(*args, **kwargs):
              try:
                  return func(*args, **kwargs)
              except Exception as e:
                  logging.error(friendly_msg)
                  raise AirflowException(friendly_msg) from None
          return wrapper
      return decorator
  
  # 使用装饰器
  @handle_task_errors("数据同步任务失败：目标数据库连接超时")
  def sync_data_task():
      # 任务逻辑
      pass
  

这样处理后，日志里只会显示你设置的友好提示和简洁的AirflowException消息，不会再被一堆栈信息刷屏啦。

备注：内容来源于stack exchange，提问作者jcgrowley