53 lines
1.1 KiB
Python
53 lines
1.1 KiB
Python
import asyncio
|
|
import random
|
|
|
|
from falyx import Action, ChainedAction, Falyx
|
|
from falyx.utils import setup_logging
|
|
|
|
setup_logging()
|
|
|
|
|
|
# A flaky async step that fails randomly
|
|
async def flaky_step():
|
|
await asyncio.sleep(0.2)
|
|
if random.random() < 0.5:
|
|
raise RuntimeError("Random failure!")
|
|
return "ok"
|
|
|
|
|
|
step1 = Action(name="step_1", action=flaky_step, retry=True)
|
|
step2 = Action(name="step_2", action=flaky_step, retry=True)
|
|
|
|
# Chain the actions
|
|
chain = ChainedAction(name="my_pipeline", actions=[step1, step2])
|
|
|
|
# Create the CLI menu
|
|
falyx = Falyx("🚀 Falyx Demo")
|
|
falyx.add_command(
|
|
key="R",
|
|
description="Run My Pipeline",
|
|
action=chain,
|
|
logging_hooks=True,
|
|
preview_before_confirm=True,
|
|
confirm=True,
|
|
)
|
|
|
|
# Create a submenu
|
|
submenu = Falyx("Submenu")
|
|
submenu.add_command(
|
|
key="T",
|
|
description="Test",
|
|
action=lambda: "test",
|
|
logging_hooks=True,
|
|
preview_before_confirm=True,
|
|
confirm=True,
|
|
)
|
|
falyx.add_submenu(
|
|
key="S",
|
|
description="Submenu",
|
|
submenu=submenu,
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(falyx.run())
|