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 == '2':
        add()
        continue
    if x == '0':
        showall()
        continue
    if x == '5':
        print "謝謝使用！"
        exit()
        continue
    if x == '1':
        name = raw_input("請輸入所要查找的聯系人姓名:")
        search(name)
        continue
    if x == '3':
        name = raw_input("請輸入所要刪除的聯系人姓名:")
        delete(name)
        continue
    if x == '4':
        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