用Python实现的简单通讯录

花了一个下午和半个晚上写了这个简单通讯录：
哈哈，第一次写这么长的Python代码，自认为结构还是挺合理的。
代码如下：

#-*- coding:utf-8 -*-
# file :addrList.py
# date :2011-10-24 15:40:13
# 设计一个基本的通讯录管理程序，使其具有添加，删除，编辑，查找等功能。
# 要求使用C/C++,java,javascript,python中任意一种语言实现。字符界面即可。
# 不需要有GUI（图形界面）
import sys
import os
import string
import re
from datetime import datetime

QUIT_FLAG = False
ADDRS_LIST = "addr_list.txt"
_addrs_dict = {} 
_addrs_count = 0
DEBUG=2
def info(message):
	global DEBUG
	if DEBUG > 0:
		print message

def debug(message):
	global DEBUG
	if DEBUG > 1:
		print message

def warn(message):
	global DEBUG
	if DEBUG > 0:
		print message
	
def error(message):
	print message

def help():
	print "用法：输入菜单中的命令即可执行相应操作！"
	print "可用菜单命令如下："
	showMenu()

def showMenu():
	print "+******************操作菜单***********************+"
	print "|查看所有联系人(all) | 查找联系人(find)           |"
	print "|添加联系人(add)     | 删除联系人(remove)         |"
	print "|编辑联系人(edit)    | 保存并退出(save)           |"
	print "|使用帮助(help)      | 退出但不保存(quit)         |"
	print "+_________________________________________________+"

def showError(info):
	print info


def doSwitch(op):
	if op == "all":
		doAll()
	elif op == "find":
		doFind()
	elif op == "add":
		doAdd()
	elif op == "remove":
		doRemove()
	elif op == "edit":
		doEdit()
	elif op == "save":
		doSave()
	elif op == "help":
		help()
	elif op == "quit":
		doQuit()
	else:
		showError("错误：您输入的命令有误，请重新输入。需要帮助请输入help！")

def verifyInput(items):
	_flag = True
	_rPhone = re.compile(r'1[0-9]{10}')
	_rQQ    = re.compile(r'[1-9][0-9]{4,9}')
	if len(items[0]) > 10:
		_flag = False
		print "姓名太长了！"
	if not _rPhone.match(items[1]):
		_flag = False
		print "手机号码格式不正确"
	if not _rQQ.match(items[2]):
		_flag = False
		print "QQ号码输入有误"
	return _flag
	

def buildAddr(addr):
	_addr={}
	items=addr.split()
	if len(items) < 3:
		print "您输入的信息太少了"
		return None
	if not	verifyInput(items):
		return None

	_addr['name']=items[0]
	_addr['phone'] = items[1]
	_addr['QQ'] = items[2]
	return _addr

def addAddr(addr):
	global _addrs_count,_addrs_dict
	_addrs_count+=1
	_addr=buildAddr(addr)
	if not _addr:
		return None
	_addrs_dict[_addrs_count]=_addr

def init():
	if not os.path.isfile(ADDRS_LIST):
		return None

	faddr=open(ADDRS_LIST,"r")
	for line in faddr:	
		if len(line) == 0:
			continue
		addAddr(line)
	faddr.close()


def save():
	global _addrs_dict
	faddr=open(ADDRS_LIST,"w+")
	for addr in _addrs_dict.values():
		faddr.write("{0}\t{1}\t{2}\n".format(addr['name'],addr['phone'],addr['QQ']))
	faddr.flush()
	faddr.close()


def doAll():
	global _addrs_dict
	if len(_addrs_dict) < 1:
		print "联系人当中暂无记录！"
		return None
	printHead()
	for key,addr in _addrs_dict.items():
		printAddr(key,addr)

def doFind():
	_flag=False
	flag1=flag2=flag3=False
	cond=raw_input("请输入查询信息：>")
	debug("DEBUG:{0}".format(cond))
	if len(cond) == 0:
		return None
	if cond.isdigit():
		flag1=findById(int(cond,10))
		flag2=findByPhone(cond)
		flag3=findByQQ(cond)
	else:
		flag1=findByName(cond)
	
	_flag = flag1 or flag2 or flag3 
	if not _flag:
		print "没有查找到任何联系人！"

