python 写的通讯录

 ''' start model book ''' import os,sys import cPickle from optparse import OptionParser class book: def __init__(self,fileName='defaultFileName.txt'): self.defaultAllowFunc=['add','select','update','insert','delete','clear','output'] self.data=[] self.fileName=fileName self.initFile() def initFile(self): try: self.data=cPickle.load(open(self.fileName,'r')) except: self.clear() def readData(self): return cPickle.load(open(self.fileName,'r')) def setData(self,data): self.data=data return cPickle.dump(self.data,open(self.fileName,'w')) def main(self): parser=OptionParser() parser.add_option('-f','--func',dest='func',help='help me',default='None') parser.add_option('-n','--name',dest='name',help='help me',default='None') parser.add_option('-a','--address',dest='address',help='help me1',default='None') options,_=parser.parse_args() if (options.func in self.defaultAllowFunc) and callable(getattr(self,options.func)): getattr(self,options.func)(name=options.name,address=options.address) else: print 'not exists this function' sys.exit() def add(self,**args): self.data.append((args["name"],args["address"])) self.setData(self.data) self.output() def select(self,**args): if args["address"]!='None' and args["name"]!='None': print [ (n,a) for n,a in self.data if (a==args["address"]) and (n==args["name"]) ] sys.exit() if args["name"]!='None': print [ (n,a) for n,a in self.data if n==args["name"] ] sys.exit() if args["address"]!='None': print [ (n,a) for n,a in self.data if a==args["address"] ] sys.exit() if args["address"]=='None' and args["name"]=='None': print self.data sys.exit() def update(self,**args): iindex=0 for n,a in self.data: iindex+=1 if args["name"]!='None' and args["name"]==n : try: self.data.remove((args["name"],a)) self.data.insert(iindex,(args["name"],args["address"])) self.setData(self.data) break except: break self.output() def insert(self,**args): self.add(**args) def delete(self,**args): for n,a in self.data: if args["address"]!='None' and args["name"]!='None' and args["address"]==a and args["name"]==n : try: self.data.remove((args["name"],args["address"])) self.setData(self.data) break except: break if args["name"]!='None' and args["name"]==n : try: self.data.remove((args["name"],a)) self.setData(self.data) break except: break if args["address"]!='None' and args["address"]==a : try: self.data.remove((n,args["address"])) self.setData(self.data) break except: break self.output() def clear(self,**args): try: data=[] cPickle.dump(data,open(self.fileName,'w')) self.data=data except: print "not write!" sys.exit() def output(self,**args): print self.data if __name__=='__main__': book("myfile.txt").main()  

 

保存为 book.py

 

执行命令:

python book.py -f add -n name1  -a address1  #新增 用户，地址
python book.py -f insert -n name1  -a address1 #新增 根据 用户，地址
python book.py -f select -n name1 #查询 根据 用户
python book.py -f select -a address1 #查询 根据 地址
python book.py -f select -n name1 -a address1  #查询 根据 用户，地址
python book.py -f update -n name1 -a address2 #更新 用户 地址
python book.py -f delete -n name1 -a address2  #删除 用户 地址

python book.py -f clear  #清除 所有用户记录

python book.py -f output #输出记事本数据

 

联系邮箱:wzm112@126.com ,QQ:67272740