python数据库程序_python简单模拟数据库程序

做个小程序，简单的模拟数据库，代码比较长，觉得可以学习的地方很多，记下来

#database.py

import sys,shelve

def store_person(db):

"""

Query user for data and store it in the shelf object

"""

pid = raw_input("Enter unique ID number:")

person = {}

person['name'] = raw_input("Enter name: ")

person['age'] = raw_input("Enter age: ")

person['phone']= raw_input("Enter phone")

db[pid] = person

def lookup_person(db):

"""

Query user for ID

"""

pid = raw_input("Enter ID number:")

field = raw_input("what would you like to know ? name,age,phont")

field = field.strip().lower()

print field.capitalize() + ':',db[pid][field]

def print_help():

print "the available commands are:"

print "store : store informantion about a person"

print "lookup: look up a person fron ID number"

print "quit : Save changes and exit"

print "? : Prints this message"

def enter_command():

cmd = raw_input("Enter command (? for help):")

cmd = cmd.strip().lower()

return cmd

def main():

database = shelve.open('/home/weichong/database.txt')

try:

while True:

cmd = enter_command()

if cmd == 'store':

store_person(database)

elif cmd == 'lookup':

lookup_person(database)

elif cmd == '?':

print_help()

elif cmd == 'quit':

return

finally:

database.close()

if __name__ == '__main__':main()