Synnefo » snf-ganeti

           "cli_option", "GenerateTable", "AskUser",

def AskUser(text, choices=None):

  """Ask the user a question.

    text - the question to ask.

    choices - list with elements tuples (input_char, return_value,

    description); if not given, it will default to: [('y', True,

    'Perform the operation'), ('n', False, 'Do no do the operation')];

    note that the '?' char is reserved for help

  Returns: one of the return values from the choices list; if input is

  not possible (i.e. not running with a tty, we return the last entry

  from the list

  if choices is None:

    choices = [('y', True, 'Perform the operation'),

               ('n', False, 'Do not perform the operation')]

  if not choices or not isinstance(choices, list):

    raise errors.ProgrammerError("Invalid choiches argument to AskUser")

  for entry in choices:

    if not isinstance(entry, tuple) or len(entry) < 3 or entry[0] == '?':

      raise errors.ProgrammerError("Invalid choiches element to AskUser")

  answer = choices[-1][1]

  new_text = []

  for line in text.splitlines():

    new_text.append(textwrap.fill(line, 70, replace_whitespace=False))

  text = "\n".join(new_text)

    return answer

    chars = [entry[0] for entry in choices]

    chars[-1] = "[%s]" % chars[-1]

    chars.append('?')

    maps = dict([(entry[0], entry[1]) for entry in choices])

    while True:

      f.write(text)

      f.write('\n')

      f.write("/".join(chars))

      f.write(": ")

      line = f.readline(2).strip().lower()

      if line in maps:

        answer = maps[line]

        break

      elif line == '?':

        for entry in choices:

          f.write(" %s - %s\n" % (entry[0], entry[2]))

        f.write("\n")

        continue