Roland Thomas
f37aee568d
- Added new `SpinnerManager` module for centralized spinner rendering using Rich `Live`. - Introduced `spinner`, `spinner_message`, `spinner_type`, `spinner_style`, and `spinner_speed` to `BaseAction` and subclasses (`Action`, `ProcessAction`, `HTTPAction`, `ActionGroup`, `ChainedAction`). - Registered `spinner_before_hook` and `spinner_teardown_hook` automatically when `spinner=True`. - Reworked `Command` spinner logic to use the new hook-based system instead of `console.status`. - Updated `OptionsManager` to include a `SpinnerManager` instance for global state. - Enhanced pipeline demo to showcase spinners across chained and grouped actions. - Bumped version to 0.1.77. This commit unifies spinner handling across commands, actions, and groups, making spinners consistent and automatically managed by hooks.
109 lines
2.6 KiB
Python
109 lines
2.6 KiB
Python
import asyncio
|
|
import random
|
|
import time
|
|
|
|
from falyx import Falyx
|
|
from falyx.action import Action, ActionGroup, ChainedAction, ProcessAction
|
|
from falyx.retry import RetryHandler, RetryPolicy
|
|
|
|
|
|
# Step 1: Fast I/O-bound setup (standard Action)
|
|
async def checkout_code():
|
|
print("📥 Checking out code...")
|
|
await asyncio.sleep(0.5)
|
|
|
|
|
|
# Step 2: CPU-bound task (ProcessAction)
|
|
def run_static_analysis():
|
|
print("🧠 Running static analysis (CPU-bound)...")
|
|
total = 0
|
|
for i in range(10_000_000):
|
|
total += i % 3
|
|
time.sleep(5)
|
|
return total
|
|
|
|
|
|
# Step 3: Simulated flaky test with retry
|
|
async def flaky_tests():
|
|
await asyncio.sleep(0.3)
|
|
if random.random() < 0.3:
|
|
raise RuntimeError("❌ Random test failure!")
|
|
print("🧪 Tests passed.")
|
|
return "ok"
|
|
|
|
|
|
# Step 4: Multiple deploy targets (parallel ActionGroup)
|
|
async def deploy_to(target: str):
|
|
print(f"🚀 Deploying to {target}...")
|
|
await asyncio.sleep(random.randint(2, 6))
|
|
return f"{target} complete"
|
|
|
|
|
|
def build_pipeline():
|
|
retry_handler = RetryHandler(RetryPolicy(max_retries=3, delay=0.5))
|
|
|
|
# Base actions
|
|
checkout = Action("Checkout", checkout_code)
|
|
analysis = ProcessAction(
|
|
"Static Analysis",
|
|
run_static_analysis,
|
|
spinner=True,
|
|
spinner_message="Analyzing code...",
|
|
)
|
|
tests = Action("Run Tests", flaky_tests)
|
|
tests.hooks.register("on_error", retry_handler.retry_on_error)
|
|
|
|
# Parallel deploys
|
|
deploy_group = ActionGroup(
|
|
"Deploy to All",
|
|
[
|
|
Action(
|
|
"Deploy US",
|
|
deploy_to,
|
|
args=("us-west",),
|
|
spinner=True,
|
|
spinner_message="Deploying US...",
|
|
),
|
|
Action(
|
|
"Deploy EU",
|
|
deploy_to,
|
|
args=("eu-central",),
|
|
spinner=True,
|
|
spinner_message="Deploying EU...",
|
|
),
|
|
Action(
|
|
"Deploy Asia",
|
|
deploy_to,
|
|
args=("asia-east",),
|
|
spinner=True,
|
|
spinner_message="Deploying Asia...",
|
|
),
|
|
],
|
|
)
|
|
|
|
# Full pipeline
|
|
return ChainedAction("CI/CD Pipeline", [checkout, analysis, tests, deploy_group])
|
|
|
|
|
|
pipeline = build_pipeline()
|
|
|
|
|
|
# Run the pipeline
|
|
async def main():
|
|
|
|
flx = Falyx()
|
|
flx.add_command(
|
|
"A",
|
|
"Action Thing",
|
|
pipeline,
|
|
spinner=True,
|
|
spinner_type="line",
|
|
spinner_message="Running pipeline...",
|
|
)
|
|
|
|
await flx.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|