diff options
Diffstat (limited to 'test_task.py')
-rwxr-xr-x | test_task.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/test_task.py b/test_task.py index ac44201..1ad9bdc 100755 --- a/test_task.py +++ b/test_task.py @@ -28,16 +28,16 @@ DEFAULT_LANGUAGE = 'si' def print_header(title, spacing=1): print('\n'*spacing + '> {}'.format(title)) +def rlinput(prompt, prefill=''): + readline.set_startup_hook(lambda: readline.insert_text(prefill)) + try: + return input(prompt) + finally: + readline.set_startup_hook() + # get the parameters for a task either from the user or from a file def get_params(params, params_meta, language=None): # prefill input() prompt with given text - def rlinput(prompt, prefill=''): - readline.set_startup_hook(lambda: readline.insert_text(prefill)) - try: - return input(prompt) - finally: - readline.set_startup_hook() - if language is None: language = params.get('language', DEFAULT_LANGUAGE) |