Python命令行地址簿小程序代码

# wwei's address book python project
import cPickle as p
import os
import sys
filename = 'addressbook.data'
class Members(object):
	def __init__(self,name,email,tel):
		self.name = name
		self.email = email
		self.tel = tel

def tell():
	f = file(filename)
	conlist = p.load(f)
	print conlist
	f.close()
	del conlist

def add_or_mod():
	s = raw_input("pls input a member's msg as 'name,email,tel':")
	s1 = s.split(',')
	member = Members(s1[0], s1[1], s1[2])
	f = file(filename)
	conlist = p.load(f)
	conlist[member.name] = member.email +','+ member.tel
	f = file(filename, 'w')
	p.dump(conlist,f)
	f.close()
	del conlist

	# show the member's msg
	f = file(filename)
	conlist = p.load(f)
	print conlist
	f.close()

def delete():
	f = file(filename)
	conlist = p.load(f)
	d = raw_input("pls enter the name of member you want to delete:")
	if d in conlist:
	    del conlist[d]
	    f = file(filename,'w')
	    p.dump(conlist,f)
	    print conlist
	    f.close()
	    del conlist
	else:
		print 'the member you enter is not in your address book, pls check'
		delete()

def search():	
	f = file(filename)
	conlist = p.load(f)
	s = raw_input("pls input the name of member you want to search:")
	if s in conlist:
	    print s,':',conlist[s]
	    f.close()
	    del conlist
	else:
		print 'the member you enter is not in your address book, pls check'
		search()

def main():
	while True:
		print '-' * 20
		menu = raw_input('0.show\n1.add or modify\n2.delete\n3.search\nx.EXIT\n')
		print '-' * 20
		if menu == '0':
			tell()
		elif menu == '1':
			add_or_mod()
		elif menu == '2':
			delete()
		elif menu == '3':
			search()
		elif menu == 'x':
			sys.exit()
		else:
			print 'no this option. pls check'

if os.path.exists('addressbook.data'):
	main()
else:
	f = file(filename, 'w')
	conlist = {'wwei':'nuaa_2w@163.com,18661616161'}
	p.dump(conlist,f)
	f.close()
	del conlist
	main()