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