用Python做命令行地址簿

用Python做命令行地址簿

阅读《简明Python教程》有一段时间了，书里提出一个小练习，实现了下。

问题：命令行地址簿程序，可以用来浏览、添加、编辑、删除联系人，需要有邮件地址、电话号码等多种信息。

程序设计

1、数据存储位置，程序主目录下address.data文件

存储方式：编号,,,姓名,,,年龄,,,地址,,,电话

2、功能： 浏览：查看所有信息

添加：添加一条信息

编辑：编辑指定的信息

删除：删除指定信息

3、实现代码

dao.py

# coding=UTF-8
#数据层

#数据库文件
datafile='address.data'
#连接符
contact=',,,'

__version__='1.0'

class AddressDao:
'''地址簿数据层'''
def __init__(self):
self.addressMap={}
self.number = '0'
def readaddressMap(self):
'''读取地址簿所有内容'''
with open(datafile) as f:
for line in f:
address = line.split(contact)
self.addressMap[address[0]]=address
self.number=address[0]

def saveAddress(self,name,age,mail,phone):
'''保存一条记录'''
newSeq = str(int(self.number)+1)
addressNew = [newSeq,name,age,mail,phone]
self.addressMap[newSeq]=addressNew
self.saveAddressToFile()

def saveAddressToFile(self):
'''持久化到文件'''
with open(datafile,'w') as f:
keys = list(self.addressMap.keys())
keys.sort(key=lambda i:i)
for key in keys:
f.writelines(contact.join(self.addressMap[key]))

def updateAddress(self,code,name,age,mail,phone):
'''修改一条记录'''
addressNew = [code,name,age,mail,phone]
self.addressMap[code]=addressNew
self.saveAddressToFile()

def delAddress(self,code):
'''删除一条记录'''
del(self.addressMap[code])
self.saveAddressToFile()

service.py

# coding=UTF-8
import dao

__version__='1.0'

menumain = '''
******我的地址簿******

1-浏览
2-新建
3-编辑
4-删除

----------------------
'''
title=['序号','姓名','年龄','邮箱','电话']
class AddressService:
'''地址簿服务层'''
def __init__(self):
'''程序加载时读取数据库现有的地址簿'''
self.add = dao.AddressDao()
self.add.readaddressMap()

def showMainMenu(self):
'''展示主菜单'''
print(menumain)

def inputCode(self):
'''接收用户输入指令'''
try:
code = input('请选择需要进行的操作:')
except EOFError:
print('谢谢使用,再见')
except KeyboardInterrupt:
print('谢谢使用,再见')
else:
if code=='1':
self.readAddress()
elif code=='2':
self.newAddress()
elif code=='3':
self.editAddress()
elif code=='4':
self.delAddress()
self.inputCode()

def readAddress(self):
'''浏览功能'''
print('\n浏览\n')
print('\t\t'.join(title))
keys = list(self.add.addressMap.keys())
keys.sort(key=lambda i:i)
for key in keys:
print('\t\t'.join(self.add.addressMap[key]).replace('\n',''))
self.showMainMenu()

def newAddress(self):
'''新建功能'''
print('\n新建\n')
name = ""
while name=="":
name=input('请输入*姓名:')
age = ""
while age=="":
age=input('请输入年龄:')
mail = ""
while mail=="":
mail=input('请输入邮箱:')
phone = ""
while phone=="":
phone=input('请输入电话')
self.add.saveAddress(name,age,mail,phone)
print('保存成功')
self.showMainMenu()

def editAddress(self):
'''编辑功能'''
print('\n编辑\n')
willEdit = input('请输入想要编辑的序号:')
if willEdit in self.add.addressMap:
name = ""
while name=="":
name=input('请输入*姓名:')
age = ""
while age=="":
age=input('请输入年龄:')
mail = ""
while mail=="":
mail=input('请输入邮箱:')
phone = ""
while phone=="":
phone=input('请输入电话:')
self.add.updateAddress(willEdit,name,age,mail,phone)
print('修改成功')
else:
print('找不到要修改的序号')
self.showMainMenu()

def delAddress(self):
'''删除功能'''
print('\n删除\n')
willDel = input('请输入想要删除的序号:')
if willDel in self.add.addressMap:
self.add.delAddress(willDel)
print('删除成功')
else:
print('找不到想要删除的序号')
self.showMainMenu()

main.py

# coding=UTF-8
#程序入口
import service

ads=service.AddressService()
ads.showMainMenu()
ads.inputCode()

运行：–>python main.py