利用shelve模块写数据库文件在程序关闭时仍然能存储之前的数据。
将所有内容都放到函数中会让程序更加结构化。
主程序放在main函数中，只有在if__name__=='__main__'条件成立的时候才被调用
意味着可以在其他程序中将这个程序作为模块导入，然后调用main函数。
我在main函数中打开数据库(shelf)，然后将其作为参数传给另外需要它的函数。
我也可以使用全局变量，毕竟这个程序很小。不过，在大多数情况下最好避免用全局变
量，除非有充足的理由要使用它。
我使用try/finally确保数据库能够正确关闭。（）我们永远不知道什么时候会出错程序会抛出异常)。如果程序在没有正确关闭数据库的情况下终止，那么，数据库文件可能被损坏了，这样的数据文件是毫无用处的。
下面是整个数据库文件代码：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
#©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() |
