utils: Add check for Python 2

This adds a check in case Python 2 is used, since it doesn't have
input(). It assigns raw_input to input if this is the case.

1 files changed, 6 insertions, 0 deletions

diff --git a/rumba/utils.py b/rumba/utils.py
index 0e8867d..7417074 100644
--- a/rumba/utils.py
+++ b/rumba/utils.py
@@ -29,6 +29,12 @@ import time
 import rumba.log as log
 import rumba.model as model
 
+# Fix Python 2.x.
+try:
+    input = raw_input
+except NameError:
+    pass
+
 logger = log.get_logger(__name__)