Add menu

menu/callbacks.py

@@ -0,0 +1,73 @@
+import asyncio
+import functools
+import inspect
+import logging
+import random
+import time
+from logging_utils import setup_logging
+from rich.console import Console
+
+console = Console()
+setup_logging()
+logger = logging.getLogger("menu")
+
+def retry(max_retries=3, delay=1, backoff=2, exceptions=(Exception,), logger=None, spinner_text=None):
+    def decorator(func):
+        is_coroutine = inspect.iscoroutinefunction(func)
+
+        @functools.wraps(func)
+        async def async_wrapper(*args, **kwargs):
+            retries, current_delay = 0, delay
+            while retries <= max_retries:
+                if logger:
+                    logger.debug(f"Retrying {retries + 1}/{max_retries} for '{func.__name__}' after {current_delay}s due to '{exceptions}'.")
+                try:
+                    with console.status(spinner_text, spinner="dots"):
+                        return await func(*args, **kwargs)
+                except exceptions as e:
+                    if retries == max_retries:
+                        if logger:
+                            logger.exception(f"❌ Max retries reached for '{func.__name__}': {e}")
+                        raise
+                    if logger:
+                        logger.warning(
+                            f"🔄 Retry {retries + 1}/{max_retries} for '{func.__name__}' after {current_delay}s due to '{e}'."
+                        )
+                    await asyncio.sleep(current_delay)
+                    retries += 1
+                    current_delay *= backoff
+
+        @functools.wraps(func)
+        def sync_wrapper(*args, **kwargs):
+            retries, current_delay = 0, delay
+            while retries <= max_retries:
+                if logger:
+                    logger.debug(f"Retrying {retries + 1}/{max_retries} for '{func.__name__}' after {current_delay}s due to '{exceptions}'.")
+                try:
+                    with console.status(spinner_text, spinner="dots"):
+                        return func(*args, **kwargs)
+                except exceptions as e:
+                    if retries == max_retries:
+                        if logger:
+                            logger.exception(f"❌ Max retries reached for '{func.__name__}': {e}")
+                        raise
+                    if logger:
+                        logger.warning(
+                            f"🔄 Retry {retries + 1}/{max_retries} for '{func.__name__}' after {current_delay}s due to '{e}'."
+                        )
+                    time.sleep(current_delay)
+                    retries += 1
+                    current_delay *= backoff
+
+        return async_wrapper if is_coroutine else sync_wrapper
+    return decorator
+
+@retry(max_retries=10, delay=1, logger=logger, spinner_text="Trying risky thing...")
+def might_fail():
+    time.sleep(4)
+    if random.random() < 0.6:
+        raise ValueError("Simulated failure")
+    return "🎉 Success!"
+
+result = might_fail()
+print(result)