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 
    ##----------------------------------------------------