Re-use is_main_thread

sebastinas · sebastinas · commit 052b3a3620d9 · 2017-09-14T00:31:31.000+02:00
Signed-off-by: Sebastian Ramacher &lt;sebastian@ramacher.at&gt;
diff --git a/bpython/curtsiesfrontend/coderunner.py b/bpython/curtsiesfrontend/coderunner.py
@@ -18,7 +18,7 @@
 import greenlet
 import logging
 
-from bpython._py3compat import py3
+from bpython._py3compat import py3, is_main_thread
 from bpython.config import getpreferredencoding
 
 logger = logging.getLogger(__name__)
@@ -131,18 +131,17 @@ def run_code(self, for_code=None):
         if source code is complete, returns "done"
         if source code is incomplete, returns "unfinished"
         """
-        is_main_thread = isinstance(threading.current_thread(), threading._MainThread)
         if self.code_context is None:
             assert self.source is not None
             self.code_context = greenlet.greenlet(self._blocking_run_code)
-            if is_main_thread:
+            if is_main_thread():
                 self.orig_sigint_handler = signal.getsignal(signal.SIGINT)
                 signal.signal(signal.SIGINT, self.sigint_handler)
             request = self.code_context.switch()
         else:
             assert self.code_is_waiting
             self.code_is_waiting = False
-            if is_main_thread:
+            if is_main_thread():
                 signal.signal(signal.SIGINT, self.sigint_handler)
             if self.sigint_happened_in_main_context:
                 self.sigint_happened_in_main_context = False
@@ -161,7 +160,7 @@ def run_code(self, for_code=None):
             return False
         elif isinstance(request, (Done, Unfinished)):
             self._unload_code()
-            if is_main_thread:
+            if is_main_thread():
                 signal.signal(signal.SIGINT, self.orig_sigint_handler)
             self.orig_sigint_handler = None
             return request