def doAdd():
	line = raw_input("请依次输入联系人的姓名，手机号码，QQ号码>")
	if len(line) == 0:
		return None
	addAddr(line)	

def existsId(_id):
	global _addrs_dict
	return _addrs_dict.has_key(_id)
#	if _id > _addrs_count or _id < 1:
#		return False 
#	else:
#		return True

def doRemove():
	FLAG = True
	while FLAG:
		key=raw_input("请输入要删除的联系人的编号(取消请输入#）")
		if key == '#':
			FLAG = False
			continue
		if not existsId(int(key,10)):
			print "不存在您所输入编号的联系人。请确认"
			continue
	
		print "编号为 {0} 的联系人信息如下：".format(key)
		printById(int(key,10))
		yesOrNo=raw_input("您确定要删除上述联系人吗？(y/n)")
		if yesOrNo in "yY":
			removeById(int(key,10))
			print "删除成功！"
		yesOrNo=raw_input("您还需要继续删除联系人吗？(y/n)")
		if not yesOrNo in "yY":
			FLAG = False



def doEdit():
	FLAG = True
	while FLAG:
		key=raw_input("请输入要编辑的联系人的编号(取消请输入#）")
		print "DEBUG:key:{0}".format(key)
		if key == '#':
			FLAG = False
			continue
		if not existsId(int(key,10)):
			print "不存在您所输入编号的联系人。请确认"
			continue
	
		print "编号为 {0} 的联系人信息如下：".format(key)
		printById(int(key,10))
		updateById(int(key,10))
		FLAG = False


def doSave():
	save()
	doQuit()

def doQuit():
	global QUIT_FLAG
	QUIT_FLAG = True
	print "正在退出……"
#	exit(0)

def printHead():
	print "+-----+----------+---------------+---------------+"
	print "|编号 |   姓名   |  手机号码     |     QQ号码    |"
	print "+-----+----------+---------------+---------------+"

def printAddr(key,addr):
#	print "+-----+----------+---------------+---------------+"
	print "|{0:^5}|{1:^10}|{2:^15}|{3:^15}|".format(key,addr['name'],addr['phone'],addr['QQ'])
	print "+-----+----------+---------------+---------------+"



def printById(_id):
	global _addrs_dict
	printHead()
	printAddr(_id,_addrs_dict[_id])

def removeById(_id):
	global _addrs_dict
	_addrs_dict.pop(_id)

def updateById(_id):
	global _addrs_dict
	_addr= _addrs_dict.get(_id)

	print "请输入该联系人的新信息，相应留空则保留原有信息"
	name=raw_input("请输入新姓名：>")
	if len(name) > 0:
		_addr['name']=name

	phone=raw_input("请输入新手机号码：>")
	if len(phone) > 0:
		_addr['phone']=phone

	qq=raw_input("请输入新的QQ号码：>")
	if len(qq) > 0 :
		_addr['QQ']=qq

	_addrs_dict[_id]=_addr
	print "更新成功！"
	print "联系人新的信息如下："
	printById(_id)

def findById(_id):
	if existsId(_id):
		printById(_id)
		return True
	else:
		return False

def findByField(cond,field='name'):
	global _addrs_dict
	_flag = False
	for key,addr in  _addrs_dict.items():
		if addr[field].find(cond) != -1:
			printAddr(key,addr)
			_flag=True
	return _flag

def findByName(name):
	return findByField(name,'name')

def findByPhone(phone):
	return findByField(phone,'phone')

def findByQQ(qq):
	return findByField(qq,'QQ')



def main():
	init()
	showMenu()
	while(not QUIT_FLAG):
		operation=raw_input("请在此处输入菜单命令>")
		doSwitch(operation)


if __name__=='__main__':
	main()
	## do something
##----------------------------------------------------