aboutsummaryrefslogtreecommitdiff
path: root/rumba/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'rumba/utils.py')
-rw-r--r--rumba/utils.py16
1 files changed, 12 insertions, 4 deletions
diff --git a/rumba/utils.py b/rumba/utils.py
index 7e3ec80..33ba71c 100644
--- a/rumba/utils.py
+++ b/rumba/utils.py
@@ -29,11 +29,19 @@ import os
import rumba.log as log
import rumba.model as model
-# Fix Python 2.x.
+# Fix input reordering
try:
- input = raw_input
-except NameError:
- pass
+ import builtins # Only in Python 3
+
+ def input(prompt=''):
+ log.flush_log()
+ return builtins.input(prompt)
+except ImportError: # We are in Python 2
+ import __builtin__
+
+ def input(prompt=''):
+ log.flush_log()
+ return __builtin__.raw_input(prompt)
logger = log.get_logger(__name__)