Как перезапустить программу на основе пользовательского ввода? ⇐ Python

[Anonymous]

Я пытаюсь перезапустить программу, используя if-тест на основе ввода пользователя.
Этот код не работает, но это примерно то, что мне нужно:

Код: Выделить всё

answer = raw_input('Run again? (y/n): ')

if answer == 'n':
print 'Goodbye'
break
elif answer == 'y':
#restart_program???
else:
print 'Invalid input.'

Я пытаюсь сделать следующее:

• если вы ответите y - программа перезапустится сверху
• если вы ответите n - программа завершится (эта часть работает)
• если вы введете что-нибудь еще, она должна напечатать «неверный ввод». пожалуйста, введите y или n...' или что-то в этом роде и снова попросите вас ввести новые данные.

Я очень близко подошел к решению с циклом «пока true», но программа либо просто перезапускается независимо от того, что вы нажимаете (кроме n), либо завершает работу независимо от того, что вы нажимаете (кроме y). Есть идеи?