Move compatible input to py2compat module

2 files changed, 10 insertions, 6 deletions

diff --git a/pandora/py2compat.py b/pandora/py2compat.py
index fe45462..b8767ba 100644
--- a/pandora/py2compat.py
+++ b/pandora/py2compat.py
@@ -18,6 +18,15 @@ except ImportError:
             return self.readfp(fp)
 
 
+# Used in pydora
+def input(prompt):
+    try:
+        return raw_input(prompt)
+    except NameError:
+        import builtins
+        return builtins.input(prompt)
+
+
 # Only used in tests
 try:
     from unittest.mock import Mock, MagicMock, call, patch  # noqa: F401
diff --git a/pydora/utils.py b/pydora/utils.py
index 3c48117..9a6d2a1 100644
--- a/pydora/utils.py
+++ b/pydora/utils.py
@@ -4,6 +4,7 @@ import os
 import sys
 import getpass
 import subprocess
+from pandora.py2compat import input
 
 try:
     import termios
@@ -12,12 +13,6 @@ except ImportError:
     termios = None
 
 
-def input(prompt):
-    try:
-        return raw_input(prompt)
-    except NameError:
-        import builtins
-        return builtins.input(prompt)
 
 
 class Colors(object):