python通讯录程序_Python练手程序——通讯录

1 #Filename:MyAddressBook.py

2 importcPickle as p3 importos4 #Class Item

5 classItem:6 def __init__(self,name,age,gender):7 self.name =name8 self.age =age9 self.gender =gender10 #the main menu of address book

11 defmenu():12 '''the main menu of address book'''

13 print ''

14 print '1.Insert an item'

15 print '2.Delete an item'

16 print '3.Modify an item'

17 print '4.Display all items'

18 print '5.Sort all items'

19 print '6.Exit the program'

20 print 'What do you want to do?'

21

22 #initialization of system, load the member list

23 defbegin():24 '''initialization of system, load the member list'''

25 globalitemlist26 if os.path.exists('memberlist.data') == True:#to judge whether the file exists

27 listfile = file('memberlist.data','r')28 if len(listfile.read())!=0:#to judge whether the file is empty

29 listfile.seek(0);30 itemlist =p.load(listfile)31 listfile.close()32 #exitance of system, store the menber list

33 defend():34 '''exitance of system, store the menber list'''

35 globalitemlist36 listfile = file('memberlist.data','w+')37 p.dump(itemlist,listfile)38 listfile.close()39 #insert an item into the member list

40 definsert():41 '''insert an item into the member list'''

42 name = raw_input('Enter name:')43 age = int(raw_input('Enter age:'))44 gender = raw_input('Enter gender:')45 item =Item(name,age,gender)46 globalitemlist47 itemlist.append(item)48 #print an item

49 defoutput(item):50 '''print an item'''

51 print '%-15s%-5d%s'%(item.name,item.age,item.gender)52 #print all items

53 defdisplay():54 '''print all items'''

55 globalitemlist56 l =len(itemlist)57 print 'name age gender'

58 for i inrange(0,l):59 output(itemlist[i])60 print ''

61 #delete an item by name from member list

62 defdelete():63 '''delete an item by name from member list'''

64 name = raw_input('Enter the name you want to delete:')65 globalitemlist66 l =len(itemlist)67 for i inrange(0,l):68 if (itemlist[i].name ==name):69 itemlist.pop(i)70 break

71 #update an item

72 defupdate(item):73 '''update an item'''

74 item.name = raw_input('Enter name:')75 item.age = int(raw_input('Enter age:'))76 item.gender = raw_input('Enter gender:')77 #update an item's information by name

78 defmodify():79 '''update an item's information by name'''

80 name = raw_input('Enter the name you want to modify:')81 globalitemlist82 l =len(itemlist)83 for i inrange(0,l):84 if (itemlist[i].name ==name):85 update(itemlist[i])86 print 'Update done!'

87 #sort all items by name

88 defsort():89 globalitemlist90 itemlist.sort(None,key = lambdaitem:item.name)91 print 'Sort done!'

92 #Here are the scripts

93 itemlist = [] #Notice here!!!

94 begin()95 whileTrue:96 menu()97 sel =int(raw_input())98 if sel == 1:99 insert()100 elif sel == 2:101 delete()102 elif sel == 3:103 modify()104 elif sel == 4:105 display()106 elif sel == 5:107 sort()108 else:109 break

110 end()111 print 'Good Bye!'