Move is_main_thread to bpython._py3compat

sebastinas · sebastinas · commit a975340d8917 · 2017-09-14T00:30:59.000+02:00
Signed-off-by: Sebastian Ramacher &lt;sebastian@ramacher.at&gt;
diff --git a/bpython/_py3compat.py b/bpython/_py3compat.py
@@ -37,14 +37,17 @@
 from __future__ import absolute_import
 
 import sys
+import threading
 
 py3 = (sys.version_info[0] == 3)
 
+
 if py3:
     from pygments.lexers import Python3Lexer as PythonLexer
 else:
     from pygments.lexers import PythonLexer
 
+
 if py3 or sys.version_info[:3] >= (2, 7, 3):
     def prepare_for_exec(arg, encoding=None):
         return arg
@@ -65,3 +68,12 @@ def try_decode(s, encoding):
             except UnicodeDecodeError:
                 return None
         return s
+
+
+if py3:
+    def is_main_thread():
+        return threading.main_thread() == threading.current_thread()
+else:
+    def is_main_thread():
+        return isinstance(threading.current_thread(), threading._MainThread)
+
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -34,7 +34,7 @@
 from bpython.formatter import BPythonFormatter
 from bpython import autocomplete
 from bpython.translations import _
-from bpython._py3compat import py3
+from bpython._py3compat import py3, is_main_thread
 from bpython.pager import get_pager_command
 
 from bpython.curtsiesfrontend import replpainter as paint
@@ -99,14 +99,6 @@
     unicode = str
 
 
-if py3:
-    def is_main_thread():
-        return threading.main_thread() == threading.current_thread()
-else:
-    def is_main_thread():
-        return isinstance(threading.current_thread(), threading._MainThread)
-
-
 class FakeStdin(object):
     """The stdin object user code will reference
 
