python简明教程练习--命令行 地址簿 程序

    在这个程序中，你可以添加、修改、删除和搜索你的联系人（朋友、家人和同事等等）以及它们的信息（诸如电子邮件地址和/或电话号码）。这些详细信息应该被保存下来以便以后提取。

    由于上个程序中的代码没有把信息从文件中读取出来，我们可以把相同的信息保存在info.txt文件中去，这使得info.txt文件会有多余的东西。以下是我修改后的代码

代码：

#!/usr/bin/env python
# -*-encoding:utf-8 -*-
'''
Created on 2011-10-16
@author: Descusr
'''

import sys
import os
import cPickle
class Person(object):
    filename = 'info.txt'
    def __init__(self, name, phone):
        self.name = name
        self.phone = phone
        self.personlist = {self.name:self.phone}

    def add(self):
        self.name = raw_input("Enter name:")
        if self.name in self.personlist.keys():
            print "The name is exist!"
        else:
            self.phone=raw_input("Enter phone:")
            self.personlist[self.name] = self.phone
            print "Contact saved!"

    def Modify(self):
        self.name = raw_input("Enter name:")
        if self.name in self.personlist.keys():
            self.phone = raw_input("Enter new phone:")
            self.personlist[self.name] = self.phone
            print "Contact saved!"
        else:
            print "The name is not in here!!"

    def delete(self):
        self.name = raw_input("Enter name:")
        if self.name in self.personlist.keys():
            del self.personlist[self.name]
            print "Deleted!!"
        else:
            print "The name is not in here!!"

    def search(self):
        self.name = raw_input("Enter name:")
        if self.name in self.personlist.keys():
            print 'Name:%s,Phone:%s'%(self.name,self.phone)
        else:
            print "The name is not in here!!"

    def save(self):
        f = file(self.filename,'w')
        cPickle.dump(self.personlist,f)
        f.close()
        print "Your contacts list has been saved to file:%s successfully~" %self.filename

    def load(self):
        if os.path.exists(self.filename):
            f = file(self.filename)
            self.personlist = cPickle.load(f)
            f.close()

    def show(self):
        for self.name,self.phone in self.personlist.items():
            print "Name:%s  Phone number:%s" %(self.name,self.phone)

if __name__ == '__main__':
    os.system('clear')
    command = ['add','modify','search','delete','quit' ,'show']
    person = Person('','')
    person.load()
    while True:
        print "The contact person:"
        str = raw_input('What are you going to do(add/modify/search/delete/show/quit)?')
        if str in command:
            if str == 'add':
                person.add()
            elif str == 'modify':
                person.Modify()
            elif str == 'search':
                person.search()
            elif str == 'delete':
                person.delete()
            elif str == 'show':
                person.show()
            else:
                ch = raw_input("Your contacts list hasn't  been saved,save it now?(Y/N)")
                if ch == 'y':
                    person.save()
                    sys.exit()
                else:
                    sys.exit()
                print "Exit the System"
                break
        else:
            print "Please input the command!"