Add pudb.b

This lets you just run import pudb.b to trigger a set_trace(). It works by monkey-patching __builtins__.__import__. It does not yet work in Python 3.3, due to the new import mechanism. See #109.
inducer · Feb 27, 2014 · f6492a5 · f6492a5
1 parent 254630e
commit f6492a5
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/pudb/b.py b/pudb/b.py
@@ -0,0 +1,19 @@
+import sys
+
+from pudb import _get_debugger, set_interrupt_handler
+
+def __myimport__(name, *args, **kwargs):
+    if name == 'pudb.b':
+        set_trace()
+    return __origimport__(name, *args, **kwargs)
+
+# Will only be run on first import
+__builtins__['__origimport__'] = __import__
+__builtins__['__import__'] = __myimport__
+
+def set_trace():
+    dbg = _get_debugger()
+    set_interrupt_handler()
+    dbg.set_trace(sys._getframe().f_back.f_back)
+
+set_trace()