lint

Author: dacort

src/pyssm_client/utils/command.py

@@ -12,7 +12,6 @@
 from ..communicator.utils import create_websocket_config
 
 
-
 def _filter_shell_output(data: bytes, original_command: str) -> bytes:
     """Filter shell prompts, command echoes, and ANSI sequences from output."""
     import re
@@ -148,35 +147,37 @@ async def run_command(
     stderr_buf = bytearray()
     session_done = asyncio.Event()
     exit_code = 0
-    
+
     # Line buffers for streaming
     stdout_line_buf = bytearray()
     stderr_line_buf = bytearray()
-    
+
     def handle_stdout(data: bytes) -> None:
         """Handle stdout from shell."""
         nonlocal stdout_buf, exit_code, stdout_line_buf
 
         # Add to line buffer for streaming
         stdout_line_buf.extend(data)
-        
+
         # Process complete lines for streaming
-        while b'\n' in stdout_line_buf:
-            line_end = stdout_line_buf.index(b'\n')
-            line = stdout_line_buf[:line_end + 1]
-            stdout_line_buf = stdout_line_buf[line_end + 1:]
-            
+        while b"\n" in stdout_line_buf:
+            line_end = stdout_line_buf.index(b"\n")
+            line = stdout_line_buf[: line_end + 1]
+            stdout_line_buf = stdout_line_buf[line_end + 1 :]
+
             # Apply existing filter to the line and stream if requested
             if stream_output:
                 filtered = _filter_shell_output(line, command)
                 if filtered and filtered.strip():
                     try:
                         import sys
+
                         sys.stdout.buffer.write(filtered)
                         sys.stdout.buffer.flush()
                     except Exception:
                         try:
                             import sys
+
                             sys.stdout.write(filtered.decode("utf-8", errors="replace"))
                             sys.stdout.flush()
                         except Exception:
@@ -205,32 +206,34 @@ def handle_stdout(data: bytes) -> None:
     def handle_stderr(data: bytes) -> None:
         """Handle stderr from shell."""
         nonlocal stderr_buf, stderr_line_buf
-        
+
         # Add to line buffer for streaming
         stderr_line_buf.extend(data)
-        
+
         # Process complete lines for streaming
-        while b'\n' in stderr_line_buf:
-            line_end = stderr_line_buf.index(b'\n')
-            line = stderr_line_buf[:line_end + 1]
-            stderr_line_buf = stderr_line_buf[line_end + 1:]
-            
+        while b"\n" in stderr_line_buf:
+            line_end = stderr_line_buf.index(b"\n")
+            line = stderr_line_buf[: line_end + 1]
+            stderr_line_buf = stderr_line_buf[line_end + 1 :]
+
             # Apply existing filter to stderr line and stream if requested
             if stream_output:
                 filtered = _filter_shell_output(line, command)
                 if filtered and filtered.strip():
                     try:
                         import sys
+
                         sys.stderr.buffer.write(filtered)
                         sys.stderr.buffer.flush()
                     except Exception:
                         try:
                             import sys
+
                             sys.stderr.write(filtered.decode("utf-8", errors="replace"))
                             sys.stderr.flush()
                         except Exception:
                             pass
-        
+
         stderr_buf.extend(data)
 
     def handle_closed() -> None:
@@ -281,7 +284,6 @@ def handle_closed() -> None:
         else:
             final_stdout = _filter_shell_output(bytes(stdout_buf), command)
 
-
         return CommandResult(
             stdout=final_stdout,
             stderr=bytes(stderr_buf),  # Now using proper stderr separation