Allowing passing NativeReportingMode.LAST and ALL to print_thread

Author: Kretzer6019

src/pystack/__main__.py

@@ -292,8 +292,7 @@ def process_remote(parser: argparse.ArgumentParser, args: argparse.Namespace) ->
         locals=args.locals,
         method=StackMethod.ALL if args.exhaustive else StackMethod.AUTO,
     ):
-        native = args.native_mode != NativeReportingMode.OFF
-        print_thread(thread, native)
+        print_thread(thread, args.native_mode)
 
 
 def format_psinfo_information(psinfo: Dict[str, Any]) -> str:
@@ -423,8 +422,7 @@ def process_core(parser: argparse.ArgumentParser, args: argparse.Namespace) -> N
         locals=args.locals,
         method=StackMethod.ALL if args.exhaustive else StackMethod.AUTO,
     ):
-        native = args.native_mode != NativeReportingMode.OFF
-        print_thread(thread, native)
+        print_thread(thread, args.native_mode)
 
 
 if __name__ == "__main__":  # pragma: no cover

src/pystack/_pystack.pyi

@@ -24,10 +24,10 @@ class CoreFileAnalyzer:
     def missing_modules(self) -> List[str]: ...
 
 class NativeReportingMode(enum.Enum):
-    ALL = 1
-    OFF = 2
-    PYTHON = 3
-    LAST = 4
+    ALL = ...
+    OFF = ...
+    PYTHON = ...
+    LAST = ...
 
 class StackMethod(enum.Enum):
     ALL = 1

src/pystack/traceback_formatter.py

@@ -4,15 +4,16 @@
 from typing import Optional
 
 from .colors import colored
+from .engine import NativeReportingMode
 from .types import NativeFrame
 from .types import PyCodeObject
 from .types import PyFrame
 from .types import PyThread
 from .types import frame_type
 
 
-def print_thread(thread: PyThread, native: bool) -> None:
-    for line in format_thread(thread, native):
+def print_thread(thread: PyThread, native_mode: NativeReportingMode) -> None:
+    for line in format_thread(thread, native_mode):
         print(line, file=sys.stdout, flush=True)
 
 
@@ -62,7 +63,8 @@ def _are_the_stacks_mergeable(thread: PyThread) -> bool:
     return n_eval_frames == n_entry_frames
 
 
-def format_thread(thread: PyThread, native: bool) -> Iterable[str]:
+def format_thread(thread: PyThread, native_mode: NativeReportingMode) -> Iterable[str]:
+    native = native_mode != NativeReportingMode.OFF
     current_frame: Optional[PyFrame] = thread.first_frame
     if current_frame is None and not native:
         yield f"The frame stack for thread {thread.tid} is empty"