From 06d2451f950961072a1dbb92e69944334ec966b2 Mon Sep 17 00:00:00 2001 From: Marco Capitani Date: Tue, 27 Feb 2018 14:14:10 +0100 Subject: log: add log to file and flush logger functionalities --- rumba/utils.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'rumba/utils.py') 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__) -- cgit v1.2.3