Python利用shelve模块设计简单数据库程序

欢迎访问我的网站omegaxyz.com
利用shelve模块写数据库文件在程序关闭时仍然能存储之前的数据。

将所有内容都放到函数中会让程序更加结构化。
主程序放在main函数中，只有在if__name__==’main’条件成立的时候才被调用
意味着可以在其他程序中将这个程序作为模块导入，然后调用main函数。

我在main函数中打开数据库(shelf)，然后将其作为参数传给另外需要它的函数。
我也可以使用全局变量，毕竟这个程序很小。不过，在大多数情况下最好避免用全局变
量，除非有充足的理由要使用它。

我使用try/finally确保数据库能够正确关闭。（）我们永远不知道什么时候会出错程序会抛出异常)。如果程序在没有正确关闭数据库的情况下终止，那么，数据库文件可能被损坏了，这样的数据文件是毫无用处的。

下面是整个数据库文件代码：

#©OmegaXYZ
import shelve

def store_person(db):
    "Query user for data and store it in the shelf object"

    pid = input("Enter unique ID number: ")
    person = {}
    person['name'] = input("input your name: ")
    person['age'] = input("input your age: ")
    person['phone'] = input("Enter your phone number: ")

    db[pid] = person

def lookup_person(db):
    "Query user for ID and desired field. and fetch the corresponding data from the shelf object"

    pid = input("Enter ID number")
    field = input("What would you like to know?  (name, age, phone)")
    field = field.strip().lower()
    print(field.capitalize() + ':' + db[pid][field])

def print_help():
    print("The available commands are: ")
    print("store : Stores information about a person")
    print("lookup : Look up a person from ID number")
    print("quit : Save changes and exit")
    print("? : Print this message")

def enter_command():
    cmd = input("Enter command (? for help): ")
    cmd = cmd.strip().lower()
    return cmd

def main():
    database = shelve.open('E:\\Temporary\\database.dat')# You may want to change this name
    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()