python phonebook management example

大家讲道理
Release: 2016-11-09 11:36:39
Original
1410 people have browsed it

#!/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,email,telephone]
            teleDict[name] = infoList     
            self.writefile(teleDict)
        else:
            teleDict = {}
            infoList = [name,email,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,name,emial,telephone):
        """ 修改信息"""
        teleDict = self.getDictData()
        if name in teleDict.keys():
            infoList = [name,email,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,email,telephone)
        elif userInput == 3:
            name = raw_input("enter name:")
            email = raw_input("enter emai:")
            telephone = raw_input("enter telephone:")
            people.editPeople(name,email,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')
Copy after login

source:php.cn
Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn
Popular Tutorials
More>
Latest Downloads
More>
Web Effects
Website Source Code
Website Materials
Front End Template