horsies 0.1.0a5__py3-none-any.whl → 0.1.0a6__py3-none-any.whl

horsies/core/banner.py

@@ -9,6 +9,33 @@ if TYPE_CHECKING:
     from horsies.core.app import Horsies
 
 
+# ---------------------------------------------------------------------------
+# ANSI color codes (matching Rust owo-colors output)
+# ---------------------------------------------------------------------------
+
+class Colors:
+    """ANSI escape codes for terminal colors."""
+
+    RESET = '\033[0m'
+    BOLD = '\033[1m'
+    DIMMED = '\033[2m'
+
+    # Standard colors
+    WHITE = '\033[37m'
+    BRIGHT_WHITE = '\033[97m'
+    CYAN = '\033[36m'
+    BRIGHT_CYAN = '\033[96m'
+    YELLOW = '\033[33m'
+    BRIGHT_YELLOW = '\033[93m'
+
+
+def _color(text: str, *codes: str) -> str:
+    """Apply ANSI color codes to text."""
+    if not codes:
+        return text
+    return ''.join(codes) + text + Colors.RESET
+
+
 # Braille art of galloping horse - converted from the golden horse image
 # Each braille character represents a 2x4 pixel block for higher resolution
 HORSE_BRAILLE = """
@@ -28,13 +55,13 @@ HORSE_BRAILLE = """
 ⠚⠒⠒⠒⡶⠚⠒⠒⠒⡰⠓⠒⠒⠒⢲⠓⠒⠒⠒⢻⠿⠀⠀⠚⢿⡷⠐⠒⠒⠚⡆⠒⠒⠒⠚⣖⠒⠒⠒⠚⢶⠒⠒⠒⠚⢶⠂⠐⠒⠛
 """
 
-# Figlet-style "horsies" text
+# Figlet-style "horsies" text with version and tagline
 LOGO_TEXT = r"""
     __                    _
    / /_  ____  __________(_)__  _____
