Python 实现简单的电话本功能

myPhoneBook2.py

#!/usr/bin/python/' target='_blank'>python# -*- coding: utf-8 -*-import reclass 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 ##----------------------------------------------------