用python实现数据库的添加和查找,以人员添加和查找为例
基于python3.6
代码
import sys,shelve
def store_person(db):
number=input("请输入此人的ID:")
person={}
person['name']=input("请输入此人的名字:")
person['age']=input("请输入此人年龄:")
person['tel']=input("请输入此人的电话号码:")
db[number]=person
def search_person(db):
num=input("请输入要查找的ID:")
middle=input("你想查找name或者age或者tel:")
middle=middle.strip().lower() #对输入的字符串进行处理,去掉多余的空格和将字符串大写变成小写
print(middle.capitalize()+":",db[num][middle]) #capitalize函数将字符串第一个字符大写
def help():
print("相关操作及说明:")
print('store ',"录入人员信息")
print("search ","查找人员信息")
print("quit","退出")
print("help","查看帮助说明")
def main():
datebase=shelve.open('G:\\人员数据库')
help()
a=1
try:
while a!='quit':
a=input("请输入相关操作:")
if a=='store':
store_person(datebase)
elif a=='search':
search_person(datebase)
elif a=='help':
help()
finally:
datebase.close()
if __name__=='__main__':
main()
运行截图: