Cristian Toma
02/03/2022, 2:47 PMKevin Kho
prefect.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
Kevin Kho
from 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()