Python字典应用的一个例子

链接地址： http://blog.csdn.net/buaa_shang/article/details/8296791

[python]  view plain  copy

1. #!/usr/bin/env python  
2.   
3. #定义一个空字典  
4. db = {}  
5.   
6. def newuser():  
7.     prompt = '--login desired: '  
8.     while True:  
9.         name = raw_input(prompt)  
10.         #根据关键字name看字典中是否已存在此键值对  
11.         if db.has_key(name):  
12.             prompt = '--name taken, try another: '  
13.             continue  
14.         else:  
15.             break  
16.     pwd = raw_input('passwd: ')  
17.     #存密码到对应的名字字典中  
18.     db[name] = pwd  
19.   
20. def olduser():  
21.     name = raw_input('login: ')  
22.     pwd = raw_input('passwd: ')  
23.     passwd = db.get(name)  
24.     if passwd == pwd:  
25.         print '--welcome back --', name  
26.     else:  
27.         print '--login incorrect--'  
28.   
29. def showmenu():  
30.     prompt = """  
31. (N)ew User Login 
32. (E)xiting User Login 
33. (Q)uit 
34.     Enter choice : """  
35.     while True:  
36.         while  True:  
37.             try:  
38.                 choice = raw_input(prompt).strip()[0].lower()  
39.             except(EOFError, KeyboardInterrupt):  
40.                 choice = 'q'   
41.             print '\n--You picked: [%s]' % choice  
42.             if choice not in 'neq':  
43.                 print '--invalid option:, try again --'  
44.             else:  
45.                 if choice == 'n':  
46.                     newuser()  
47.                 elif choice == 'e':  
48.                     olduser()  
49.                 else:  
50.                     print 'quit!'  
51.                     return  
52. if __name__ == '__main__':  
53.     showmenu()  

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