Fix -p / --parallel handling of tasks.

CHANGES.md

@@ -1,5 +1,9 @@
 # Release Notes
 
+## 0.10.4
+
+Fix `-p` / `--parallel` handling of tasks - no longer flatten them.
+
 ## 0.10.3
 
 Gracefully handle keyboard interrupt and kill in-flight processes with logging.

dev_cmd/__init__.py

@@ -1,4 +1,4 @@
 # Copyright 2024 John Sirois.
 # Licensed under the Apache License, Version 2.0 (see LICENSE).
 
-__version__ = "0.10.3"
+__version__ = "0.10.4"

dev_cmd/invoke.py

@@ -4,15 +4,14 @@
 from __future__ import annotations
 
 import asyncio
-import itertools
 import os
 import sys
 from asyncio import CancelledError
 from asyncio.subprocess import Process
 from asyncio.tasks import Task as AsyncTask
 from contextlib import asynccontextmanager
 from dataclasses import dataclass, field
-from typing import Any, AsyncIterator, Container, Iterator
+from typing import Any, AsyncIterator, Container
 
 from dev_cmd import color
 from dev_cmd.color import USE_COLOR
@@ -21,14 +20,6 @@
 from dev_cmd.model import Command, ExitStyle, Group, Task
 
 
-def _flatten(step: Command | Group | Task) -> Iterator[Command | Group]:
-    if isinstance(step, Task):
-        for step in step.steps.members:
-            yield from _flatten(step)
-    else:
-        yield step
-
-
 def _step_prefix(step_name: str) -> str:
     return color.cyan(f"dev-cmd {color.bold(step_name)}]")
 
@@ -134,15 +125,7 @@ async def invoke_parallel(
     ) -> None:
         async with _guarded_stdin(), self._guarded_ctrl_c():
             if error := await self._invoke_group(
-                "*",
-                Group(
-                    members=tuple(
-                        itertools.chain.from_iterable(_flatten(task) for task in self.steps)
-                    )
-                ),
-                *extra_args,
-                serial=False,
-                exit_style=exit_style,
+                "*", Group(members=self.steps), *extra_args, serial=False, exit_style=exit_style
             ):
                 await self._terminate_in_flight_processes()
                 raise error