pyallel 0.3.4__tar.gz → 0.4.0__tar.gz

{pyallel-0.3.4 → pyallel-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyallel
-Version: 0.3.4
+Version: 0.4.0
 Summary: Run and handle the output of multiple executables in pyallel (as in parallel)
 Home-page: https://github.com/Danthewaann/pyallel
 License: MIT
@@ -53,6 +53,7 @@ pyallel "black --color --check --diff ." "mypy ." "ruff check --no-fix ."
 - [ ] Allow a single main command output to be streamed to stdout, while all other
       commands will only get outputted after the main command has completed (such as running
       `pytest` as the main command, whilst running `mypy`, `ruff` etc. as other commands)
+- [x] Provide a way to set environment variables for each command to run with
 - [ ] Allow list of files to be provided to supply as input arguments to each command
 - [ ] Allow input to be piped into `pyallel` via stdin to supply as standard input to each
       command

{pyallel-0.3.4 → pyallel-0.4.0}/README.md

@@ -32,6 +32,7 @@ pyallel "black --color --check --diff ." "mypy ." "ruff check --no-fix ."
 - [ ] Allow a single main command output to be streamed to stdout, while all other
       commands will only get outputted after the main command has completed (such as running
       `pytest` as the main command, whilst running `mypy`, `ruff` etc. as other commands)
+- [x] Provide a way to set environment variables for each command to run with
 - [ ] Allow list of files to be provided to supply as input arguments to each command
 - [ ] Allow input to be piped into `pyallel` via stdin to supply as standard input to each
       command

{pyallel-0.3.4 → pyallel-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyallel"
-version = "0.3.4"
+version = "0.4.0"
 description = "Run and handle the output of multiple executables in pyallel (as in parallel)"
 authors = ["Daniel Black <danielcrblack@gmail.com>"]
 license = "MIT"

{pyallel-0.3.4 → pyallel-0.4.0}/src/pyallel/main.py

@@ -1,18 +1,12 @@
 from __future__ import annotations
-import io
 
-import os
 import sys
 import time
-import shutil
-from dataclasses import dataclass
-import shlex
 import importlib.metadata
-import subprocess
-from typing import IO
 
 from pyallel.errors import InvalidExecutableErrors, InvalidExecutableError
 from pyallel.parser import Arguments, create_parser
+from pyallel.process import Process
 
 IN_TTY = sys.__stdin__.isatty()
 
@@ -42,57 +36,13 @@ TICK = "\u2713"
 X = "\u2717"
 
 
-@dataclass
-class Process:
-    name: str
-    args: list[str]
-    start: float = 0.0
-    process: subprocess.Popen[bytes] | None = None
-
-    def run(self) -> None:
-        env = os.environ.copy()
-        # TODO: need to provide a way to supply environment variablesreturn
-        # for each provided command
-        env["MYPY_FORCE_COLOR"] = "1" if IN_TTY else "0"
-        self.start = time.perf_counter()
-        self.process = subprocess.Popen(
-            [self.name, *self.args],
-            stdout=subprocess.PIPE,
-            stderr=subprocess.STDOUT,
-            env=env,
-        )
-
-    def poll(self) -> int | None:
-        if self.process:
-            return self.process.poll()
-        return None
-
-    def stdout(self) -> IO[bytes]:
-        if self.process and self.process.stdout:
-            return self.process.stdout
-        return io.BytesIO(b"")
-
-    def return_code(self) -> int | None:
-        if self.process:
-            return self.process.returncode
-        return None
-
-
-def parse_command(command: str) -> Process:
-    executable, *args = command.split(maxsplit=1)
-    if not shutil.which(executable):
-        raise InvalidExecutableError(executable)
-    args = shlex.split(" ".join(args))
-    return Process(name=executable, args=args)
-
-
 def run_commands(commands: list[str]) -> list[Process]:
     processes: list[Process] = []
     errors: list[InvalidExecutableError] = []
 
     for command in commands:
         try:
-            processes.append(parse_command(command))
+            processes.append(Process.from_command(command))
         except InvalidExecutableError as e:
             errors.append(e)
 
@@ -257,3 +207,7 @@ def run(*args: str) -> int:
 
 def entry_point() -> None:
     sys.exit(run(*sys.argv[1:]))
+
+
+if __name__ == "__main__":
+    entry_point()

pyallel-0.4.0/src/pyallel/process.py

@@ -0,0 +1,63 @@
+from __future__ import annotations
+
+import time
+import subprocess
+import shlex
+import io
+import shutil
+import os
+from typing import IO
+
+from dataclasses import dataclass, field
+from pyallel.errors import InvalidExecutableError
+
+
+@dataclass
+class Process:
+    name: str
+    args: list[str]
+    env: dict[str, str] = field(default_factory=dict)
+    start: float = 0.0
+    process: subprocess.Popen[bytes] | None = None
+
+    def run(self) -> None:
+        self.start = time.perf_counter()
+        self.process = subprocess.Popen(
+            [self.name, *self.args],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            env=self.env,
+        )
+
+    def poll(self) -> int | None:
+        if self.process:
+            return self.process.poll()
+        return None
+
+    def stdout(self) -> IO[bytes]:
+        if self.process and self.process.stdout:
+            return self.process.stdout
+        return io.BytesIO(b"")
+
+    def return_code(self) -> int | None:
+        if self.process:
+            return self.process.returncode
+        return None
+
+    @classmethod
+    def from_command(cls, command: str) -> Process:
+        env = os.environ.copy()
+        args = command.split()
+        parsed_args: list[str] = []
+        for arg in args:
+            if "=" in arg:
+                name, value = arg.split("=")
+                env[name] = value
+            else:
+                parsed_args.append(arg)
+
+        if not shutil.which(parsed_args[0]):
+            raise InvalidExecutableError(parsed_args[0])
+
+        args = shlex.split(" ".join(parsed_args[1:]))
+        return Process(name=parsed_args[0], args=args, env=env)