mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-21 20:03:55 +08:00
547a033a3c
* Add ProcessOutputProxy * Move launcher to its own field * Move ProcessOutputCapture to its own file * Return itself from __enter__ of OpenpilotPrefix * Integrate ProcessOutputCapture into process_replay * Add note about capture_output_store to README * ipykernel import is optional * Disable type checking for link_with_current_proc * Remove assertion * Decode outputs to utf-8 * read(self): return empty buf if its none * Fix type annotations * Replace fifo with regular file, to avoid hitting fifo size limit
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import os
|
|
import sys
|
|
|
|
from typing import Tuple, no_type_check
|
|
|
|
class FdRedirect:
|
|
def __init__(self, file_prefix: str, fd: int):
|
|
fname = os.path.join("/tmp", f"{file_prefix}.{fd}")
|
|
if os.path.exists(fname):
|
|
os.unlink(fname)
|
|
self.dest_fd = os.open(fname, os.O_WRONLY | os.O_CREAT)
|
|
self.dest_fname = fname
|
|
self.source_fd = fd
|
|
os.set_inheritable(self.dest_fd, True)
|
|
|
|
def __del__(self):
|
|
os.close(self.dest_fd)
|
|
|
|
def purge(self) -> None:
|
|
os.unlink(self.dest_fname)
|
|
|
|
def read(self) -> bytes:
|
|
with open(self.dest_fname, "rb") as f:
|
|
return f.read() or b""
|
|
|
|
def link(self) -> None:
|
|
os.dup2(self.dest_fd, self.source_fd)
|
|
|
|
|
|
class ProcessOutputCapture:
|
|
def __init__(self, proc_name: str, prefix: str):
|
|
prefix = f"{proc_name}_{prefix}"
|
|
self.stdout_redirect = FdRedirect(prefix, 1)
|
|
self.stderr_redirect = FdRedirect(prefix, 2)
|
|
|
|
def __del__(self):
|
|
self.stdout_redirect.purge()
|
|
self.stderr_redirect.purge()
|
|
|
|
@no_type_check # ipython classes have incompatible signatures
|
|
def link_with_current_proc(self) -> None:
|
|
try:
|
|
# prevent ipykernel from redirecting stdout/stderr of python subprocesses
|
|
from ipykernel.iostream import OutStream
|
|
if isinstance(sys.stdout, OutStream):
|
|
sys.stdout = sys.__stdout__
|
|
if isinstance(sys.stderr, OutStream):
|
|
sys.stderr = sys.__stderr__
|
|
except ImportError:
|
|
pass
|
|
|
|
# link stdout/stderr to the fifo
|
|
self.stdout_redirect.link()
|
|
self.stderr_redirect.link()
|
|
|
|
def read_outerr(self) -> Tuple[str, str]:
|
|
out_str = self.stdout_redirect.read().decode()
|
|
err_str = self.stderr_redirect.read().decode()
|
|
return out_str, err_str
|