Example #1
0
def prompt_user():
    while True:
        master_pw = getpass('Master password: '******'Confirm password: '******'Confirmation didn\'t match.  Try again.')
        else:
            break

    service = input('Service name (e.g. facebook): ').lower()
    number = input('number (leave blank to generate): ')

    if number == '':
        number = SystemRandom().choice(range(MAX_NUM + 1))
    else:
        assert number.isdigit(), "Must enter a valid digit.  You entered: {}".format(number)
        number = int(number)
        assert 0 <= number <= MAX_NUM, "Number must be >= 0 and <= {}.  You entered: {}".format(
            str(MAX_NUM), str(number))

    return master_pw, service, number