Python初学者笔记(4)-简单的通讯录

要求：

编写一个简单的通讯录

1、通讯录包含至少包含姓名、电话号码、电子邮箱；2、通讯录的信息能够保存在本地磁盘；3、通讯录查找特定人员的信息；4、通讯录能够修改特定人员的信息；5、通讯录能够删除特定人员的信息；6、通讯录能完全显示所有人的信息

方法：Python 2.7.9 / Windows

 # -*- coding: gbk -*-

 import os,sys
 import cPickle as p

 #判断通讯录是否存在，不存在则创建通讯录
 if os.path.exists('contacts.dat')==False:
     f = file('contacts.dat','wb')
     temp = {'total':0}
     p.dump(temp,f)
     f.close()
     print "\n本地磁盘尚无通讯录，新通讯录创建成功！\n"
 else:
     pass

 #添加联系人
 def add():
     f = file('contacts.dat','rb')
     a = p.load(f)
     f.close()
     name = raw_input('请输入所要添加联系人的姓名:')
     if a.has_key(name):
         print "\n联系人已存在，添加失败！请改名~"
     else:
         tel = raw_input('请输入电话号码:')
         email = raw_input('请输入所要添加联系人的电子邮件([e-mail]):')
         list = [tel,email]
         information = {name:list}
         a['total'] += 1
         a.update(information)
         f = file('contacts.dat','wb')
         p.dump(a,f)
         f.close()
         print '添加成功！'

 #显示所以联系人
 def showall():
     f = file('contacts.dat','rb')
     a = p.load(f)
     print "一共有{}个联系人.\n".format(a['total'])
     print "{:<8}\t{:<15}\t{:<30}".format('姓名','电话号码','电子邮箱')
     for key in a.keys():
         if key != 'total':
             print "{:<8}\t{:<15}\t{:<30}".format(key,a[key][0],a[key][1])
     f.close()

 #退出通讯录
 def exit():
     sys.exit()

 #查找
 def search(name):
     f = file('contacts.dat','rb')
     a = p.load(f)
     if a.has_key(name):
         print "{}的号码是:{}".format(name,a[name][0])
         print "{}的电子邮箱是:s{}".format(name,a[name][1])
     else:
         print "联系人不存在！"
     f.close()

 #删除
 def delete(name):
     f = file('contacts.dat','rb')
     a = p.load(f)
     f.close()
     if a.has_key(name):
         a.pop(name)
         a['total'] -= 1
         f = file('contacts.dat','wb')
         p.dump(a,f)
         f.close()
         print "删除成功！"
     else:
         print "联系人不存在！无法删除！"

 #修改
 def modify(name):
     f = file('contacts.dat','rb')
     a = p.load(f)
     f.close()
     if a.has_key(name):
         tel = raw_input("请输入修改后的号码:")
         a[name][0] = tel
         email = raw_input("请输入修改后的邮箱:")
         a[name][1] = email
         f = file('contacts.dat','wb')
         p.dump(a,f)
         f.close()
         print "修改成功！"
     else:
         print "联系人不存在！"

 #界面
 def menu():
     print
     print "****************************"
     print "     显示菜单提示信息：*"
     print "     显示所有联系人:0"
     print "     查找联系人:1"
     print "     添加联系人:2"
     print "     删除联系人:3"
     print "     更改联系人资料:4"
     print "     退出通讯录:5"
     print "****************************"

 #主程序
 while True:
     menu()
     x=raw_input("\n请输入您的选择菜单号:")
     if x == '':
         add()
         continue
     if x == '':
         showall()
         continue
     if x == '':
         print "谢谢使用！"
         exit()
         continue
     if x == '':
         name = raw_input("请输入所要查找的联系人姓名:")
         search(name)
         continue
     if x == '':
         name = raw_input("请输入所要删除的联系人姓名:")
         delete(name)
         continue
     if x == '':
         name = raw_input("请输入所要修改的联系人姓名:")
         modify(name)
         continue
     if x == '*':
         menu()
     else:
         print "输入的选项不存在，请重新输入！"
         continue

结果：

>>>

****************************
显示菜单提示信息：*
显示所有联系人:0
查找联系人:1
添加联系人:2
删除联系人:3
更改联系人资料:4
退出通讯录:5
****************************

其余功能显示正常。

说明：Windows平台下，开头声明# -*- coding: gbk -*-显示中文正常

本来网上查询中文乱码问题，应该是使用# -*- coding: utf-8 -*-的开头声明，后发现这个声明不管用，后改用gbk起作用。

更深入的内容有待后续研究，欢迎大家讨论！

源自：DuShuSir.com