Olivér Atanaszov
12/08/2022, 11:27 AMon_handle
argument of Flow
)?Anna Geller
12/08/2022, 11:58 AMOlivér Atanaszov
12/08/2022, 12:00 PMon_failure
sryAnna Geller
12/08/2022, 12:08 PMfrom prefect import flow, task
import random
@task
def on_failure_task():
print("Taking action on failure here")
@task
def on_success_task():
print("Taking action on success here")
@flow
def bad_subflow():
if random.random() > 0.5:
raise ValueError("Non-deterministic error has occured.")
@flow(log_prints=True, name="Conditional subflows")
def main():
state = bad_subflow(return_state=True)
if state.name == "Failed":
print("Subflow failed! Run on_failure_task")
on_failure_task()
else:
print("Subflow succeeded! Run on_success_task")
on_success_task()
if __name__ == "__main__":
main()
Olivér Atanaszov
12/08/2022, 12:09 PMAnna Geller
12/08/2022, 12:10 PMOlivér Atanaszov
12/08/2022, 12:12 PMAnna Geller
12/08/2022, 12:15 PM