From c90095f3356e4085120272fd909cc860ace578a9 Mon Sep 17 00:00:00 2001 From: nabeelsherazi Date: Thu, 12 Nov 2020 20:55:57 -0500 Subject: [PATCH] Fix message function not working #96 --- PyInquirer/prompt.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/PyInquirer/prompt.py b/PyInquirer/prompt.py index e4f3811..34c8cc1 100644 --- a/PyInquirer/prompt.py +++ b/PyInquirer/prompt.py @@ -8,7 +8,6 @@ from prompt_toolkit.application import Application - def prompt(questions, answers=None, **kwargs): from . import prompts @@ -36,6 +35,10 @@ def prompt(questions, answers=None, **kwargs): if choices is not None and callable(choices): question['choices'] = choices(answers) + message = question.get('message') + if message is not None and callable(message): + question['message'] = message(answers) + _kwargs = {} _kwargs.update(kwargs) _kwargs.update(question) @@ -56,12 +59,12 @@ def prompt(questions, answers=None, **kwargs): 'Problem in \'when\' check of %s question: %s' % (name, e)) else: - raise ValueError('\'when\' needs to be function that ' \ + raise ValueError('\'when\' needs to be function that ' 'accepts a dict argument') if filter: # at least a little sanity check! if not callable(question['filter']): - raise ValueError('\'filter\' needs to be function that ' \ + raise ValueError('\'filter\' needs to be function that ' 'accepts an argument') if callable(question.get('default')): @@ -70,7 +73,6 @@ def prompt(questions, answers=None, **kwargs): with pt_patch_stdout() if patch_stdout else _dummy_context_manager(): result = getattr(prompts, type_).question(message, **_kwargs) - if isinstance(result, PromptSession): answer = result.prompt() elif isinstance(result, Application): @@ -106,6 +108,7 @@ def prompt(questions, answers=None, **kwargs): return {} return answers + @contextmanager def _dummy_context_manager(): yield