mytype

Author: joyguoguo

fuzz/command_util.py

@@ -4,11 +4,10 @@
 import time
 import logging
 from pathlib import Path
-from typing import Optional
+from typing import Optional, cast
 from returns.maybe import Maybe
 from errors import CommandError
 
-
 def _run_subprocess(
     cmd: str,
     cwd: Optional[Path] = None,
@@ -39,17 +38,21 @@ def _run_subprocess(
 
     try:
         if capture_output:
-            assert process.stdout is not None  # ✅ MyPy static check
+            # 类型断言确保 stdout 不为 None
+            stdout = cast(Optional[subprocess.PIPE], process.stdout)
+            if stdout is None:
+                raise RuntimeError("Stdout should not be None when capture_output is True")
+                
             while True:
-                line = process.stdout.readline()
+                line = stdout.readline()
                 if line:
                     output_lines.append(line.rstrip())
                     if logger:
                         logger.debug(line.rstrip())
                 elif process.poll() is not None:
                     break
 
-                if timeout and (time.time() - start_time) > timeout:
+                if timeout is not None and (time.time() - start_time) > timeout:
                     if logger:
                         logger.error(f"⌛ Command timed out after {timeout} seconds")
                     process.terminate()
@@ -107,4 +110,4 @@ def run_command_fuzz_all_targets(
     if exit_code not in [0, *allowed_codes]:
         logger.error(f"❌ Command execution failed, exit code: {exit_code}")
         return False
-    return True
+    return True