def yes_or_no()

in tools/add_module.py [0:0]

• 17 lines of code
• 6 McCabe index (conditional complexity)

def yes_or_no(question, default):
  if default:
    question += " [Y/n]: "
  else:
    question += " [y/N]: "

  var = None
  while var is None:
    user_input = ask_input(question).strip().lower()
    if user_input == "y":
      var = True
    elif user_input == "n":
      var = False
    elif not user_input:
      var = default
    else:
      print("Invalid selection: {}".format(user_input))
  return var