Merge pull request #38 from JungWinter/fix-ipython

#35: Improve keyboard interrupt handling in iPython

halo/halo.py

@@ -5,6 +5,7 @@
 from __future__ import unicode_literals, absolute_import, print_function
 
 import sys
+import signal
 import threading
 import time
 import functools
@@ -64,6 +65,13 @@ def __init__(self, text='', color='cyan', spinner=None, animation=None, interval
         self._spinner_id = None
         self._enabled = enabled  # Need to check for stream
 
+        def handle_keyboard_interrupt(signal, frame):
+            """Handle KeyboardInterrupt without try-except statement"""
+            self.fail(self.text)
+            raise KeyboardInterrupt
+
+        signal.signal(signal.SIGINT, handle_keyboard_interrupt)
+
     def __enter__(self):
         """Starts the spinner on a separate thread. For use in context managers.
         Returns

tests/test_halo.py

@@ -5,6 +5,7 @@
 import unittest
 import time
 import sys
+import signal
 import io
 import logging
 import os