Bring back the good part of fe4fd3a

ixdiagnose/utils/command.py

@@ -1,6 +1,5 @@
 import subprocess
-
-from typing import Optional, Union
+from typing import Any, Optional, Union
 
 from .run import run
 
@@ -14,12 +13,14 @@ def __init__(
         self.command: Union[str, list] = command
         self.description: str = description
         self.env: Optional[dict] = env
-        self.max_length: Optional[int] = max_lines
+        self.max_lines: Optional[int] = max_lines
         self.serializable: bool = serializable
         self.safe_returncodes: list = safe_returncodes or [0]
+        self.execution_context: Any = None
 
     def execute(self) -> subprocess.CompletedProcess:
         cp = run(self.command, check=False, env=self.env)
-        if self.max_length and cp.returncode in self.safe_returncodes:
-            cp.stdout = cp.stdout.splitlines()[:self.max_length]
+        if cp.returncode in self.safe_returncodes:
+            if self.max_lines:
+                cp.stdout = "\n".join(cp.stdout.splitlines()[:self.max_lines])
         return cp