except ImportError as _readline_import_error:
_readline = None
-from . import UserInterface
+try:
+ from pygments.console import colorize as _colorize
+except ImportError as e:
+ def _colorize(color_key=None, text=None):
+ return text
+ print(e)
+
+from .. import error as _error
+from .. import question as _question
+from . import UserInterface as _UserInterface
class QuestionCommandLine (_cmd.Cmd):
def __init__(self, ui):
super(QuestionCommandLine, self).__init__()
self.ui = ui
+ if self.ui.quiz.introduction:
+ self.intro = '\n\n'.join([self.intro, self.ui.quiz.introduction])
+ self._tempdir = None
+
+ def get_question(self):
+ self.question = self.ui.get_question(user=self.ui.user)
+ if self.question:
+ self._reset()
+ else:
+ return True # out of questions
def preloop(self):
- self.question = self.ui.get_question()
- self._reset()
+ self.get_question()
def _reset(self):
self.answers = []
+ if self._tempdir:
+ self._tempdir.cleanup() # occasionally redundant, but that's ok
+ self._tempdir = None
self._set_ps1()
def _set_ps1(self):
"Pose a question and prompt"
if self.question:
- self.prompt = '\n{}\n{}'.format(
- self.question.format_prompt(), self._prompt)
+ lines = [
+ '',
+ _colorize(
+ self.ui.colors['question'], self.question.format_prompt()),
+ ]
+ lines.extend(
+ _colorize(self.ui.colors['prompt'], line)
+ for line in self._extra_ps1_lines())
+ lines.append(_colorize(self.ui.colors['prompt'], self._prompt))
+ self.prompt = '\n'.join(lines)
else:
- self.prompt = self._prompt
+ self.prompt = _colorize(self.ui.colors['prompt'], self._prompt)
def _set_ps2(self):
"Just prompt (without the question, e.g. for multi-line answers)"
- self.prompt = self._prompt
+ self.prompt = _colorize(self.ui.colors['prompt'], self._prompt)
+
+ def _extra_ps1_lines(self):
+ if (isinstance(self.question, _question.ChoiceQuestion) and
+ self.question.display_choices):
+ for i,choice in enumerate(self.question.answer):
+ yield '{}) {}'.format(i, choice)
+ if self.question.accept_all:
+ yield 'or fill in something else'
+ return []
+
+ def _process_answer(self, answer):
+ "Back out any mappings suggested by _extra_ps1_lines()"
+ if (isinstance(self.question, _question.ChoiceQuestion) and
+ self.question.display_choices):
+ try:
+ a = int(answer)
+ return self.question.answer[a]
+ except (ValueError, IndexError):
+ pass
+ return answer
def default(self, line):
self.answers.append(line)
if self.question.multiline:
self._set_ps2()
else:
- self._answer()
+ return self._answer()
def emptyline(self):
- self._answer()
+ return self._answer()
def _answer(self):
if self.question.multiline:
answer = self.answers[0]
else:
answer = ''
- correct = self.ui.process_answer(question=self.question, answer=answer)
+ kwargs = {}
+ if self._tempdir:
+ kwargs['tempdir'] = self._tempdir
+ answer = self._process_answer(answer=answer)
+ correct,details = self.ui.process_answer(
+ question=self.question, answer=answer, **kwargs)
if correct:
- print('correct\n')
+ print(_colorize(self.ui.colors['correct'], 'correct\n'))
else:
- print('incorrect\n')
- self.question = self.ui.get_question()
- if not self.question:
- return True # out of questions
- self._reset()
+ print(_colorize(self.ui.colors['incorrect'], 'incorrect'))
+ if details:
+ print(_colorize(
+ self.ui.colors['incorrect'], '{}\n'.format(details)))
+ else:
+ print('')
+ return self.get_question()
def do_answer(self, arg):
"""Explicitly add a line to your answer
"""
return self.default(arg)
+ def do_shell(self, arg):
+ """Run a shell command in the question temporary directory
+
+ For example, you can spawn an interactive session with:
+
+ quizzer? !bash
+
+ If the question does not allow interactive sessions, this
+ action is a no-op.
+ """
+ if getattr(self.question, 'allow_interactive', False):
+ if not self._tempdir:
+ self._tempdir = self.question.setup_tempdir()
+ try:
+ self._tempdir.invoke(
+ interpreter='/bin/sh', text=arg, stdout=None, stderr=None,
+ universal_newlines=False,
+ env=self.question.get_environment())
+ except (KeyboardInterrupt, _error.CommandError) as e:
+ if isinstance(e, KeyboardInterrupt):
+ LOG.warning('KeyboardInterrupt')
+ else:
+ LOG.warning(e)
+ self._tempdir.cleanup()
+ self._tempdir = None
+
def do_quit(self, arg):
"Stop taking the quiz"
self._reset()
return True
+ def do_skip(self, arg):
+ "Skip the current question, and continue with the quiz"
+ self.ui.stack[self.ui.user].append(self.question)
+ return self.get_question()
+
def do_hint(self, arg):
"Show a hint for the current question"
self._reset()
print(self.question.format_help())
+ def do_copyright(self, arg):
+ "Print the quiz copyright notice"
+ if self.ui.quiz.copyright:
+ print('\n'.join(self.ui.quiz.copyright))
+ else:
+ print(self.ui.quiz.copyright)
+
def do_help(self, arg):
'List available commands with "help" or detailed help with "help cmd"'
if not arg:
super(QuestionCommandLine, self).do_help(arg)
-class CommandLineInterface (UserInterface):
+class CommandLineInterface (_UserInterface):
+ colors = { # listed in pygments.console.light_colors
+ 'question': 'turquoise',
+ 'prompt': 'blue',
+ 'correct': 'green',
+ 'incorrect': 'red',
+ 'result': 'fuchsia',
+ }
+
def run(self):
- cmd = QuestionCommandLine(ui=self)
- cmd.cmdloop()
- print()
+ self.user = None
+ if self.stack[self.user]:
+ cmd = QuestionCommandLine(ui=self)
+ cmd.cmdloop()
+ print()
+ self._display_results()
- def display_results(self):
- print('results:')
+ def _display_results(self):
+ print(_colorize(self.colors['result'], 'results:'))
+ answers = self.answers.get_answers(user=self.user)
for question in self.quiz:
- if question.id in self.answers:
- self.display_result(question=question)
+ if question.id in answers:
+ self._display_result(question=question)
print()
- self.display_totals()
+ self._display_totals()
- def display_result(self, question):
- answers = self.answers.get(question.id, [])
+ def _display_result(self, question):
+ answers = self.answers.get_answers(user=self.user).get(question.id, [])
print('question:')
- print(' {}'.format(question.format_prompt(newline='\n ')))
+ print(' {}'.format(
+ _colorize(
+ self.colors['question'],
+ question.format_prompt(newline='\n '))))
la = len(answers)
lc = len([a for a in answers if a['correct']])
- print('answers: {}/{} ({:.2f})'.format(lc, la, float(lc)/la))
+ if la:
+ print('answers: {}/{} ({:.2f})'.format(lc, la, float(lc)/la))
for answer in answers:
if answer['correct']:
correct = 'correct'
else:
correct = 'incorrect'
- print(' you answered: {}'.format(answer['answer']))
+ correct = _colorize(self.colors[correct], correct)
+ ans = answer['answer']
+ if question.multiline:
+ ans = '\n '.join(ans)
+ print(' you answered: {}'.format(ans))
print(' which was: {}'.format(correct))
- def display_totals(self):
- answered = self.answers.get_answered(questions=self.quiz)
+ def _display_totals(self):
+ answered = self.answers.get_answered(
+ questions=self.quiz, user=self.user)
correctly_answered = self.answers.get_correctly_answered(
- questions=self.quiz)
+ questions=self.quiz, user=self.user)
la = len(answered)
lc = len(correctly_answered)
print('answered {} of {} questions'.format(la, len(self.quiz)))
- print(('of the answered questions, {} ({:.2f}) were answered correctly'
- ).format(lc, float(lc)/la))
+ if la:
+ print(('of the answered questions, '
+ '{} ({:.2f}) were answered correctly'
+ ).format(lc, float(lc)/la))