Python 之 地址簿程序

最近在看《简明 Python 教程》，作者在16章留了个问题，通过前面章节的学习，设计一个“ 地址簿程序 ， 在这个程序中，你可以添加、修改、删除和搜索你的联系人（朋友、家人和同事等等）以及它们的信息（诸如电子邮件地址和/或电话号码）。这些详细信息应该被保存下来以便以后提取“，网上发现已经有很多网友读了 《简明 Python 教程》，并贴出了自己的”杰作“。我也在此班门弄斧，望大家多多指教！

#!/usr/bin/python
#encoding: utf-8
#Filename: member_ver2.py


import cPickle as p
import os
import sys


class Person:
	filename = 'address.txt'
	addressList = {}
	
	def __init__(self):
		if os.path.exists(self.filename):
			f = file(self.filename)
			self.addressList = p.load(f)
			f.close()

	def save(self):
		f = file(self.filename, 'w')
		p.dump(self.addressList, f)
		f.close()
		print('saved successfully.')

	def add(self, name, tel, email, type = 'friend'):
		if name in self.addressList.keys():
			print('The name is exist')
		else:
			self.addressList[name] = [tel, email, type]
			self.save()
		
		print('Added successfully!')
	
	def modify(self, name, tel, email, types = 'friend'):
		if name in self.addressList.keys():
			self.addressList[name] = [tel, email, types]
			self.save()
			print('Modified successfully!')
		else:
			print('The name is not exists!')

	def search(self, name):
		if name in self.addressList.keys():
			print('Name: %s\t Tel: %s\t email: %s\t types: %s ' % (name, self.addressList[name][0], self.addressList[name][1], self.addressList[name][2]))
		else:
			print('The name is not exist')

	def show(self):
		print('Name	\t	Tel	\t	Email	\t Type')
		for name, info in self.addressList.items():
			print(name, info[0], info[1], info[2])


	def run(self):
		while True:
			cmd = raw_input('Please input command(add/modify/search/delete/show/quit) ' )
			if cmd  == 'add':
				while True:
					name = raw_input('Enter name: ')
					if name not in self.addressList.keys():
						tel = raw_input('Enter tel: ')
						email = raw_input('Enter email: ')
						types = raw_input('Enter type(default type is "friend"): ')
						if types != '':
							self.add(name, tel, email, types)
						else:
							self.add(name, tel, email)
					else:
						print('The name you added has exist!')
					flag = raw_input('Do you want to continue to add?(y/n)')
					if flag == 'y' or flag == 'Y':
						continue
					break
			elif cmd == 'modify':
				while True:
					name = raw_input('Enter name: ')
					if name in self.addressList.keys():
						tel = raw_input('Enter tel: ')
						email = raw_input('Enter email: ')
						types = raw_input('Enter type(default type is "friend"): ')
						if tel == '':
							tel = self.addressList[name][0]
						if email == '':
							email = self.addressList[name][1]
						if types != '':	
							self.modify(name, tel, email, types)
						else:
							self.modify(name,tel,email)
					else:
						print('The name you modifed is not exist!')
					flag = raw_input('Do you want to continue to modify?(y/n)')
					if flag == 'y' or flag == 'Y':
						continue
					break
			elif cmd == 'search':
				while True:
					name = raw_input('Enter name: ')
					self.search(name)
					flag = raw_input('Do you want to continue to search?(y/n)')
					if flag == 'y' or flag == 'Y':
						continue
					break
			elif cmd == 'delete':
				while True:
					name = raw_input('Enter name: ')	
					self.delete(name)
					flag = raw_input('Do you want to continue to delete?(y/n)')
					if flag == 'y' or flag == 'Y':
						continue
					break

			elif cmd == 'show':
				self.show()	
			elif cmd == 'quit' or cmd == 'q':
				print('quit')
				sys.exit()
			else:
				print('Wrong command,try again')


if __name__ == '__main__':
	
	per = Person()
	per.run()