Python字典应用的一个例子

#!/usr/bin/env python

#定义一个空字典
db = {}

def newuser():
    prompt = '--login desired: '
    while True:
        name = raw_input(prompt)
        #根据关键字name看字典中是否已存在此键值对
        if db.has_key(name):
            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)xiting User Login
(Q)uit
    Enter choice : """
    while True:
        while  True:
            try:
                choice = raw_input(prompt).strip()[0].lower()
            except(EOFError, KeyboardInterrupt):
                choice = 'q' 
            print '\n--You picked: [%s]' % choice
            if choice not in 'neq':
                print '--invalid option:, try again --'
            else:
                if choice == 'n':
                    newuser()
                elif choice == 'e':
                    olduser()
                else:
                    print 'quit!'
                    return
if __name__ == '__main__':
    showmenu()

这个程序管理用于登录系统的用户信息:登录名字和密码。登录用户帐号建立后,已存在用户可以用登录名字和密码重返系统。新用户不能用别人的登录名建立用户帐号




你可能感兴趣的:(Python字典应用的一个例子)