更改动态任务中自动生成的任务名称

如何解决更改动态任务中自动生成的任务名称

我正在尝试创建一个动态工作流，但想要更改气流自动生成的任务名称并分配给列表中的任务。我尝试访问上下文并手动更改 taskid，但这在 UI 中的管道呈现期间也不起作用。

我的代码

    def get_the_route(router_ip,taskid):
        dev1 = junos_ops()
        dev1.open_fabric_connection()
        result = dev1.dev_handler.rpc.get_route_information(destination="10.0.0.3",normalize=True)

        logger.info("result is: {}".format(pformat(result)))
        dev1.close_fabric_connection()
       # <--do-some-logic-->
        return {"result": result}

    for dev in dev_list:
        get_the_route_dev_list.append(get_the_route(router_ip=dev,taskid=dev))
    start >> hello_task >> get_the_route_dev_list >> bye_task >> end

生成的图表

无论如何要为动态任务赋予不同的名称？我知道这可以使用 Pythonoperator。但我正在尝试使用 TaskFlow API 来做到这一点。

谢谢

解决方法

是的，这是可能的。例如：

@task(task_id="different_name")
def foo():
    pass

with DAG(
    'test_foo',start_date=days_ago(1),schedule_interval=None,) as dag:
    foo()

导致：