python 电话本管理例子

以下代码由PHP站长网 52php.cn收集自互联网

现在PHP站长网小编把它分享给大家，仅供参考

#!/usr/bin/env python
#coding=utf-8
#电话本管理，可以进行添加，删除，修改，查询用户信息
import cPickle as p 

class Telephone:
    def __init__(self):
        '''构造方法 '''
        
    def addPeople(self,name,email,telephone):
        """添加用户"""
        teleDict = self.getDictData()
        if teleDict:
            infoList = [name,telephone]
            teleDict[name] = infoList     
            self.writefile(teleDict)
        else:
            teleDict = {}
            infoList = [name,telephone]
            teleDict[name] = infoList       
            self.writefile(teleDict)
            
    
    def delPeople(self,name):
        """删除用户"""
        teleDict = self.getDictData()
        if name in teleDict.keys():
            del teleDict[name]
            self.writefile(teleDict)
        else:
            print name,'is not in dict'
    
    def editPeople(self,emial,telephone):
        """ 修改信息"""
        teleDict = self.getDictData()
        if name in teleDict.keys():
            infoList = [name,telephone]
            teleDict[name] = infoList     
            self.writefile(teleDict)
            print name+'edit success'
        else:
            print name,'is not in dict'
        
    
    def getPeople(self,name):
        """获取用户信息"""
        teleDict = self.getDictData()
        if teleDict:
            if name in teleDict.keys():
                people = teleDict[name]
                print people
            else:
                print name,'is not in dict'
        else:
            print 'people is empty'
            
        
    def writefile(self,dictData):
        """ 写入文件"""
        f = file('dict.data','w')
        p.dump(dictData,f)
        f.close()
        
    def getDictData(self):  
        """ 获取文件内容"""
        fileName = 'dict.data'
        try:
            f = file(fileName)
            teleDict = p.load(f)
            return teleDict
        except:
            print 'open file error'
 
 # 提示 信息
def notice():
    print "please enter 1-get people 2-add people 3-edit pelole 4-del people 5-get all people 0-break"
    
if __name__ == "__main__":
    while(True):
        notice()
        userInput = int(raw_input())
        
        people = Telephone()
        if userInput == 1:
            name = raw_input("please enter user name:")
            people.getPeople(name)
        elif userInput == 2:
            name = raw_input("enter name:")
            email = raw_input("enter emai:")
            telephone = raw_input("enter telephone:")
            people.addPeople(name,telephone)
        elif userInput == 3:
            name = raw_input("enter name:")
            email = raw_input("enter emai:")
            telephone = raw_input("enter telephone:")
            people.editPeople(name,telephone)
        elif userInput == 4:
            name = raw_input("enter del people name:")
            people.delPeople(name)
        elif userInput == 5:
            allpeople = people.getDictData()
            if allpeople:
                for key in allpeople:
                    print key,allpeople[key]
            else:
                print 'there is no people'
        elif userInput == 0:
            break
        else:
            print 'you select number is wrong'
    
    
    raw_input('press enter')
              
        
        

以上内容由PHP站长网【52php.cn】收集整理供大家参考研究

如果以上内容对您有帮助，欢迎收藏、点赞、推荐、分享。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!