Cristian Toma
02/03/2022, 2:47 PMKevin Kho
02/03/2022, 2:50 PMprefect.context.get("flow_run_name")
Cristian Toma
02/03/2022, 4:22 PMprefect.context.get("flow_run_name")
can run only inside a task, right? So I need to create a task X and inside of it return the flow name and after in the with Flow section to pass it to the RenameFlowRun task.Kevin Kho
02/03/2022, 4:26 PMfrom prefect import Flow, task
from prefect.tasks.prefect import RenameFlowRun
def rename_handler(obj, new_state, old_state):
if new_state.is_running():
RenameFlowRun().run(flow_run_name="new_name")
return
@task
def first_task():
return 1
with Flow("test-flow", state_handlers=[rename_handler]) as flow:
first_task()