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

horsies/core/app.py

@@ -82,7 +82,9 @@ class Horsies:
                 f'horsies subprocess initialized with {config.queue_mode.name} mode (pid={os.getpid()})'
             )
         else:
-            self.logger.info(f'horsies initialized as {self._role} with {config.queue_mode.name} mode')
+            self.logger.info(
+                f'horsies initialized as {self._role} with {config.queue_mode.name} mode'
+            )
 
     def set_role(self, role: str) -> None:
         """Set the role and log it. Called by CLI after discovery."""
@@ -192,7 +194,7 @@ class Horsies:
             # Register task with this app, passing source for duplicate detection
             # Normalize path with realpath to handle symlinks and relative paths
             source_str = (
-                f"{os.path.realpath(fn_location.file)}:{fn_location.line}"
+                f'{os.path.realpath(fn_location.file)}:{fn_location.line}'
                 if fn_location
                 else None
             )
@@ -247,15 +249,19 @@ class Horsies:
                     if module_path.endswith('.py') or os.path.sep in module_path:
                         abs_path = os.path.realpath(module_path)
                         if not os.path.exists(abs_path):
-                            errors.append(_no_location(ConfigurationError(
-                                message=f'task module not found: {module_path}',
-                                code=ErrorCode.CLI_INVALID_ARGS,
-                                notes=[f'resolved path: {abs_path}'],
-                                help_text=(
-                                    'remove it from app.discover_tasks([...]) or fix the path; \n'
-                                    'if using globs, run app.expand_module_globs([...]) first'
-                                ),
-                            )))
+                            errors.append(
+                                _no_location(
+                                    ConfigurationError(
+                                        message=f'task module not found: {module_path}',
+                                        code=ErrorCode.CLI_INVALID_ARGS,
+                                        notes=[f'resolved path: {abs_path}'],
+                                        help_text=(
+                                            'remove it from app.discover_tasks([...]) or fix the path; \n'
+                                            'if using globs, run app.expand_module_globs([...]) first'
+                                        ),
+                                    )
+                                )
+                            )
                             continue
                         import_by_path(abs_path)
                     else:
@@ -265,16 +271,20 @@ class Horsies:
                 except HorsiesError as exc:
                     errors.append(exc)
                 except Exception as exc:
-                    errors.append(_no_location(ConfigurationError(
-                        message=f'failed to import module: {module_path}',
-                        code=ErrorCode.CLI_INVALID_ARGS,
-                        notes=[str(exc)],
-                        help_text=(
-                            'ensure the module is importable; '
-                            'for file paths include .py and a valid path, '
-                            'for dotted paths verify PYTHONPATH or run from the project root'
-                        ),
-                    )))
+                    errors.append(
+                        _no_location(
+                            ConfigurationError(
+                                message=f'failed to import module: {module_path}',
+                                code=ErrorCode.CLI_INVALID_ARGS,
+                                notes=[str(exc)],
+                                help_text=(
+                                    'ensure the module is importable; '
+                                    'for file paths include .py and a valid path, '
+                                    'for dotted paths verify PYTHONPATH or run from the project root'
+                                ),
+                            )
+                        )
+                    )
         finally:
             self.suppress_sends(prev_suppress)
         return errors
@@ -283,26 +293,32 @@ class Horsies:
         """Check broker connectivity via SELECT 1."""
         import asyncio
 
+        from sqlalchemy import text
+
+        HEALTH_CHECK_SQL = text("""SELECT 1""")
+
         errors: list[HorsiesError] = []
         try:
             broker = self.get_broker()
 
             async def _test_connection() -> None:
-                from sqlalchemy import text
-
                 async with broker.session_factory() as session:
-                    await session.execute(text('SELECT 1'))
+                    await session.execute(HEALTH_CHECK_SQL)
 
             asyncio.run(_test_connection())
         except HorsiesError as exc:
             errors.append(exc)
         except Exception as exc:
-            errors.append(_no_location(ConfigurationError(
-                message='broker connectivity check failed',
-                code=ErrorCode.BROKER_INVALID_URL,
-                notes=[str(exc)],
-                help_text='check database_url in PostgresConfig',
-            )))
+            errors.append(
+                _no_location(
+                    ConfigurationError(
+                        message='broker connectivity check failed',
+                        code=ErrorCode.BROKER_INVALID_URL,
+                        notes=[str(exc)],
+                        help_text='check database_url in PostgresConfig',
+                    )
+                )
+            )
         return errors
 
     def list_tasks(self) -> list[str]:
