fork download
  1. #!/usr/bin/env python3
  2. import sys
  3.  
  4.  
  5. def exit():
  6. input('Press <Enter> to exit')
  7. sys.exit()
  8.  
  9.  
  10. def expect_digits():
  11. options = {'4': exit}
  12. for d in '123':
  13. options[d] = lambda d=d: print(d)
  14. expect(options)
  15.  
  16.  
  17. def expect(options):
  18. while True:
  19. answer = input('Input ' + '|'.join(options)).strip()
  20. if answer in options:
  21. options[answer]()
  22. else:
  23. print('Unknown command. Try again')
  24.  
  25.  
  26. expect({'y': expect_digits, 'n': exit})
Success #stdin #stdout 0.04s 9408KB
stdin
invalid
y
1
invalid
2
1
4

stdout
Input n|yUnknown command. Try again
Input n|yInput 4|2|3|11
Input 4|2|3|1Unknown command. Try again
Input 4|2|3|12
Input 4|2|3|11
Input 4|2|3|1Press <Enter> to exit