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