python手机通讯录管理,python 电话本管理例子

#!/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')