核心编程答案（第七章）

#!/usr/bin/env python
# encoding: utf-8
import ipdb


db ={}


def newuser():
    prompt = ‘login desierd: ‘
    while True:
        name = raw_input(prompt)
        if name in db:
            prompt = ‘name taken, try another: ‘
            continue
        else:
            break
    pwd = raw_input(‘passwd: ‘)
    db[name] = pwd


def olduser():
    name = raw_input(‘login: ‘)
    pwd = raw_input(‘passwd: ‘)
    passwd = db.get(name)
    if passwd == pwd:
        print ‘welcome back‘, name
    else:
        print ‘login incorrect‘


def showmenu():
    prompt = """
    (N)ew User Login
    (E)xisting User Login
    (Q)uit
    Enter choice: """

    done = False
    while not done:
        chosen = False
        while not chosen:
            try:
                choice =raw_input(prompt).strip()[0].lower()
            except (EOFError, KeyboardInterrupt):
                choice = ‘q‘
            print ‘\nYou picked: [%s]‘ % choice
            if choice not in ‘neq‘:
                print ‘invalid option, try again‘
            else:
                chosen = True

        if choice == ‘q‘:
            done = True
        if choice == ‘n‘:
            newuser()
        if choice == ‘e‘:
            olduser()

if __name__ == "__main__":
    showmenu()