Merge pull request #23 from gregoil/feature/add-breakpoint-signal

Added breakpoint signal

ipdbugger/__init__.py

@@ -25,12 +25,15 @@
 import types
 import inspect
 import traceback
+from bdb import BdbQuit
 
 import colorama
 from termcolor import colored
 from future.utils import raise_
 from IPython.terminal.debugger import TerminalPdb
 
+from .signals import register_break_signal
+
 # Enable color printing on screen.
 colorama.init()
 
@@ -287,7 +290,8 @@ def get_last_lineno(node):
     return max_lineno
 
 
-def debug(victim=None, ignore_exceptions=(), catch_exception=None, depth=0):
+def debug(victim=None, ignore_exceptions=(BdbQuit,),
+          catch_exception=None, depth=0):
     """A decorator function to catch exceptions and enter debug mode.
 
     Args:
@@ -313,6 +317,7 @@ def wrapper(real_victim):
 
         return wrapper
 
+    register_break_signal()
     if inspect.isfunction(victim):
         if hasattr(victim, '_ipdebug_wrapped'):
             # Don't wrap the function more than once

ipdbugger/signals.py

@@ -0,0 +1,31 @@
+"""Signals usage in Rotest."""
+# pylint: disable=global-statement,no-member
+import sys
+import signal
+
+
+BREAKPOINT_SIGNAL_REGISTERED = False
+
+
+class BreakPointException(Exception):
+    """An exception to raise on the break signal."""
+
+
+def raise_exception_handler(_signum, _frame):
+    """Raise a break-point exception."""
+    raise BreakPointException()
+
+
+def register_break_signal():
+    """Register raising an exception on break signal if needed."""
+    global BREAKPOINT_SIGNAL_REGISTERED
+    if BREAKPOINT_SIGNAL_REGISTERED:
+        return
+
+    BREAKPOINT_SIGNAL_REGISTERED = True
+
+    if sys.platform == "win32":
+        signal.signal(signal.SIGBREAK, raise_exception_handler)
+
+    else:
+        signal.signal(signal.SIGQUIT, raise_exception_handler)

setup.py

@@ -1,7 +1,7 @@
 """Setup file for handling packaging and distribution."""
 from setuptools import setup
 
-__version__ = "2.4.1"
+__version__ = "2.5.0"
 
 setup(
     name="ipdbugger",

tests/test_debug.py

@@ -352,6 +352,21 @@ def func_upper():
         assert name_saver.func_name == "func_lowest"
 
 
+def test_depth_and_catch_attribute():
+    """Test wrapping a function one more call level."""
+    def func_lower():
+        raise utils.TestError()
+
+    def func_upper():
+        try:
+            func_lower()
+        except (utils.TestError, ValueError):
+            pass
+
+    func_upper = debug(func_upper, depth=1)
+    func_upper()
+
+
 def test_ignore_all_exceptions():
     """Test ignoring all exceptions."""
     def func():