Prefect 2.0 | Accessing current flow state
Hi! :wa...
# prefect-communityd
Dominik Wagner
07/26/2022, 1:28 PMPrefect 2.0 | Accessing current flow state
Hi! 👋 Possibly simple question which I can’t seem to find an answer for:
Can I access the state of the current flow from within that flow?
In particular I’d like to check if the flow is retrying, and if so pass an extra argument to my task(s) i.e.
Copy code
@flow
def my_flow():
if <somehow_get_state_of_this_flow?> == "Retrying":
do_something()✅ 1
a
Anna Geller
07/26/2022, 1:32 PM
when running tasks, you can do:
Copy code
if task.get_state().is_failed():Anna Geller
07/26/2022, 1:32 PM
or rather:
Copy code
if task.get_state().is_retrying():Anna Geller
07/26/2022, 1:33 PM
this assumes that task is submitted to task runner
Anna Geller
07/26/2022, 1:34 PM
Copy code
@task
def do_sth():
...
@flow
def my_flow():
if do_sth.submit().get_state().is_retrying():
do_something()d
Dominik Wagner
07/26/2022, 1:37 PMThanks, but maybe it wasn’t clear, I’m retrying the flow, not a task (added in 2.0b8).
All my tasks are using
shell_run_command()prefect_shella
Anna Geller
07/26/2022, 1:38 PM
for flow there is no way currently, but will be after GA release
🦜 1
Anna Geller
07/26/2022, 1:39 PM
for 2.0b8, this syntax works for tasks https://discourse.prefect.io/t/how-to-take-action-on-a-state-change-of-a-task-run-task-level-state-handler/82
d
Dominik Wagner
07/26/2022, 1:41 PMOkay, thanks! Will hold off until tomorrow then 😁
🙏 1
🙌 1
3 Views