return without calling _cmdloop() after error

a quick fix for #1
simonsobs · Feb 18, 2022 · c93e93f · c93e93f
1 parent f866f09
commit c93e93f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/nextline/pdb/custom.py b/nextline/pdb/custom.py
@@ -37,7 +37,10 @@ def trace_dispatch(self, frame, event, arg):
 
     def _cmdloop(self):
         """Prompt user input"""
-        self._registrar.entering_cmdloop()
+        try:
+            self._registrar.entering_cmdloop()
+        except RuntimeError:
+            return
         super()._cmdloop()
         self._registrar.exited_cmdloop()
 

diff --git a/nextline/pdb/proxy.py b/nextline/pdb/proxy.py
@@ -113,6 +113,9 @@ def exited_trace(self) -> None:
         self._current_trace_args = None
 
     def entering_cmdloop(self) -> None:
+        if not self._current_trace_args:
+            raise RuntimeError("calling_trace() must be called first")
+
         frame, event, _ = self._current_trace_args
 
         module_name = frame.f_globals.get("__name__")