-  / __ \/ __ \/ ___/ ___/ / _ \/ ___/
- / / / / /_/ / /  (__  ) /  __(__  )
-/_/ /_/\____/_/  /____/_/\___/____/
+  / __ \/ __ \/ ___/ ___/ / _ \/ ___/      {version}
+ / / / / /_/ / /  (__  ) /  __(__  )       distributed task queue
+/_/ /_/\____/_/  /____/_/\___/____/        and workflow engine
 """
 
 # Full banner combining horse and logo
@@ -51,14 +78,12 @@ BANNER = (
 
 
 def get_version() -> str:
-    """Get horsies version."""
+    """Get horsies version from package metadata."""
     try:
-        import horsies
-
-        version = getattr(horsies, '__version__', None)
-        return str(version) if version else '0.1.0'
-    except ImportError:
-        return '0.1.0'
+        from importlib.metadata import version
+        return version('horsies')
+    except Exception:
+        return 'dev'
 
 
 def format_banner(version: str | None = None) -> str:
@@ -68,6 +93,48 @@ def format_banner(version: str | None = None) -> str:
     return BANNER.format(version=f'v{version}')
 
 
+# ---------------------------------------------------------------------------
+# Formatting helpers (matching Rust banner.rs)
+# ---------------------------------------------------------------------------
+
+def _format_ms(ms: int) -> str:
+    """Format milliseconds into a human-readable string."""
+    if ms >= 60_000:
+        mins = ms // 60_000
+        remainder_s = (ms % 60_000) // 1_000
+        if remainder_s > 0:
+            return f'{mins}m{remainder_s}s'
+        return f'{mins}m'
+    elif ms >= 1_000:
+        secs = ms // 1_000
+        remainder_ms = ms % 1_000
+        if remainder_ms > 0:
+            return f'{secs}.{remainder_ms // 100}s'
+        return f'{secs}s'
+    return f'{ms}ms'
+
+
+def _format_bool(val: bool) -> str:
+    """Format a boolean for display."""
+    return 'yes' if val else 'no'
+
+
+def _write_section_header(lines: list[str], name: str) -> None:
+    """Write a colored [section] header line."""
+    header = _color(name, Colors.BRIGHT_CYAN, Colors.BOLD)
+    lines.append(f'[{header}]')
+
+
+def _write_kv(lines: list[str], key: str, value: str) -> None:
+    """Write a colored '.> key: value' line with consistent alignment."""
+    prefix = _color('.>', Colors.DIMMED)
+    # Pad key before coloring to maintain alignment
+    padded_key = f'{key}:'.ljust(22)
+    colored_key = _color(padded_key, Colors.WHITE, Colors.BOLD)
+    colored_value = _color(value, Colors.BRIGHT_WHITE)
+    lines.append(f'  {prefix} {colored_key} {colored_value}')
+
+
 def print_banner(
     app: 'Horsies',
     role: str = 'worker',
@@ -91,45 +158,82 @@ def print_banner(
     # Build the banner
     lines: list[str] = []
 
-    # ASCII art header
-    banner = format_banner(version)
-    lines.append(banner)
+    # ASCII art header - cyan colored
+    colored_horse = _color(HORSE_BRAILLE, Colors.CYAN)
+    lines.append(colored_horse)
+
+    # Logo with version - bright yellow bold
+    logo = LOGO_TEXT.replace('{version}', f'v{version}')
+    colored_logo = _color(logo, Colors.BRIGHT_YELLOW, Colors.BOLD)
+    lines.append(colored_logo)
 
-    # Configuration section
-    lines.append('[config]')
-    lines.append(f'  .> app:         {app.__class__.__name__}')
-    lines.append(f'  .> role:        {role}')
-    lines.append(f'  .> queue_mode:  {app.config.queue_mode.name}')
+    # [config] section
+    _write_section_header(lines, 'config')
+    _write_kv(lines, 'app', app.__class__.__name__)
+    _write_kv(lines, 'role', role)
+    _write_kv(lines, 'queue_mode', app.config.queue_mode.name.lower())
 
     # Queue info
     if app.config.queue_mode.name == 'CUSTOM' and app.config.custom_queues:
         queues_str = ', '.join(q.name for q in app.config.custom_queues)
-        lines.append(f'  .> queues:      {queues_str}')
     else:
-        lines.append(f'  .> queues:      default')
+        queues_str = 'default'
+    _write_kv(lines, 'queues', queues_str)
 
-    # Broker info
+    # Broker info (mask password)
     broker_url = app.config.broker.database_url
-    # Mask password in URL
     if '@' in broker_url:
         pre, post = broker_url.split('@', 1)
         if ':' in pre:
             scheme_user = pre.rsplit(':', 1)[0]
             broker_url = f'{scheme_user}:****@{post}'
-    lines.append(f'  .> broker:      {broker_url}')
+    _write_kv(lines, 'broker', broker_url)
 
-    # Concurrency info (if available)
+    # Cluster-wide cap
     if hasattr(app.config, 'cluster_wide_cap') and app.config.cluster_wide_cap:
-        lines.append(f'  .> cap:         {app.config.cluster_wide_cap} (cluster-wide)')
+        _write_kv(lines, 'cluster_cap', f'{app.config.cluster_wide_cap} (cluster-wide)')
+
+    # Prefetch buffer
+    if hasattr(app.config, 'prefetch_buffer') and app.config.prefetch_buffer > 0:
+        _write_kv(lines, 'prefetch', str(app.config.prefetch_buffer))
 
     lines.append('')
 
-    # Tasks section
+    # [recovery] section
+    if hasattr(app.config, 'recovery') and app.config.recovery:
+        recovery = app.config.recovery
+        _write_section_header(lines, 'recovery')
+        _write_kv(
+            lines,
+            'requeue_stale_claimed',
+            _format_bool(recovery.auto_requeue_stale_claimed),
+        )
+        _write_kv(
+            lines,
+            'fail_stale_running',
+            _format_bool(recovery.auto_fail_stale_running),
+        )
+        _write_kv(
+            lines,
+            'check_interval',
+            _format_ms(recovery.check_interval_ms),
+        )
+        _write_kv(
+            lines,
+            'heartbeat_interval',
+            _format_ms(recovery.runner_heartbeat_interval_ms),
+        )
+        lines.append('')
+
+    # [tasks] section
     if show_tasks:
         task_names = app.list_tasks()
-        lines.append(f'[tasks] ({len(task_names)} registered)')
+        tasks_header = _color('tasks', Colors.BRIGHT_CYAN, Colors.BOLD)
+        lines.append(f'[{tasks_header}] ({len(task_names)} registered)')
         for task_name in sorted(task_names):
-            lines.append(f'  . {task_name}')
+            bullet = _color('.', Colors.DIMMED)
+            name = _color(task_name, Colors.WHITE)
+            lines.append(f'  {bullet} {name}')
         lines.append('')
 
     # Print everything
@@ -141,4 +245,14 @@ def print_simple_banner(file: TextIO | None = None) -> None:
     """Print just the ASCII art banner without config."""
     if file is None:
         file = sys.stdout
-    print(format_banner(), file=file)
+
+    version = get_version()
+
+    # Horse art - cyan
+    colored_horse = _color(HORSE_BRAILLE, Colors.CYAN)
+    print(colored_horse, file=file)
+
+    # Logo with version - bright yellow bold
+    logo = LOGO_TEXT.replace('{version}', f'v{version}')
+    colored_logo = _color(logo, Colors.BRIGHT_YELLOW, Colors.BOLD)
+    print(colored_logo, file=file)

{horsies-0.1.0a5.dist-info → horsies-0.1.0a6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: horsies
-Version: 0.1.0a5
+Version: 0.1.0a6
 Summary: A Python library for distributed task execution
 Author: Suleyman Ozkeskin
 License-Expression: MIT

{horsies-0.1.0a5.dist-info → horsies-0.1.0a6.dist-info}/RECORD

@@ -2,7 +2,7 @@ horsies/__init__.py,sha256=rLxes5IuyzggohtXKRsHoddeyqEy0LsgwDlk09itKXk,2769
 horsies/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 horsies/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 horsies/core/app.py,sha256=5HkGil3sZvJAZAerRuPulKUJO-U-vgtrmQxtrVGKc2w,22218
-horsies/core/banner.py,sha256=lgdM5x9aa7kfD0a91wLCSOh61zXZdlA4phq5HagPHZQ,5716
+horsies/core/banner.py,sha256=UuPylHfAutn_Vmxso2qGT_mR0iTeq6NzwBr34iOCmHE,9547
 horsies/core/cli.py,sha256=irlRO9pbpGPZT7r0hSi1d0mEQkAB5M54zLb-6oB6GnY,21078
 horsies/core/errors.py,sha256=3jcu4TcI1P9d2w-Tgz1yTsh4NFPOqvrd5S58cnUEKjE,16768
 horsies/core/logging.py,sha256=p2utHDeOHgvwtSzKTKXh0RwUNCk38ODmgQQkmsD1tWA,2934
@@ -35,8 +35,8 @@ horsies/core/workflows/__init__.py,sha256=JA-8wcYUHp-wF5OCEqJwKElT8PaZZB1G7iglDZ
 horsies/core/workflows/engine.py,sha256=ByIpZnPhQvjydVZSAJuV3Yj6eLzDfZQ3KHtPw93VojI,85894
 horsies/core/workflows/recovery.py,sha256=x-jhYwIZpBxByYyFb5g2vDMtjmUFt33HgOB2J6mgPfs,19377
 horsies/core/workflows/registry.py,sha256=ItpjTN8yK9NNSV8Q8OwDnHdQSO-hxDzxeWAyGvExpRk,3194
-horsies-0.1.0a5.dist-info/METADATA,sha256=rCIhp6_TP-PZEhX630Sv970eChRIyCEldmlVXEQPtFw,1321
-horsies-0.1.0a5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-horsies-0.1.0a5.dist-info/entry_points.txt,sha256=6b_OhuNbJ1ky9WSOt3UfNQXETG2YuH0lKimcibILrEE,50
-horsies-0.1.0a5.dist-info/top_level.txt,sha256=ZQ_NurgT-yr3pE4a1svlO_VAhGJ6NFA31vJDLT1yyOA,8
-horsies-0.1.0a5.dist-info/RECORD,,
+horsies-0.1.0a6.dist-info/METADATA,sha256=W3o2wh8r9BWMryc0t_Reo5D8_OwUhLtCuCaee88PCsI,1321
+horsies-0.1.0a6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+horsies-0.1.0a6.dist-info/entry_points.txt,sha256=6b_OhuNbJ1ky9WSOt3UfNQXETG2YuH0lKimcibILrEE,50
+horsies-0.1.0a6.dist-info/top_level.txt,sha256=ZQ_NurgT-yr3pE4a1svlO_VAhGJ6NFA31vJDLT1yyOA,8
+horsies-0.1.0a6.dist-info/RECORD,,