Python使用shelve模組實現簡單資料存放區的方法

本文執行個體講述了Python使用shelve模組實現簡單資料存放區的方法。分享給大家供大家參考。具體分析如下：

Python的shelve模組提供了一種簡單的資料存放區方案，以dict（字典）的形式來操作資料。

#!/usr/bin/pythonimport sys, shelvedef 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 number:')  db[pid] = persondef lookup_person(db):  """  Query user for ID and desired field,   and fetch the corresponding data   from the shelf object  """  pid = raw_input('Enter unique ID number:')  temp = db[pid]  field = raw_input('Please enter name, age or phone:')  field.strip().lower()  print field.capitalize() + ': ', temp[field]def print_help():  print 'The avaliable commands are:'  print 'store  :Stores infomation about a person'  print 'lookup  :Looks up a person form 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 cmddef main():  database = shelve.open('database')  # database stores in current directory  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()    # Close database in any conditionif __name__ == '__main__':  main()

希望本文所述對大家的Python程式設計有所協助。

