钉钉操作符

先决条件

要使用此操作符,您必须执行以下几项操作
  • 将自定义机器人添加到您要发送消息的聊天群组中。

  • 获取 钉钉自定义机器人 webhook token

  • 将访问令牌放入 dingding_default 连接的密码字段中。注意:您只需要令牌值,而不是整个 webhook 字符串。

基本用法

使用 DingdingOperator 通过 钉钉自定义机器人 发送消息

tests/system/dingding/example_dingding.py[源码]

text_msg_remind_none = DingdingOperator(
    task_id="text_msg_remind_none",
    message_type="text",
    message="Airflow dingding text message remind none",
    at_mobiles=None,
    at_all=False,
)

在消息中提醒用户

使用参数 at_mobilesat_all 在发送消息时提醒特定用户,当 at_all 设置为 True 时,at_mobiles 将被忽略。

tests/system/dingding/example_dingding.py[源码]

text_msg_remind_all = DingdingOperator(
    task_id="text_msg_remind_all",
    message_type="text",
    message="Airflow dingding text message remind all users in group",
    # list of user phone/email here in the group
    # when at_all is specific will cover at_mobiles
    at_mobiles=["156XXXXXXXX", "130XXXXXXXX"],
    at_all=True,
)

发送富文本消息

DingdingOperator 可以通过 钉钉自定义机器人 发送包括链接、markdown、actionCard 和 feedCard 的富文本消息。富文本消息不能提醒特定用户,除非使用 markdown 类型消息。

tests/system/dingding/example_dingding.py[源码]

markdown_msg = DingdingOperator(
    task_id="markdown_msg",
    message_type="markdown",
    message={
        "title": "Airflow dingding markdown message",
        "text": "# Markdown message title\n"
        "content content .. \n"
        "### sub-title\n"
        "![logo](https://airflow.org.cn/_images/pin_large.png)",
    },
    at_mobiles=["156XXXXXXXX"],
    at_all=False,
)

从任务回调发送消息

DingdingHook 可以通过编写一个回调函数,然后将该函数传递给 sla_miss_callback, on_success_callback, on_failure_callback, 或 on_retry_callback 来处理任务回调。这里我们使用 on_failure_callback 作为示例

tests/system/dingding/example_dingding.py[源码]

def failure_callback(context):
    """
    The function that will be executed on failure.

    :param context: The context of the executed task.
    """
    message = f"The task {context['ti'].task_id} failed"
    DingdingHook(message_type="text", message=message, at_all=True).send()


如果需要,更改连接主机

DingdingOperator 操作符使用默认主机 https://oapi.dingtalk.com 发送 http 请求,如果需要更改使用的主机,可以设置连接的主机字段。

这个条目有帮助吗?