簡易Python電話本（Simple Python Telephone Book）

1. #!/usr/bin/python
2. '''''
3. Compiler: Python 2.6
4. Filename: tb.py
5. Version: 1.00 @20081205
6. Author: t0nsha
7.                 (liaodunxia{at}gmail.com)
8. Remark: Just a simple telephone book program 
9.                 for my one week python study. 
10. '''
12. import cPickle as p
13. tbDict={}
14. tbdat = 'tb.dat'
16. class Person:
17.     def __init__(self, name, phone, email, address):
18.         self.name = name
19.         self.phone = phone
20.         self.email = email
21.         self.address = address
23. def tbAdd(a):
24.     while True:
25.         name = raw_input('Enter a name for Add:')
26.         if name == 'Q':
27.             break
28.         if tbDict.has_key(name):
29.             print '%s already exists.' % name
30.             continue
31.         phone = raw_input('phone:')
32.         email = raw_input('email:')
33.         address = raw_input('address:')
34.         person = Person(name = name, /
35.                         phone = phone, /
36.                         email = email, /
37.                         address = address)
38.         tbDict[name] = person
39.         print '%s added, total: %d.' % (name, len(tbDict))
40.         if a != 'A':
41.             break
43. def tbRemove(r):
44.     while True:
45.         name = raw_input('Enter a name for Delete:')
46.         if name == 'Q':
47.             break
48.         if not tbDict.has_key(name):
49.             print '%s not exist.' % name
50.         else:
51.             del tbDict[name]
52.             print '%s removed.' % name
53.         if r != 'R':
54.             break
56. def tbSearch(s):
57.     while True:
58.         name = raw_input('Enter a name for Search:')
59.         if name == 'Q':
60.             break
61.         if tbDict.has_key(name):
62.             print 'name:'.rjust(12),tbDict[name].name
63.             print 'phone:'.rjust(12),tbDict[name].phone
64.             print 'email:'.rjust(12),tbDict[name].email
65.             print 'address:'.rjust(12),tbDict[name].address
66.         else:
67.             print '%s not found.' % name
68.         if s != 'S':
69.             break
70.     
71. def tbList(l):
72.     if l == 'l':
73.         for key in tbDict.keys():
74.             print key
75.     elif l == 'L':
76.         for name in tbDict.keys():
77.             print 'name:'.rjust(12),tbDict[name].name
78.             print 'phone:'.rjust(12),tbDict[name].phone
79.             print 'email:'.rjust(12),tbDict[name].email
80.             print 'address:'.rjust(12),tbDict[name].address
81.     print 'total: %d.' % len(tbDict)
83. def tbModify(m):
84.     while True:
85.         name = raw_input('Enter a name for Modify:')
86.         if name == 'Q':
87.             break
88.         if not tbDict.has_key(name):
89.             print '%s not exist.' % name
90.             return
91.         phone = raw_input('new phone:')
92.         email = raw_input('new email:')
93.         address = raw_input('new address:')
94.         person = Person(name = name, /
95.                         phone = phone, /
96.                         email = email, /
97.                         address = address)
98.         tbDict[name] = person
99.         print '%s modified.' % (name)
100.         if m != 'M':
101.             break
103. def tbUsage():
104.     print '''''
105. Usage:
106.     a/A --> add a new person/loop add
107.     l/L --> list all names/list all details
108.     r/S --> remove one/loop remove
109.     s/S --> search/loop search
110.     q/Q --> quit & save/quit (loop) but not save
111. '''
113. def tbQuit(q):
114.     if q == 'q':
115.         p.dump(tbDict, file(tbdat, 'w+'))
116.         
117. def tbLoop():
118.     while True:
119.         c = raw_input('Ready to work:')
120.         if   c == 'a' or c == 'A':
121.             tbAdd(c)
122.         elif c == 'l' or c == 'L':
123.             tbList(c)
124.         elif c == 'm' or c == 'M':
125.             tbModify(c)
126.         elif c == 'r' or c == 'R':
127.             tbRemove(c)
128.         elif c == 's' or c == 'S':
129.             tbSearch(c)
130.         elif c == 'q' or c == 'Q':
131.             tbQuit(c)
132.             break
133.         else:
134.             tbUsage()
136. #While first run, file "tb.dat" doesn't exist, this
137. #will raise an IOError exception, but we igored it.
138. try:
139.     tbDict = p.load(file(tbdat))
140. except IOError:
141.     pass
142. tbLoop()