@@ -508,26 +524,30 @@ class Horsies:
             if self.config.queue_mode == QueueMode.CUSTOM:
                 valid_queues = self.get_valid_queue_names()
                 if resolved_queue not in valid_queues:
-                    report.add(ConfigurationError(
-                        message='TaskNode queue not in app config',
-                        code=ErrorCode.TASK_INVALID_QUEUE,
+                    report.add(
+                        ConfigurationError(
+                            message='TaskNode queue not in app config',
+                            code=ErrorCode.TASK_INVALID_QUEUE,
+                            notes=[
+                                f"TaskNode '{task.name}' has queue '{resolved_queue}'",
+                                f'valid queues: {valid_queues}',
+                            ],
+                            help_text='use one of the configured queue names or add this queue to app config',
+                        )
+                    )
+                    queue_valid = False
+            elif resolved_queue != 'default':
+                report.add(
+                    ConfigurationError(
+                        message='TaskNode has non-default queue in DEFAULT mode',
+                        code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
                         notes=[
                             f"TaskNode '{task.name}' has queue '{resolved_queue}'",
-                            f'valid queues: {valid_queues}',
+                            "app is in DEFAULT mode (only 'default' queue allowed)",
                         ],
-                        help_text='use one of the configured queue names or add this queue to app config',
-                    ))
-                    queue_valid = False
-            elif resolved_queue != 'default':
-                report.add(ConfigurationError(
-                    message='TaskNode has non-default queue in DEFAULT mode',
-                    code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
-                    notes=[
-                        f"TaskNode '{task.name}' has queue '{resolved_queue}'",
-                        "app is in DEFAULT mode (only 'default' queue allowed)",
-                    ],
-                    help_text='either remove queue override or switch to QueueMode.CUSTOM',
-                ))
+                        help_text='either remove queue override or switch to QueueMode.CUSTOM',
+                    )
+                )
                 queue_valid = False
 
             if queue_valid:

horsies/core/banner.py

@@ -55,22 +55,22 @@ def get_version() -> str:
     try:
         import horsies
 
-        version = getattr(horsies, "__version__", None)
-        return str(version) if version else "0.1.0"
+        version = getattr(horsies, '__version__', None)
+        return str(version) if version else '0.1.0'
     except ImportError:
-        return "0.1.0"
+        return '0.1.0'
 
 
 def format_banner(version: str | None = None) -> str:
     """Format the banner string with version."""
     if version is None:
         version = get_version()
-    return BANNER.format(version=f"v{version}")
+    return BANNER.format(version=f'v{version}')
 
 
 def print_banner(
-    app: "Horsies",
-    role: str = "worker",
+    app: 'Horsies',
+    role: str = 'worker',
     show_tasks: bool = True,
     file: TextIO | None = None,
 ) -> None:
@@ -96,44 +96,44 @@ def print_banner(
     lines.append(banner)
 
     # 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}")
+    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}')
 
     # 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}")
+    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")
+        lines.append(f'  .> queues:      default')
 
     # Broker info
     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}")
+    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}')
 
     # Concurrency info (if available)
-    if hasattr(app.config, "cluster_wide_cap") and app.config.cluster_wide_cap:
-        lines.append(f"  .> cap:         {app.config.cluster_wide_cap} (cluster-wide)")
+    if hasattr(app.config, 'cluster_wide_cap') and app.config.cluster_wide_cap:
+        lines.append(f'  .> cap:         {app.config.cluster_wide_cap} (cluster-wide)')
 
-    lines.append("")
+    lines.append('')
 
     # Tasks section
     if show_tasks:
         task_names = app.list_tasks()
-        lines.append(f"[tasks] ({len(task_names)} registered)")
+        lines.append(f'[tasks] ({len(task_names)} registered)')
         for task_name in sorted(task_names):
-            lines.append(f"  . {task_name}")
-        lines.append("")
+            lines.append(f'  . {task_name}')
+        lines.append('')
 
     # Print everything
-    output = "\n".join(lines)
+    output = '\n'.join(lines)
     print(output, file=file)