Python 实现简单的电话本功能

myPhoneBook2.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
import re
class PhoneBook(object):
 '''这是一个电话簿脚本。

该脚本能够实现
 AddContact：添加联系人信息
 ShowContact：查找姓名显示联系人
 SaveContacts：存储联系人到 TXT 文档（存储格式——姓名:号码/号码)
 LoadContacts：从 txt 文档中载入联系人
 '''
 def __init__(self):
   self.contactsDict = {}

def AddContact(self):
   while True:
     name = raw_input('请输入姓名>>>')
     name = name.strip()
     # 姓名必须包含有效字符
     if name != '':
       break
     print '***姓名不能为空'

while True:
     number = raw_input('请输入号码>>>')
     number = re.sub(r'\D', '', number)
     # 删除号码中的非数字字符
     if number != '':
       break
     print '***号码只能是数字'

cover = True #若联系人已存在，是否覆盖
   if self.contactsDict.has_key(name):
     print '***联系人已存在'
     self.ShowContact(name)
     while True:
       control = raw_input('''
输入 "c"：覆盖原来号码
输入 "n"：保留原来号码并存储新号码
输入 "q" 退出\n>>>''')
       if control.lower() == 'n':
         cover = False
         break
       if control.lower() == 'c':
         break
       if control.lower() == 'q':
         return None
       print '***输入错误'

if cover:
     self.contactsDict[name] = number
   else:
     if number in self.contactsDict[name]:
       print '***号码已经存在'
     else:
       self.contactsDict[name] = self.contactsDict[name] + '/' + number

def ShowContact(self, name):
   print '++++++++++++++++++++++++'
   if self.contactsDict.has_key(name):
     print '【联系人信息】'
     print '【姓名： ％s】' ％ name
     numberList = self.contactsDict[name].split('/')
     for num in range(len(numberList)):
       print '【号码％d： ％s】' ％ (num+1, numberList[num])

else:
     print '【找不到联系人％s】' ％ name
   print '++++++++++++++++++++++++'

def DeleteNumber(self, name):
   if self.contactsDict.has_key(name):
     self.ShowContact(name)
     number = self.contactsDict[name].split('/')
     while True:
       print '''
请输入要删除号码的序号
或者输入 "a" 删除该联系人
或者输入 "q" 退出（不删除）
（若联系人号码被全部删除，该联系人也将被删除）'''
       control = raw_input('>>>')
       if control.lower() == 'q':
         break
       elif control.lower() == 'a':
         del self.contactsDict[name]
         break
       elif control.isdigit() and int(control) <= len(number):
         del number[int(control)-1]
         self.contactsDict[name] = '/'.join(number)
         break
       else:
         print '***输入有误'

def LoadContacts(self):
   '''
   try:
     PhoneBook = open('PhoneBook.txt', 'a+')
     contacts = PhoneBook.read()
     if contacts == '':
       print '***电话簿为空'
     else:
       ContactsList = contacts.split('\n')
       for contact in ContactsList:
         if not contact == '':
           contact = contact.split(':')
           name = contact[0]
           number = contact[1]
           self.contactsDict[name] = number
   finally:
     PhoneBook.close()
   '''
   self.contactsDict = {line.split(':')[0]: line.split(':')[1] for line in open('PhoneBook.txt','a+').readlines()}

def SaveContacts(self):
   try:  
     if self.contactsDict:
       PhoneBook = open('PhoneBook.txt', 'w')
       for name, number in self.contactsDict.items():
         line = name + ':' + number
         PhoneBook.write(line)
         PhoneBook.write('\n')

else:
       print '***没有联系人信息'

finally:
     PhoneBook.close()

if __name__ == '__main__':
 myPhoneBook = PhoneBook()
 myPhoneBook.LoadContacts()
 try:
   while True:
     raw_input('按回车键继续')
     print '''
--------------------------------
 输入 a：添加联系人
 输入 s：显示联系人信息
 输入 d：删除联系人
 输入 q：退出
--------------------------------'''
     control = raw_input('>>>')
     if control.lower() == 'a':
       myPhoneBook.AddContact()
     elif control.lower() == 's':
       name = raw_input('请输入要查找的联系人姓名\n>>>')
       myPhoneBook.ShowContact(name)
     elif control.lower() == 'd':
       name = raw_input('请输入要删除的联系人姓名\n>>>')
       myPhoneBook.DeleteNumber(name)
     elif control.lower() == 'q':
       break
     else:
       print '***输入有误'

finally:
   myPhoneBook.SaveContacts()

花了一个下午和半个晚上写了这个简单通讯录：
哈哈，第一次写这么长的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
##----------------------------------------------------